chalkpy 2.89.22__py3-none-any.whl → 2.95.3__py3-none-any.whl

chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.py

@@ -25,6 +25,11 @@ class SqlServiceStub(object):
             request_serializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PlanSqlQueryRequest.SerializeToString,
             response_deserializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PlanSqlQueryResponse.FromString,
         )
+        self.PollSqlQuery = channel.unary_unary(
+            "/chalk.protosql.v1.SqlService/PollSqlQuery",
+            request_serializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PollSqlQueryRequest.SerializeToString,
+            response_deserializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PollSqlQueryResponse.FromString,
+        )
         self.GetDbCatalogs = channel.unary_unary(
             "/chalk.protosql.v1.SqlService/GetDbCatalogs",
             request_serializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.GetDbCatalogsRequest.SerializeToString,
@@ -57,6 +62,12 @@ class SqlServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def PollSqlQuery(self, request, context):
+        """Poll for the status and results of an asynchronous SQL query"""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
     def GetDbCatalogs(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -88,6 +99,11 @@ def add_SqlServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PlanSqlQueryRequest.FromString,
             response_serializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PlanSqlQueryResponse.SerializeToString,
         ),
+        "PollSqlQuery": grpc.unary_unary_rpc_method_handler(
+            servicer.PollSqlQuery,
+            request_deserializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PollSqlQueryRequest.FromString,
+            response_serializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PollSqlQueryResponse.SerializeToString,
+        ),
         "GetDbCatalogs": grpc.unary_unary_rpc_method_handler(
             servicer.GetDbCatalogs,
             request_deserializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.GetDbCatalogsRequest.FromString,
@@ -170,6 +186,35 @@ class SqlService(object):
             metadata,
         )
 
+    @staticmethod
+    def PollSqlQuery(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chalk.protosql.v1.SqlService/PollSqlQuery",
+            chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PollSqlQueryRequest.SerializeToString,
+            chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PollSqlQueryResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
     @staticmethod
     def GetDbCatalogs(
         request,

chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.pyi

@@ -18,6 +18,8 @@ from chalk._gen.chalk.protosql.v1.sql_service_pb2 import (
     GetTablesResponse,
     PlanSqlQueryRequest,
     PlanSqlQueryResponse,
+    PollSqlQueryRequest,
+    PollSqlQueryResponse,
 )
 from grpc import (
     Channel,
@@ -36,6 +38,11 @@ class SqlServiceStub:
         PlanSqlQueryRequest,
         PlanSqlQueryResponse,
     ]
+    PollSqlQuery: UnaryUnaryMultiCallable[
+        PollSqlQueryRequest,
+        PollSqlQueryResponse,
+    ]
+    """Poll for the status and results of an asynchronous SQL query"""
     GetDbCatalogs: UnaryUnaryMultiCallable[
         GetDbCatalogsRequest,
         GetDbCatalogsResponse,
@@ -63,6 +70,13 @@ class SqlServiceServicer(metaclass=ABCMeta):
         context: ServicerContext,
     ) -> PlanSqlQueryResponse: ...
     @abstractmethod
+    def PollSqlQuery(
+        self,
+        request: PollSqlQueryRequest,
+        context: ServicerContext,
+    ) -> PollSqlQueryResponse:
+        """Poll for the status and results of an asynchronous SQL query"""
+    @abstractmethod
     def GetDbCatalogs(
         self,
         request: GetDbCatalogsRequest,

chalk/_gen/chalk/python/v1/types_pb2.py

@@ -17,7 +17,7 @@ from chalk._gen.chalk.arrow.v1 import arrow_pb2 as chalk_dot_arrow_dot_v1_dot_ar
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n\x1b\x63halk/python/v1/types.proto\x12\x0f\x63halk.python.v1\x1a\x1a\x63halk/arrow/v1/arrow.proto"2\n\x05TySet\x12)\n\x05items\x18\x01 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x05items"3\n\x06TyList\x12)\n\x05items\x18\x01 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x05items"7\n\nTyIterable\x12)\n\x05items\x18\x01 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x05items"8\n\x0bTyGenerator\x12)\n\x05items\x18\x01 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x05items"(\n\nTyDatetime\x12\x13\n\x02tz\x18\x01 \x01(\tH\x00R\x02tz\x88\x01\x01\x42\x05\n\x03_tz"Z\n\x06TyDict\x12%\n\x03key\x18\x01 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x03key\x12)\n\x05value\x18\x02 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x05value"U\n\x07TyTuple\x12)\n\x05\x66ixed\x18\x01 \x03(\x0b\x32\x13.chalk.python.v1.TyR\x05\x66ixed\x12\x1f\n\x0bis_variable\x18\x02 \x01(\x08R\nisVariable"\xa7\x01\n\x0fTyLogicalStruct\x12\x44\n\x06\x66ields\x18\x01 \x03(\x0b\x32,.chalk.python.v1.TyLogicalStruct.FieldsEntryR\x06\x66ields\x1aN\n\x0b\x46ieldsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12)\n\x05value\x18\x02 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x05value:\x02\x38\x01"\xfb\x01\n\x0eTyFeatureClass\x12\x33\n\x15\x63onstructor_namespace\x18\x01 \x01(\tR\x14\x63onstructorNamespace\x12\\\n\x0f\x61ssigned_fields\x18\x02 \x03(\x0b\x32\x33.chalk.python.v1.TyFeatureClass.AssignedFieldsEntryR\x0e\x61ssignedFields\x1aV\n\x13\x41ssignedFieldsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12)\n\x05value\x18\x02 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x05value:\x02\x38\x01"\x84\x01\n\x06TyEnum\x12\x16\n\x06module\x18\x01 \x01(\tR\x06module\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12)\n\x05\x62\x61ses\x18\x03 \x03(\x0b\x32\x13.chalk.python.v1.TyR\x05\x62\x61ses\x12#\n\x02ty\x18\x04 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x02ty"\x89\t\n\x02Ty\x12\x1a\n\x08nullable\x18\x01 \x01(\x08R\x08nullable\x12\x31\n\x03int\x18\x02 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x03int\x12\x31\n\x03str\x18\x03 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x03str\x12\x33\n\x04\x62ool\x18\x04 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x04\x62ool\x12\x35\n\x05\x66loat\x18\x05 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x05\x66loat\x12-\n\x04list\x18\x06 \x01(\x0b\x32\x17.chalk.python.v1.TyListH\x00R\x04list\x12*\n\x03set\x18\x07 \x01(\x0b\x32\x16.chalk.python.v1.TySetH\x00R\x03set\x12\x33\n\x04none\x18\x08 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x04none\x12\x31\n\x03\x61ny\x18\t \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x03\x61ny\x12\x39\n\x08iterable\x18\n \x01(\x0b\x32\x1b.chalk.python.v1.TyIterableH\x00R\x08iterable\x12\x39\n\x08\x64\x61tetime\x18\x0b \x01(\x0b\x32\x1b.chalk.python.v1.TyDatetimeH\x00R\x08\x64\x61tetime\x12=\n\ttimedelta\x18\x0c \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\ttimedelta\x12\x33\n\x04\x64\x61te\x18\r \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x04\x64\x61te\x12\x30\n\x05tuple\x18\x0e \x01(\x0b\x32\x18.chalk.python.v1.TyTupleH\x00R\x05tuple\x12-\n\x04\x64ict\x18\x0f \x01(\x0b\x32\x17.chalk.python.v1.TyDictH\x00R\x04\x64ict\x12\x35\n\x05\x62ytes\x18\x10 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x05\x62ytes\x12I\n\x0elogical_struct\x18\x11 \x01(\x0b\x32 .chalk.python.v1.TyLogicalStructH\x00R\rlogicalStruct\x12\x46\n\rfeature_class\x18\x12 \x01(\x0b\x32\x1f.chalk.python.v1.TyFeatureClassH\x00R\x0c\x66\x65\x61tureClass\x12-\n\x04\x65num\x18\x13 \x01(\x0b\x32\x17.chalk.python.v1.TyEnumH\x00R\x04\x65num\x12J\n\x10sequence_matcher\x18\x14 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x0fsequenceMatcher\x12<\n\tgenerator\x18\x15 \x01(\x0b\x32\x1c.chalk.python.v1.TyGeneratorH\x00R\tgeneratorB\x04\n\x02ty"\x0e\n\x0c\x45mptyMessage"g\n\rSymbolicConst\x12#\n\x02ty\x18\x01 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x02ty\x12\x31\n\x05value\x18\x02 \x01(\x0b\x32\x1b.chalk.arrow.v1.ScalarValueR\x05value":\n\x0c\x43odeVariable\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x16\n\x06module\x18\x02 \x01(\tR\x06module"\x84\x01\n\x11\x43odeVariableValue\x12\x39\n\x08variable\x18\x01 \x01(\x0b\x32\x1d.chalk.python.v1.CodeVariableR\x08variable\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32\x1e.chalk.python.v1.SymbolicConstR\x05value"\x9e\x02\n\x13GlobalVariablesInfo\x12I\n\x0e\x63ode_variables\x18\x01 \x03(\x0b\x32".chalk.python.v1.CodeVariableValueR\rcodeVariables\x12s\n\x15\x65nvironment_variables\x18\x02 \x03(\x0b\x32>.chalk.python.v1.GlobalVariablesInfo.EnvironmentVariablesEntryR\x14\x65nvironmentVariables\x1aG\n\x19\x45nvironmentVariablesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x7f\n\x13\x63om.chalk.python.v1B\nTypesProtoP\x01\xa2\x02\x03\x43PX\xaa\x02\x0f\x43halk.Python.V1\xca\x02\x0f\x43halk\\Python\\V1\xe2\x02\x1b\x43halk\\Python\\V1\\GPBMetadata\xea\x02\x11\x43halk::Python::V1b\x06proto3'
+    b'\n\x1b\x63halk/python/v1/types.proto\x12\x0f\x63halk.python.v1\x1a\x1a\x63halk/arrow/v1/arrow.proto"2\n\x05TySet\x12)\n\x05items\x18\x01 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x05items"3\n\x06TyList\x12)\n\x05items\x18\x01 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x05items"7\n\nTyIterable\x12)\n\x05items\x18\x01 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x05items"8\n\x0bTyGenerator\x12)\n\x05items\x18\x01 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x05items"(\n\nTyDatetime\x12\x13\n\x02tz\x18\x01 \x01(\tH\x00R\x02tz\x88\x01\x01\x42\x05\n\x03_tz"Z\n\x06TyDict\x12%\n\x03key\x18\x01 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x03key\x12)\n\x05value\x18\x02 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x05value"U\n\x07TyTuple\x12)\n\x05\x66ixed\x18\x01 \x03(\x0b\x32\x13.chalk.python.v1.TyR\x05\x66ixed\x12\x1f\n\x0bis_variable\x18\x02 \x01(\x08R\nisVariable"\xa7\x01\n\x0fTyLogicalStruct\x12\x44\n\x06\x66ields\x18\x01 \x03(\x0b\x32,.chalk.python.v1.TyLogicalStruct.FieldsEntryR\x06\x66ields\x1aN\n\x0b\x46ieldsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12)\n\x05value\x18\x02 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x05value:\x02\x38\x01"\xfb\x01\n\x0eTyFeatureClass\x12\x33\n\x15\x63onstructor_namespace\x18\x01 \x01(\tR\x14\x63onstructorNamespace\x12\\\n\x0f\x61ssigned_fields\x18\x02 \x03(\x0b\x32\x33.chalk.python.v1.TyFeatureClass.AssignedFieldsEntryR\x0e\x61ssignedFields\x1aV\n\x13\x41ssignedFieldsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12)\n\x05value\x18\x02 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x05value:\x02\x38\x01"\x84\x01\n\x06TyEnum\x12\x16\n\x06module\x18\x01 \x01(\tR\x06module\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12)\n\x05\x62\x61ses\x18\x03 \x03(\x0b\x32\x13.chalk.python.v1.TyR\x05\x62\x61ses\x12#\n\x02ty\x18\x04 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x02ty"\x97\n\n\x02Ty\x12\x1a\n\x08nullable\x18\x01 \x01(\x08R\x08nullable\x12\x31\n\x03int\x18\x02 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x03int\x12\x31\n\x03str\x18\x03 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x03str\x12\x33\n\x04\x62ool\x18\x04 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x04\x62ool\x12\x35\n\x05\x66loat\x18\x05 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x05\x66loat\x12-\n\x04list\x18\x06 \x01(\x0b\x32\x17.chalk.python.v1.TyListH\x00R\x04list\x12*\n\x03set\x18\x07 \x01(\x0b\x32\x16.chalk.python.v1.TySetH\x00R\x03set\x12\x33\n\x04none\x18\x08 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x04none\x12\x31\n\x03\x61ny\x18\t \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x03\x61ny\x12\x39\n\x08iterable\x18\n \x01(\x0b\x32\x1b.chalk.python.v1.TyIterableH\x00R\x08iterable\x12\x39\n\x08\x64\x61tetime\x18\x0b \x01(\x0b\x32\x1b.chalk.python.v1.TyDatetimeH\x00R\x08\x64\x61tetime\x12=\n\ttimedelta\x18\x0c \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\ttimedelta\x12\x33\n\x04\x64\x61te\x18\r \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x04\x64\x61te\x12\x30\n\x05tuple\x18\x0e \x01(\x0b\x32\x18.chalk.python.v1.TyTupleH\x00R\x05tuple\x12-\n\x04\x64ict\x18\x0f \x01(\x0b\x32\x17.chalk.python.v1.TyDictH\x00R\x04\x64ict\x12\x35\n\x05\x62ytes\x18\x10 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x05\x62ytes\x12I\n\x0elogical_struct\x18\x11 \x01(\x0b\x32 .chalk.python.v1.TyLogicalStructH\x00R\rlogicalStruct\x12\x46\n\rfeature_class\x18\x12 \x01(\x0b\x32\x1f.chalk.python.v1.TyFeatureClassH\x00R\x0c\x66\x65\x61tureClass\x12-\n\x04\x65num\x18\x13 \x01(\x0b\x32\x17.chalk.python.v1.TyEnumH\x00R\x04\x65num\x12J\n\x10sequence_matcher\x18\x14 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x0fsequenceMatcher\x12<\n\tgenerator\x18\x15 \x01(\x0b\x32\x1c.chalk.python.v1.TyGeneratorH\x00R\tgenerator\x12\x35\n\x05never\x18\x16 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x05never\x12U\n\x16requests_http_response\x18\x17 \x01(\x0b\x32\x1d.chalk.python.v1.EmptyMessageH\x00R\x14requestsHttpResponseB\x04\n\x02ty"\x0e\n\x0c\x45mptyMessage"g\n\rSymbolicConst\x12#\n\x02ty\x18\x01 \x01(\x0b\x32\x13.chalk.python.v1.TyR\x02ty\x12\x31\n\x05value\x18\x02 \x01(\x0b\x32\x1b.chalk.arrow.v1.ScalarValueR\x05value":\n\x0c\x43odeVariable\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x16\n\x06module\x18\x02 \x01(\tR\x06module"\x84\x01\n\x11\x43odeVariableValue\x12\x39\n\x08variable\x18\x01 \x01(\x0b\x32\x1d.chalk.python.v1.CodeVariableR\x08variable\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32\x1e.chalk.python.v1.SymbolicConstR\x05value"\x9e\x02\n\x13GlobalVariablesInfo\x12I\n\x0e\x63ode_variables\x18\x01 \x03(\x0b\x32".chalk.python.v1.CodeVariableValueR\rcodeVariables\x12s\n\x15\x65nvironment_variables\x18\x02 \x03(\x0b\x32>.chalk.python.v1.GlobalVariablesInfo.EnvironmentVariablesEntryR\x14\x65nvironmentVariables\x1aG\n\x19\x45nvironmentVariablesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x7f\n\x13\x63om.chalk.python.v1B\nTypesProtoP\x01\xa2\x02\x03\x43PX\xaa\x02\x0f\x43halk.Python.V1\xca\x02\x0f\x43halk\\Python\\V1\xe2\x02\x1b\x43halk\\Python\\V1\\GPBMetadata\xea\x02\x11\x43halk::Python::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -59,17 +59,17 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_TYENUM"]._serialized_start = 942
     _globals["_TYENUM"]._serialized_end = 1074
     _globals["_TY"]._serialized_start = 1077
-    _globals["_TY"]._serialized_end = 2238
-    _globals["_EMPTYMESSAGE"]._serialized_start = 2240
-    _globals["_EMPTYMESSAGE"]._serialized_end = 2254
-    _globals["_SYMBOLICCONST"]._serialized_start = 2256
-    _globals["_SYMBOLICCONST"]._serialized_end = 2359
-    _globals["_CODEVARIABLE"]._serialized_start = 2361
-    _globals["_CODEVARIABLE"]._serialized_end = 2419
-    _globals["_CODEVARIABLEVALUE"]._serialized_start = 2422
-    _globals["_CODEVARIABLEVALUE"]._serialized_end = 2554
-    _globals["_GLOBALVARIABLESINFO"]._serialized_start = 2557
-    _globals["_GLOBALVARIABLESINFO"]._serialized_end = 2843
-    _globals["_GLOBALVARIABLESINFO_ENVIRONMENTVARIABLESENTRY"]._serialized_start = 2772
-    _globals["_GLOBALVARIABLESINFO_ENVIRONMENTVARIABLESENTRY"]._serialized_end = 2843
+    _globals["_TY"]._serialized_end = 2380
+    _globals["_EMPTYMESSAGE"]._serialized_start = 2382
+    _globals["_EMPTYMESSAGE"]._serialized_end = 2396
+    _globals["_SYMBOLICCONST"]._serialized_start = 2398
+    _globals["_SYMBOLICCONST"]._serialized_end = 2501
+    _globals["_CODEVARIABLE"]._serialized_start = 2503
+    _globals["_CODEVARIABLE"]._serialized_end = 2561
+    _globals["_CODEVARIABLEVALUE"]._serialized_start = 2564
+    _globals["_CODEVARIABLEVALUE"]._serialized_end = 2696
+    _globals["_GLOBALVARIABLESINFO"]._serialized_start = 2699
+    _globals["_GLOBALVARIABLESINFO"]._serialized_end = 2985
+    _globals["_GLOBALVARIABLESINFO_ENVIRONMENTVARIABLESENTRY"]._serialized_start = 2914
+    _globals["_GLOBALVARIABLESINFO_ENVIRONMENTVARIABLESENTRY"]._serialized_end = 2985
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/python/v1/types_pb2.pyi

@@ -133,6 +133,8 @@ class Ty(_message.Message):
         "enum",
         "sequence_matcher",
         "generator",
+        "never",
+        "requests_http_response",
     )
     NULLABLE_FIELD_NUMBER: _ClassVar[int]
     INT_FIELD_NUMBER: _ClassVar[int]
@@ -155,6 +157,8 @@ class Ty(_message.Message):
     ENUM_FIELD_NUMBER: _ClassVar[int]
     SEQUENCE_MATCHER_FIELD_NUMBER: _ClassVar[int]
     GENERATOR_FIELD_NUMBER: _ClassVar[int]
+    NEVER_FIELD_NUMBER: _ClassVar[int]
+    REQUESTS_HTTP_RESPONSE_FIELD_NUMBER: _ClassVar[int]
     nullable: bool
     int: EmptyMessage
     str: EmptyMessage
@@ -176,6 +180,8 @@ class Ty(_message.Message):
     enum: TyEnum
     sequence_matcher: EmptyMessage
     generator: TyGenerator
+    never: EmptyMessage
+    requests_http_response: EmptyMessage
     def __init__(
         self,
         nullable: bool = ...,
@@ -199,6 +205,8 @@ class Ty(_message.Message):
         enum: _Optional[_Union[TyEnum, _Mapping]] = ...,
         sequence_matcher: _Optional[_Union[EmptyMessage, _Mapping]] = ...,
         generator: _Optional[_Union[TyGenerator, _Mapping]] = ...,
+        never: _Optional[_Union[EmptyMessage, _Mapping]] = ...,
+        requests_http_response: _Optional[_Union[EmptyMessage, _Mapping]] = ...,
     ) -> None: ...
 
 class EmptyMessage(_message.Message):

chalk/_gen/chalk/server/v1/benchmark_pb2.py

@@ -0,0 +1,76 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: chalk/server/v1/benchmark.proto
+# Protobuf Python Version: 4.25.3
+"""Generated protocol buffer code."""
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from chalk._gen.chalk.auth.v1 import permissions_pb2 as chalk_dot_auth_dot_v1_dot_permissions__pb2
+from chalk._gen.chalk.common.v1 import online_query_pb2 as chalk_dot_common_dot_v1_dot_online__query__pb2
+from google.protobuf import duration_pb2 as google_dot_protobuf_dot_duration__pb2
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n\x1f\x63halk/server/v1/benchmark.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a"chalk/common/v1/online_query.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto"6\n\rInputFeatures\x12%\n\x0einput_features\x18\x01 \x03(\tR\rinputFeatures"\xef\x01\n\x1cSimpleOnlineQueryBulkRequest\x12)\n\x0einput_features\x18\x01 \x03(\tB\x02\x18\x01R\rinputFeatures\x12P\n\x13input_features_list\x18\x03 \x01(\x0b\x32\x1e.chalk.server.v1.InputFeaturesH\x00R\x11inputFeaturesList\x12\x1f\n\ninput_file\x18\x04 \x01(\tH\x00R\tinputFile\x12\'\n\x0foutput_features\x18\x02 \x03(\tR\x0eoutputFeaturesB\x08\n\x06inputs"\xfc\x02\n\x16\x43reateBenchmarkRequest\x12\x1d\n\nwarmup_qps\x18\x01 \x01(\x03R\twarmupQps\x12\x42\n\x0fwarmup_duration\x18\x02 \x01(\x0b\x32\x19.google.protobuf.DurationR\x0ewarmupDuration\x12\x10\n\x03qps\x18\x03 \x01(\x03R\x03qps\x12\x35\n\x08\x64uration\x18\x04 \x01(\x0b\x32\x19.google.protobuf.DurationR\x08\x64uration\x12H\n\nquery_bulk\x18\x05 \x01(\x0b\x32\'.chalk.common.v1.OnlineQueryBulkRequestH\x00R\tqueryBulk\x12[\n\x11simple_query_bulk\x18\x06 \x01(\x0b\x32-.chalk.server.v1.SimpleOnlineQueryBulkRequestH\x00R\x0fsimpleQueryBulkB\x0f\n\rquery_request"v\n\x17\x43reateBenchmarkResponse\x12\x38\n\x06status\x18\x01 \x01(\x0e\x32 .chalk.server.v1.BenchmarkStatusR\x06status\x12!\n\x0c\x62\x65nchmark_id\x18\x02 \x01(\tR\x0b\x62\x65nchmarkId"\x99\x01\n\x12\x42\x65nchmarkInputFile\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x17\n\x04size\x18\x02 \x01(\x03H\x00R\x04size\x88\x01\x01\x12>\n\nupdated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\tupdatedAt\x88\x01\x01\x42\x07\n\x05_sizeB\r\n\x0b_updated_at"7\n\x1dGetAvailableInputFilesRequest\x12\x16\n\x06prefix\x18\x01 \x01(\tR\x06prefix"f\n\x1eGetAvailableInputFilesResponse\x12\x44\n\x0binput_files\x18\x01 \x03(\x0b\x32#.chalk.server.v1.BenchmarkInputFileR\ninputFiles"e\n\x1dGetInputFileUploadUrlsRequest\x12\x44\n\x0binput_files\x18\x01 \x03(\x0b\x32#.chalk.server.v1.BenchmarkInputFileR\ninputFiles"G\n\x1eGetInputFileUploadUrlsResponse\x12%\n\x0epresigned_urls\x18\x01 \x03(\tR\rpresignedUrls"8\n\x1eGetAvailableResultFilesRequest\x12\x16\n\x06prefix\x18\x01 \x01(\tR\x06prefix"i\n\x1fGetAvailableResultFilesResponse\x12\x46\n\x0cresult_files\x18\x01 \x03(\x0b\x32#.chalk.server.v1.BenchmarkInputFileR\x0bresultFiles"9\n\x18GetResultFileUrlsRequest\x12\x1d\n\nfile_names\x18\x01 \x03(\tR\tfileNames"B\n\x19GetResultFileUrlsResponse\x12%\n\x0epresigned_urls\x18\x01 \x03(\tR\rpresignedUrls*\xe8\x01\n\x0f\x42\x65nchmarkStatus\x12 \n\x1c\x42\x45NCHMARK_STATUS_UNSPECIFIED\x10\x00\x12\x1b\n\x17\x42\x45NCHMARK_STATUS_QUEUED\x10\x01\x12\x1c\n\x18\x42\x45NCHMARK_STATUS_WORKING\x10\x02\x12\x1e\n\x1a\x42\x45NCHMARK_STATUS_COMPLETED\x10\x03\x12\x1b\n\x17\x42\x45NCHMARK_STATUS_FAILED\x10\x04\x12\x1c\n\x18\x42\x45NCHMARK_STATUS_SKIPPED\x10\x05\x12\x1d\n\x19\x42\x45NCHMARK_STATUS_CANCELED\x10\x06\x32\xf2\x04\n\x10\x42\x65nchmarkService\x12i\n\x0f\x43reateBenchmark\x12\'.chalk.server.v1.CreateBenchmarkRequest\x1a(.chalk.server.v1.CreateBenchmarkResponse"\x03\x80}\x03\x12~\n\x16GetAvailableInputFiles\x12..chalk.server.v1.GetAvailableInputFilesRequest\x1a/.chalk.server.v1.GetAvailableInputFilesResponse"\x03\x80}\x03\x12~\n\x16GetInputFileUploadUrls\x12..chalk.server.v1.GetInputFileUploadUrlsRequest\x1a/.chalk.server.v1.GetInputFileUploadUrlsResponse"\x03\x80}\x03\x12\x81\x01\n\x17GetAvailableResultFiles\x12/.chalk.server.v1.GetAvailableResultFilesRequest\x1a\x30.chalk.server.v1.GetAvailableResultFilesResponse"\x03\x80}\x03\x12o\n\x11GetResultFileUrls\x12).chalk.server.v1.GetResultFileUrlsRequest\x1a*.chalk.server.v1.GetResultFileUrlsResponse"\x03\x80}\x03\x42\x97\x01\n\x13\x63om.chalk.server.v1B\x0e\x42\x65nchmarkProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.server.v1.benchmark_pb2", _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    _globals["DESCRIPTOR"]._options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n\023com.chalk.server.v1B\016BenchmarkProtoP\001Z\022server/v1;serverv1\242\002\003CSX\252\002\017Chalk.Server.V1\312\002\017Chalk\\Server\\V1\342\002\033Chalk\\Server\\V1\\GPBMetadata\352\002\021Chalk::Server::V1"
+    _globals["_SIMPLEONLINEQUERYBULKREQUEST"].fields_by_name["input_features"]._options = None
+    _globals["_SIMPLEONLINEQUERYBULKREQUEST"].fields_by_name["input_features"]._serialized_options = b"\030\001"
+    _globals["_BENCHMARKSERVICE"].methods_by_name["CreateBenchmark"]._options = None
+    _globals["_BENCHMARKSERVICE"].methods_by_name["CreateBenchmark"]._serialized_options = b"\200}\003"
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetAvailableInputFiles"]._options = None
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetAvailableInputFiles"]._serialized_options = b"\200}\003"
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetInputFileUploadUrls"]._options = None
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetInputFileUploadUrls"]._serialized_options = b"\200}\003"
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetAvailableResultFiles"]._options = None
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetAvailableResultFiles"]._serialized_options = b"\200}\003"
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetResultFileUrls"]._options = None
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetResultFileUrls"]._serialized_options = b"\200}\003"
+    _globals["_BENCHMARKSTATUS"]._serialized_start = 1773
+    _globals["_BENCHMARKSTATUS"]._serialized_end = 2005
+    _globals["_INPUTFEATURES"]._serialized_start = 186
+    _globals["_INPUTFEATURES"]._serialized_end = 240
+    _globals["_SIMPLEONLINEQUERYBULKREQUEST"]._serialized_start = 243
+    _globals["_SIMPLEONLINEQUERYBULKREQUEST"]._serialized_end = 482
+    _globals["_CREATEBENCHMARKREQUEST"]._serialized_start = 485
+    _globals["_CREATEBENCHMARKREQUEST"]._serialized_end = 865
+    _globals["_CREATEBENCHMARKRESPONSE"]._serialized_start = 867
+    _globals["_CREATEBENCHMARKRESPONSE"]._serialized_end = 985
+    _globals["_BENCHMARKINPUTFILE"]._serialized_start = 988
+    _globals["_BENCHMARKINPUTFILE"]._serialized_end = 1141
+    _globals["_GETAVAILABLEINPUTFILESREQUEST"]._serialized_start = 1143
+    _globals["_GETAVAILABLEINPUTFILESREQUEST"]._serialized_end = 1198
+    _globals["_GETAVAILABLEINPUTFILESRESPONSE"]._serialized_start = 1200
+    _globals["_GETAVAILABLEINPUTFILESRESPONSE"]._serialized_end = 1302
+    _globals["_GETINPUTFILEUPLOADURLSREQUEST"]._serialized_start = 1304
+    _globals["_GETINPUTFILEUPLOADURLSREQUEST"]._serialized_end = 1405
+    _globals["_GETINPUTFILEUPLOADURLSRESPONSE"]._serialized_start = 1407
+    _globals["_GETINPUTFILEUPLOADURLSRESPONSE"]._serialized_end = 1478
+    _globals["_GETAVAILABLERESULTFILESREQUEST"]._serialized_start = 1480
+    _globals["_GETAVAILABLERESULTFILESREQUEST"]._serialized_end = 1536
+    _globals["_GETAVAILABLERESULTFILESRESPONSE"]._serialized_start = 1538
+    _globals["_GETAVAILABLERESULTFILESRESPONSE"]._serialized_end = 1643
+    _globals["_GETRESULTFILEURLSREQUEST"]._serialized_start = 1645
+    _globals["_GETRESULTFILEURLSREQUEST"]._serialized_end = 1702
+    _globals["_GETRESULTFILEURLSRESPONSE"]._serialized_start = 1704
+    _globals["_GETRESULTFILEURLSRESPONSE"]._serialized_end = 1770
+    _globals["_BENCHMARKSERVICE"]._serialized_start = 2008
+    _globals["_BENCHMARKSERVICE"]._serialized_end = 2634
+# @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/benchmark_pb2.pyi

@@ -0,0 +1,156 @@
+from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
+from chalk._gen.chalk.common.v1 import online_query_pb2 as _online_query_pb2
+from google.protobuf import duration_pb2 as _duration_pb2
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import (
+    ClassVar as _ClassVar,
+    Iterable as _Iterable,
+    Mapping as _Mapping,
+    Optional as _Optional,
+    Union as _Union,
+)
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class BenchmarkStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    BENCHMARK_STATUS_UNSPECIFIED: _ClassVar[BenchmarkStatus]
+    BENCHMARK_STATUS_QUEUED: _ClassVar[BenchmarkStatus]
+    BENCHMARK_STATUS_WORKING: _ClassVar[BenchmarkStatus]
+    BENCHMARK_STATUS_COMPLETED: _ClassVar[BenchmarkStatus]
+    BENCHMARK_STATUS_FAILED: _ClassVar[BenchmarkStatus]
+    BENCHMARK_STATUS_SKIPPED: _ClassVar[BenchmarkStatus]
+    BENCHMARK_STATUS_CANCELED: _ClassVar[BenchmarkStatus]
+
+BENCHMARK_STATUS_UNSPECIFIED: BenchmarkStatus
+BENCHMARK_STATUS_QUEUED: BenchmarkStatus
+BENCHMARK_STATUS_WORKING: BenchmarkStatus
+BENCHMARK_STATUS_COMPLETED: BenchmarkStatus
+BENCHMARK_STATUS_FAILED: BenchmarkStatus
+BENCHMARK_STATUS_SKIPPED: BenchmarkStatus
+BENCHMARK_STATUS_CANCELED: BenchmarkStatus
+
+class InputFeatures(_message.Message):
+    __slots__ = ("input_features",)
+    INPUT_FEATURES_FIELD_NUMBER: _ClassVar[int]
+    input_features: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, input_features: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class SimpleOnlineQueryBulkRequest(_message.Message):
+    __slots__ = ("input_features", "input_features_list", "input_file", "output_features")
+    INPUT_FEATURES_FIELD_NUMBER: _ClassVar[int]
+    INPUT_FEATURES_LIST_FIELD_NUMBER: _ClassVar[int]
+    INPUT_FILE_FIELD_NUMBER: _ClassVar[int]
+    OUTPUT_FEATURES_FIELD_NUMBER: _ClassVar[int]
+    input_features: _containers.RepeatedScalarFieldContainer[str]
+    input_features_list: InputFeatures
+    input_file: str
+    output_features: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(
+        self,
+        input_features: _Optional[_Iterable[str]] = ...,
+        input_features_list: _Optional[_Union[InputFeatures, _Mapping]] = ...,
+        input_file: _Optional[str] = ...,
+        output_features: _Optional[_Iterable[str]] = ...,
+    ) -> None: ...
+
+class CreateBenchmarkRequest(_message.Message):
+    __slots__ = ("warmup_qps", "warmup_duration", "qps", "duration", "query_bulk", "simple_query_bulk")
+    WARMUP_QPS_FIELD_NUMBER: _ClassVar[int]
+    WARMUP_DURATION_FIELD_NUMBER: _ClassVar[int]
+    QPS_FIELD_NUMBER: _ClassVar[int]
+    DURATION_FIELD_NUMBER: _ClassVar[int]
+    QUERY_BULK_FIELD_NUMBER: _ClassVar[int]
+    SIMPLE_QUERY_BULK_FIELD_NUMBER: _ClassVar[int]
+    warmup_qps: int
+    warmup_duration: _duration_pb2.Duration
+    qps: int
+    duration: _duration_pb2.Duration
+    query_bulk: _online_query_pb2.OnlineQueryBulkRequest
+    simple_query_bulk: SimpleOnlineQueryBulkRequest
+    def __init__(
+        self,
+        warmup_qps: _Optional[int] = ...,
+        warmup_duration: _Optional[_Union[_duration_pb2.Duration, _Mapping]] = ...,
+        qps: _Optional[int] = ...,
+        duration: _Optional[_Union[_duration_pb2.Duration, _Mapping]] = ...,
+        query_bulk: _Optional[_Union[_online_query_pb2.OnlineQueryBulkRequest, _Mapping]] = ...,
+        simple_query_bulk: _Optional[_Union[SimpleOnlineQueryBulkRequest, _Mapping]] = ...,
+    ) -> None: ...
+
+class CreateBenchmarkResponse(_message.Message):
+    __slots__ = ("status", "benchmark_id")
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    BENCHMARK_ID_FIELD_NUMBER: _ClassVar[int]
+    status: BenchmarkStatus
+    benchmark_id: str
+    def __init__(
+        self, status: _Optional[_Union[BenchmarkStatus, str]] = ..., benchmark_id: _Optional[str] = ...
+    ) -> None: ...
+
+class BenchmarkInputFile(_message.Message):
+    __slots__ = ("name", "size", "updated_at")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    SIZE_FIELD_NUMBER: _ClassVar[int]
+    UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    size: int
+    updated_at: _timestamp_pb2.Timestamp
+    def __init__(
+        self,
+        name: _Optional[str] = ...,
+        size: _Optional[int] = ...,
+        updated_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+    ) -> None: ...
+
+class GetAvailableInputFilesRequest(_message.Message):
+    __slots__ = ("prefix",)
+    PREFIX_FIELD_NUMBER: _ClassVar[int]
+    prefix: str
+    def __init__(self, prefix: _Optional[str] = ...) -> None: ...
+
+class GetAvailableInputFilesResponse(_message.Message):
+    __slots__ = ("input_files",)
+    INPUT_FILES_FIELD_NUMBER: _ClassVar[int]
+    input_files: _containers.RepeatedCompositeFieldContainer[BenchmarkInputFile]
+    def __init__(self, input_files: _Optional[_Iterable[_Union[BenchmarkInputFile, _Mapping]]] = ...) -> None: ...
+
+class GetInputFileUploadUrlsRequest(_message.Message):
+    __slots__ = ("input_files",)
+    INPUT_FILES_FIELD_NUMBER: _ClassVar[int]
+    input_files: _containers.RepeatedCompositeFieldContainer[BenchmarkInputFile]
+    def __init__(self, input_files: _Optional[_Iterable[_Union[BenchmarkInputFile, _Mapping]]] = ...) -> None: ...
+
+class GetInputFileUploadUrlsResponse(_message.Message):
+    __slots__ = ("presigned_urls",)
+    PRESIGNED_URLS_FIELD_NUMBER: _ClassVar[int]
+    presigned_urls: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, presigned_urls: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class GetAvailableResultFilesRequest(_message.Message):
+    __slots__ = ("prefix",)
+    PREFIX_FIELD_NUMBER: _ClassVar[int]
+    prefix: str
+    def __init__(self, prefix: _Optional[str] = ...) -> None: ...
+
+class GetAvailableResultFilesResponse(_message.Message):
+    __slots__ = ("result_files",)
+    RESULT_FILES_FIELD_NUMBER: _ClassVar[int]
+    result_files: _containers.RepeatedCompositeFieldContainer[BenchmarkInputFile]
+    def __init__(self, result_files: _Optional[_Iterable[_Union[BenchmarkInputFile, _Mapping]]] = ...) -> None: ...
+
+class GetResultFileUrlsRequest(_message.Message):
+    __slots__ = ("file_names",)
+    FILE_NAMES_FIELD_NUMBER: _ClassVar[int]
+    file_names: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, file_names: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class GetResultFileUrlsResponse(_message.Message):
+    __slots__ = ("presigned_urls",)
+    PRESIGNED_URLS_FIELD_NUMBER: _ClassVar[int]
+    presigned_urls: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, presigned_urls: _Optional[_Iterable[str]] = ...) -> None: ...