omium-contracts 0.1.0__tar.gz
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.
- omium_contracts-0.1.0/PKG-INFO +7 -0
- omium_contracts-0.1.0/omium_contracts.egg-info/PKG-INFO +7 -0
- omium_contracts-0.1.0/omium_contracts.egg-info/SOURCES.txt +27 -0
- omium_contracts-0.1.0/omium_contracts.egg-info/dependency_links.txt +1 -0
- omium_contracts-0.1.0/omium_contracts.egg-info/requires.txt +2 -0
- omium_contracts-0.1.0/omium_contracts.egg-info/top_level.txt +1 -0
- omium_contracts-0.1.0/omiumpb/agent/v1/agent_pb2.py +57 -0
- omium_contracts-0.1.0/omiumpb/agent/v1/agent_pb2.pyi +91 -0
- omium_contracts-0.1.0/omiumpb/agent/v1/agent_pb2_grpc.py +215 -0
- omium_contracts-0.1.0/omiumpb/checkpoint/v1/checkpoint_pb2.py +65 -0
- omium_contracts-0.1.0/omiumpb/checkpoint/v1/checkpoint_pb2.pyi +200 -0
- omium_contracts-0.1.0/omiumpb/checkpoint/v1/checkpoint_pb2_grpc.py +295 -0
- omium_contracts-0.1.0/omiumpb/common/v1/common_pb2.py +49 -0
- omium_contracts-0.1.0/omiumpb/common/v1/common_pb2.pyi +73 -0
- omium_contracts-0.1.0/omiumpb/common/v1/common_pb2_grpc.py +4 -0
- omium_contracts-0.1.0/omiumpb/consensus/v1/consensus_pb2.py +89 -0
- omium_contracts-0.1.0/omiumpb/consensus/v1/consensus_pb2.pyi +345 -0
- omium_contracts-0.1.0/omiumpb/consensus/v1/consensus_pb2_grpc.py +462 -0
- omium_contracts-0.1.0/omiumpb/events/v1/events_pb2.py +49 -0
- omium_contracts-0.1.0/omiumpb/events/v1/events_pb2.pyi +99 -0
- omium_contracts-0.1.0/omiumpb/events/v1/events_pb2_grpc.py +4 -0
- omium_contracts-0.1.0/omiumpb/execution/v1/execution_pb2.py +71 -0
- omium_contracts-0.1.0/omiumpb/execution/v1/execution_pb2.pyi +253 -0
- omium_contracts-0.1.0/omiumpb/execution/v1/execution_pb2_grpc.py +295 -0
- omium_contracts-0.1.0/omiumpb/policy/v1/policy_pb2.py +105 -0
- omium_contracts-0.1.0/omiumpb/policy/v1/policy_pb2.pyi +381 -0
- omium_contracts-0.1.0/omiumpb/policy/v1/policy_pb2_grpc.py +424 -0
- omium_contracts-0.1.0/pyproject.toml +23 -0
- omium_contracts-0.1.0/setup.cfg +4 -0
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
pyproject.toml
|
|
2
|
+
omium_contracts.egg-info/PKG-INFO
|
|
3
|
+
omium_contracts.egg-info/SOURCES.txt
|
|
4
|
+
omium_contracts.egg-info/dependency_links.txt
|
|
5
|
+
omium_contracts.egg-info/requires.txt
|
|
6
|
+
omium_contracts.egg-info/top_level.txt
|
|
7
|
+
omiumpb/agent/v1/agent_pb2.py
|
|
8
|
+
omiumpb/agent/v1/agent_pb2.pyi
|
|
9
|
+
omiumpb/agent/v1/agent_pb2_grpc.py
|
|
10
|
+
omiumpb/checkpoint/v1/checkpoint_pb2.py
|
|
11
|
+
omiumpb/checkpoint/v1/checkpoint_pb2.pyi
|
|
12
|
+
omiumpb/checkpoint/v1/checkpoint_pb2_grpc.py
|
|
13
|
+
omiumpb/common/v1/common_pb2.py
|
|
14
|
+
omiumpb/common/v1/common_pb2.pyi
|
|
15
|
+
omiumpb/common/v1/common_pb2_grpc.py
|
|
16
|
+
omiumpb/consensus/v1/consensus_pb2.py
|
|
17
|
+
omiumpb/consensus/v1/consensus_pb2.pyi
|
|
18
|
+
omiumpb/consensus/v1/consensus_pb2_grpc.py
|
|
19
|
+
omiumpb/events/v1/events_pb2.py
|
|
20
|
+
omiumpb/events/v1/events_pb2.pyi
|
|
21
|
+
omiumpb/events/v1/events_pb2_grpc.py
|
|
22
|
+
omiumpb/execution/v1/execution_pb2.py
|
|
23
|
+
omiumpb/execution/v1/execution_pb2.pyi
|
|
24
|
+
omiumpb/execution/v1/execution_pb2_grpc.py
|
|
25
|
+
omiumpb/policy/v1/policy_pb2.py
|
|
26
|
+
omiumpb/policy/v1/policy_pb2.pyi
|
|
27
|
+
omiumpb/policy/v1/policy_pb2_grpc.py
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
omiumpb
|
|
@@ -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: omiumpb/agent/v1/agent.proto
|
|
5
|
+
# Protobuf Python Version: 5.29.3
|
|
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
|
+
3,
|
|
17
|
+
'',
|
|
18
|
+
'omiumpb/agent/v1/agent.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\x1comiumpb/agent/v1/agent.proto\x12\x10omiumpb.agent.v1\"\xd6\x02\n\x0e\x45xecuteRequest\x12!\n\x0c\x65xecution_id\x18\x01 \x01(\tR\x0b\x65xecutionId\x12#\n\rworkflow_type\x18\x02 \x01(\tR\x0cworkflowType\x12\x38\n\x18workflow_definition_json\x18\x03 \x01(\tR\x16workflowDefinitionJson\x12\x1f\n\x0binputs_json\x18\x04 \x01(\tR\ninputsJson\x12\x1b\n\ttenant_id\x18\x05 \x01(\tR\x08tenantId\x12H\n\x08\x65nv_vars\x18\x06 \x03(\x0b\x32-.omiumpb.agent.v1.ExecuteRequest.EnvVarsEntryR\x07\x65nvVars\x1a:\n\x0c\x45nvVarsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\"e\n\x0f\x45xecuteResponse\x12\x15\n\x06job_id\x18\x01 \x01(\tR\x05jobId\x12\x16\n\x06status\x18\x02 \x01(\tR\x06status\x12#\n\rerror_message\x18\x03 \x01(\tR\x0c\x65rrorMessage\"*\n\x11StreamLogsRequest\x12\x15\n\x06job_id\x18\x01 \x01(\tR\x05jobId\"\x88\x01\n\x08LogEntry\x12\x1c\n\ttimestamp\x18\x01 \x01(\tR\ttimestamp\x12\x14\n\x05level\x18\x02 \x01(\tR\x05level\x12\x18\n\x07message\x18\x03 \x01(\tR\x07message\x12\x12\n\x04step\x18\x04 \x01(\tR\x04step\x12\x1a\n\x08progress\x18\x05 \x01(\x02R\x08progress\")\n\x10GetStatusRequest\x12\x15\n\x06job_id\x18\x01 \x01(\tR\x05jobId\"\x88\x01\n\x11GetStatusResponse\x12\x15\n\x06job_id\x18\x01 \x01(\tR\x05jobId\x12\x16\n\x06status\x18\x02 \x01(\tR\x06status\x12\x1f\n\x0bresult_json\x18\x03 \x01(\tR\nresultJson\x12#\n\rerror_message\x18\x04 \x01(\tR\x0c\x65rrorMessage\"$\n\x0bStopRequest\x12\x15\n\x06job_id\x18\x01 \x01(\tR\x05jobId\"B\n\x0cStopResponse\x12\x18\n\x07success\x18\x01 \x01(\x08R\x07success\x12\x18\n\x07message\x18\x02 \x01(\tR\x07message2\xcc\x02\n\x0c\x41gentService\x12N\n\x07\x45xecute\x12 .omiumpb.agent.v1.ExecuteRequest\x1a!.omiumpb.agent.v1.ExecuteResponse\x12O\n\nStreamLogs\x12#.omiumpb.agent.v1.StreamLogsRequest\x1a\x1a.omiumpb.agent.v1.LogEntry0\x01\x12T\n\tGetStatus\x12\".omiumpb.agent.v1.GetStatusRequest\x1a#.omiumpb.agent.v1.GetStatusResponse\x12\x45\n\x04Stop\x12\x1d.omiumpb.agent.v1.StopRequest\x1a\x1e.omiumpb.agent.v1.StopResponseB\xd5\x01\n\x14\x63om.omiumpb.agent.v1B\nAgentProtoP\x01ZOgithub.com/Omium-ai/Omium-platform/shared/proto/gen/go/omiumpb/agent/v1;agentv1\xa2\x02\x03OAX\xaa\x02\x10Omiumpb.Agent.V1\xca\x02\x10Omiumpb\\Agent\\V1\xe2\x02\x1cOmiumpb\\Agent\\V1\\GPBMetadata\xea\x02\x12Omiumpb::Agent::V1b\x06proto3')
|
|
28
|
+
|
|
29
|
+
_globals = globals()
|
|
30
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
31
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'omiumpb.agent.v1.agent_pb2', _globals)
|
|
32
|
+
if not _descriptor._USE_C_DESCRIPTORS:
|
|
33
|
+
_globals['DESCRIPTOR']._loaded_options = None
|
|
34
|
+
_globals['DESCRIPTOR']._serialized_options = b'\n\024com.omiumpb.agent.v1B\nAgentProtoP\001ZOgithub.com/Omium-ai/Omium-platform/shared/proto/gen/go/omiumpb/agent/v1;agentv1\242\002\003OAX\252\002\020Omiumpb.Agent.V1\312\002\020Omiumpb\\Agent\\V1\342\002\034Omiumpb\\Agent\\V1\\GPBMetadata\352\002\022Omiumpb::Agent::V1'
|
|
35
|
+
_globals['_EXECUTEREQUEST_ENVVARSENTRY']._loaded_options = None
|
|
36
|
+
_globals['_EXECUTEREQUEST_ENVVARSENTRY']._serialized_options = b'8\001'
|
|
37
|
+
_globals['_EXECUTEREQUEST']._serialized_start=51
|
|
38
|
+
_globals['_EXECUTEREQUEST']._serialized_end=393
|
|
39
|
+
_globals['_EXECUTEREQUEST_ENVVARSENTRY']._serialized_start=335
|
|
40
|
+
_globals['_EXECUTEREQUEST_ENVVARSENTRY']._serialized_end=393
|
|
41
|
+
_globals['_EXECUTERESPONSE']._serialized_start=395
|
|
42
|
+
_globals['_EXECUTERESPONSE']._serialized_end=496
|
|
43
|
+
_globals['_STREAMLOGSREQUEST']._serialized_start=498
|
|
44
|
+
_globals['_STREAMLOGSREQUEST']._serialized_end=540
|
|
45
|
+
_globals['_LOGENTRY']._serialized_start=543
|
|
46
|
+
_globals['_LOGENTRY']._serialized_end=679
|
|
47
|
+
_globals['_GETSTATUSREQUEST']._serialized_start=681
|
|
48
|
+
_globals['_GETSTATUSREQUEST']._serialized_end=722
|
|
49
|
+
_globals['_GETSTATUSRESPONSE']._serialized_start=725
|
|
50
|
+
_globals['_GETSTATUSRESPONSE']._serialized_end=861
|
|
51
|
+
_globals['_STOPREQUEST']._serialized_start=863
|
|
52
|
+
_globals['_STOPREQUEST']._serialized_end=899
|
|
53
|
+
_globals['_STOPRESPONSE']._serialized_start=901
|
|
54
|
+
_globals['_STOPRESPONSE']._serialized_end=967
|
|
55
|
+
_globals['_AGENTSERVICE']._serialized_start=970
|
|
56
|
+
_globals['_AGENTSERVICE']._serialized_end=1302
|
|
57
|
+
# @@protoc_insertion_point(module_scope)
|
|
@@ -0,0 +1,91 @@
|
|
|
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, Mapping as _Mapping, Optional as _Optional
|
|
5
|
+
|
|
6
|
+
DESCRIPTOR: _descriptor.FileDescriptor
|
|
7
|
+
|
|
8
|
+
class ExecuteRequest(_message.Message):
|
|
9
|
+
__slots__ = ("execution_id", "workflow_type", "workflow_definition_json", "inputs_json", "tenant_id", "env_vars")
|
|
10
|
+
class EnvVarsEntry(_message.Message):
|
|
11
|
+
__slots__ = ("key", "value")
|
|
12
|
+
KEY_FIELD_NUMBER: _ClassVar[int]
|
|
13
|
+
VALUE_FIELD_NUMBER: _ClassVar[int]
|
|
14
|
+
key: str
|
|
15
|
+
value: str
|
|
16
|
+
def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
|
|
17
|
+
EXECUTION_ID_FIELD_NUMBER: _ClassVar[int]
|
|
18
|
+
WORKFLOW_TYPE_FIELD_NUMBER: _ClassVar[int]
|
|
19
|
+
WORKFLOW_DEFINITION_JSON_FIELD_NUMBER: _ClassVar[int]
|
|
20
|
+
INPUTS_JSON_FIELD_NUMBER: _ClassVar[int]
|
|
21
|
+
TENANT_ID_FIELD_NUMBER: _ClassVar[int]
|
|
22
|
+
ENV_VARS_FIELD_NUMBER: _ClassVar[int]
|
|
23
|
+
execution_id: str
|
|
24
|
+
workflow_type: str
|
|
25
|
+
workflow_definition_json: str
|
|
26
|
+
inputs_json: str
|
|
27
|
+
tenant_id: str
|
|
28
|
+
env_vars: _containers.ScalarMap[str, str]
|
|
29
|
+
def __init__(self, execution_id: _Optional[str] = ..., workflow_type: _Optional[str] = ..., workflow_definition_json: _Optional[str] = ..., inputs_json: _Optional[str] = ..., tenant_id: _Optional[str] = ..., env_vars: _Optional[_Mapping[str, str]] = ...) -> None: ...
|
|
30
|
+
|
|
31
|
+
class ExecuteResponse(_message.Message):
|
|
32
|
+
__slots__ = ("job_id", "status", "error_message")
|
|
33
|
+
JOB_ID_FIELD_NUMBER: _ClassVar[int]
|
|
34
|
+
STATUS_FIELD_NUMBER: _ClassVar[int]
|
|
35
|
+
ERROR_MESSAGE_FIELD_NUMBER: _ClassVar[int]
|
|
36
|
+
job_id: str
|
|
37
|
+
status: str
|
|
38
|
+
error_message: str
|
|
39
|
+
def __init__(self, job_id: _Optional[str] = ..., status: _Optional[str] = ..., error_message: _Optional[str] = ...) -> None: ...
|
|
40
|
+
|
|
41
|
+
class StreamLogsRequest(_message.Message):
|
|
42
|
+
__slots__ = ("job_id",)
|
|
43
|
+
JOB_ID_FIELD_NUMBER: _ClassVar[int]
|
|
44
|
+
job_id: str
|
|
45
|
+
def __init__(self, job_id: _Optional[str] = ...) -> None: ...
|
|
46
|
+
|
|
47
|
+
class LogEntry(_message.Message):
|
|
48
|
+
__slots__ = ("timestamp", "level", "message", "step", "progress")
|
|
49
|
+
TIMESTAMP_FIELD_NUMBER: _ClassVar[int]
|
|
50
|
+
LEVEL_FIELD_NUMBER: _ClassVar[int]
|
|
51
|
+
MESSAGE_FIELD_NUMBER: _ClassVar[int]
|
|
52
|
+
STEP_FIELD_NUMBER: _ClassVar[int]
|
|
53
|
+
PROGRESS_FIELD_NUMBER: _ClassVar[int]
|
|
54
|
+
timestamp: str
|
|
55
|
+
level: str
|
|
56
|
+
message: str
|
|
57
|
+
step: str
|
|
58
|
+
progress: float
|
|
59
|
+
def __init__(self, timestamp: _Optional[str] = ..., level: _Optional[str] = ..., message: _Optional[str] = ..., step: _Optional[str] = ..., progress: _Optional[float] = ...) -> None: ...
|
|
60
|
+
|
|
61
|
+
class GetStatusRequest(_message.Message):
|
|
62
|
+
__slots__ = ("job_id",)
|
|
63
|
+
JOB_ID_FIELD_NUMBER: _ClassVar[int]
|
|
64
|
+
job_id: str
|
|
65
|
+
def __init__(self, job_id: _Optional[str] = ...) -> None: ...
|
|
66
|
+
|
|
67
|
+
class GetStatusResponse(_message.Message):
|
|
68
|
+
__slots__ = ("job_id", "status", "result_json", "error_message")
|
|
69
|
+
JOB_ID_FIELD_NUMBER: _ClassVar[int]
|
|
70
|
+
STATUS_FIELD_NUMBER: _ClassVar[int]
|
|
71
|
+
RESULT_JSON_FIELD_NUMBER: _ClassVar[int]
|
|
72
|
+
ERROR_MESSAGE_FIELD_NUMBER: _ClassVar[int]
|
|
73
|
+
job_id: str
|
|
74
|
+
status: str
|
|
75
|
+
result_json: str
|
|
76
|
+
error_message: str
|
|
77
|
+
def __init__(self, job_id: _Optional[str] = ..., status: _Optional[str] = ..., result_json: _Optional[str] = ..., error_message: _Optional[str] = ...) -> None: ...
|
|
78
|
+
|
|
79
|
+
class StopRequest(_message.Message):
|
|
80
|
+
__slots__ = ("job_id",)
|
|
81
|
+
JOB_ID_FIELD_NUMBER: _ClassVar[int]
|
|
82
|
+
job_id: str
|
|
83
|
+
def __init__(self, job_id: _Optional[str] = ...) -> None: ...
|
|
84
|
+
|
|
85
|
+
class StopResponse(_message.Message):
|
|
86
|
+
__slots__ = ("success", "message")
|
|
87
|
+
SUCCESS_FIELD_NUMBER: _ClassVar[int]
|
|
88
|
+
MESSAGE_FIELD_NUMBER: _ClassVar[int]
|
|
89
|
+
success: bool
|
|
90
|
+
message: str
|
|
91
|
+
def __init__(self, success: bool = ..., message: _Optional[str] = ...) -> None: ...
|
|
@@ -0,0 +1,215 @@
|
|
|
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
|
+
|
|
5
|
+
from omiumpb.agent.v1 import agent_pb2 as omiumpb_dot_agent_dot_v1_dot_agent__pb2
|
|
6
|
+
|
|
7
|
+
|
|
8
|
+
class AgentServiceStub(object):
|
|
9
|
+
"""AgentService is implemented by agent Runner processes; the Execution Engine
|
|
10
|
+
is the client. (The Agent entity itself is an embedded value object — see
|
|
11
|
+
omium.execution.v1.Agent; this is the runtime runner contract.)
|
|
12
|
+
"""
|
|
13
|
+
|
|
14
|
+
def __init__(self, channel):
|
|
15
|
+
"""Constructor.
|
|
16
|
+
|
|
17
|
+
Args:
|
|
18
|
+
channel: A grpc.Channel.
|
|
19
|
+
"""
|
|
20
|
+
self.Execute = channel.unary_unary(
|
|
21
|
+
'/omiumpb.agent.v1.AgentService/Execute',
|
|
22
|
+
request_serializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.ExecuteRequest.SerializeToString,
|
|
23
|
+
response_deserializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.ExecuteResponse.FromString,
|
|
24
|
+
_registered_method=True)
|
|
25
|
+
self.StreamLogs = channel.unary_stream(
|
|
26
|
+
'/omiumpb.agent.v1.AgentService/StreamLogs',
|
|
27
|
+
request_serializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.StreamLogsRequest.SerializeToString,
|
|
28
|
+
response_deserializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.LogEntry.FromString,
|
|
29
|
+
_registered_method=True)
|
|
30
|
+
self.GetStatus = channel.unary_unary(
|
|
31
|
+
'/omiumpb.agent.v1.AgentService/GetStatus',
|
|
32
|
+
request_serializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.GetStatusRequest.SerializeToString,
|
|
33
|
+
response_deserializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.GetStatusResponse.FromString,
|
|
34
|
+
_registered_method=True)
|
|
35
|
+
self.Stop = channel.unary_unary(
|
|
36
|
+
'/omiumpb.agent.v1.AgentService/Stop',
|
|
37
|
+
request_serializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.StopRequest.SerializeToString,
|
|
38
|
+
response_deserializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.StopResponse.FromString,
|
|
39
|
+
_registered_method=True)
|
|
40
|
+
|
|
41
|
+
|
|
42
|
+
class AgentServiceServicer(object):
|
|
43
|
+
"""AgentService is implemented by agent Runner processes; the Execution Engine
|
|
44
|
+
is the client. (The Agent entity itself is an embedded value object — see
|
|
45
|
+
omium.execution.v1.Agent; this is the runtime runner contract.)
|
|
46
|
+
"""
|
|
47
|
+
|
|
48
|
+
def Execute(self, request, context):
|
|
49
|
+
"""Missing associated documentation comment in .proto file."""
|
|
50
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
51
|
+
context.set_details('Method not implemented!')
|
|
52
|
+
raise NotImplementedError('Method not implemented!')
|
|
53
|
+
|
|
54
|
+
def StreamLogs(self, request, context):
|
|
55
|
+
"""Missing associated documentation comment in .proto file."""
|
|
56
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
57
|
+
context.set_details('Method not implemented!')
|
|
58
|
+
raise NotImplementedError('Method not implemented!')
|
|
59
|
+
|
|
60
|
+
def GetStatus(self, request, context):
|
|
61
|
+
"""Missing associated documentation comment in .proto file."""
|
|
62
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
63
|
+
context.set_details('Method not implemented!')
|
|
64
|
+
raise NotImplementedError('Method not implemented!')
|
|
65
|
+
|
|
66
|
+
def Stop(self, request, context):
|
|
67
|
+
"""Missing associated documentation comment in .proto file."""
|
|
68
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
69
|
+
context.set_details('Method not implemented!')
|
|
70
|
+
raise NotImplementedError('Method not implemented!')
|
|
71
|
+
|
|
72
|
+
|
|
73
|
+
def add_AgentServiceServicer_to_server(servicer, server):
|
|
74
|
+
rpc_method_handlers = {
|
|
75
|
+
'Execute': grpc.unary_unary_rpc_method_handler(
|
|
76
|
+
servicer.Execute,
|
|
77
|
+
request_deserializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.ExecuteRequest.FromString,
|
|
78
|
+
response_serializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.ExecuteResponse.SerializeToString,
|
|
79
|
+
),
|
|
80
|
+
'StreamLogs': grpc.unary_stream_rpc_method_handler(
|
|
81
|
+
servicer.StreamLogs,
|
|
82
|
+
request_deserializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.StreamLogsRequest.FromString,
|
|
83
|
+
response_serializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.LogEntry.SerializeToString,
|
|
84
|
+
),
|
|
85
|
+
'GetStatus': grpc.unary_unary_rpc_method_handler(
|
|
86
|
+
servicer.GetStatus,
|
|
87
|
+
request_deserializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.GetStatusRequest.FromString,
|
|
88
|
+
response_serializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.GetStatusResponse.SerializeToString,
|
|
89
|
+
),
|
|
90
|
+
'Stop': grpc.unary_unary_rpc_method_handler(
|
|
91
|
+
servicer.Stop,
|
|
92
|
+
request_deserializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.StopRequest.FromString,
|
|
93
|
+
response_serializer=omiumpb_dot_agent_dot_v1_dot_agent__pb2.StopResponse.SerializeToString,
|
|
94
|
+
),
|
|
95
|
+
}
|
|
96
|
+
generic_handler = grpc.method_handlers_generic_handler(
|
|
97
|
+
'omiumpb.agent.v1.AgentService', rpc_method_handlers)
|
|
98
|
+
server.add_generic_rpc_handlers((generic_handler,))
|
|
99
|
+
server.add_registered_method_handlers('omiumpb.agent.v1.AgentService', rpc_method_handlers)
|
|
100
|
+
|
|
101
|
+
|
|
102
|
+
# This class is part of an EXPERIMENTAL API.
|
|
103
|
+
class AgentService(object):
|
|
104
|
+
"""AgentService is implemented by agent Runner processes; the Execution Engine
|
|
105
|
+
is the client. (The Agent entity itself is an embedded value object — see
|
|
106
|
+
omium.execution.v1.Agent; this is the runtime runner contract.)
|
|
107
|
+
"""
|
|
108
|
+
|
|
109
|
+
@staticmethod
|
|
110
|
+
def Execute(request,
|
|
111
|
+
target,
|
|
112
|
+
options=(),
|
|
113
|
+
channel_credentials=None,
|
|
114
|
+
call_credentials=None,
|
|
115
|
+
insecure=False,
|
|
116
|
+
compression=None,
|
|
117
|
+
wait_for_ready=None,
|
|
118
|
+
timeout=None,
|
|
119
|
+
metadata=None):
|
|
120
|
+
return grpc.experimental.unary_unary(
|
|
121
|
+
request,
|
|
122
|
+
target,
|
|
123
|
+
'/omiumpb.agent.v1.AgentService/Execute',
|
|
124
|
+
omiumpb_dot_agent_dot_v1_dot_agent__pb2.ExecuteRequest.SerializeToString,
|
|
125
|
+
omiumpb_dot_agent_dot_v1_dot_agent__pb2.ExecuteResponse.FromString,
|
|
126
|
+
options,
|
|
127
|
+
channel_credentials,
|
|
128
|
+
insecure,
|
|
129
|
+
call_credentials,
|
|
130
|
+
compression,
|
|
131
|
+
wait_for_ready,
|
|
132
|
+
timeout,
|
|
133
|
+
metadata,
|
|
134
|
+
_registered_method=True)
|
|
135
|
+
|
|
136
|
+
@staticmethod
|
|
137
|
+
def StreamLogs(request,
|
|
138
|
+
target,
|
|
139
|
+
options=(),
|
|
140
|
+
channel_credentials=None,
|
|
141
|
+
call_credentials=None,
|
|
142
|
+
insecure=False,
|
|
143
|
+
compression=None,
|
|
144
|
+
wait_for_ready=None,
|
|
145
|
+
timeout=None,
|
|
146
|
+
metadata=None):
|
|
147
|
+
return grpc.experimental.unary_stream(
|
|
148
|
+
request,
|
|
149
|
+
target,
|
|
150
|
+
'/omiumpb.agent.v1.AgentService/StreamLogs',
|
|
151
|
+
omiumpb_dot_agent_dot_v1_dot_agent__pb2.StreamLogsRequest.SerializeToString,
|
|
152
|
+
omiumpb_dot_agent_dot_v1_dot_agent__pb2.LogEntry.FromString,
|
|
153
|
+
options,
|
|
154
|
+
channel_credentials,
|
|
155
|
+
insecure,
|
|
156
|
+
call_credentials,
|
|
157
|
+
compression,
|
|
158
|
+
wait_for_ready,
|
|
159
|
+
timeout,
|
|
160
|
+
metadata,
|
|
161
|
+
_registered_method=True)
|
|
162
|
+
|
|
163
|
+
@staticmethod
|
|
164
|
+
def GetStatus(request,
|
|
165
|
+
target,
|
|
166
|
+
options=(),
|
|
167
|
+
channel_credentials=None,
|
|
168
|
+
call_credentials=None,
|
|
169
|
+
insecure=False,
|
|
170
|
+
compression=None,
|
|
171
|
+
wait_for_ready=None,
|
|
172
|
+
timeout=None,
|
|
173
|
+
metadata=None):
|
|
174
|
+
return grpc.experimental.unary_unary(
|
|
175
|
+
request,
|
|
176
|
+
target,
|
|
177
|
+
'/omiumpb.agent.v1.AgentService/GetStatus',
|
|
178
|
+
omiumpb_dot_agent_dot_v1_dot_agent__pb2.GetStatusRequest.SerializeToString,
|
|
179
|
+
omiumpb_dot_agent_dot_v1_dot_agent__pb2.GetStatusResponse.FromString,
|
|
180
|
+
options,
|
|
181
|
+
channel_credentials,
|
|
182
|
+
insecure,
|
|
183
|
+
call_credentials,
|
|
184
|
+
compression,
|
|
185
|
+
wait_for_ready,
|
|
186
|
+
timeout,
|
|
187
|
+
metadata,
|
|
188
|
+
_registered_method=True)
|
|
189
|
+
|
|
190
|
+
@staticmethod
|
|
191
|
+
def Stop(request,
|
|
192
|
+
target,
|
|
193
|
+
options=(),
|
|
194
|
+
channel_credentials=None,
|
|
195
|
+
call_credentials=None,
|
|
196
|
+
insecure=False,
|
|
197
|
+
compression=None,
|
|
198
|
+
wait_for_ready=None,
|
|
199
|
+
timeout=None,
|
|
200
|
+
metadata=None):
|
|
201
|
+
return grpc.experimental.unary_unary(
|
|
202
|
+
request,
|
|
203
|
+
target,
|
|
204
|
+
'/omiumpb.agent.v1.AgentService/Stop',
|
|
205
|
+
omiumpb_dot_agent_dot_v1_dot_agent__pb2.StopRequest.SerializeToString,
|
|
206
|
+
omiumpb_dot_agent_dot_v1_dot_agent__pb2.StopResponse.FromString,
|
|
207
|
+
options,
|
|
208
|
+
channel_credentials,
|
|
209
|
+
insecure,
|
|
210
|
+
call_credentials,
|
|
211
|
+
compression,
|
|
212
|
+
wait_for_ready,
|
|
213
|
+
timeout,
|
|
214
|
+
metadata,
|
|
215
|
+
_registered_method=True)
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
# -*- coding: utf-8 -*-
|
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
3
|
+
# NO CHECKED-IN PROTOBUF GENCODE
|
|
4
|
+
# source: omiumpb/checkpoint/v1/checkpoint.proto
|
|
5
|
+
# Protobuf Python Version: 5.29.3
|
|
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
|
+
3,
|
|
17
|
+
'',
|
|
18
|
+
'omiumpb/checkpoint/v1/checkpoint.proto'
|
|
19
|
+
)
|
|
20
|
+
# @@protoc_insertion_point(imports)
|
|
21
|
+
|
|
22
|
+
_sym_db = _symbol_database.Default()
|
|
23
|
+
|
|
24
|
+
|
|
25
|
+
from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
|
|
26
|
+
from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
|
|
27
|
+
|
|
28
|
+
|
|
29
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n&omiumpb/checkpoint/v1/checkpoint.proto\x12\x15omiumpb.checkpoint.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1cgoogle/protobuf/struct.proto\"\x88\x05\n\nCheckpoint\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12!\n\x0c\x65xecution_id\x18\x02 \x01(\tR\x0b\x65xecutionId\x12\x19\n\x08\x61gent_id\x18\x03 \x01(\tR\x07\x61gentId\x12\'\n\x0f\x63heckpoint_name\x18\x04 \x01(\tR\x0e\x63heckpointName\x12)\n\x10\x63heckpoint_index\x18\x05 \x01(\x05R\x0f\x63heckpointIndex\x12(\n\x10state_size_bytes\x18\x06 \x01(\x03R\x0estateSizeBytes\x12$\n\x0estate_blob_uri\x18\x07 \x01(\tR\x0cstateBlobUri\x12\x1a\n\x08\x63hecksum\x18\x08 \x01(\tR\x08\x63hecksum\x12$\n\rpreconditions\x18\t \x03(\tR\rpreconditions\x12&\n\x0epostconditions\x18\n \x03(\tR\x0epostconditions\x12+\n\x11validation_passed\x18\x0b \x01(\x08R\x10validationPassed\x12\x33\n\x08metadata\x18\x0c \x01(\x0b\x32\x17.google.protobuf.StructR\x08metadata\x12)\n\x10\x63ompression_type\x18\r \x01(\tR\x0f\x63ompressionType\x12\x39\n\ncreated_at\x18\x0e \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nexpires_at\x18\x0f \x01(\x0b\x32\x1a.google.protobuf.TimestampR\texpiresAt\x12\x1b\n\ttenant_id\x18\x10 \x01(\tR\x08tenantId\"\x9b\x03\n\x17\x43reateCheckpointRequest\x12!\n\x0c\x65xecution_id\x18\x01 \x01(\tR\x0b\x65xecutionId\x12\x19\n\x08\x61gent_id\x18\x02 \x01(\tR\x07\x61gentId\x12\'\n\x0f\x63heckpoint_name\x18\x03 \x01(\tR\x0e\x63heckpointName\x12-\n\x05state\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\x05state\x12$\n\rpreconditions\x18\x05 \x03(\tR\rpreconditions\x12&\n\x0epostconditions\x18\x06 \x03(\tR\x0epostconditions\x12\x33\n\x08metadata\x18\x07 \x01(\x0b\x32\x17.google.protobuf.StructR\x08metadata\x12)\n\x10\x63ompression_type\x18\x08 \x01(\tR\x0f\x63ompressionType\x12\x1f\n\x0bttl_seconds\x18\t \x01(\x03R\nttlSeconds\x12\x1b\n\ttenant_id\x18\n \x01(\tR\x08tenantId\"\x9c\x01\n\x18\x43reateCheckpointResponse\x12\x41\n\ncheckpoint\x18\x01 \x01(\x0b\x32!.omiumpb.checkpoint.v1.CheckpointR\ncheckpoint\x12\x18\n\x07success\x18\x02 \x01(\x08R\x07success\x12#\n\rerror_message\x18\x03 \x01(\tR\x0c\x65rrorMessage\"}\n\x14GetCheckpointRequest\x12#\n\rcheckpoint_id\x18\x01 \x01(\tR\x0c\x63heckpointId\x12#\n\rinclude_state\x18\x02 \x01(\x08R\x0cincludeState\x12\x1b\n\ttenant_id\x18\x03 \x01(\tR\x08tenantId\"\xc8\x01\n\x15GetCheckpointResponse\x12\x41\n\ncheckpoint\x18\x01 \x01(\x0b\x32!.omiumpb.checkpoint.v1.CheckpointR\ncheckpoint\x12-\n\x05state\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructR\x05state\x12\x18\n\x07success\x18\x03 \x01(\x08R\x07success\x12#\n\rerror_message\x18\x04 \x01(\tR\x0c\x65rrorMessage\"\xda\x01\n\x16ListCheckpointsRequest\x12!\n\x0c\x65xecution_id\x18\x01 \x01(\tR\x0b\x65xecutionId\x12\x19\n\x08\x61gent_id\x18\x02 \x01(\tR\x07\x61gentId\x12\x14\n\x05limit\x18\x03 \x01(\x05R\x05limit\x12\x16\n\x06offset\x18\x04 \x01(\x05R\x06offset\x12\x19\n\x08order_by\x18\x05 \x01(\tR\x07orderBy\x12\x1c\n\tascending\x18\x06 \x01(\x08R\tascending\x12\x1b\n\ttenant_id\x18\x07 \x01(\tR\x08tenantId\"\xbe\x01\n\x17ListCheckpointsResponse\x12\x43\n\x0b\x63heckpoints\x18\x01 \x03(\x0b\x32!.omiumpb.checkpoint.v1.CheckpointR\x0b\x63heckpoints\x12\x1f\n\x0btotal_count\x18\x02 \x01(\x05R\ntotalCount\x12\x18\n\x07success\x18\x03 \x01(\x08R\x07success\x12#\n\rerror_message\x18\x04 \x01(\tR\x0c\x65rrorMessage\"\xa2\x02\n\x1bRollbackToCheckpointRequest\x12!\n\x0c\x65xecution_id\x18\x01 \x01(\tR\x0b\x65xecutionId\x12#\n\rcheckpoint_id\x18\x02 \x01(\tR\x0c\x63heckpointId\x12!\n\x0ctriggered_by\x18\x03 \x01(\tR\x0btriggeredBy\x12%\n\x0etrigger_reason\x18\x04 \x01(\tR\rtriggerReason\x12!\n\x0ctrigger_type\x18\x05 \x01(\tR\x0btriggerType\x12\x31\n\x07options\x18\x06 \x01(\x0b\x32\x17.google.protobuf.StructR\x07options\x12\x1b\n\ttenant_id\x18\x07 \x01(\tR\x08tenantId\"\x9d\x02\n\x1cRollbackToCheckpointResponse\x12\x1f\n\x0brollback_id\x18\x01 \x01(\tR\nrollbackId\x12N\n\x11target_checkpoint\x18\x02 \x01(\x0b\x32!.omiumpb.checkpoint.v1.CheckpointR\x10targetCheckpoint\x12\x18\n\x07success\x18\x03 \x01(\x08R\x07success\x12#\n\rerror_message\x18\x04 \x01(\tR\x0c\x65rrorMessage\x12M\n\x14\x65stimated_completion\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x13\x65stimatedCompletion\"q\n\x17\x44\x65leteCheckpointRequest\x12#\n\rcheckpoint_id\x18\x01 \x01(\tR\x0c\x63heckpointId\x12\x14\n\x05\x66orce\x18\x02 \x01(\x08R\x05\x66orce\x12\x1b\n\ttenant_id\x18\x03 \x01(\tR\x08tenantId\"Y\n\x18\x44\x65leteCheckpointResponse\x12\x18\n\x07success\x18\x01 \x01(\x08R\x07success\x12#\n\rerror_message\x18\x02 \x01(\tR\x0c\x65rrorMessage\"[\n\x17VerifyCheckpointRequest\x12#\n\rcheckpoint_id\x18\x01 \x01(\tR\x0c\x63heckpointId\x12\x1b\n\ttenant_id\x18\x02 \x01(\tR\x08tenantId\"\x9d\x01\n\x18VerifyCheckpointResponse\x12\x14\n\x05valid\x18\x01 \x01(\x08R\x05valid\x12%\n\x0e\x63hecksum_match\x18\x02 \x01(\tR\rchecksumMatch\x12\x1f\n\x0b\x62lob_exists\x18\x03 \x01(\x08R\nblobExists\x12#\n\rerror_message\x18\x04 \x01(\tR\x0c\x65rrorMessage2\xd1\x05\n\x11\x43heckpointService\x12s\n\x10\x43reateCheckpoint\x12..omiumpb.checkpoint.v1.CreateCheckpointRequest\x1a/.omiumpb.checkpoint.v1.CreateCheckpointResponse\x12j\n\rGetCheckpoint\x12+.omiumpb.checkpoint.v1.GetCheckpointRequest\x1a,.omiumpb.checkpoint.v1.GetCheckpointResponse\x12p\n\x0fListCheckpoints\x12-.omiumpb.checkpoint.v1.ListCheckpointsRequest\x1a..omiumpb.checkpoint.v1.ListCheckpointsResponse\x12\x7f\n\x14RollbackToCheckpoint\x12\x32.omiumpb.checkpoint.v1.RollbackToCheckpointRequest\x1a\x33.omiumpb.checkpoint.v1.RollbackToCheckpointResponse\x12s\n\x10\x44\x65leteCheckpoint\x12..omiumpb.checkpoint.v1.DeleteCheckpointRequest\x1a/.omiumpb.checkpoint.v1.DeleteCheckpointResponse\x12s\n\x10VerifyCheckpoint\x12..omiumpb.checkpoint.v1.VerifyCheckpointRequest\x1a/.omiumpb.checkpoint.v1.VerifyCheckpointResponseB\xfd\x01\n\x19\x63om.omiumpb.checkpoint.v1B\x0f\x43heckpointProtoP\x01ZYgithub.com/Omium-ai/Omium-platform/shared/proto/gen/go/omiumpb/checkpoint/v1;checkpointv1\xa2\x02\x03OCX\xaa\x02\x15Omiumpb.Checkpoint.V1\xca\x02\x15Omiumpb\\Checkpoint\\V1\xe2\x02!Omiumpb\\Checkpoint\\V1\\GPBMetadata\xea\x02\x17Omiumpb::Checkpoint::V1b\x06proto3')
|
|
30
|
+
|
|
31
|
+
_globals = globals()
|
|
32
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
33
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'omiumpb.checkpoint.v1.checkpoint_pb2', _globals)
|
|
34
|
+
if not _descriptor._USE_C_DESCRIPTORS:
|
|
35
|
+
_globals['DESCRIPTOR']._loaded_options = None
|
|
36
|
+
_globals['DESCRIPTOR']._serialized_options = b'\n\031com.omiumpb.checkpoint.v1B\017CheckpointProtoP\001ZYgithub.com/Omium-ai/Omium-platform/shared/proto/gen/go/omiumpb/checkpoint/v1;checkpointv1\242\002\003OCX\252\002\025Omiumpb.Checkpoint.V1\312\002\025Omiumpb\\Checkpoint\\V1\342\002!Omiumpb\\Checkpoint\\V1\\GPBMetadata\352\002\027Omiumpb::Checkpoint::V1'
|
|
37
|
+
_globals['_CHECKPOINT']._serialized_start=129
|
|
38
|
+
_globals['_CHECKPOINT']._serialized_end=777
|
|
39
|
+
_globals['_CREATECHECKPOINTREQUEST']._serialized_start=780
|
|
40
|
+
_globals['_CREATECHECKPOINTREQUEST']._serialized_end=1191
|
|
41
|
+
_globals['_CREATECHECKPOINTRESPONSE']._serialized_start=1194
|
|
42
|
+
_globals['_CREATECHECKPOINTRESPONSE']._serialized_end=1350
|
|
43
|
+
_globals['_GETCHECKPOINTREQUEST']._serialized_start=1352
|
|
44
|
+
_globals['_GETCHECKPOINTREQUEST']._serialized_end=1477
|
|
45
|
+
_globals['_GETCHECKPOINTRESPONSE']._serialized_start=1480
|
|
46
|
+
_globals['_GETCHECKPOINTRESPONSE']._serialized_end=1680
|
|
47
|
+
_globals['_LISTCHECKPOINTSREQUEST']._serialized_start=1683
|
|
48
|
+
_globals['_LISTCHECKPOINTSREQUEST']._serialized_end=1901
|
|
49
|
+
_globals['_LISTCHECKPOINTSRESPONSE']._serialized_start=1904
|
|
50
|
+
_globals['_LISTCHECKPOINTSRESPONSE']._serialized_end=2094
|
|
51
|
+
_globals['_ROLLBACKTOCHECKPOINTREQUEST']._serialized_start=2097
|
|
52
|
+
_globals['_ROLLBACKTOCHECKPOINTREQUEST']._serialized_end=2387
|
|
53
|
+
_globals['_ROLLBACKTOCHECKPOINTRESPONSE']._serialized_start=2390
|
|
54
|
+
_globals['_ROLLBACKTOCHECKPOINTRESPONSE']._serialized_end=2675
|
|
55
|
+
_globals['_DELETECHECKPOINTREQUEST']._serialized_start=2677
|
|
56
|
+
_globals['_DELETECHECKPOINTREQUEST']._serialized_end=2790
|
|
57
|
+
_globals['_DELETECHECKPOINTRESPONSE']._serialized_start=2792
|
|
58
|
+
_globals['_DELETECHECKPOINTRESPONSE']._serialized_end=2881
|
|
59
|
+
_globals['_VERIFYCHECKPOINTREQUEST']._serialized_start=2883
|
|
60
|
+
_globals['_VERIFYCHECKPOINTREQUEST']._serialized_end=2974
|
|
61
|
+
_globals['_VERIFYCHECKPOINTRESPONSE']._serialized_start=2977
|
|
62
|
+
_globals['_VERIFYCHECKPOINTRESPONSE']._serialized_end=3134
|
|
63
|
+
_globals['_CHECKPOINTSERVICE']._serialized_start=3137
|
|
64
|
+
_globals['_CHECKPOINTSERVICE']._serialized_end=3858
|
|
65
|
+
# @@protoc_insertion_point(module_scope)
|