fixes

Author: rustyconover

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "query-farm-flight-server"
-version = "0.1.13"
+version = "0.1.14"
 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/flight_inventory.py

@@ -71,6 +71,8 @@ def serialize(self) -> bytes:
 
 FlightInventoryWithMetadata = tuple[flight.FlightInfo, FlightSchemaMetadata]
 
+ScalarFunctionStability = Literal["consistent", "volatile", "consistent_within_query"]
+
 
 class ScalarFunctionMetadata(FlightSchemaMetadata):
     """
@@ -87,7 +89,7 @@ def __init__(
         comment: str | None,
         action_name: str | None = None,
         input_schema: pa.Schema,
-        stability: Literal["consistent", "volatile", "consistent_within_query"] = "volatile",
+        stability: ScalarFunctionStability = "volatile",
     ):
         extra_data = msgpack.packb({"stability": stability})
         assert extra_data