fix: expand create table

Author: rustyconover

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "query-farm-flight-server"
-version = "0.1.8"
+version = "0.1.9"
 description = "A framework for building Arrow Flight servers for the DuckDB Airport extension with robust authentication, schema management, and data handling capabilities."
 authors = [{ name = "Rusty Conover", email = "hello@query.farm" }]
 dependencies = [

src/query_farm_flight_server/parameter_types.py

@@ -118,6 +118,11 @@ class CreateTable(BaseModel):
     unique_constraints: list[int]
     check_constraints: list[str]
 
+    primary_key_columns: list[str]
+    unique_columns: list[str]
+    multi_key_primary_keys: list[str]
+    extra_constraints: list[str]
+
 
 T = TypeVar("T", bound=BaseModel)
 

src/query_farm_flight_server/test_action_decoders.py

@@ -41,6 +41,10 @@ def test_decode_create_table() -> None:
             "not_null_constraints": [],
             "unique_constraints": [],
             "check_constraints": ["test1"],
+            "primary_key_columns": [],
+            "unique_columns": [],
+            "multi_key_primary_keys": [],
+            "extra_constraints": [],
         },
     )