sapiopycommons 2025.7.22a631__py3-none-any.whl → 2025.7.22a637__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of sapiopycommons might be problematic. Click here for more details.

Files changed (33) hide show
  1. sapiopycommons/ai/__init__.py +0 -0
  2. sapiopycommons/ai/api/fielddefinitions/proto/fields_pb2.py +43 -0
  3. sapiopycommons/ai/api/fielddefinitions/proto/fields_pb2.pyi +31 -0
  4. sapiopycommons/ai/api/fielddefinitions/proto/fields_pb2_grpc.py +24 -0
  5. sapiopycommons/ai/api/fielddefinitions/proto/velox_field_def_pb2.py +123 -0
  6. sapiopycommons/ai/api/fielddefinitions/proto/velox_field_def_pb2.pyi +598 -0
  7. sapiopycommons/ai/api/fielddefinitions/proto/velox_field_def_pb2_grpc.py +24 -0
  8. sapiopycommons/ai/api/plan/proto/step_output_pb2.py +45 -0
  9. sapiopycommons/ai/api/plan/proto/step_output_pb2.pyi +42 -0
  10. sapiopycommons/ai/api/plan/proto/step_output_pb2_grpc.py +24 -0
  11. sapiopycommons/ai/api/plan/proto/step_pb2.py +43 -0
  12. sapiopycommons/ai/api/plan/proto/step_pb2.pyi +43 -0
  13. sapiopycommons/ai/api/plan/proto/step_pb2_grpc.py +24 -0
  14. sapiopycommons/ai/api/plan/script/proto/script_pb2.py +55 -0
  15. sapiopycommons/ai/api/plan/script/proto/script_pb2.pyi +115 -0
  16. sapiopycommons/ai/api/plan/script/proto/script_pb2_grpc.py +153 -0
  17. sapiopycommons/ai/api/plan/tool/proto/entry_pb2.py +57 -0
  18. sapiopycommons/ai/api/plan/tool/proto/entry_pb2.pyi +96 -0
  19. sapiopycommons/ai/api/plan/tool/proto/entry_pb2_grpc.py +24 -0
  20. sapiopycommons/ai/api/plan/tool/proto/tool_pb2.py +71 -0
  21. sapiopycommons/ai/api/plan/tool/proto/tool_pb2.pyi +250 -0
  22. sapiopycommons/ai/api/plan/tool/proto/tool_pb2_grpc.py +154 -0
  23. sapiopycommons/ai/api/session/proto/sapio_conn_info_pb2.py +39 -0
  24. sapiopycommons/ai/api/session/proto/sapio_conn_info_pb2.pyi +32 -0
  25. sapiopycommons/ai/api/session/proto/sapio_conn_info_pb2_grpc.py +24 -0
  26. sapiopycommons/ai/protobuf_utils.py +509 -0
  27. sapiopycommons/ai/test_client.py +260 -0
  28. sapiopycommons/ai/tool_service_base.py +919 -0
  29. sapiopycommons/callbacks/callback_util.py +2 -5
  30. {sapiopycommons-2025.7.22a631.dist-info → sapiopycommons-2025.7.22a637.dist-info}/METADATA +1 -1
  31. {sapiopycommons-2025.7.22a631.dist-info → sapiopycommons-2025.7.22a637.dist-info}/RECORD +33 -5
  32. {sapiopycommons-2025.7.22a631.dist-info → sapiopycommons-2025.7.22a637.dist-info}/WHEEL +0 -0
  33. {sapiopycommons-2025.7.22a631.dist-info → sapiopycommons-2025.7.22a637.dist-info}/licenses/LICENSE +0 -0
@@ -0,0 +1,42 @@
1
+ from google.protobuf.internal import containers as _containers
2
+ from google.protobuf import descriptor as _descriptor
3
+ from google.protobuf import message as _message
4
+ from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
5
+
6
+ DESCRIPTOR: _descriptor.FileDescriptor
7
+
8
+ class StepJsonSingletonItemPbo(_message.Message):
9
+ __slots__ = ("item",)
10
+ ITEM_FIELD_NUMBER: _ClassVar[int]
11
+ item: str
12
+ def __init__(self, item: _Optional[str] = ...) -> None: ...
13
+
14
+ class StepTextSingletonItemPbo(_message.Message):
15
+ __slots__ = ("item",)
16
+ ITEM_FIELD_NUMBER: _ClassVar[int]
17
+ item: str
18
+ def __init__(self, item: _Optional[str] = ...) -> None: ...
19
+
20
+ class StepCsvSingletonItemPbo(_message.Message):
21
+ __slots__ = ("cells",)
22
+ CELLS_FIELD_NUMBER: _ClassVar[int]
23
+ cells: _containers.RepeatedScalarFieldContainer[str]
24
+ def __init__(self, cells: _Optional[_Iterable[str]] = ...) -> None: ...
25
+
26
+ class StepBinarySingletonItemPbo(_message.Message):
27
+ __slots__ = ("item",)
28
+ ITEM_FIELD_NUMBER: _ClassVar[int]
29
+ item: bytes
30
+ def __init__(self, item: _Optional[bytes] = ...) -> None: ...
31
+
32
+ class StepSingletonItemPbo(_message.Message):
33
+ __slots__ = ("json_singleton", "text_singleton", "csv_singleton", "binary_singleton")
34
+ JSON_SINGLETON_FIELD_NUMBER: _ClassVar[int]
35
+ TEXT_SINGLETON_FIELD_NUMBER: _ClassVar[int]
36
+ CSV_SINGLETON_FIELD_NUMBER: _ClassVar[int]
37
+ BINARY_SINGLETON_FIELD_NUMBER: _ClassVar[int]
38
+ json_singleton: StepJsonSingletonItemPbo
39
+ text_singleton: StepTextSingletonItemPbo
40
+ csv_singleton: StepCsvSingletonItemPbo
41
+ binary_singleton: StepBinarySingletonItemPbo
42
+ def __init__(self, json_singleton: _Optional[_Union[StepJsonSingletonItemPbo, _Mapping]] = ..., text_singleton: _Optional[_Union[StepTextSingletonItemPbo, _Mapping]] = ..., csv_singleton: _Optional[_Union[StepCsvSingletonItemPbo, _Mapping]] = ..., binary_singleton: _Optional[_Union[StepBinarySingletonItemPbo, _Mapping]] = ...) -> None: ...
@@ -0,0 +1,24 @@
1
+ # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
+ """Client and server classes corresponding to protobuf-defined services."""
3
+ import grpc
4
+ import warnings
5
+
6
+
7
+ GRPC_GENERATED_VERSION = '1.70.0'
8
+ GRPC_VERSION = grpc.__version__
9
+ _version_not_supported = False
10
+
11
+ try:
12
+ from grpc._utilities import first_version_is_lower
13
+ _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
14
+ except ImportError:
15
+ _version_not_supported = True
16
+
17
+ if _version_not_supported:
18
+ raise RuntimeError(
19
+ f'The grpc package installed is at version {GRPC_VERSION},'
20
+ + f' but the generated code in sapiopycommons/ai/api/plan/proto/step_output_pb2_grpc.py depends on'
21
+ + f' grpcio>={GRPC_GENERATED_VERSION}.'
22
+ + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
23
+ + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
24
+ )
@@ -0,0 +1,43 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: sapiopycommons/ai/api/plan/proto/step.proto
5
+ # Protobuf Python Version: 5.29.0
6
+ """Generated protocol buffer code."""
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import runtime_version as _runtime_version
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ from google.protobuf.internal import builder as _builder
12
+ _runtime_version.ValidateProtobufRuntimeVersion(
13
+ _runtime_version.Domain.PUBLIC,
14
+ 5,
15
+ 29,
16
+ 0,
17
+ '',
18
+ 'sapiopycommons/ai/api/plan/proto/step.proto'
19
+ )
20
+ # @@protoc_insertion_point(imports)
21
+
22
+ _sym_db = _symbol_database.Default()
23
+
24
+
25
+
26
+
27
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+sapiopycommons/ai/api/plan/proto/step.proto\"N\n\rStepIoInfoPbo\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\"r\n\x10StepIoDetailsPbo\x12$\n\x0cstep_io_info\x18\x01 \x01(\x0b\x32\x0e.StepIoInfoPbo\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\"p\n\x13StepInputDetailsPbo\x12*\n\x0fstep_io_details\x18\x01 \x01(\x0b\x32\x11.StepIoDetailsPbo\x12\x18\n\x10paging_supported\x18\x02 \x01(\x08\x12\x13\n\x0bmax_entries\x18\x03 \x01(\x05\"B\n\x14StepOutputDetailsPbo\x12*\n\x0fstep_io_details\x18\x01 \x01(\x0b\x32\x11.StepIoDetailsPboB\x1c\n\x18\x63om.velox.api.plan.protoP\x01\x62\x06proto3')
28
+
29
+ _globals = globals()
30
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
31
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'sapiopycommons.ai.api.plan.proto.step_pb2', _globals)
32
+ if not _descriptor._USE_C_DESCRIPTORS:
33
+ _globals['DESCRIPTOR']._loaded_options = None
34
+ _globals['DESCRIPTOR']._serialized_options = b'\n\030com.velox.api.plan.protoP\001'
35
+ _globals['_STEPIOINFOPBO']._serialized_start=47
36
+ _globals['_STEPIOINFOPBO']._serialized_end=125
37
+ _globals['_STEPIODETAILSPBO']._serialized_start=127
38
+ _globals['_STEPIODETAILSPBO']._serialized_end=241
39
+ _globals['_STEPINPUTDETAILSPBO']._serialized_start=243
40
+ _globals['_STEPINPUTDETAILSPBO']._serialized_end=355
41
+ _globals['_STEPOUTPUTDETAILSPBO']._serialized_start=357
42
+ _globals['_STEPOUTPUTDETAILSPBO']._serialized_end=423
43
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,43 @@
1
+ from google.protobuf import descriptor as _descriptor
2
+ from google.protobuf import message as _message
3
+ from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
4
+
5
+ DESCRIPTOR: _descriptor.FileDescriptor
6
+
7
+ class StepIoInfoPbo(_message.Message):
8
+ __slots__ = ("step_io_number", "content_type", "io_name")
9
+ STEP_IO_NUMBER_FIELD_NUMBER: _ClassVar[int]
10
+ CONTENT_TYPE_FIELD_NUMBER: _ClassVar[int]
11
+ IO_NAME_FIELD_NUMBER: _ClassVar[int]
12
+ step_io_number: int
13
+ content_type: str
14
+ io_name: str
15
+ def __init__(self, step_io_number: _Optional[int] = ..., content_type: _Optional[str] = ..., io_name: _Optional[str] = ...) -> None: ...
16
+
17
+ class StepIoDetailsPbo(_message.Message):
18
+ __slots__ = ("step_io_info", "description", "example", "validation")
19
+ STEP_IO_INFO_FIELD_NUMBER: _ClassVar[int]
20
+ DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
21
+ EXAMPLE_FIELD_NUMBER: _ClassVar[int]
22
+ VALIDATION_FIELD_NUMBER: _ClassVar[int]
23
+ step_io_info: StepIoInfoPbo
24
+ description: str
25
+ example: str
26
+ validation: str
27
+ def __init__(self, step_io_info: _Optional[_Union[StepIoInfoPbo, _Mapping]] = ..., description: _Optional[str] = ..., example: _Optional[str] = ..., validation: _Optional[str] = ...) -> None: ...
28
+
29
+ class StepInputDetailsPbo(_message.Message):
30
+ __slots__ = ("step_io_details", "paging_supported", "max_entries")
31
+ STEP_IO_DETAILS_FIELD_NUMBER: _ClassVar[int]
32
+ PAGING_SUPPORTED_FIELD_NUMBER: _ClassVar[int]
33
+ MAX_ENTRIES_FIELD_NUMBER: _ClassVar[int]
34
+ step_io_details: StepIoDetailsPbo
35
+ paging_supported: bool
36
+ max_entries: int
37
+ def __init__(self, step_io_details: _Optional[_Union[StepIoDetailsPbo, _Mapping]] = ..., paging_supported: bool = ..., max_entries: _Optional[int] = ...) -> None: ...
38
+
39
+ class StepOutputDetailsPbo(_message.Message):
40
+ __slots__ = ("step_io_details",)
41
+ STEP_IO_DETAILS_FIELD_NUMBER: _ClassVar[int]
42
+ step_io_details: StepIoDetailsPbo
43
+ def __init__(self, step_io_details: _Optional[_Union[StepIoDetailsPbo, _Mapping]] = ...) -> None: ...
@@ -0,0 +1,24 @@
1
+ # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
+ """Client and server classes corresponding to protobuf-defined services."""
3
+ import grpc
4
+ import warnings
5
+
6
+
7
+ GRPC_GENERATED_VERSION = '1.70.0'
8
+ GRPC_VERSION = grpc.__version__
9
+ _version_not_supported = False
10
+
11
+ try:
12
+ from grpc._utilities import first_version_is_lower
13
+ _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
14
+ except ImportError:
15
+ _version_not_supported = True
16
+
17
+ if _version_not_supported:
18
+ raise RuntimeError(
19
+ f'The grpc package installed is at version {GRPC_VERSION},'
20
+ + f' but the generated code in sapiopycommons/ai/api/plan/proto/step_pb2_grpc.py depends on'
21
+ + f' grpcio>={GRPC_GENERATED_VERSION}.'
22
+ + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
23
+ + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
24
+ )
@@ -0,0 +1,55 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: sapiopycommons/ai/api/plan/script/proto/script.proto
5
+ # Protobuf Python Version: 5.29.0
6
+ """Generated protocol buffer code."""
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import runtime_version as _runtime_version
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ from google.protobuf.internal import builder as _builder
12
+ _runtime_version.ValidateProtobufRuntimeVersion(
13
+ _runtime_version.Domain.PUBLIC,
14
+ 5,
15
+ 29,
16
+ 0,
17
+ '',
18
+ 'sapiopycommons/ai/api/plan/script/proto/script.proto'
19
+ )
20
+ # @@protoc_insertion_point(imports)
21
+
22
+ _sym_db = _symbol_database.Default()
23
+
24
+
25
+ from sapiopycommons.ai.api.plan.tool.proto import entry_pb2 as sapiopycommons_dot_ai_dot_api_dot_plan_dot_tool_dot_proto_dot_entry__pb2
26
+ from sapiopycommons.ai.api.plan.proto import step_pb2 as sapiopycommons_dot_ai_dot_api_dot_plan_dot_proto_dot_step__pb2
27
+ from sapiopycommons.ai.api.session.proto import sapio_conn_info_pb2 as sapiopycommons_dot_ai_dot_api_dot_session_dot_proto_dot_sapio__conn__info__pb2
28
+
29
+ from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import *
30
+ from sapiopycommons.ai.api.plan.proto.step_pb2 import *
31
+ from sapiopycommons.ai.api.session.proto.sapio_conn_info_pb2 import *
32
+
33
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n4sapiopycommons/ai/api/plan/script/proto/script.proto\x1a\x31sapiopycommons/ai/api/plan/tool/proto/entry.proto\x1a+sapiopycommons/ai/api/plan/proto/step.proto\x1a\x39sapiopycommons/ai/api/session/proto/sapio_conn_info.proto\"A\n\x15ScriptFileContentsPbo\x12\x11\n\tfile_name\x18\x01 \x01(\t\x12\x15\n\rfile_contents\x18\x02 \x01(\x0c\"\xc2\x03\n\x19\x43reateScriptJobRequestPbo\x12+\n\nsapio_user\x18\x01 \x01(\x0b\x32\x17.SapioConnectionInfoPbo\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\x0e.StepIoInfoPbo\x12&\n\x0eoutput_configs\x18\x07 \x03(\x0b\x32\x0e.StepIoInfoPbo\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\x0binput_files\x18\x84\x08 \x03(\x0b\x32\x16.ScriptFileContentsPbo\x12\x1c\n\x13\x64ownload_file_names\x18\x85\x08 \x03(\t\x12\"\n\x05input\x18\xff\x0f \x03(\x0b\x32\x12.StepInputBatchPbo\",\n\x1a\x43reateScriptJobResponsePbo\x12\x0e\n\x06job_id\x18\x01 \x01(\t\"N\n\x10GetJobRequestPbo\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12\x12\n\nlog_offset\x18\x02 \x01(\x05\x12\x16\n\x0e\x64ownload_files\x18\x03 \x01(\x08\"\xa4\x01\n\x11GetJobResponsePbo\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\x0coutput_files\x18\x84\x08 \x03(\x0b\x32\x16.ScriptFileContentsPbo\x12$\n\x06output\x18\xff\x0f \x03(\x0b\x32\x13.StepOutputBatchPbo*@\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\x8c\x01\n\rScriptService\x12J\n\x0f\x43reateScriptJob\x12\x1a.CreateScriptJobRequestPbo\x1a\x1b.CreateScriptJobResponsePbo\x12/\n\x06GetJob\x12\x11.GetJobRequestPbo\x1a\x12.GetJobResponsePboB!\n\x1f\x63om.velox.api.plan.script.protoP\x00P\x01P\x02\x62\x06proto3')
34
+
35
+ _globals = globals()
36
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
37
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'sapiopycommons.ai.api.plan.script.proto.script_pb2', _globals)
38
+ if not _descriptor._USE_C_DESCRIPTORS:
39
+ _globals['DESCRIPTOR']._loaded_options = None
40
+ _globals['DESCRIPTOR']._serialized_options = b'\n\037com.velox.api.plan.script.proto'
41
+ _globals['_JOBSTATUS']._serialized_start=1024
42
+ _globals['_JOBSTATUS']._serialized_end=1088
43
+ _globals['_SCRIPTFILECONTENTSPBO']._serialized_start=211
44
+ _globals['_SCRIPTFILECONTENTSPBO']._serialized_end=276
45
+ _globals['_CREATESCRIPTJOBREQUESTPBO']._serialized_start=279
46
+ _globals['_CREATESCRIPTJOBREQUESTPBO']._serialized_end=729
47
+ _globals['_CREATESCRIPTJOBRESPONSEPBO']._serialized_start=731
48
+ _globals['_CREATESCRIPTJOBRESPONSEPBO']._serialized_end=775
49
+ _globals['_GETJOBREQUESTPBO']._serialized_start=777
50
+ _globals['_GETJOBREQUESTPBO']._serialized_end=855
51
+ _globals['_GETJOBRESPONSEPBO']._serialized_start=858
52
+ _globals['_GETJOBRESPONSEPBO']._serialized_end=1022
53
+ _globals['_SCRIPTSERVICE']._serialized_start=1091
54
+ _globals['_SCRIPTSERVICE']._serialized_end=1231
55
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,115 @@
1
+ from sapiopycommons.ai.api.plan.tool.proto import entry_pb2 as _entry_pb2
2
+ from sapiopycommons.ai.api.plan.proto import step_pb2 as _step_pb2
3
+ from sapiopycommons.ai.api.session.proto import sapio_conn_info_pb2 as _sapio_conn_info_pb2
4
+ from google.protobuf.internal import containers as _containers
5
+ from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
6
+ from google.protobuf import descriptor as _descriptor
7
+ from google.protobuf import message as _message
8
+ from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
9
+ from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepCsvHeaderRowPbo as StepCsvHeaderRowPbo
10
+ from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepCsvRowPbo as StepCsvRowPbo
11
+ from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepCsvContainerPbo as StepCsvContainerPbo
12
+ from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepJsonContainerPbo as StepJsonContainerPbo
13
+ from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepTextContainerPbo as StepTextContainerPbo
14
+ from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepBinaryContainerPbo as StepBinaryContainerPbo
15
+ from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepImageContainerPbo as StepImageContainerPbo
16
+ from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepItemContainerPbo as StepItemContainerPbo
17
+ from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepInputBatchPbo as StepInputBatchPbo
18
+ from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepOutputBatchPbo as StepOutputBatchPbo
19
+ from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import DataTypePbo as DataTypePbo
20
+ from sapiopycommons.ai.api.plan.proto.step_pb2 import StepIoInfoPbo as StepIoInfoPbo
21
+ from sapiopycommons.ai.api.plan.proto.step_pb2 import StepIoDetailsPbo as StepIoDetailsPbo
22
+ from sapiopycommons.ai.api.plan.proto.step_pb2 import StepInputDetailsPbo as StepInputDetailsPbo
23
+ from sapiopycommons.ai.api.plan.proto.step_pb2 import StepOutputDetailsPbo as StepOutputDetailsPbo
24
+ from sapiopycommons.ai.api.session.proto.sapio_conn_info_pb2 import SapioConnectionInfoPbo as SapioConnectionInfoPbo
25
+ from sapiopycommons.ai.api.session.proto.sapio_conn_info_pb2 import SapioUserSecretTypePbo as SapioUserSecretTypePbo
26
+
27
+ DESCRIPTOR: _descriptor.FileDescriptor
28
+ BINARY: _entry_pb2.DataTypePbo
29
+ JSON: _entry_pb2.DataTypePbo
30
+ CSV: _entry_pb2.DataTypePbo
31
+ TEXT: _entry_pb2.DataTypePbo
32
+ IMAGE: _entry_pb2.DataTypePbo
33
+ SESSION_TOKEN: _sapio_conn_info_pb2.SapioUserSecretTypePbo
34
+ PASSWORD: _sapio_conn_info_pb2.SapioUserSecretTypePbo
35
+
36
+ class JobStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
37
+ __slots__ = ()
38
+ PENDING: _ClassVar[JobStatus]
39
+ RUNNING: _ClassVar[JobStatus]
40
+ COMPLETED: _ClassVar[JobStatus]
41
+ FAILED: _ClassVar[JobStatus]
42
+ PENDING: JobStatus
43
+ RUNNING: JobStatus
44
+ COMPLETED: JobStatus
45
+ FAILED: JobStatus
46
+
47
+ class ScriptFileContentsPbo(_message.Message):
48
+ __slots__ = ("file_name", "file_contents")
49
+ FILE_NAME_FIELD_NUMBER: _ClassVar[int]
50
+ FILE_CONTENTS_FIELD_NUMBER: _ClassVar[int]
51
+ file_name: str
52
+ file_contents: bytes
53
+ def __init__(self, file_name: _Optional[str] = ..., file_contents: _Optional[bytes] = ...) -> None: ...
54
+
55
+ class CreateScriptJobRequestPbo(_message.Message):
56
+ __slots__ = ("sapio_user", "script_language", "plan_instance_id", "step_instance_id", "invocation_id", "input_configs", "output_configs", "script", "timeout", "max_memory_mb", "working_directory", "input_files", "download_file_names", "input")
57
+ SAPIO_USER_FIELD_NUMBER: _ClassVar[int]
58
+ SCRIPT_LANGUAGE_FIELD_NUMBER: _ClassVar[int]
59
+ PLAN_INSTANCE_ID_FIELD_NUMBER: _ClassVar[int]
60
+ STEP_INSTANCE_ID_FIELD_NUMBER: _ClassVar[int]
61
+ INVOCATION_ID_FIELD_NUMBER: _ClassVar[int]
62
+ INPUT_CONFIGS_FIELD_NUMBER: _ClassVar[int]
63
+ OUTPUT_CONFIGS_FIELD_NUMBER: _ClassVar[int]
64
+ SCRIPT_FIELD_NUMBER: _ClassVar[int]
65
+ TIMEOUT_FIELD_NUMBER: _ClassVar[int]
66
+ MAX_MEMORY_MB_FIELD_NUMBER: _ClassVar[int]
67
+ WORKING_DIRECTORY_FIELD_NUMBER: _ClassVar[int]
68
+ INPUT_FILES_FIELD_NUMBER: _ClassVar[int]
69
+ DOWNLOAD_FILE_NAMES_FIELD_NUMBER: _ClassVar[int]
70
+ INPUT_FIELD_NUMBER: _ClassVar[int]
71
+ sapio_user: _sapio_conn_info_pb2.SapioConnectionInfoPbo
72
+ script_language: str
73
+ plan_instance_id: int
74
+ step_instance_id: int
75
+ invocation_id: int
76
+ input_configs: _containers.RepeatedCompositeFieldContainer[_step_pb2.StepIoInfoPbo]
77
+ output_configs: _containers.RepeatedCompositeFieldContainer[_step_pb2.StepIoInfoPbo]
78
+ script: str
79
+ timeout: int
80
+ max_memory_mb: int
81
+ working_directory: str
82
+ input_files: _containers.RepeatedCompositeFieldContainer[ScriptFileContentsPbo]
83
+ download_file_names: _containers.RepeatedScalarFieldContainer[str]
84
+ input: _containers.RepeatedCompositeFieldContainer[_entry_pb2.StepInputBatchPbo]
85
+ def __init__(self, sapio_user: _Optional[_Union[_sapio_conn_info_pb2.SapioConnectionInfoPbo, _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.StepIoInfoPbo, _Mapping]]] = ..., output_configs: _Optional[_Iterable[_Union[_step_pb2.StepIoInfoPbo, _Mapping]]] = ..., script: _Optional[str] = ..., timeout: _Optional[int] = ..., max_memory_mb: _Optional[int] = ..., working_directory: _Optional[str] = ..., input_files: _Optional[_Iterable[_Union[ScriptFileContentsPbo, _Mapping]]] = ..., download_file_names: _Optional[_Iterable[str]] = ..., input: _Optional[_Iterable[_Union[_entry_pb2.StepInputBatchPbo, _Mapping]]] = ...) -> None: ...
86
+
87
+ class CreateScriptJobResponsePbo(_message.Message):
88
+ __slots__ = ("job_id",)
89
+ JOB_ID_FIELD_NUMBER: _ClassVar[int]
90
+ job_id: str
91
+ def __init__(self, job_id: _Optional[str] = ...) -> None: ...
92
+
93
+ class GetJobRequestPbo(_message.Message):
94
+ __slots__ = ("job_id", "log_offset", "download_files")
95
+ JOB_ID_FIELD_NUMBER: _ClassVar[int]
96
+ LOG_OFFSET_FIELD_NUMBER: _ClassVar[int]
97
+ DOWNLOAD_FILES_FIELD_NUMBER: _ClassVar[int]
98
+ job_id: str
99
+ log_offset: int
100
+ download_files: bool
101
+ def __init__(self, job_id: _Optional[str] = ..., log_offset: _Optional[int] = ..., download_files: bool = ...) -> None: ...
102
+
103
+ class GetJobResponsePbo(_message.Message):
104
+ __slots__ = ("status", "log", "exception", "output_files", "output")
105
+ STATUS_FIELD_NUMBER: _ClassVar[int]
106
+ LOG_FIELD_NUMBER: _ClassVar[int]
107
+ EXCEPTION_FIELD_NUMBER: _ClassVar[int]
108
+ OUTPUT_FILES_FIELD_NUMBER: _ClassVar[int]
109
+ OUTPUT_FIELD_NUMBER: _ClassVar[int]
110
+ status: JobStatus
111
+ log: str
112
+ exception: str
113
+ output_files: _containers.RepeatedCompositeFieldContainer[ScriptFileContentsPbo]
114
+ output: _containers.RepeatedCompositeFieldContainer[_entry_pb2.StepOutputBatchPbo]
115
+ def __init__(self, status: _Optional[_Union[JobStatus, str]] = ..., log: _Optional[str] = ..., exception: _Optional[str] = ..., output_files: _Optional[_Iterable[_Union[ScriptFileContentsPbo, _Mapping]]] = ..., output: _Optional[_Iterable[_Union[_entry_pb2.StepOutputBatchPbo, _Mapping]]] = ...) -> None: ...
@@ -0,0 +1,153 @@
1
+ # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
+ """Client and server classes corresponding to protobuf-defined services."""
3
+ import grpc
4
+ import warnings
5
+
6
+ from sapiopycommons.ai.api.plan.script.proto import script_pb2 as sapiopycommons_dot_ai_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2
7
+
8
+ GRPC_GENERATED_VERSION = '1.70.0'
9
+ GRPC_VERSION = grpc.__version__
10
+ _version_not_supported = False
11
+
12
+ try:
13
+ from grpc._utilities import first_version_is_lower
14
+ _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
15
+ except ImportError:
16
+ _version_not_supported = True
17
+
18
+ if _version_not_supported:
19
+ raise RuntimeError(
20
+ f'The grpc package installed is at version {GRPC_VERSION},'
21
+ + f' but the generated code in sapiopycommons/ai/api/plan/script/proto/script_pb2_grpc.py depends on'
22
+ + f' grpcio>={GRPC_GENERATED_VERSION}.'
23
+ + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
24
+ + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
25
+ )
26
+
27
+
28
+ class ScriptServiceStub(object):
29
+ """
30
+ This is the generic service interface that all tools must implement. We will use hostname routing to route the request to the correct tool.
31
+ 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
32
+ a pain to manage. So we will use hostname routing for now.
33
+ """
34
+
35
+ def __init__(self, channel):
36
+ """Constructor.
37
+
38
+ Args:
39
+ channel: A grpc.Channel.
40
+ """
41
+ self.CreateScriptJob = channel.unary_unary(
42
+ '/ScriptService/CreateScriptJob',
43
+ request_serializer=sapiopycommons_dot_ai_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.CreateScriptJobRequestPbo.SerializeToString,
44
+ response_deserializer=sapiopycommons_dot_ai_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.CreateScriptJobResponsePbo.FromString,
45
+ _registered_method=True)
46
+ self.GetJob = channel.unary_unary(
47
+ '/ScriptService/GetJob',
48
+ request_serializer=sapiopycommons_dot_ai_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.GetJobRequestPbo.SerializeToString,
49
+ response_deserializer=sapiopycommons_dot_ai_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.GetJobResponsePbo.FromString,
50
+ _registered_method=True)
51
+
52
+
53
+ class ScriptServiceServicer(object):
54
+ """
55
+ This is the generic service interface that all tools must implement. We will use hostname routing to route the request to the correct tool.
56
+ 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
57
+ a pain to manage. So we will use hostname routing for now.
58
+ """
59
+
60
+ def CreateScriptJob(self, request, context):
61
+ """The main entry point for script
62
+ """
63
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
64
+ context.set_details('Method not implemented!')
65
+ raise NotImplementedError('Method not implemented!')
66
+
67
+ def GetJob(self, request, context):
68
+ """Missing associated documentation comment in .proto file."""
69
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
70
+ context.set_details('Method not implemented!')
71
+ raise NotImplementedError('Method not implemented!')
72
+
73
+
74
+ def add_ScriptServiceServicer_to_server(servicer, server):
75
+ rpc_method_handlers = {
76
+ 'CreateScriptJob': grpc.unary_unary_rpc_method_handler(
77
+ servicer.CreateScriptJob,
78
+ request_deserializer=sapiopycommons_dot_ai_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.CreateScriptJobRequestPbo.FromString,
79
+ response_serializer=sapiopycommons_dot_ai_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.CreateScriptJobResponsePbo.SerializeToString,
80
+ ),
81
+ 'GetJob': grpc.unary_unary_rpc_method_handler(
82
+ servicer.GetJob,
83
+ request_deserializer=sapiopycommons_dot_ai_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.GetJobRequestPbo.FromString,
84
+ response_serializer=sapiopycommons_dot_ai_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.GetJobResponsePbo.SerializeToString,
85
+ ),
86
+ }
87
+ generic_handler = grpc.method_handlers_generic_handler(
88
+ 'ScriptService', rpc_method_handlers)
89
+ server.add_generic_rpc_handlers((generic_handler,))
90
+ server.add_registered_method_handlers('ScriptService', rpc_method_handlers)
91
+
92
+
93
+ # This class is part of an EXPERIMENTAL API.
94
+ class ScriptService(object):
95
+ """
96
+ This is the generic service interface that all tools must implement. We will use hostname routing to route the request to the correct tool.
97
+ 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
98
+ a pain to manage. So we will use hostname routing for now.
99
+ """
100
+
101
+ @staticmethod
102
+ def CreateScriptJob(request,
103
+ target,
104
+ options=(),
105
+ channel_credentials=None,
106
+ call_credentials=None,
107
+ insecure=False,
108
+ compression=None,
109
+ wait_for_ready=None,
110
+ timeout=None,
111
+ metadata=None):
112
+ return grpc.experimental.unary_unary(
113
+ request,
114
+ target,
115
+ '/ScriptService/CreateScriptJob',
116
+ sapiopycommons_dot_ai_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.CreateScriptJobRequestPbo.SerializeToString,
117
+ sapiopycommons_dot_ai_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.CreateScriptJobResponsePbo.FromString,
118
+ options,
119
+ channel_credentials,
120
+ insecure,
121
+ call_credentials,
122
+ compression,
123
+ wait_for_ready,
124
+ timeout,
125
+ metadata,
126
+ _registered_method=True)
127
+
128
+ @staticmethod
129
+ def GetJob(request,
130
+ target,
131
+ options=(),
132
+ channel_credentials=None,
133
+ call_credentials=None,
134
+ insecure=False,
135
+ compression=None,
136
+ wait_for_ready=None,
137
+ timeout=None,
138
+ metadata=None):
139
+ return grpc.experimental.unary_unary(
140
+ request,
141
+ target,
142
+ '/ScriptService/GetJob',
143
+ sapiopycommons_dot_ai_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.GetJobRequestPbo.SerializeToString,
144
+ sapiopycommons_dot_ai_dot_api_dot_plan_dot_script_dot_proto_dot_script__pb2.GetJobResponsePbo.FromString,
145
+ options,
146
+ channel_credentials,
147
+ insecure,
148
+ call_credentials,
149
+ compression,
150
+ wait_for_ready,
151
+ timeout,
152
+ metadata,
153
+ _registered_method=True)
@@ -0,0 +1,57 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: sapiopycommons/ai/api/plan/tool/proto/entry.proto
5
+ # Protobuf Python Version: 5.29.0
6
+ """Generated protocol buffer code."""
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import runtime_version as _runtime_version
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ from google.protobuf.internal import builder as _builder
12
+ _runtime_version.ValidateProtobufRuntimeVersion(
13
+ _runtime_version.Domain.PUBLIC,
14
+ 5,
15
+ 29,
16
+ 0,
17
+ '',
18
+ 'sapiopycommons/ai/api/plan/tool/proto/entry.proto'
19
+ )
20
+ # @@protoc_insertion_point(imports)
21
+
22
+ _sym_db = _symbol_database.Default()
23
+
24
+
25
+
26
+
27
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1sapiopycommons/ai/api/plan/tool/proto/entry.proto\"$\n\x13StepCsvHeaderRowPbo\x12\r\n\x05\x63\x65lls\x18\x01 \x03(\t\"\x1e\n\rStepCsvRowPbo\x12\r\n\x05\x63\x65lls\x18\x01 \x03(\t\"Z\n\x13StepCsvContainerPbo\x12$\n\x06header\x18\x0e \x01(\x0b\x32\x14.StepCsvHeaderRowPbo\x12\x1d\n\x05items\x18\x0f \x03(\x0b\x32\x0e.StepCsvRowPbo\"%\n\x14StepJsonContainerPbo\x12\r\n\x05items\x18\x0f \x03(\t\"%\n\x14StepTextContainerPbo\x12\r\n\x05items\x18\x0f \x03(\t\"\'\n\x16StepBinaryContainerPbo\x12\r\n\x05items\x18\x0f \x03(\x0c\"<\n\x15StepImageContainerPbo\x12\x14\n\x0cimage_format\x18\x01 \x01(\t\x12\r\n\x05items\x18\x0f \x03(\x0c\"\xc1\x02\n\x14StepItemContainerPbo\x12\x1e\n\x08\x64\x61taType\x18\x01 \x01(\x0e\x32\x0c.DataTypePbo\x12\x34\n\x10\x62inary_container\x18\xff\x0f \x01(\x0b\x32\x17.StepBinaryContainerPboH\x00\x12.\n\rcsv_container\x18\xfe\x0f \x01(\x0b\x32\x14.StepCsvContainerPboH\x00\x12\x30\n\x0ejson_container\x18\xfd\x0f \x01(\x0b\x32\x15.StepJsonContainerPboH\x00\x12\x30\n\x0etext_container\x18\xfc\x0f \x01(\x0b\x32\x15.StepTextContainerPboH\x00\x12\x32\n\x0fimage_container\x18\xfb\x0f \x01(\x0b\x32\x16.StepImageContainerPboH\x00\x42\x0b\n\tcontainer\"V\n\x11StepInputBatchPbo\x12\x12\n\nis_partial\x18\x02 \x01(\x08\x12-\n\x0eitem_container\x18\x01 \x01(\x0b\x32\x15.StepItemContainerPbo\"C\n\x12StepOutputBatchPbo\x12-\n\x0eitem_container\x18\x01 \x01(\x0b\x32\x15.StepItemContainerPbo*A\n\x0b\x44\x61taTypePbo\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')
28
+
29
+ _globals = globals()
30
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
31
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'sapiopycommons.ai.api.plan.tool.proto.entry_pb2', _globals)
32
+ if not _descriptor._USE_C_DESCRIPTORS:
33
+ _globals['DESCRIPTOR']._loaded_options = None
34
+ _globals['DESCRIPTOR']._serialized_options = b'\n\035com.velox.api.plan.tool.protoP\001'
35
+ _globals['_DATATYPEPBO']._serialized_start=877
36
+ _globals['_DATATYPEPBO']._serialized_end=942
37
+ _globals['_STEPCSVHEADERROWPBO']._serialized_start=53
38
+ _globals['_STEPCSVHEADERROWPBO']._serialized_end=89
39
+ _globals['_STEPCSVROWPBO']._serialized_start=91
40
+ _globals['_STEPCSVROWPBO']._serialized_end=121
41
+ _globals['_STEPCSVCONTAINERPBO']._serialized_start=123
42
+ _globals['_STEPCSVCONTAINERPBO']._serialized_end=213
43
+ _globals['_STEPJSONCONTAINERPBO']._serialized_start=215
44
+ _globals['_STEPJSONCONTAINERPBO']._serialized_end=252
45
+ _globals['_STEPTEXTCONTAINERPBO']._serialized_start=254
46
+ _globals['_STEPTEXTCONTAINERPBO']._serialized_end=291
47
+ _globals['_STEPBINARYCONTAINERPBO']._serialized_start=293
48
+ _globals['_STEPBINARYCONTAINERPBO']._serialized_end=332
49
+ _globals['_STEPIMAGECONTAINERPBO']._serialized_start=334
50
+ _globals['_STEPIMAGECONTAINERPBO']._serialized_end=394
51
+ _globals['_STEPITEMCONTAINERPBO']._serialized_start=397
52
+ _globals['_STEPITEMCONTAINERPBO']._serialized_end=718
53
+ _globals['_STEPINPUTBATCHPBO']._serialized_start=720
54
+ _globals['_STEPINPUTBATCHPBO']._serialized_end=806
55
+ _globals['_STEPOUTPUTBATCHPBO']._serialized_start=808
56
+ _globals['_STEPOUTPUTBATCHPBO']._serialized_end=875
57
+ # @@protoc_insertion_point(module_scope)