Skip to content

Commit 4b64813

Browse files
committed
fix: add time travel parameters
1 parent c53214c commit 4b64813

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

src/query_farm_server_base/parameter_types.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -212,6 +212,9 @@ class TableFunctionParameters(BaseModel):
212212
column_ids: list[int]
213213
parameters: pa.RecordBatch | None
214214

215+
at_unit: str | None = None
216+
at_value: str | None = None
217+
215218
_validate_parameters = field_validator("parameters", mode="before")(
216219
deserialize_record_batch_or_none
217220
)
@@ -226,6 +229,9 @@ class EndpointsParameters(BaseModel):
226229
table_function_parameters: pa.RecordBatch | None
227230
table_function_input_schema: pa.Schema | None
228231

232+
at_unit: str | None = None
233+
at_value: str | None = None
234+
229235
_validate_table_function_parameters = field_validator(
230236
"table_function_parameters", mode="before"
231237
)(deserialize_record_batch_or_none)
@@ -304,6 +310,9 @@ class TableFunctionFlightInfo(BaseModel):
304310
# part of an in-out table returning function.
305311
table_input_schema: pa.Schema | None
306312

313+
at_unit: str | None = None
314+
at_value: str | None = None
315+
307316
_validate_flight_descriptor = field_validator("descriptor", mode="before")(
308317
deserialize_flight_descriptor
309318
)

0 commit comments

Comments
 (0)