sapiopycommons 2025.4.17a489__py3-none-any.whl → 2025.4.23a491__py3-none-any.whl

sapiopycommons/ai/api/plan/proto/step_output_pb2.pyi

@@ -0,0 +1,42 @@
+from google.protobuf.internal import containers as _containers
+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 StepJsonOutputEntry(_message.Message):
+    __slots__ = ("entry",)
+    ENTRY_FIELD_NUMBER: _ClassVar[int]
+    entry: str
+    def __init__(self, entry: _Optional[str] = ...) -> None: ...
+
+class StepTextOutputEntry(_message.Message):
+    __slots__ = ("entry",)
+    ENTRY_FIELD_NUMBER: _ClassVar[int]
+    entry: str
+    def __init__(self, entry: _Optional[str] = ...) -> None: ...
+
+class StepCsvOutputEntry(_message.Message):
+    __slots__ = ("cells",)
+    CELLS_FIELD_NUMBER: _ClassVar[int]
+    cells: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, cells: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class StepBinaryOutputEntry(_message.Message):
+    __slots__ = ("entry",)
+    ENTRY_FIELD_NUMBER: _ClassVar[int]
+    entry: bytes
+    def __init__(self, entry: _Optional[bytes] = ...) -> None: ...
+
+class StepOutputEntry(_message.Message):
+    __slots__ = ("json_output", "text_output", "csv_output", "binary_output")
+    JSON_OUTPUT_FIELD_NUMBER: _ClassVar[int]
+    TEXT_OUTPUT_FIELD_NUMBER: _ClassVar[int]
+    CSV_OUTPUT_FIELD_NUMBER: _ClassVar[int]
+    BINARY_OUTPUT_FIELD_NUMBER: _ClassVar[int]
+    json_output: StepJsonOutputEntry
+    text_output: StepTextOutputEntry
+    csv_output: StepCsvOutputEntry
+    binary_output: StepBinaryOutputEntry
+    def __init__(self, json_output: _Optional[_Union[StepJsonOutputEntry, _Mapping]] = ..., text_output: _Optional[_Union[StepTextOutputEntry, _Mapping]] = ..., csv_output: _Optional[_Union[StepCsvOutputEntry, _Mapping]] = ..., binary_output: _Optional[_Union[StepBinaryOutputEntry, _Mapping]] = ...) -> None: ...

sapiopycommons/ai/api/plan/proto/step_output_pb2_grpc.py

@@ -0,0 +1,24 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+
+GRPC_GENERATED_VERSION = '1.70.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in sapiopycommons/ai/api/plan/proto/step_output_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )

sapiopycommons/ai/api/plan/proto/step_pb2.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: sapiopycommons/ai/api/plan/proto/step.proto
+# Protobuf Python Version: 5.29.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'sapiopycommons/ai/api/plan/proto/step.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#sapiopycommons/ai/api/plan/proto/step.proto\"K\n\nStepIoInfo\x12\x16\n\x0estep_io_number\x18\x01 \x01(\x05\x12\x14\n\x0c\x63ontent_type\x18\x02 \x01(\t\x12\x0f\n\x07io_name\x18\x03 \x01(\t\"l\n\rStepIoDetails\x12!\n\x0cstep_io_info\x18\x01 \x01(\x0b\x32\x0b.StepIoInfo\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x0f\n\x07\x65xample\x18\x03 \x01(\t\x12\x12\n\nvalidation\x18\x04 \x01(\t\"j\n\x10StepInputDetails\x12\'\n\x0fstep_io_details\x18\x01 \x01(\x0b\x32\x0e.StepIoDetails\x12\x18\n\x10paging_supported\x18\x02 \x01(\x08\x12\x13\n\x0bmax_entries\x18\x03 \x01(\x05\"<\n\x11StepOutputDetails\x12\'\n\x0fstep_io_details\x18\x01 \x01(\x0b\x32\x0e.StepIoDetailsB\x1c\n\x18\x63om.velox.api.plan.protoP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'sapiopycommons.ai.api.plan.proto.step_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\030sapiopycommons.ai.api.plan.protoP\001'
+  _globals['_STEPIOINFO']._serialized_start=39
+  _globals['_STEPIOINFO']._serialized_end=114
+  _globals['_STEPIODETAILS']._serialized_start=116
+  _globals['_STEPIODETAILS']._serialized_end=224
+  _globals['_STEPINPUTDETAILS']._serialized_start=226
+  _globals['_STEPINPUTDETAILS']._serialized_end=332
+  _globals['_STEPOUTPUTDETAILS']._serialized_start=334
+  _globals['_STEPOUTPUTDETAILS']._serialized_end=394
+# @@protoc_insertion_point(module_scope)

sapiopycommons/ai/api/plan/proto/step_pb2.pyi

@@ -0,0 +1,43 @@
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class StepIoInfo(_message.Message):
+    __slots__ = ("step_io_number", "content_type", "io_name")
+    STEP_IO_NUMBER_FIELD_NUMBER: _ClassVar[int]
+    CONTENT_TYPE_FIELD_NUMBER: _ClassVar[int]
+    IO_NAME_FIELD_NUMBER: _ClassVar[int]
+    step_io_number: int
+    content_type: str
+    io_name: str
+    def __init__(self, step_io_number: _Optional[int] = ..., content_type: _Optional[str] = ..., io_name: _Optional[str] = ...) -> None: ...
+
+class StepIoDetails(_message.Message):
+    __slots__ = ("step_io_info", "description", "example", "validation")
+    STEP_IO_INFO_FIELD_NUMBER: _ClassVar[int]
+    DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
+    EXAMPLE_FIELD_NUMBER: _ClassVar[int]
+    VALIDATION_FIELD_NUMBER: _ClassVar[int]
+    step_io_info: StepIoInfo
+    description: str
+    example: str
+    validation: str
+    def __init__(self, step_io_info: _Optional[_Union[StepIoInfo, _Mapping]] = ..., description: _Optional[str] = ..., example: _Optional[str] = ..., validation: _Optional[str] = ...) -> None: ...
+
+class StepInputDetails(_message.Message):
+    __slots__ = ("step_io_details", "paging_supported", "max_entries")
+    STEP_IO_DETAILS_FIELD_NUMBER: _ClassVar[int]
+    PAGING_SUPPORTED_FIELD_NUMBER: _ClassVar[int]
+    MAX_ENTRIES_FIELD_NUMBER: _ClassVar[int]
+    step_io_details: StepIoDetails
+    paging_supported: bool
+    max_entries: int
+    def __init__(self, step_io_details: _Optional[_Union[StepIoDetails, _Mapping]] = ..., paging_supported: bool = ..., max_entries: _Optional[int] = ...) -> None: ...
+
+class StepOutputDetails(_message.Message):
+    __slots__ = ("step_io_details",)
+    STEP_IO_DETAILS_FIELD_NUMBER: _ClassVar[int]
+    step_io_details: StepIoDetails
+    def __init__(self, step_io_details: _Optional[_Union[StepIoDetails, _Mapping]] = ...) -> None: ...

sapiopycommons/ai/api/plan/proto/step_pb2_grpc.py

@@ -0,0 +1,24 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+
+GRPC_GENERATED_VERSION = '1.70.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in sapiopycommons/ai/api/plan/proto/step_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )

sapiopycommons/ai/api/plan/script/proto/script_pb2.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: sapiopycommons/ai/api/plan/script/proto/script.proto
+# Protobuf Python Version: 5.29.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'sapiopycommons/ai/api/plan/script/proto/script.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from sapiopycommons.ai.api.plan.tool.proto import entry_pb2 as com_dot_velox_dot_api_dot_plan_dot_tool_dot_proto_dot_entry__pb2
+from sapiopycommons.ai.api.plan.proto import step_pb2 as com_dot_velox_dot_api_dot_plan_dot_proto_dot_step__pb2
+from sapiopycommons.ai.api.session.proto import sapio_conn_info_pb2 as com_dot_velox_dot_api_dot_session_dot_proto_dot_sapio__conn__info__pb2
+
+from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import *
+from sapiopycommons.ai.api.plan.proto.step_pb2 import *
+from sapiopycommons.ai.api.session.proto.sapio_conn_info_pb2 import *
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,sapiopycommons/ai/api/plan/script/proto/script.proto\x1a)sapiopycommons/ai/api/plan/tool/proto/entry.proto\x1a#sapiopycommons/ai/api/plan/proto/step.proto\x1a\x31\x63om/velox/api/session/proto/sapio_conn_info.proto\"\xf0\x02\n\x16\x43reateScriptJobRequest\x12(\n\nsapio_user\x18\x01 \x01(\x0b\x32\x14.SapioConnectionInfo\x12\x17\n\x0fscript_language\x18\x02 \x01(\t\x12\x18\n\x10plan_instance_id\x18\x03 \x01(\x03\x12\x18\n\x10step_instance_id\x18\x04 \x01(\x03\x12\x15\n\rinvocation_id\x18\x05 \x01(\x03\x12\"\n\rinput_configs\x18\x06 \x03(\x0b\x32\x0b.StepIoInfo\x12#\n\x0eoutput_configs\x18\x07 \x03(\x0b\x32\x0b.StepIoInfo\x12\x0f\n\x06script\x18\x80\x08 \x01(\t\x12\x10\n\x07timeout\x18\x81\x08 \x01(\x05\x12\x16\n\rmax_memory_mb\x18\x82\x08 \x01(\x05\x12\x1a\n\x11working_directory\x18\x83\x08 \x01(\t\x12(\n\nentry_data\x18\xff\x0f \x03(\x0b\x32\x13.StepEntryInputData\")\n\x17\x43reateScriptJobResponse\x12\x0e\n\x06job_id\x18\x01 \x01(\t\"3\n\rGetJobRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12\x12\n\nlog_offset\x18\x02 \x01(\x05\"w\n\x0eGetJobResponse\x12\x1a\n\x06status\x18\x01 \x01(\x0e\x32\n.JobStatus\x12\x0b\n\x03log\x18\x02 \x01(\t\x12\x11\n\texception\x18\x03 \x01(\t\x12)\n\nentry_data\x18\xff\x0f \x03(\x0b\x32\x14.StepEntryOutputData*@\n\tJobStatus\x12\x0b\n\x07PENDING\x10\x00\x12\x0b\n\x07RUNNING\x10\x01\x12\r\n\tCOMPLETED\x10\x02\x12\n\n\x06\x46\x41ILED\x10\x03\x32\x80\x01\n\rScriptService\x12\x44\n\x0f\x43reateScriptJob\x12\x17.CreateScriptJobRequest\x1a\x18.CreateScriptJobResponse\x12)\n\x06GetJob\x12\x0e.GetJobRequest\x1a\x0f.GetJobResponseB!\n\x1f\x63om.velox.api.plan.script.protoP\x00P\x01P\x02\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'sapiopycommons.ai.api.plan.script.proto.script_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\037sapiopycommons.ai.api.plan.script.proto'
+  _globals['_JOBSTATUS']._serialized_start=767
+  _globals['_JOBSTATUS']._serialized_end=831
+  _globals['_CREATESCRIPTJOBREQUEST']._serialized_start=180
+  _globals['_CREATESCRIPTJOBREQUEST']._serialized_end=548
+  _globals['_CREATESCRIPTJOBRESPONSE']._serialized_start=550
+  _globals['_CREATESCRIPTJOBRESPONSE']._serialized_end=591
+  _globals['_GETJOBREQUEST']._serialized_start=593
+  _globals['_GETJOBREQUEST']._serialized_end=644
+  _globals['_GETJOBRESPONSE']._serialized_start=646
+  _globals['_GETJOBRESPONSE']._serialized_end=765
+  _globals['_SCRIPTSERVICE']._serialized_start=834
+  _globals['_SCRIPTSERVICE']._serialized_end=962
+# @@protoc_insertion_point(module_scope)

sapiopycommons/ai/api/plan/script/proto/script_pb2.pyi

@@ -0,0 +1,99 @@
+from sapiopycommons.ai.api.plan.tool.proto import entry_pb2 as _entry_pb2
+from sapiopycommons.ai.api.plan.proto import step_pb2 as _step_pb2
+from sapiopycommons.ai.api.session.proto import sapio_conn_info_pb2 as _sapio_conn_info_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
+from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepCsvHeaderRow as StepCsvHeaderRow
+from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepCsvRow as StepCsvRow
+from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepCsvData as StepCsvData
+from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepJsonData as StepJsonData
+from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepTextData as StepTextData
+from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepBinaryData as StepBinaryData
+from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepImageData as StepImageData
+from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepEntryData as StepEntryData
+from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepEntryInputData as StepEntryInputData
+from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepEntryOutputData as StepEntryOutputData
+from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import DataType as DataType
+from sapiopycommons.ai.api.plan.proto.step_pb2 import StepIoInfo as StepIoInfo
+from sapiopycommons.ai.api.plan.proto.step_pb2 import StepIoDetails as StepIoDetails
+from sapiopycommons.ai.api.plan.proto.step_pb2 import StepInputDetails as StepInputDetails
+from sapiopycommons.ai.api.plan.proto.step_pb2 import StepOutputDetails as StepOutputDetails
+from sapiopycommons.ai.api.session.proto.sapio_conn_info_pb2 import SapioConnectionInfo as SapioConnectionInfo
+from sapiopycommons.ai.api.session.proto.sapio_conn_info_pb2 import SapioUserSecretType as SapioUserSecretType
+
+DESCRIPTOR: _descriptor.FileDescriptor
+BINARY: _entry_pb2.DataType
+JSON: _entry_pb2.DataType
+CSV: _entry_pb2.DataType
+TEXT: _entry_pb2.DataType
+IMAGE: _entry_pb2.DataType
+SESSION_TOKEN: _sapio_conn_info_pb2.SapioUserSecretType
+PASSWORD: _sapio_conn_info_pb2.SapioUserSecretType
+
+class JobStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    PENDING: _ClassVar[JobStatus]
+    RUNNING: _ClassVar[JobStatus]
+    COMPLETED: _ClassVar[JobStatus]
+    FAILED: _ClassVar[JobStatus]
+PENDING: JobStatus
+RUNNING: JobStatus
+COMPLETED: JobStatus
+FAILED: JobStatus
+
+class CreateScriptJobRequest(_message.Message):
+    __slots__ = ("sapio_user", "script_language", "plan_instance_id", "step_instance_id", "invocation_id", "input_configs", "output_configs", "script", "timeout", "max_memory_mb", "working_directory", "entry_data")
+    SAPIO_USER_FIELD_NUMBER: _ClassVar[int]
+    SCRIPT_LANGUAGE_FIELD_NUMBER: _ClassVar[int]
+    PLAN_INSTANCE_ID_FIELD_NUMBER: _ClassVar[int]
+    STEP_INSTANCE_ID_FIELD_NUMBER: _ClassVar[int]
+    INVOCATION_ID_FIELD_NUMBER: _ClassVar[int]
+    INPUT_CONFIGS_FIELD_NUMBER: _ClassVar[int]
+    OUTPUT_CONFIGS_FIELD_NUMBER: _ClassVar[int]
+    SCRIPT_FIELD_NUMBER: _ClassVar[int]
+    TIMEOUT_FIELD_NUMBER: _ClassVar[int]
+    MAX_MEMORY_MB_FIELD_NUMBER: _ClassVar[int]
+    WORKING_DIRECTORY_FIELD_NUMBER: _ClassVar[int]
+    ENTRY_DATA_FIELD_NUMBER: _ClassVar[int]
+    sapio_user: _sapio_conn_info_pb2.SapioConnectionInfo
+    script_language: str
+    plan_instance_id: int
+    step_instance_id: int
+    invocation_id: int
+    input_configs: _containers.RepeatedCompositeFieldContainer[_step_pb2.StepIoInfo]
+    output_configs: _containers.RepeatedCompositeFieldContainer[_step_pb2.StepIoInfo]
+    script: str
+    timeout: int
+    max_memory_mb: int
+    working_directory: str
+    entry_data: _containers.RepeatedCompositeFieldContainer[_entry_pb2.StepEntryInputData]
+    def __init__(self, sapio_user: _Optional[_Union[_sapio_conn_info_pb2.SapioConnectionInfo, _Mapping]] = ..., script_language: _Optional[str] = ..., plan_instance_id: _Optional[int] = ..., step_instance_id: _Optional[int] = ..., invocation_id: _Optional[int] = ..., input_configs: _Optional[_Iterable[_Union[_step_pb2.StepIoInfo, _Mapping]]] = ..., output_configs: _Optional[_Iterable[_Union[_step_pb2.StepIoInfo, _Mapping]]] = ..., script: _Optional[str] = ..., timeout: _Optional[int] = ..., max_memory_mb: _Optional[int] = ..., working_directory: _Optional[str] = ..., entry_data: _Optional[_Iterable[_Union[_entry_pb2.StepEntryInputData, _Mapping]]] = ...) -> None: ...
+
+class CreateScriptJobResponse(_message.Message):
+    __slots__ = ("job_id",)
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    job_id: str
+    def __init__(self, job_id: _Optional[str] = ...) -> None: ...
+
+class GetJobRequest(_message.Message):
+    __slots__ = ("job_id", "log_offset")
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    LOG_OFFSET_FIELD_NUMBER: _ClassVar[int]
+    job_id: str
+    log_offset: int
+    def __init__(self, job_id: _Optional[str] = ..., log_offset: _Optional[int] = ...) -> None: ...
+
+class GetJobResponse(_message.Message):
+    __slots__ = ("status", "log", "exception", "entry_data")
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    LOG_FIELD_NUMBER: _ClassVar[int]
+    EXCEPTION_FIELD_NUMBER: _ClassVar[int]
+    ENTRY_DATA_FIELD_NUMBER: _ClassVar[int]
+    status: JobStatus
+    log: str
+    exception: str
+    entry_data: _containers.RepeatedCompositeFieldContainer[_entry_pb2.StepEntryOutputData]
+    def __init__(self, status: _Optional[_Union[JobStatus, str]] = ..., log: _Optional[str] = ..., exception: _Optional[str] = ..., entry_data: _Optional[_Iterable[_Union[_entry_pb2.StepEntryOutputData, _Mapping]]] = ...) -> None: ...

sapiopycommons/ai/api/plan/script/proto/script_pb2_grpc.py

@@ -0,0 +1,153 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+from sapiopycommons.ai.api.plan.script.proto import script_pb2 as com_dot_velox_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2
+
+GRPC_GENERATED_VERSION = '1.70.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in sapiopycommons/ai/api/plan/script/proto/script_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
+
+class ScriptServiceStub(object):
+    """
+    This is the generic service interface that all tools must implement. We will use hostname routing to route the request to the correct tool.
+    The other option is for us to use gRPC metadata to route the request to the correct tool. We don't want to expose a bunch of ports and that's
+    a pain to manage. So we will use hostname routing for now.
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.CreateScriptJob = channel.unary_unary(
+                '/ScriptService/CreateScriptJob',
+                request_serializer=com_dot_velox_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.CreateScriptJobRequest.SerializeToString,
+                response_deserializer=com_dot_velox_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.CreateScriptJobResponse.FromString,
+                _registered_method=True)
+        self.GetJob = channel.unary_unary(
+                '/ScriptService/GetJob',
+                request_serializer=com_dot_velox_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.GetJobRequest.SerializeToString,
+                response_deserializer=com_dot_velox_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.GetJobResponse.FromString,
+                _registered_method=True)
+
+
+class ScriptServiceServicer(object):
+    """
+    This is the generic service interface that all tools must implement. We will use hostname routing to route the request to the correct tool.
+    The other option is for us to use gRPC metadata to route the request to the correct tool. We don't want to expose a bunch of ports and that's
+    a pain to manage. So we will use hostname routing for now.
+    """
+
+    def CreateScriptJob(self, request, context):
+        """The main entry point for script
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetJob(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_ScriptServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'CreateScriptJob': grpc.unary_unary_rpc_method_handler(
+                    servicer.CreateScriptJob,
+                    request_deserializer=com_dot_velox_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.CreateScriptJobRequest.FromString,
+                    response_serializer=com_dot_velox_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.CreateScriptJobResponse.SerializeToString,
+            ),
+            'GetJob': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetJob,
+                    request_deserializer=com_dot_velox_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.GetJobRequest.FromString,
+                    response_serializer=com_dot_velox_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.GetJobResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'ScriptService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('ScriptService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class ScriptService(object):
+    """
+    This is the generic service interface that all tools must implement. We will use hostname routing to route the request to the correct tool.
+    The other option is for us to use gRPC metadata to route the request to the correct tool. We don't want to expose a bunch of ports and that's
+    a pain to manage. So we will use hostname routing for now.
+    """
+
+    @staticmethod
+    def CreateScriptJob(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,
+            '/ScriptService/CreateScriptJob',
+            com_dot_velox_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.CreateScriptJobRequest.SerializeToString,
+            com_dot_velox_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.CreateScriptJobResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def GetJob(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,
+            '/ScriptService/GetJob',
+            com_dot_velox_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.GetJobRequest.SerializeToString,
+            com_dot_velox_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.GetJobResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

sapiopycommons/ai/api/plan/tool/proto/entry_pb2.py

@@ -0,0 +1,57 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: sapiopycommons/ai/api/plan/tool/proto/entry.proto
+# Protobuf Python Version: 5.29.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'sapiopycommons/ai/api/plan/tool/proto/entry.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)sapiopycommons/ai/api/plan/tool/proto/entry.proto\"!\n\x10StepCsvHeaderRow\x12\r\n\x05\x63\x65lls\x18\x01 \x03(\t\"\x1b\n\nStepCsvRow\x12\r\n\x05\x63\x65lls\x18\x01 \x03(\t\"N\n\x0bStepCsvData\x12!\n\x06header\x18\x0e \x01(\x0b\x32\x11.StepCsvHeaderRow\x12\x1c\n\x07\x65ntries\x18\x0f \x03(\x0b\x32\x0b.StepCsvRow\"\x1f\n\x0cStepJsonData\x12\x0f\n\x07\x65ntries\x18\x0f \x03(\t\"\x1f\n\x0cStepTextData\x12\x0f\n\x07\x65ntries\x18\x0f \x03(\t\"!\n\x0eStepBinaryData\x12\x0f\n\x07\x65ntries\x18\x0f \x03(\x0c\"6\n\rStepImageData\x12\x14\n\x0cimage_format\x18\x01 \x01(\t\x12\x0f\n\x07\x65ntries\x18\x0f \x03(\x0c\"\xf1\x01\n\rStepEntryData\x12\x1b\n\x08\x64\x61taType\x18\x01 \x01(\x0e\x32\t.DataType\x12\'\n\x0b\x62inary_data\x18\xff\x0f \x01(\x0b\x32\x0f.StepBinaryDataH\x00\x12!\n\x08\x63sv_data\x18\xfe\x0f \x01(\x0b\x32\x0c.StepCsvDataH\x00\x12#\n\tjson_data\x18\xfd\x0f \x01(\x0b\x32\r.StepJsonDataH\x00\x12#\n\ttext_data\x18\xfc\x0f \x01(\x0b\x32\r.StepTextDataH\x00\x12%\n\nimage_data\x18\xfb\x0f \x01(\x0b\x32\x0e.StepImageDataH\x00\x42\x06\n\x04\x64\x61ta\"L\n\x12StepEntryInputData\x12\x12\n\nis_partial\x18\x02 \x01(\x08\x12\"\n\nentry_data\x18\x01 \x01(\x0b\x32\x0e.StepEntryData\"9\n\x13StepEntryOutputData\x12\"\n\nentry_data\x18\x01 \x01(\x0b\x32\x0e.StepEntryData*>\n\x08\x44\x61taType\x12\n\n\x06\x42INARY\x10\x00\x12\x08\n\x04JSON\x10\x01\x12\x07\n\x03\x43SV\x10\x02\x12\x08\n\x04TEXT\x10\x03\x12\t\n\x05IMAGE\x10\x04\x42!\n\x1d\x63om.velox.api.plan.tool.protoP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'sapiopycommons.ai.api.plan.tool.proto.entry_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\035sapiopycommons.ai.api.plan.tool.protoP\001'
+  _globals['_DATATYPE']._serialized_start=727
+  _globals['_DATATYPE']._serialized_end=789
+  _globals['_STEPCSVHEADERROW']._serialized_start=45
+  _globals['_STEPCSVHEADERROW']._serialized_end=78
+  _globals['_STEPCSVROW']._serialized_start=80
+  _globals['_STEPCSVROW']._serialized_end=107
+  _globals['_STEPCSVDATA']._serialized_start=109
+  _globals['_STEPCSVDATA']._serialized_end=187
+  _globals['_STEPJSONDATA']._serialized_start=189
+  _globals['_STEPJSONDATA']._serialized_end=220
+  _globals['_STEPTEXTDATA']._serialized_start=222
+  _globals['_STEPTEXTDATA']._serialized_end=253
+  _globals['_STEPBINARYDATA']._serialized_start=255
+  _globals['_STEPBINARYDATA']._serialized_end=288
+  _globals['_STEPIMAGEDATA']._serialized_start=290
+  _globals['_STEPIMAGEDATA']._serialized_end=344
+  _globals['_STEPENTRYDATA']._serialized_start=347
+  _globals['_STEPENTRYDATA']._serialized_end=588
+  _globals['_STEPENTRYINPUTDATA']._serialized_start=590
+  _globals['_STEPENTRYINPUTDATA']._serialized_end=666
+  _globals['_STEPENTRYOUTPUTDATA']._serialized_start=668
+  _globals['_STEPENTRYOUTPUTDATA']._serialized_end=725
+# @@protoc_insertion_point(module_scope)

sapiopycommons/ai/api/plan/tool/proto/entry_pb2.pyi

@@ -0,0 +1,96 @@
+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 DataType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    BINARY: _ClassVar[DataType]
+    JSON: _ClassVar[DataType]
+    CSV: _ClassVar[DataType]
+    TEXT: _ClassVar[DataType]
+    IMAGE: _ClassVar[DataType]
+BINARY: DataType
+JSON: DataType
+CSV: DataType
+TEXT: DataType
+IMAGE: DataType
+
+class StepCsvHeaderRow(_message.Message):
+    __slots__ = ("cells",)
+    CELLS_FIELD_NUMBER: _ClassVar[int]
+    cells: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, cells: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class StepCsvRow(_message.Message):
+    __slots__ = ("cells",)
+    CELLS_FIELD_NUMBER: _ClassVar[int]
+    cells: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, cells: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class StepCsvData(_message.Message):
+    __slots__ = ("header", "entries")
+    HEADER_FIELD_NUMBER: _ClassVar[int]
+    ENTRIES_FIELD_NUMBER: _ClassVar[int]
+    header: StepCsvHeaderRow
+    entries: _containers.RepeatedCompositeFieldContainer[StepCsvRow]
+    def __init__(self, header: _Optional[_Union[StepCsvHeaderRow, _Mapping]] = ..., entries: _Optional[_Iterable[_Union[StepCsvRow, _Mapping]]] = ...) -> None: ...
+
+class StepJsonData(_message.Message):
+    __slots__ = ("entries",)
+    ENTRIES_FIELD_NUMBER: _ClassVar[int]
+    entries: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, entries: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class StepTextData(_message.Message):
+    __slots__ = ("entries",)
+    ENTRIES_FIELD_NUMBER: _ClassVar[int]
+    entries: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, entries: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class StepBinaryData(_message.Message):
+    __slots__ = ("entries",)
+    ENTRIES_FIELD_NUMBER: _ClassVar[int]
+    entries: _containers.RepeatedScalarFieldContainer[bytes]
+    def __init__(self, entries: _Optional[_Iterable[bytes]] = ...) -> None: ...
+
+class StepImageData(_message.Message):
+    __slots__ = ("image_format", "entries")
+    IMAGE_FORMAT_FIELD_NUMBER: _ClassVar[int]
+    ENTRIES_FIELD_NUMBER: _ClassVar[int]
+    image_format: str
+    entries: _containers.RepeatedScalarFieldContainer[bytes]
+    def __init__(self, image_format: _Optional[str] = ..., entries: _Optional[_Iterable[bytes]] = ...) -> None: ...
+
+class StepEntryData(_message.Message):
+    __slots__ = ("dataType", "binary_data", "csv_data", "json_data", "text_data", "image_data")
+    DATATYPE_FIELD_NUMBER: _ClassVar[int]
+    BINARY_DATA_FIELD_NUMBER: _ClassVar[int]
+    CSV_DATA_FIELD_NUMBER: _ClassVar[int]
+    JSON_DATA_FIELD_NUMBER: _ClassVar[int]
+    TEXT_DATA_FIELD_NUMBER: _ClassVar[int]
+    IMAGE_DATA_FIELD_NUMBER: _ClassVar[int]
+    dataType: DataType
+    binary_data: StepBinaryData
+    csv_data: StepCsvData
+    json_data: StepJsonData
+    text_data: StepTextData
+    image_data: StepImageData
+    def __init__(self, dataType: _Optional[_Union[DataType, str]] = ..., binary_data: _Optional[_Union[StepBinaryData, _Mapping]] = ..., csv_data: _Optional[_Union[StepCsvData, _Mapping]] = ..., json_data: _Optional[_Union[StepJsonData, _Mapping]] = ..., text_data: _Optional[_Union[StepTextData, _Mapping]] = ..., image_data: _Optional[_Union[StepImageData, _Mapping]] = ...) -> None: ...
+
+class StepEntryInputData(_message.Message):
+    __slots__ = ("is_partial", "entry_data")
+    IS_PARTIAL_FIELD_NUMBER: _ClassVar[int]
+    ENTRY_DATA_FIELD_NUMBER: _ClassVar[int]
+    is_partial: bool
+    entry_data: StepEntryData
+    def __init__(self, is_partial: bool = ..., entry_data: _Optional[_Union[StepEntryData, _Mapping]] = ...) -> None: ...
+
+class StepEntryOutputData(_message.Message):
+    __slots__ = ("entry_data",)
+    ENTRY_DATA_FIELD_NUMBER: _ClassVar[int]
+    entry_data: StepEntryData
+    def __init__(self, entry_data: _Optional[_Union[StepEntryData, _Mapping]] = ...) -> None: ...