sapiopycommons 2025.4.17a488__py3-none-any.whl → 2025.4.22a490__py3-none-any.whl

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: ...

sapiopycommons/ai/api/plan/tool/proto/entry_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.71.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 com/velox/api/plan/tool/proto/entry_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/tool/proto/tool_pb2.py

@@ -0,0 +1,61 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: com/velox/api/plan/tool/proto/tool.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,
+    '',
+    'com/velox/api/plan/tool/proto/tool.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(com/velox/api/plan/tool/proto/tool.proto\x1a)com/velox/api/plan/tool/proto/entry.proto\x1a#com/velox/api/plan/proto/step.proto\x1a\x31\x63om/velox/api/session/proto/sapio_conn_info.proto\"\x8f\x02\n\x12ProcessStepRequest\x12(\n\nsapio_user\x18\x01 \x01(\x0b\x32\x14.SapioConnectionInfo\x12\x11\n\ttool_name\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(\n\nentry_data\x18\xff\x0f \x03(\x0b\x32\x13.StepEntryInputData\"q\n\x13ProcessStepResponse\x12!\n\x0bnew_records\x18\xfd\x0f \x03(\x0b\x32\x0b.StepRecord\x12\x0c\n\x03log\x18\xfe\x0f \x03(\t\x12)\n\nentry_data\x18\xff\x0f \x03(\x0b\x32\x14.StepEntryOutputData\"C\n\x12ToolDetailsRequest\x12-\n\x0fsapio_conn_info\x18\x01 \x01(\x0b\x32\x14.SapioConnectionInfo\"z\n\x14StepRecordFieldValue\x12\x16\n\x0cstring_value\x18\x0f \x01(\tH\x00\x12\x13\n\tint_value\x18\x0e \x01(\x03H\x00\x12\x16\n\x0c\x64ouble_value\x18\r \x01(\x01H\x00\x12\x14\n\nbool_value\x18\x0c \x01(\x08H\x00\x42\x07\n\x05\x66ield\"{\n\nStepRecord\x12\'\n\x06\x66ields\x18\x01 \x03(\x0b\x32\x17.StepRecord.FieldsEntry\x1a\x44\n\x0b\x46ieldsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12$\n\x05value\x18\x02 \x01(\x0b\x32\x15.StepRecordFieldValue:\x02\x38\x01\"\x7f\n\x0bToolDetails\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12%\n\rinput_configs\x18\x04 \x03(\x0b\x32\x0e.StepIoDetails\x12&\n\x0eoutput_configs\x18\x05 \x03(\x0b\x32\x0e.StepIoDetails\"Y\n\x13ToolDetailsResponse\x12\x1e\n\x16tool_framework_version\x18\x01 \x01(\x05\x12\"\n\x0ctool_details\x18\x02 \x03(\x0b\x32\x0c.ToolDetails2\x84\x01\n\x0bToolService\x12;\n\x0eGetToolDetails\x12\x13.ToolDetailsRequest\x1a\x14.ToolDetailsResponse\x12\x38\n\x0bProcessData\x12\x13.ProcessStepRequest\x1a\x14.ProcessStepResponseB!\n\x1d\x63om.velox.api.plan.tool.protoP\x01P\x00P\x01P\x02\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'sapiopycommons.ai.api.plan.tool.proto.tool_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['_STEPRECORD_FIELDSENTRY']._loaded_options = None
+  _globals['_STEPRECORD_FIELDSENTRY']._serialized_options = b'8\001'
+  _globals['_PROCESSSTEPREQUEST']._serialized_start=176
+  _globals['_PROCESSSTEPREQUEST']._serialized_end=447
+  _globals['_PROCESSSTEPRESPONSE']._serialized_start=449
+  _globals['_PROCESSSTEPRESPONSE']._serialized_end=562
+  _globals['_TOOLDETAILSREQUEST']._serialized_start=564
+  _globals['_TOOLDETAILSREQUEST']._serialized_end=631
+  _globals['_STEPRECORDFIELDVALUE']._serialized_start=633
+  _globals['_STEPRECORDFIELDVALUE']._serialized_end=755
+  _globals['_STEPRECORD']._serialized_start=757
+  _globals['_STEPRECORD']._serialized_end=880
+  _globals['_STEPRECORD_FIELDSENTRY']._serialized_start=812
+  _globals['_STEPRECORD_FIELDSENTRY']._serialized_end=880
+  _globals['_TOOLDETAILS']._serialized_start=882
+  _globals['_TOOLDETAILS']._serialized_end=1009
+  _globals['_TOOLDETAILSRESPONSE']._serialized_start=1011
+  _globals['_TOOLDETAILSRESPONSE']._serialized_end=1100
+  _globals['_TOOLSERVICE']._serialized_start=1103
+  _globals['_TOOLSERVICE']._serialized_end=1235
+# @@protoc_insertion_point(module_scope)

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

@@ -0,0 +1,114 @@
+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 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 ProcessStepRequest(_message.Message):
+    __slots__ = ("sapio_user", "tool_name", "plan_instance_id", "step_instance_id", "invocation_id", "input_configs", "output_configs", "entry_data")
+    SAPIO_USER_FIELD_NUMBER: _ClassVar[int]
+    TOOL_NAME_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]
+    ENTRY_DATA_FIELD_NUMBER: _ClassVar[int]
+    sapio_user: _sapio_conn_info_pb2.SapioConnectionInfo
+    tool_name: 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]
+    entry_data: _containers.RepeatedCompositeFieldContainer[_entry_pb2.StepEntryInputData]
+    def __init__(self, sapio_user: _Optional[_Union[_sapio_conn_info_pb2.SapioConnectionInfo, _Mapping]] = ..., tool_name: _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]]] = ..., entry_data: _Optional[_Iterable[_Union[_entry_pb2.StepEntryInputData, _Mapping]]] = ...) -> None: ...
+
+class ProcessStepResponse(_message.Message):
+    __slots__ = ("new_records", "log", "entry_data")
+    NEW_RECORDS_FIELD_NUMBER: _ClassVar[int]
+    LOG_FIELD_NUMBER: _ClassVar[int]
+    ENTRY_DATA_FIELD_NUMBER: _ClassVar[int]
+    new_records: _containers.RepeatedCompositeFieldContainer[StepRecord]
+    log: _containers.RepeatedScalarFieldContainer[str]
+    entry_data: _containers.RepeatedCompositeFieldContainer[_entry_pb2.StepEntryOutputData]
+    def __init__(self, new_records: _Optional[_Iterable[_Union[StepRecord, _Mapping]]] = ..., log: _Optional[_Iterable[str]] = ..., entry_data: _Optional[_Iterable[_Union[_entry_pb2.StepEntryOutputData, _Mapping]]] = ...) -> None: ...
+
+class ToolDetailsRequest(_message.Message):
+    __slots__ = ("sapio_conn_info",)
+    SAPIO_CONN_INFO_FIELD_NUMBER: _ClassVar[int]
+    sapio_conn_info: _sapio_conn_info_pb2.SapioConnectionInfo
+    def __init__(self, sapio_conn_info: _Optional[_Union[_sapio_conn_info_pb2.SapioConnectionInfo, _Mapping]] = ...) -> None: ...
+
+class StepRecordFieldValue(_message.Message):
+    __slots__ = ("string_value", "int_value", "double_value", "bool_value")
+    STRING_VALUE_FIELD_NUMBER: _ClassVar[int]
+    INT_VALUE_FIELD_NUMBER: _ClassVar[int]
+    DOUBLE_VALUE_FIELD_NUMBER: _ClassVar[int]
+    BOOL_VALUE_FIELD_NUMBER: _ClassVar[int]
+    string_value: str
+    int_value: int
+    double_value: float
+    bool_value: bool
+    def __init__(self, string_value: _Optional[str] = ..., int_value: _Optional[int] = ..., double_value: _Optional[float] = ..., bool_value: bool = ...) -> None: ...
+
+class StepRecord(_message.Message):
+    __slots__ = ("fields",)
+    class FieldsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: StepRecordFieldValue
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[StepRecordFieldValue, _Mapping]] = ...) -> None: ...
+    FIELDS_FIELD_NUMBER: _ClassVar[int]
+    fields: _containers.MessageMap[str, StepRecordFieldValue]
+    def __init__(self, fields: _Optional[_Mapping[str, StepRecordFieldValue]] = ...) -> None: ...
+
+class ToolDetails(_message.Message):
+    __slots__ = ("name", "description", "input_configs", "output_configs")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
+    INPUT_CONFIGS_FIELD_NUMBER: _ClassVar[int]
+    OUTPUT_CONFIGS_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    description: str
+    input_configs: _containers.RepeatedCompositeFieldContainer[_step_pb2.StepIoDetails]
+    output_configs: _containers.RepeatedCompositeFieldContainer[_step_pb2.StepIoDetails]
+    def __init__(self, name: _Optional[str] = ..., description: _Optional[str] = ..., input_configs: _Optional[_Iterable[_Union[_step_pb2.StepIoDetails, _Mapping]]] = ..., output_configs: _Optional[_Iterable[_Union[_step_pb2.StepIoDetails, _Mapping]]] = ...) -> None: ...
+
+class ToolDetailsResponse(_message.Message):
+    __slots__ = ("tool_framework_version", "tool_details")
+    TOOL_FRAMEWORK_VERSION_FIELD_NUMBER: _ClassVar[int]
+    TOOL_DETAILS_FIELD_NUMBER: _ClassVar[int]
+    tool_framework_version: int
+    tool_details: _containers.RepeatedCompositeFieldContainer[ToolDetails]
+    def __init__(self, tool_framework_version: _Optional[int] = ..., tool_details: _Optional[_Iterable[_Union[ToolDetails, _Mapping]]] = ...) -> None: ...

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

@@ -0,0 +1,154 @@
+# 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.tool.proto import tool_pb2 as com_dot_velox_dot_api_dot_plan_dot_tool_dot_proto_dot_tool__pb2
+
+GRPC_GENERATED_VERSION = '1.71.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 com/velox/api/plan/tool/proto/tool_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 ToolServiceStub(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.GetToolDetails = channel.unary_unary(
+                '/ToolService/GetToolDetails',
+                request_serializer=com_dot_velox_dot_api_dot_plan_dot_tool_dot_proto_dot_tool__pb2.ToolDetailsRequest.SerializeToString,
+                response_deserializer=com_dot_velox_dot_api_dot_plan_dot_tool_dot_proto_dot_tool__pb2.ToolDetailsResponse.FromString,
+                _registered_method=True)
+        self.ProcessData = channel.unary_unary(
+                '/ToolService/ProcessData',
+                request_serializer=com_dot_velox_dot_api_dot_plan_dot_tool_dot_proto_dot_tool__pb2.ProcessStepRequest.SerializeToString,
+                response_deserializer=com_dot_velox_dot_api_dot_plan_dot_tool_dot_proto_dot_tool__pb2.ProcessStepResponse.FromString,
+                _registered_method=True)
+
+
+class ToolServiceServicer(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 GetToolDetails(self, request, context):
+        """This allows for a tool to self-describe itself, which allows for the UI to be automatically populated with this information
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ProcessData(self, request, context):
+        """This is the main entry point for a tool to be called
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_ToolServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'GetToolDetails': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetToolDetails,
+                    request_deserializer=com_dot_velox_dot_api_dot_plan_dot_tool_dot_proto_dot_tool__pb2.ToolDetailsRequest.FromString,
+                    response_serializer=com_dot_velox_dot_api_dot_plan_dot_tool_dot_proto_dot_tool__pb2.ToolDetailsResponse.SerializeToString,
+            ),
+            'ProcessData': grpc.unary_unary_rpc_method_handler(
+                    servicer.ProcessData,
+                    request_deserializer=com_dot_velox_dot_api_dot_plan_dot_tool_dot_proto_dot_tool__pb2.ProcessStepRequest.FromString,
+                    response_serializer=com_dot_velox_dot_api_dot_plan_dot_tool_dot_proto_dot_tool__pb2.ProcessStepResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'ToolService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('ToolService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class ToolService(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 GetToolDetails(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,
+            '/ToolService/GetToolDetails',
+            com_dot_velox_dot_api_dot_plan_dot_tool_dot_proto_dot_tool__pb2.ToolDetailsRequest.SerializeToString,
+            com_dot_velox_dot_api_dot_plan_dot_tool_dot_proto_dot_tool__pb2.ToolDetailsResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def ProcessData(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,
+            '/ToolService/ProcessData',
+            com_dot_velox_dot_api_dot_plan_dot_tool_dot_proto_dot_tool__pb2.ProcessStepRequest.SerializeToString,
+            com_dot_velox_dot_api_dot_plan_dot_tool_dot_proto_dot_tool__pb2.ProcessStepResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

sapiopycommons/ai/api/session/proto/sapio_conn_info_pb2.py

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: com/velox/api/session/proto/sapio_conn_info.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,
+    '',
+    'com/velox/api/session/proto/sapio_conn_info.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1com/velox/api/session/proto/sapio_conn_info.proto\"\xb0\x01\n\x13SapioConnectionInfo\x12\x16\n\x0ewebservice_url\x18\x01 \x01(\t\x12\x10\n\x08rmi_host\x18\x02 \x03(\t\x12\x10\n\x08rmi_port\x18\x03 \x01(\x05\x12\x10\n\x08\x61pp_guid\x18\x04 \x01(\t\x12\x10\n\x08username\x18\x05 \x01(\t\x12)\n\x0bsecret_type\x18\x06 \x01(\x0e\x32\x14.SapioUserSecretType\x12\x0e\n\x06secret\x18\x07 \x01(\t*6\n\x13SapioUserSecretType\x12\x11\n\rSESSION_TOKEN\x10\x00\x12\x0c\n\x08PASSWORD\x10\x01\x42\x1f\n\x1b\x63om.velox.api.session.protoP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'sapiopycommons.ai.api.session.proto.sapio_conn_info_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\033sapiopycommons.ai.api.session.protoP\001'
+  _globals['_SAPIOUSERSECRETTYPE']._serialized_start=232
+  _globals['_SAPIOUSERSECRETTYPE']._serialized_end=286
+  _globals['_SAPIOCONNECTIONINFO']._serialized_start=54
+  _globals['_SAPIOCONNECTIONINFO']._serialized_end=230
+# @@protoc_insertion_point(module_scope)

sapiopycommons/ai/api/session/proto/sapio_conn_info_pb2.pyi

@@ -0,0 +1,32 @@
+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, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class SapioUserSecretType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    SESSION_TOKEN: _ClassVar[SapioUserSecretType]
+    PASSWORD: _ClassVar[SapioUserSecretType]
+SESSION_TOKEN: SapioUserSecretType
+PASSWORD: SapioUserSecretType
+
+class SapioConnectionInfo(_message.Message):
+    __slots__ = ("webservice_url", "rmi_host", "rmi_port", "app_guid", "username", "secret_type", "secret")
+    WEBSERVICE_URL_FIELD_NUMBER: _ClassVar[int]
+    RMI_HOST_FIELD_NUMBER: _ClassVar[int]
+    RMI_PORT_FIELD_NUMBER: _ClassVar[int]
+    APP_GUID_FIELD_NUMBER: _ClassVar[int]
+    USERNAME_FIELD_NUMBER: _ClassVar[int]
+    SECRET_TYPE_FIELD_NUMBER: _ClassVar[int]
+    SECRET_FIELD_NUMBER: _ClassVar[int]
+    webservice_url: str
+    rmi_host: _containers.RepeatedScalarFieldContainer[str]
+    rmi_port: int
+    app_guid: str
+    username: str
+    secret_type: SapioUserSecretType
+    secret: str
+    def __init__(self, webservice_url: _Optional[str] = ..., rmi_host: _Optional[_Iterable[str]] = ..., rmi_port: _Optional[int] = ..., app_guid: _Optional[str] = ..., username: _Optional[str] = ..., secret_type: _Optional[_Union[SapioUserSecretType, str]] = ..., secret: _Optional[str] = ...) -> None: ...

sapiopycommons/ai/api/session/proto/sapio_conn_info_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.71.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 com/velox/api/session/proto/sapio_conn_info_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}.'
+    )