sondera-harness 0.6.0__py3-none-any.whl

sondera/proto/sondera/core/v1/primitives_pb2.pyi

@@ -0,0 +1,259 @@
+import datetime
+
+from google.protobuf import struct_pb2 as _struct_pb2
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class TrajectoryStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    TRAJECTORY_STATUS_UNSPECIFIED: _ClassVar[TrajectoryStatus]
+    TRAJECTORY_STATUS_PENDING: _ClassVar[TrajectoryStatus]
+    TRAJECTORY_STATUS_RUNNING: _ClassVar[TrajectoryStatus]
+    TRAJECTORY_STATUS_COMPLETED: _ClassVar[TrajectoryStatus]
+    TRAJECTORY_STATUS_SUSPENDED: _ClassVar[TrajectoryStatus]
+    TRAJECTORY_STATUS_FAILED: _ClassVar[TrajectoryStatus]
+
+class Stage(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    STAGE_UNSPECIFIED: _ClassVar[Stage]
+    STAGE_PRE_RUN: _ClassVar[Stage]
+    STAGE_PRE_MODEL: _ClassVar[Stage]
+    STAGE_POST_MODEL: _ClassVar[Stage]
+    STAGE_PRE_TOOL: _ClassVar[Stage]
+    STAGE_POST_TOOL: _ClassVar[Stage]
+    STAGE_POST_RUN: _ClassVar[Stage]
+
+class Role(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    ROLE_UNSPECIFIED: _ClassVar[Role]
+    ROLE_USER: _ClassVar[Role]
+    ROLE_MODEL: _ClassVar[Role]
+    ROLE_TOOL: _ClassVar[Role]
+    ROLE_SYSTEM: _ClassVar[Role]
+
+class Decision(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    DECISION_UNSPECIFIED: _ClassVar[Decision]
+    DECISION_ALLOW: _ClassVar[Decision]
+    DECISION_DENY: _ClassVar[Decision]
+    DECISION_ESCALATE: _ClassVar[Decision]
+TRAJECTORY_STATUS_UNSPECIFIED: TrajectoryStatus
+TRAJECTORY_STATUS_PENDING: TrajectoryStatus
+TRAJECTORY_STATUS_RUNNING: TrajectoryStatus
+TRAJECTORY_STATUS_COMPLETED: TrajectoryStatus
+TRAJECTORY_STATUS_SUSPENDED: TrajectoryStatus
+TRAJECTORY_STATUS_FAILED: TrajectoryStatus
+STAGE_UNSPECIFIED: Stage
+STAGE_PRE_RUN: Stage
+STAGE_PRE_MODEL: Stage
+STAGE_POST_MODEL: Stage
+STAGE_PRE_TOOL: Stage
+STAGE_POST_TOOL: Stage
+STAGE_POST_RUN: Stage
+ROLE_UNSPECIFIED: Role
+ROLE_USER: Role
+ROLE_MODEL: Role
+ROLE_TOOL: Role
+ROLE_SYSTEM: Role
+DECISION_UNSPECIFIED: Decision
+DECISION_ALLOW: Decision
+DECISION_DENY: Decision
+DECISION_ESCALATE: Decision
+
+class SourceCode(_message.Message):
+    __slots__ = ("language", "code")
+    LANGUAGE_FIELD_NUMBER: _ClassVar[int]
+    CODE_FIELD_NUMBER: _ClassVar[int]
+    language: str
+    code: str
+    def __init__(self, language: _Optional[str] = ..., code: _Optional[str] = ...) -> None: ...
+
+class Parameter(_message.Message):
+    __slots__ = ("name", "description", "type")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
+    TYPE_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    description: str
+    type: str
+    def __init__(self, name: _Optional[str] = ..., description: _Optional[str] = ..., type: _Optional[str] = ...) -> None: ...
+
+class Tool(_message.Message):
+    __slots__ = ("name", "description", "parameters", "response", "source")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
+    PARAMETERS_FIELD_NUMBER: _ClassVar[int]
+    RESPONSE_FIELD_NUMBER: _ClassVar[int]
+    SOURCE_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    description: str
+    parameters: _containers.RepeatedCompositeFieldContainer[Parameter]
+    response: str
+    source: SourceCode
+    def __init__(self, name: _Optional[str] = ..., description: _Optional[str] = ..., parameters: _Optional[_Iterable[_Union[Parameter, _Mapping]]] = ..., response: _Optional[str] = ..., source: _Optional[_Union[SourceCode, _Mapping]] = ...) -> None: ...
+
+class Agent(_message.Message):
+    __slots__ = ("id", "provider_id", "name", "description", "instruction", "tools")
+    ID_FIELD_NUMBER: _ClassVar[int]
+    PROVIDER_ID_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
+    INSTRUCTION_FIELD_NUMBER: _ClassVar[int]
+    TOOLS_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    provider_id: str
+    name: str
+    description: str
+    instruction: str
+    tools: _containers.RepeatedCompositeFieldContainer[Tool]
+    def __init__(self, id: _Optional[str] = ..., provider_id: _Optional[str] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., instruction: _Optional[str] = ..., tools: _Optional[_Iterable[_Union[Tool, _Mapping]]] = ...) -> None: ...
+
+class Prompt(_message.Message):
+    __slots__ = ("text",)
+    TEXT_FIELD_NUMBER: _ClassVar[int]
+    text: str
+    def __init__(self, text: _Optional[str] = ...) -> None: ...
+
+class ToolRequest(_message.Message):
+    __slots__ = ("tool_id", "args")
+    TOOL_ID_FIELD_NUMBER: _ClassVar[int]
+    ARGS_FIELD_NUMBER: _ClassVar[int]
+    tool_id: str
+    args: _struct_pb2.Value
+    def __init__(self, tool_id: _Optional[str] = ..., args: _Optional[_Union[_struct_pb2.Value, _Mapping]] = ...) -> None: ...
+
+class ToolResponse(_message.Message):
+    __slots__ = ("tool_id", "response")
+    TOOL_ID_FIELD_NUMBER: _ClassVar[int]
+    RESPONSE_FIELD_NUMBER: _ClassVar[int]
+    tool_id: str
+    response: _struct_pb2.Value
+    def __init__(self, tool_id: _Optional[str] = ..., response: _Optional[_Union[_struct_pb2.Value, _Mapping]] = ...) -> None: ...
+
+class Content(_message.Message):
+    __slots__ = ("prompt", "tool_request", "tool_response")
+    PROMPT_FIELD_NUMBER: _ClassVar[int]
+    TOOL_REQUEST_FIELD_NUMBER: _ClassVar[int]
+    TOOL_RESPONSE_FIELD_NUMBER: _ClassVar[int]
+    prompt: Prompt
+    tool_request: ToolRequest
+    tool_response: ToolResponse
+    def __init__(self, prompt: _Optional[_Union[Prompt, _Mapping]] = ..., tool_request: _Optional[_Union[ToolRequest, _Mapping]] = ..., tool_response: _Optional[_Union[ToolResponse, _Mapping]] = ...) -> None: ...
+
+class TrajectoryStep(_message.Message):
+    __slots__ = ("stage", "role", "created_at", "content")
+    STAGE_FIELD_NUMBER: _ClassVar[int]
+    ROLE_FIELD_NUMBER: _ClassVar[int]
+    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
+    CONTENT_FIELD_NUMBER: _ClassVar[int]
+    stage: Stage
+    role: Role
+    created_at: _timestamp_pb2.Timestamp
+    content: Content
+    def __init__(self, stage: _Optional[_Union[Stage, str]] = ..., role: _Optional[_Union[Role, str]] = ..., created_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., content: _Optional[_Union[Content, _Mapping]] = ...) -> None: ...
+
+class Trajectory(_message.Message):
+    __slots__ = ("id", "agent_id", "status", "metadata", "created_at", "updated_at", "started_at", "ended_at", "steps")
+    class MetadataEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: _struct_pb2.Value
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[_struct_pb2.Value, _Mapping]] = ...) -> None: ...
+    ID_FIELD_NUMBER: _ClassVar[int]
+    AGENT_ID_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
+    UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
+    STARTED_AT_FIELD_NUMBER: _ClassVar[int]
+    ENDED_AT_FIELD_NUMBER: _ClassVar[int]
+    STEPS_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    agent_id: str
+    status: TrajectoryStatus
+    metadata: _containers.MessageMap[str, _struct_pb2.Value]
+    created_at: _timestamp_pb2.Timestamp
+    updated_at: _timestamp_pb2.Timestamp
+    started_at: _timestamp_pb2.Timestamp
+    ended_at: _timestamp_pb2.Timestamp
+    steps: _containers.RepeatedCompositeFieldContainer[TrajectoryStep]
+    def __init__(self, id: _Optional[str] = ..., agent_id: _Optional[str] = ..., status: _Optional[_Union[TrajectoryStatus, str]] = ..., metadata: _Optional[_Mapping[str, _struct_pb2.Value]] = ..., created_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., updated_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., started_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., ended_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., steps: _Optional[_Iterable[_Union[TrajectoryStep, _Mapping]]] = ...) -> None: ...
+
+class Check(_message.Message):
+    __slots__ = ("name", "flagged", "message")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    FLAGGED_FIELD_NUMBER: _ClassVar[int]
+    MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    flagged: bool
+    message: str
+    def __init__(self, name: _Optional[str] = ..., flagged: bool = ..., message: _Optional[str] = ...) -> None: ...
+
+class GuardrailContext(_message.Message):
+    __slots__ = ("checks",)
+    class ChecksEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: Check
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Check, _Mapping]] = ...) -> None: ...
+    CHECKS_FIELD_NUMBER: _ClassVar[int]
+    checks: _containers.MessageMap[str, Check]
+    def __init__(self, checks: _Optional[_Mapping[str, Check]] = ...) -> None: ...
+
+class PolicyAnnotations(_message.Message):
+    __slots__ = ("id", "description", "custom")
+    class CustomEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    ID_FIELD_NUMBER: _ClassVar[int]
+    DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
+    CUSTOM_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    description: str
+    custom: _containers.ScalarMap[str, str]
+    def __init__(self, id: _Optional[str] = ..., description: _Optional[str] = ..., custom: _Optional[_Mapping[str, str]] = ...) -> None: ...
+
+class Adjudication(_message.Message):
+    __slots__ = ("decision", "reason", "policy_ids", "annotations")
+    DECISION_FIELD_NUMBER: _ClassVar[int]
+    REASON_FIELD_NUMBER: _ClassVar[int]
+    POLICY_IDS_FIELD_NUMBER: _ClassVar[int]
+    ANNOTATIONS_FIELD_NUMBER: _ClassVar[int]
+    decision: Decision
+    reason: str
+    policy_ids: _containers.RepeatedScalarFieldContainer[str]
+    annotations: _containers.RepeatedCompositeFieldContainer[PolicyAnnotations]
+    def __init__(self, decision: _Optional[_Union[Decision, str]] = ..., reason: _Optional[str] = ..., policy_ids: _Optional[_Iterable[str]] = ..., annotations: _Optional[_Iterable[_Union[PolicyAnnotations, _Mapping]]] = ...) -> None: ...
+
+class AdjudicatedStep(_message.Message):
+    __slots__ = ("step", "guardrails", "adjudication")
+    STEP_FIELD_NUMBER: _ClassVar[int]
+    GUARDRAILS_FIELD_NUMBER: _ClassVar[int]
+    ADJUDICATION_FIELD_NUMBER: _ClassVar[int]
+    step: TrajectoryStep
+    guardrails: GuardrailContext
+    adjudication: Adjudication
+    def __init__(self, step: _Optional[_Union[TrajectoryStep, _Mapping]] = ..., guardrails: _Optional[_Union[GuardrailContext, _Mapping]] = ..., adjudication: _Optional[_Union[Adjudication, _Mapping]] = ...) -> None: ...
+
+class AdjudicatedTrajectory(_message.Message):
+    __slots__ = ("trajectory", "steps")
+    TRAJECTORY_FIELD_NUMBER: _ClassVar[int]
+    STEPS_FIELD_NUMBER: _ClassVar[int]
+    trajectory: Trajectory
+    steps: _containers.RepeatedCompositeFieldContainer[AdjudicatedStep]
+    def __init__(self, trajectory: _Optional[_Union[Trajectory, _Mapping]] = ..., steps: _Optional[_Iterable[_Union[AdjudicatedStep, _Mapping]]] = ...) -> None: ...

sondera/proto/sondera/core/v1/primitives_pb2_grpc.py

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

sondera/proto/sondera/harness/v1/harness_pb2.py

@@ -0,0 +1,81 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: sondera/harness/v1/harness.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'sondera/harness/v1/harness.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+from ...core.v1 import primitives_pb2 as sondera_dot_core_dot_v1_dot_primitives__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n sondera/harness/v1/harness.proto\x12\x12sondera.harness.v1\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a sondera/core/v1/primitives.proto\"\x89\x01\n\x14RegisterAgentRequest\x12\x13\n\x0bprovider_id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12\x13\n\x0binstruction\x18\x04 \x01(\t\x12$\n\x05tools\x18\x05 \x03(\x0b\x32\x15.sondera.core.v1.Tool\">\n\x15RegisterAgentResponse\x12%\n\x05\x61gent\x18\x01 \x01(\x0b\x32\x16.sondera.core.v1.Agent\"#\n\x0fGetAgentRequest\x12\x10\n\x08\x61gent_id\x18\x01 \x01(\t\"9\n\x10GetAgentResponse\x12%\n\x05\x61gent\x18\x01 \x01(\x0b\x32\x16.sondera.core.v1.Agent\"+\n\x17\x43reateTrajectoryRequest\x12\x10\n\x08\x61gent_id\x18\x01 \x01(\t\"K\n\x18\x43reateTrajectoryResponse\x12/\n\ntrajectory\x18\x01 \x01(\x0b\x32\x1b.sondera.core.v1.Trajectory\"\xa8\x01\n\x18\x41\x64\x64TrajectoryStepRequest\x12\x15\n\rtrajectory_id\x18\x01 \x01(\t\x12%\n\x05stage\x18\x02 \x01(\x0e\x32\x16.sondera.core.v1.Stage\x12#\n\x04role\x18\x03 \x01(\x0e\x32\x15.sondera.core.v1.Role\x12)\n\x07\x63ontent\x18\x04 \x01(\x0b\x32\x18.sondera.core.v1.Content\"W\n\x19\x41\x64\x64TrajectoryStepResponse\x12:\n\x10\x61\x64judicated_step\x18\x01 \x01(\x0b\x32 .sondera.core.v1.AdjudicatedStep\"i\n\x1dUpdateTrajectoryStatusRequest\x12\x15\n\rtrajectory_id\x18\x01 \x01(\t\x12\x31\n\x06status\x18\x02 \x01(\x0e\x32!.sondera.core.v1.TrajectoryStatus\"Q\n\x1eUpdateTrajectoryStatusResponse\x12/\n\ntrajectory\x18\x01 \x01(\x0b\x32\x1b.sondera.core.v1.Trajectory\"-\n\x14GetTrajectoryRequest\x12\x15\n\rtrajectory_id\x18\x01 \x01(\t\"y\n\x15GetTrajectoryResponse\x12/\n\ntrajectory\x18\x01 \x01(\x0b\x32\x1b.sondera.core.v1.Trajectory\x12/\n\x05steps\x18\x02 \x03(\x0b\x32 .sondera.core.v1.AdjudicatedStep\"\x95\x01\n\x17ListTrajectoriesRequest\x12\x10\n\x08\x61gent_id\x18\x01 \x01(\t\x12\x36\n\x06status\x18\x02 \x01(\x0e\x32!.sondera.core.v1.TrajectoryStatusH\x00\x88\x01\x01\x12\x11\n\tpage_size\x18\x03 \x01(\x05\x12\x12\n\npage_token\x18\x04 \x01(\tB\t\n\x07_status\"f\n\x18ListTrajectoriesResponse\x12\x31\n\x0ctrajectories\x18\x01 \x03(\x0b\x32\x1b.sondera.core.v1.Trajectory\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"d\n\x11ListAgentsRequest\x12\x18\n\x0bprovider_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x11\n\tpage_size\x18\x02 \x01(\x05\x12\x12\n\npage_token\x18\x03 \x01(\tB\x0e\n\x0c_provider_id\"U\n\x12ListAgentsResponse\x12&\n\x06\x61gents\x18\x01 \x03(\x0b\x32\x16.sondera.core.v1.Agent\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"\x9f\x01\n\x1a\x41nalyzeTrajectoriesRequest\x12\x10\n\x08\x61gent_id\x18\x01 \x01(\t\x12.\n\nstart_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x11\n\tanalytics\x18\x04 \x03(\t\"\x94\x01\n\x1b\x41nalyzeTrajectoriesResponse\x12*\n\tanalytics\x18\x01 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x18\n\x10trajectory_count\x18\x02 \x01(\x05\x12/\n\x0b\x63omputed_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"e\n\x18ListAdjudicationsRequest\x12\x15\n\x08\x61gent_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x11\n\tpage_size\x18\x02 \x01(\x05\x12\x12\n\npage_token\x18\x03 \x01(\tB\x0b\n\t_agent_id\"\x83\x01\n\x12\x41\x64judicationRecord\x12\x10\n\x08\x61gent_id\x18\x01 \x01(\t\x12\x15\n\rtrajectory_id\x18\x02 \x01(\t\x12\x0f\n\x07step_id\x18\x03 \x01(\t\x12\x33\n\x0c\x61\x64judication\x18\x04 \x01(\x0b\x32\x1d.sondera.core.v1.Adjudication\"s\n\x19ListAdjudicationsResponse\x12=\n\radjudications\x18\x01 \x03(\x0b\x32&.sondera.harness.v1.AdjudicationRecord\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t2\xcb\x08\n\x0eHarnessService\x12\x64\n\rRegisterAgent\x12(.sondera.harness.v1.RegisterAgentRequest\x1a).sondera.harness.v1.RegisterAgentResponse\x12U\n\x08GetAgent\x12#.sondera.harness.v1.GetAgentRequest\x1a$.sondera.harness.v1.GetAgentResponse\x12m\n\x10\x43reateTrajectory\x12+.sondera.harness.v1.CreateTrajectoryRequest\x1a,.sondera.harness.v1.CreateTrajectoryResponse\x12p\n\x11\x41\x64\x64TrajectoryStep\x12,.sondera.harness.v1.AddTrajectoryStepRequest\x1a-.sondera.harness.v1.AddTrajectoryStepResponse\x12\x7f\n\x16UpdateTrajectoryStatus\x12\x31.sondera.harness.v1.UpdateTrajectoryStatusRequest\x1a\x32.sondera.harness.v1.UpdateTrajectoryStatusResponse\x12\x64\n\rGetTrajectory\x12(.sondera.harness.v1.GetTrajectoryRequest\x1a).sondera.harness.v1.GetTrajectoryResponse\x12m\n\x10ListTrajectories\x12+.sondera.harness.v1.ListTrajectoriesRequest\x1a,.sondera.harness.v1.ListTrajectoriesResponse\x12[\n\nListAgents\x12%.sondera.harness.v1.ListAgentsRequest\x1a&.sondera.harness.v1.ListAgentsResponse\x12v\n\x13\x41nalyzeTrajectories\x12..sondera.harness.v1.AnalyzeTrajectoriesRequest\x1a/.sondera.harness.v1.AnalyzeTrajectoriesResponse\x12p\n\x11ListAdjudications\x12,.sondera.harness.v1.ListAdjudicationsRequest\x1a-.sondera.harness.v1.ListAdjudicationsResponseb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'sondera.harness.v1.harness_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_REGISTERAGENTREQUEST']._serialized_start=154
+  _globals['_REGISTERAGENTREQUEST']._serialized_end=291
+  _globals['_REGISTERAGENTRESPONSE']._serialized_start=293
+  _globals['_REGISTERAGENTRESPONSE']._serialized_end=355
+  _globals['_GETAGENTREQUEST']._serialized_start=357
+  _globals['_GETAGENTREQUEST']._serialized_end=392
+  _globals['_GETAGENTRESPONSE']._serialized_start=394
+  _globals['_GETAGENTRESPONSE']._serialized_end=451
+  _globals['_CREATETRAJECTORYREQUEST']._serialized_start=453
+  _globals['_CREATETRAJECTORYREQUEST']._serialized_end=496
+  _globals['_CREATETRAJECTORYRESPONSE']._serialized_start=498
+  _globals['_CREATETRAJECTORYRESPONSE']._serialized_end=573
+  _globals['_ADDTRAJECTORYSTEPREQUEST']._serialized_start=576
+  _globals['_ADDTRAJECTORYSTEPREQUEST']._serialized_end=744
+  _globals['_ADDTRAJECTORYSTEPRESPONSE']._serialized_start=746
+  _globals['_ADDTRAJECTORYSTEPRESPONSE']._serialized_end=833
+  _globals['_UPDATETRAJECTORYSTATUSREQUEST']._serialized_start=835
+  _globals['_UPDATETRAJECTORYSTATUSREQUEST']._serialized_end=940
+  _globals['_UPDATETRAJECTORYSTATUSRESPONSE']._serialized_start=942
+  _globals['_UPDATETRAJECTORYSTATUSRESPONSE']._serialized_end=1023
+  _globals['_GETTRAJECTORYREQUEST']._serialized_start=1025
+  _globals['_GETTRAJECTORYREQUEST']._serialized_end=1070
+  _globals['_GETTRAJECTORYRESPONSE']._serialized_start=1072
+  _globals['_GETTRAJECTORYRESPONSE']._serialized_end=1193
+  _globals['_LISTTRAJECTORIESREQUEST']._serialized_start=1196
+  _globals['_LISTTRAJECTORIESREQUEST']._serialized_end=1345
+  _globals['_LISTTRAJECTORIESRESPONSE']._serialized_start=1347
+  _globals['_LISTTRAJECTORIESRESPONSE']._serialized_end=1449
+  _globals['_LISTAGENTSREQUEST']._serialized_start=1451
+  _globals['_LISTAGENTSREQUEST']._serialized_end=1551
+  _globals['_LISTAGENTSRESPONSE']._serialized_start=1553
+  _globals['_LISTAGENTSRESPONSE']._serialized_end=1638
+  _globals['_ANALYZETRAJECTORIESREQUEST']._serialized_start=1641
+  _globals['_ANALYZETRAJECTORIESREQUEST']._serialized_end=1800
+  _globals['_ANALYZETRAJECTORIESRESPONSE']._serialized_start=1803
+  _globals['_ANALYZETRAJECTORIESRESPONSE']._serialized_end=1951
+  _globals['_LISTADJUDICATIONSREQUEST']._serialized_start=1953
+  _globals['_LISTADJUDICATIONSREQUEST']._serialized_end=2054
+  _globals['_ADJUDICATIONRECORD']._serialized_start=2057
+  _globals['_ADJUDICATIONRECORD']._serialized_end=2188
+  _globals['_LISTADJUDICATIONSRESPONSE']._serialized_start=2190
+  _globals['_LISTADJUDICATIONSRESPONSE']._serialized_end=2305
+  _globals['_HARNESSSERVICE']._serialized_start=2308
+  _globals['_HARNESSSERVICE']._serialized_end=3407
+# @@protoc_insertion_point(module_scope)

sondera/proto/sondera/harness/v1/harness_pb2.pyi

@@ -0,0 +1,192 @@
+import datetime
+
+from google.protobuf import struct_pb2 as _struct_pb2
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from sondera.core.v1 import primitives_pb2 as _primitives_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class RegisterAgentRequest(_message.Message):
+    __slots__ = ("provider_id", "name", "description", "instruction", "tools")
+    PROVIDER_ID_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
+    INSTRUCTION_FIELD_NUMBER: _ClassVar[int]
+    TOOLS_FIELD_NUMBER: _ClassVar[int]
+    provider_id: str
+    name: str
+    description: str
+    instruction: str
+    tools: _containers.RepeatedCompositeFieldContainer[_primitives_pb2.Tool]
+    def __init__(self, provider_id: _Optional[str] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., instruction: _Optional[str] = ..., tools: _Optional[_Iterable[_Union[_primitives_pb2.Tool, _Mapping]]] = ...) -> None: ...
+
+class RegisterAgentResponse(_message.Message):
+    __slots__ = ("agent",)
+    AGENT_FIELD_NUMBER: _ClassVar[int]
+    agent: _primitives_pb2.Agent
+    def __init__(self, agent: _Optional[_Union[_primitives_pb2.Agent, _Mapping]] = ...) -> None: ...
+
+class GetAgentRequest(_message.Message):
+    __slots__ = ("agent_id",)
+    AGENT_ID_FIELD_NUMBER: _ClassVar[int]
+    agent_id: str
+    def __init__(self, agent_id: _Optional[str] = ...) -> None: ...
+
+class GetAgentResponse(_message.Message):
+    __slots__ = ("agent",)
+    AGENT_FIELD_NUMBER: _ClassVar[int]
+    agent: _primitives_pb2.Agent
+    def __init__(self, agent: _Optional[_Union[_primitives_pb2.Agent, _Mapping]] = ...) -> None: ...
+
+class CreateTrajectoryRequest(_message.Message):
+    __slots__ = ("agent_id",)
+    AGENT_ID_FIELD_NUMBER: _ClassVar[int]
+    agent_id: str
+    def __init__(self, agent_id: _Optional[str] = ...) -> None: ...
+
+class CreateTrajectoryResponse(_message.Message):
+    __slots__ = ("trajectory",)
+    TRAJECTORY_FIELD_NUMBER: _ClassVar[int]
+    trajectory: _primitives_pb2.Trajectory
+    def __init__(self, trajectory: _Optional[_Union[_primitives_pb2.Trajectory, _Mapping]] = ...) -> None: ...
+
+class AddTrajectoryStepRequest(_message.Message):
+    __slots__ = ("trajectory_id", "stage", "role", "content")
+    TRAJECTORY_ID_FIELD_NUMBER: _ClassVar[int]
+    STAGE_FIELD_NUMBER: _ClassVar[int]
+    ROLE_FIELD_NUMBER: _ClassVar[int]
+    CONTENT_FIELD_NUMBER: _ClassVar[int]
+    trajectory_id: str
+    stage: _primitives_pb2.Stage
+    role: _primitives_pb2.Role
+    content: _primitives_pb2.Content
+    def __init__(self, trajectory_id: _Optional[str] = ..., stage: _Optional[_Union[_primitives_pb2.Stage, str]] = ..., role: _Optional[_Union[_primitives_pb2.Role, str]] = ..., content: _Optional[_Union[_primitives_pb2.Content, _Mapping]] = ...) -> None: ...
+
+class AddTrajectoryStepResponse(_message.Message):
+    __slots__ = ("adjudicated_step",)
+    ADJUDICATED_STEP_FIELD_NUMBER: _ClassVar[int]
+    adjudicated_step: _primitives_pb2.AdjudicatedStep
+    def __init__(self, adjudicated_step: _Optional[_Union[_primitives_pb2.AdjudicatedStep, _Mapping]] = ...) -> None: ...
+
+class UpdateTrajectoryStatusRequest(_message.Message):
+    __slots__ = ("trajectory_id", "status")
+    TRAJECTORY_ID_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    trajectory_id: str
+    status: _primitives_pb2.TrajectoryStatus
+    def __init__(self, trajectory_id: _Optional[str] = ..., status: _Optional[_Union[_primitives_pb2.TrajectoryStatus, str]] = ...) -> None: ...
+
+class UpdateTrajectoryStatusResponse(_message.Message):
+    __slots__ = ("trajectory",)
+    TRAJECTORY_FIELD_NUMBER: _ClassVar[int]
+    trajectory: _primitives_pb2.Trajectory
+    def __init__(self, trajectory: _Optional[_Union[_primitives_pb2.Trajectory, _Mapping]] = ...) -> None: ...
+
+class GetTrajectoryRequest(_message.Message):
+    __slots__ = ("trajectory_id",)
+    TRAJECTORY_ID_FIELD_NUMBER: _ClassVar[int]
+    trajectory_id: str
+    def __init__(self, trajectory_id: _Optional[str] = ...) -> None: ...
+
+class GetTrajectoryResponse(_message.Message):
+    __slots__ = ("trajectory", "steps")
+    TRAJECTORY_FIELD_NUMBER: _ClassVar[int]
+    STEPS_FIELD_NUMBER: _ClassVar[int]
+    trajectory: _primitives_pb2.Trajectory
+    steps: _containers.RepeatedCompositeFieldContainer[_primitives_pb2.AdjudicatedStep]
+    def __init__(self, trajectory: _Optional[_Union[_primitives_pb2.Trajectory, _Mapping]] = ..., steps: _Optional[_Iterable[_Union[_primitives_pb2.AdjudicatedStep, _Mapping]]] = ...) -> None: ...
+
+class ListTrajectoriesRequest(_message.Message):
+    __slots__ = ("agent_id", "status", "page_size", "page_token")
+    AGENT_ID_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    PAGE_SIZE_FIELD_NUMBER: _ClassVar[int]
+    PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
+    agent_id: str
+    status: _primitives_pb2.TrajectoryStatus
+    page_size: int
+    page_token: str
+    def __init__(self, agent_id: _Optional[str] = ..., status: _Optional[_Union[_primitives_pb2.TrajectoryStatus, str]] = ..., page_size: _Optional[int] = ..., page_token: _Optional[str] = ...) -> None: ...
+
+class ListTrajectoriesResponse(_message.Message):
+    __slots__ = ("trajectories", "next_page_token")
+    TRAJECTORIES_FIELD_NUMBER: _ClassVar[int]
+    NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
+    trajectories: _containers.RepeatedCompositeFieldContainer[_primitives_pb2.Trajectory]
+    next_page_token: str
+    def __init__(self, trajectories: _Optional[_Iterable[_Union[_primitives_pb2.Trajectory, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ...
+
+class ListAgentsRequest(_message.Message):
+    __slots__ = ("provider_id", "page_size", "page_token")
+    PROVIDER_ID_FIELD_NUMBER: _ClassVar[int]
+    PAGE_SIZE_FIELD_NUMBER: _ClassVar[int]
+    PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
+    provider_id: str
+    page_size: int
+    page_token: str
+    def __init__(self, provider_id: _Optional[str] = ..., page_size: _Optional[int] = ..., page_token: _Optional[str] = ...) -> None: ...
+
+class ListAgentsResponse(_message.Message):
+    __slots__ = ("agents", "next_page_token")
+    AGENTS_FIELD_NUMBER: _ClassVar[int]
+    NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
+    agents: _containers.RepeatedCompositeFieldContainer[_primitives_pb2.Agent]
+    next_page_token: str
+    def __init__(self, agents: _Optional[_Iterable[_Union[_primitives_pb2.Agent, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ...
+
+class AnalyzeTrajectoriesRequest(_message.Message):
+    __slots__ = ("agent_id", "start_time", "end_time", "analytics")
+    AGENT_ID_FIELD_NUMBER: _ClassVar[int]
+    START_TIME_FIELD_NUMBER: _ClassVar[int]
+    END_TIME_FIELD_NUMBER: _ClassVar[int]
+    ANALYTICS_FIELD_NUMBER: _ClassVar[int]
+    agent_id: str
+    start_time: _timestamp_pb2.Timestamp
+    end_time: _timestamp_pb2.Timestamp
+    analytics: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, agent_id: _Optional[str] = ..., start_time: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., end_time: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., analytics: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class AnalyzeTrajectoriesResponse(_message.Message):
+    __slots__ = ("analytics", "trajectory_count", "computed_at")
+    ANALYTICS_FIELD_NUMBER: _ClassVar[int]
+    TRAJECTORY_COUNT_FIELD_NUMBER: _ClassVar[int]
+    COMPUTED_AT_FIELD_NUMBER: _ClassVar[int]
+    analytics: _struct_pb2.Struct
+    trajectory_count: int
+    computed_at: _timestamp_pb2.Timestamp
+    def __init__(self, analytics: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., trajectory_count: _Optional[int] = ..., computed_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
+
+class ListAdjudicationsRequest(_message.Message):
+    __slots__ = ("agent_id", "page_size", "page_token")
+    AGENT_ID_FIELD_NUMBER: _ClassVar[int]
+    PAGE_SIZE_FIELD_NUMBER: _ClassVar[int]
+    PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
+    agent_id: str
+    page_size: int
+    page_token: str
+    def __init__(self, agent_id: _Optional[str] = ..., page_size: _Optional[int] = ..., page_token: _Optional[str] = ...) -> None: ...
+
+class AdjudicationRecord(_message.Message):
+    __slots__ = ("agent_id", "trajectory_id", "step_id", "adjudication")
+    AGENT_ID_FIELD_NUMBER: _ClassVar[int]
+    TRAJECTORY_ID_FIELD_NUMBER: _ClassVar[int]
+    STEP_ID_FIELD_NUMBER: _ClassVar[int]
+    ADJUDICATION_FIELD_NUMBER: _ClassVar[int]
+    agent_id: str
+    trajectory_id: str
+    step_id: str
+    adjudication: _primitives_pb2.Adjudication
+    def __init__(self, agent_id: _Optional[str] = ..., trajectory_id: _Optional[str] = ..., step_id: _Optional[str] = ..., adjudication: _Optional[_Union[_primitives_pb2.Adjudication, _Mapping]] = ...) -> None: ...
+
+class ListAdjudicationsResponse(_message.Message):
+    __slots__ = ("adjudications", "next_page_token")
+    ADJUDICATIONS_FIELD_NUMBER: _ClassVar[int]
+    NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
+    adjudications: _containers.RepeatedCompositeFieldContainer[AdjudicationRecord]
+    next_page_token: str
+    def __init__(self, adjudications: _Optional[_Iterable[_Union[AdjudicationRecord, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ...