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.
Files changed (29) hide show
  1. omium_contracts-0.1.0/PKG-INFO +7 -0
  2. omium_contracts-0.1.0/omium_contracts.egg-info/PKG-INFO +7 -0
  3. omium_contracts-0.1.0/omium_contracts.egg-info/SOURCES.txt +27 -0
  4. omium_contracts-0.1.0/omium_contracts.egg-info/dependency_links.txt +1 -0
  5. omium_contracts-0.1.0/omium_contracts.egg-info/requires.txt +2 -0
  6. omium_contracts-0.1.0/omium_contracts.egg-info/top_level.txt +1 -0
  7. omium_contracts-0.1.0/omiumpb/agent/v1/agent_pb2.py +57 -0
  8. omium_contracts-0.1.0/omiumpb/agent/v1/agent_pb2.pyi +91 -0
  9. omium_contracts-0.1.0/omiumpb/agent/v1/agent_pb2_grpc.py +215 -0
  10. omium_contracts-0.1.0/omiumpb/checkpoint/v1/checkpoint_pb2.py +65 -0
  11. omium_contracts-0.1.0/omiumpb/checkpoint/v1/checkpoint_pb2.pyi +200 -0
  12. omium_contracts-0.1.0/omiumpb/checkpoint/v1/checkpoint_pb2_grpc.py +295 -0
  13. omium_contracts-0.1.0/omiumpb/common/v1/common_pb2.py +49 -0
  14. omium_contracts-0.1.0/omiumpb/common/v1/common_pb2.pyi +73 -0
  15. omium_contracts-0.1.0/omiumpb/common/v1/common_pb2_grpc.py +4 -0
  16. omium_contracts-0.1.0/omiumpb/consensus/v1/consensus_pb2.py +89 -0
  17. omium_contracts-0.1.0/omiumpb/consensus/v1/consensus_pb2.pyi +345 -0
  18. omium_contracts-0.1.0/omiumpb/consensus/v1/consensus_pb2_grpc.py +462 -0
  19. omium_contracts-0.1.0/omiumpb/events/v1/events_pb2.py +49 -0
  20. omium_contracts-0.1.0/omiumpb/events/v1/events_pb2.pyi +99 -0
  21. omium_contracts-0.1.0/omiumpb/events/v1/events_pb2_grpc.py +4 -0
  22. omium_contracts-0.1.0/omiumpb/execution/v1/execution_pb2.py +71 -0
  23. omium_contracts-0.1.0/omiumpb/execution/v1/execution_pb2.pyi +253 -0
  24. omium_contracts-0.1.0/omiumpb/execution/v1/execution_pb2_grpc.py +295 -0
  25. omium_contracts-0.1.0/omiumpb/policy/v1/policy_pb2.py +105 -0
  26. omium_contracts-0.1.0/omiumpb/policy/v1/policy_pb2.pyi +381 -0
  27. omium_contracts-0.1.0/omiumpb/policy/v1/policy_pb2_grpc.py +424 -0
  28. omium_contracts-0.1.0/pyproject.toml +23 -0
  29. omium_contracts-0.1.0/setup.cfg +4 -0
@@ -0,0 +1,7 @@
1
+ Metadata-Version: 2.4
2
+ Name: omium-contracts
3
+ Version: 0.1.0
4
+ Summary: Generated Omium protobuf/gRPC contracts (single source of truth).
5
+ Requires-Python: >=3.11
6
+ Requires-Dist: protobuf<6,>=5.27
7
+ Requires-Dist: grpcio>=1.66
@@ -0,0 +1,7 @@
1
+ Metadata-Version: 2.4
2
+ Name: omium-contracts
3
+ Version: 0.1.0
4
+ Summary: Generated Omium protobuf/gRPC contracts (single source of truth).
5
+ Requires-Python: >=3.11
6
+ Requires-Dist: protobuf<6,>=5.27
7
+ Requires-Dist: grpcio>=1.66
@@ -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,2 @@
1
+ protobuf<6,>=5.27
2
+ grpcio>=1.66
@@ -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)