ankka 0.4.0__py3-none-any.whl

ankka/__init__.py

@@ -0,0 +1,18 @@
+"""The ankka SDK for Python: components in Python, hosted by the ankka sidecar."""
+
+__version__ = "0.4.0"
+
+from ankka.codec import Codec, Done, DONE, json_codec  # noqa: E402
+from ankka.context import CommandContext, Metadata, Principal, RequestContext  # noqa: E402
+from ankka.effects.common import DeleteNow, Error, ErrorCode, ExpireAfter  # noqa: E402
+from ankka.effects.event_sourced import EventSourcedEffect, ReadOnlyEffect  # noqa: E402
+from ankka.endpoint import Acl, Endpoint, HttpProblem, delete, get, patch, post, put, sse  # noqa: E402
+from ankka.event_sourced_entity import EventSourcedEntity, RegistrationError, command, query  # noqa: E402
+from ankka.service import Ankka, ServiceBuilder  # noqa: E402
+
+__all__ = [
+    "Acl", "Ankka", "Codec", "CommandContext", "DeleteNow", "Done", "DONE", "Endpoint", "Error",
+    "ErrorCode", "EventSourcedEffect", "EventSourcedEntity", "ExpireAfter", "HttpProblem", "Metadata",
+    "Principal", "ReadOnlyEffect", "RegistrationError", "RequestContext", "ServiceBuilder",
+    "command", "delete", "get", "json_codec", "patch", "post", "put", "query", "sse",
+]

ankka/_conformance.py

@@ -0,0 +1,56 @@
+"""`uv run conformance`: serve the Python reference service and run the platform's conformance
+suite against it, exiting with sbt's status. Needs Docker (the suite starts Postgres) and sbt on
+PATH. The reference service is the SDK's example, so the SDK checkout is put on the path."""
+
+from __future__ import annotations
+
+import asyncio
+import logging
+import os
+import subprocess
+import sys
+import threading
+from pathlib import Path
+
+SDK = Path(__file__).resolve().parents[2]
+REPO = SDK.parent.parent
+PORT = int(os.environ.get("ANKKA_PROCESS_PORT", "9010"))
+
+
+def _serve(ready: threading.Event, stop: threading.Event) -> None:
+    sys.path.insert(0, str(SDK))
+    from examples.shopping_cart.conformance import reference_service
+
+    async def run() -> None:
+        server = reference_service().server()
+        await server.start("127.0.0.1", PORT)
+        ready.set()
+        while not stop.is_set():
+            await asyncio.sleep(0.2)
+        await server.stop(grace=1.0)
+
+    asyncio.run(run())
+
+
+def main() -> int:
+    logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s %(name)s - %(message)s")
+    ready, stop = threading.Event(), threading.Event()
+    thread = threading.Thread(target=_serve, args=(ready, stop), daemon=True)
+    thread.start()
+    if not ready.wait(30):
+        print("the reference service did not start", file=sys.stderr)
+        return 1
+    command = [
+        "sbt",
+        f"-Dankka.conformance.target=127.0.0.1:{PORT}",
+        *(["-Dankka.benchmarks=on"] if os.environ.get("ANKKA_BENCHMARKS") else []),
+        "-Dankka.cluster.tests=off",
+        "-Dankka.template.tests=off",
+        "sidecar/testOnly *ConformanceSuite" + (f" -- {only}" if (only := os.environ.get("ANKKA_CONFORMANCE_ONLY")) else ""),
+    ]
+    print("running:", " ".join(command), "in", REPO, flush=True)
+    try:
+        return subprocess.run(command, cwd=REPO).returncode
+    finally:
+        stop.set()
+        thread.join(5)

ankka/_proto/ankka/protocol/v1/agent_pb2.py

@@ -0,0 +1,59 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: ankka/protocol/v1/agent.proto
+# Protobuf Python Version: 7.35.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,
+    7,
+    35,
+    1,
+    '',
+    'ankka/protocol/v1/agent.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from ankka._proto.ankka.protocol.v1 import payload_pb2 as ankka_dot_protocol_dot_v1_dot_payload__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1d\x61nkka/protocol/v1/agent.proto\x12\x11\x61nkka.protocol.v1\x1a\x1f\x61nkka/protocol/v1/payload.proto\"\xa1\x01\n\x0bPlanRequest\x12\x14\n\x0c\x63omponent_id\x18\x01 \x01(\t\x12\x12\n\nsession_id\x18\x02 \x01(\t\x12\x0c\n\x04name\x18\x03 \x01(\t\x12+\n\x07payload\x18\x04 \x01(\x0b\x32\x1a.ankka.protocol.v1.Payload\x12-\n\x08metadata\x18\x05 \x01(\x0b\x32\x1b.ankka.protocol.v1.Metadata\"s\n\tPlanReply\x12,\n\x04plan\x18\x01 \x01(\x0b\x32\x1c.ankka.protocol.v1.AgentPlanH\x00\x12-\n\x07\x66\x61ilure\x18\x02 \x01(\x0b\x32\x1a.ankka.protocol.v1.FailureH\x00\x42\t\n\x07message\"\x92\x04\n\tAgentPlan\x12\x12\n\x05model\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x13\n\x06system\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x11\n\x04user\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x0f\n\x07\x63ontext\x18\x04 \x03(\t\x12\x33\n\x06memory\x18\x05 \x01(\x0e\x32#.ankka.protocol.v1.AgentPlan.Memory\x12\r\n\x05tools\x18\x06 \x03(\t\x12\x42\n\x0eresponse_shape\x18\x07 \x01(\x0b\x32*.ankka.protocol.v1.AgentPlan.ResponseShape\x12\x12\n\nguardrails\x18\x08 \x03(\t\x12.\n\x07\x66\x61ilure\x18\t \x01(\x0b\x32\x18.ankka.protocol.v1.ErrorH\x03\x88\x01\x01\x1a\xa0\x01\n\rResponseShape\x12(\n\x04text\x18\x01 \x01(\x0b\x32\x18.ankka.protocol.v1.EmptyH\x00\x12?\n\x04json\x18\x02 \x01(\x0b\x32/.ankka.protocol.v1.AgentPlan.ResponseShape.JsonH\x00\x1a\x1b\n\x04Json\x12\x13\n\x0bschema_hint\x18\x01 \x01(\tB\x07\n\x05shape\"\x1f\n\x06Memory\x12\x0b\n\x07SESSION\x10\x00\x12\x08\n\x04NONE\x10\x01\x42\x08\n\x06_modelB\t\n\x07_systemB\x07\n\x05_userB\n\n\x08_failure\"]\n\x0bToolRequest\x12\x14\n\x0c\x63omponent_id\x18\x01 \x01(\t\x12\x12\n\nsession_id\x18\x02 \x01(\t\x12\x0c\n\x04tool\x18\x03 \x01(\t\x12\x16\n\x0e\x61rguments_json\x18\x04 \x01(\t\"5\n\nToolResult\x12\x0c\n\x02ok\x18\x01 \x01(\tH\x00\x12\x0f\n\x05\x65rror\x18\x02 \x01(\tH\x00\x42\x08\n\x06result\"\xb7\x01\n\x10GuardrailRequest\x12\x14\n\x0c\x63omponent_id\x18\x01 \x01(\t\x12\x12\n\nsession_id\x18\x02 \x01(\t\x12\x11\n\tguardrail\x18\x03 \x01(\t\x12\x38\n\x05stage\x18\x04 \x01(\x0e\x32).ankka.protocol.v1.GuardrailRequest.Stage\x12\x0c\n\x04text\x18\x05 \x01(\t\"\x1e\n\x05Stage\x12\t\n\x05INPUT\x10\x00\x12\n\n\x06OUTPUT\x10\x01\"V\n\x0fGuardrailResult\x12(\n\x04pass\x18\x01 \x01(\x0b\x32\x18.ankka.protocol.v1.EmptyH\x00\x12\x0f\n\x05\x62lock\x18\x02 \x01(\tH\x00\x42\x08\n\x06result2\xf5\x01\n\x05\x41gent\x12\x44\n\x04Plan\x12\x1e.ankka.protocol.v1.PlanRequest\x1a\x1c.ankka.protocol.v1.PlanReply\x12K\n\nInvokeTool\x12\x1e.ankka.protocol.v1.ToolRequest\x1a\x1d.ankka.protocol.v1.ToolResult\x12Y\n\x0e\x43heckGuardrail\x12#.ankka.protocol.v1.GuardrailRequest\x1a\".ankka.protocol.v1.GuardrailResultb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'ankka.protocol.v1.agent_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_PLANREQUEST']._serialized_start=86
+  _globals['_PLANREQUEST']._serialized_end=247
+  _globals['_PLANREPLY']._serialized_start=249
+  _globals['_PLANREPLY']._serialized_end=364
+  _globals['_AGENTPLAN']._serialized_start=367
+  _globals['_AGENTPLAN']._serialized_end=897
+  _globals['_AGENTPLAN_RESPONSESHAPE']._serialized_start=662
+  _globals['_AGENTPLAN_RESPONSESHAPE']._serialized_end=822
+  _globals['_AGENTPLAN_RESPONSESHAPE_JSON']._serialized_start=786
+  _globals['_AGENTPLAN_RESPONSESHAPE_JSON']._serialized_end=813
+  _globals['_AGENTPLAN_MEMORY']._serialized_start=824
+  _globals['_AGENTPLAN_MEMORY']._serialized_end=855
+  _globals['_TOOLREQUEST']._serialized_start=899
+  _globals['_TOOLREQUEST']._serialized_end=992
+  _globals['_TOOLRESULT']._serialized_start=994
+  _globals['_TOOLRESULT']._serialized_end=1047
+  _globals['_GUARDRAILREQUEST']._serialized_start=1050
+  _globals['_GUARDRAILREQUEST']._serialized_end=1233
+  _globals['_GUARDRAILREQUEST_STAGE']._serialized_start=1203
+  _globals['_GUARDRAILREQUEST_STAGE']._serialized_end=1233
+  _globals['_GUARDRAILRESULT']._serialized_start=1235
+  _globals['_GUARDRAILRESULT']._serialized_end=1321
+  _globals['_AGENT']._serialized_start=1324
+  _globals['_AGENT']._serialized_end=1569
+# @@protoc_insertion_point(module_scope)

ankka/_proto/ankka/protocol/v1/agent_pb2.pyi

@@ -0,0 +1,118 @@
+from ankka._proto.ankka.protocol.v1 import payload_pb2 as _payload_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 PlanRequest(_message.Message):
+    __slots__ = ("component_id", "session_id", "name", "payload", "metadata")
+    COMPONENT_ID_FIELD_NUMBER: _ClassVar[int]
+    SESSION_ID_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    PAYLOAD_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    component_id: str
+    session_id: str
+    name: str
+    payload: _payload_pb2.Payload
+    metadata: _payload_pb2.Metadata
+    def __init__(self, component_id: _Optional[str] = ..., session_id: _Optional[str] = ..., name: _Optional[str] = ..., payload: _Optional[_Union[_payload_pb2.Payload, _Mapping]] = ..., metadata: _Optional[_Union[_payload_pb2.Metadata, _Mapping]] = ...) -> None: ...
+
+class PlanReply(_message.Message):
+    __slots__ = ("plan", "failure")
+    PLAN_FIELD_NUMBER: _ClassVar[int]
+    FAILURE_FIELD_NUMBER: _ClassVar[int]
+    plan: AgentPlan
+    failure: _payload_pb2.Failure
+    def __init__(self, plan: _Optional[_Union[AgentPlan, _Mapping]] = ..., failure: _Optional[_Union[_payload_pb2.Failure, _Mapping]] = ...) -> None: ...
+
+class AgentPlan(_message.Message):
+    __slots__ = ("model", "system", "user", "context", "memory", "tools", "response_shape", "guardrails", "failure")
+    class Memory(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        SESSION: _ClassVar[AgentPlan.Memory]
+        NONE: _ClassVar[AgentPlan.Memory]
+    SESSION: AgentPlan.Memory
+    NONE: AgentPlan.Memory
+    class ResponseShape(_message.Message):
+        __slots__ = ("text", "json")
+        class Json(_message.Message):
+            __slots__ = ("schema_hint",)
+            SCHEMA_HINT_FIELD_NUMBER: _ClassVar[int]
+            schema_hint: str
+            def __init__(self, schema_hint: _Optional[str] = ...) -> None: ...
+        TEXT_FIELD_NUMBER: _ClassVar[int]
+        JSON_FIELD_NUMBER: _ClassVar[int]
+        text: _payload_pb2.Empty
+        json: AgentPlan.ResponseShape.Json
+        def __init__(self, text: _Optional[_Union[_payload_pb2.Empty, _Mapping]] = ..., json: _Optional[_Union[AgentPlan.ResponseShape.Json, _Mapping]] = ...) -> None: ...
+    MODEL_FIELD_NUMBER: _ClassVar[int]
+    SYSTEM_FIELD_NUMBER: _ClassVar[int]
+    USER_FIELD_NUMBER: _ClassVar[int]
+    CONTEXT_FIELD_NUMBER: _ClassVar[int]
+    MEMORY_FIELD_NUMBER: _ClassVar[int]
+    TOOLS_FIELD_NUMBER: _ClassVar[int]
+    RESPONSE_SHAPE_FIELD_NUMBER: _ClassVar[int]
+    GUARDRAILS_FIELD_NUMBER: _ClassVar[int]
+    FAILURE_FIELD_NUMBER: _ClassVar[int]
+    model: str
+    system: str
+    user: str
+    context: _containers.RepeatedScalarFieldContainer[str]
+    memory: AgentPlan.Memory
+    tools: _containers.RepeatedScalarFieldContainer[str]
+    response_shape: AgentPlan.ResponseShape
+    guardrails: _containers.RepeatedScalarFieldContainer[str]
+    failure: _payload_pb2.Error
+    def __init__(self, model: _Optional[str] = ..., system: _Optional[str] = ..., user: _Optional[str] = ..., context: _Optional[_Iterable[str]] = ..., memory: _Optional[_Union[AgentPlan.Memory, str]] = ..., tools: _Optional[_Iterable[str]] = ..., response_shape: _Optional[_Union[AgentPlan.ResponseShape, _Mapping]] = ..., guardrails: _Optional[_Iterable[str]] = ..., failure: _Optional[_Union[_payload_pb2.Error, _Mapping]] = ...) -> None: ...
+
+class ToolRequest(_message.Message):
+    __slots__ = ("component_id", "session_id", "tool", "arguments_json")
+    COMPONENT_ID_FIELD_NUMBER: _ClassVar[int]
+    SESSION_ID_FIELD_NUMBER: _ClassVar[int]
+    TOOL_FIELD_NUMBER: _ClassVar[int]
+    ARGUMENTS_JSON_FIELD_NUMBER: _ClassVar[int]
+    component_id: str
+    session_id: str
+    tool: str
+    arguments_json: str
+    def __init__(self, component_id: _Optional[str] = ..., session_id: _Optional[str] = ..., tool: _Optional[str] = ..., arguments_json: _Optional[str] = ...) -> None: ...
+
+class ToolResult(_message.Message):
+    __slots__ = ("ok", "error")
+    OK_FIELD_NUMBER: _ClassVar[int]
+    ERROR_FIELD_NUMBER: _ClassVar[int]
+    ok: str
+    error: str
+    def __init__(self, ok: _Optional[str] = ..., error: _Optional[str] = ...) -> None: ...
+
+class GuardrailRequest(_message.Message):
+    __slots__ = ("component_id", "session_id", "guardrail", "stage", "text")
+    class Stage(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        INPUT: _ClassVar[GuardrailRequest.Stage]
+        OUTPUT: _ClassVar[GuardrailRequest.Stage]
+    INPUT: GuardrailRequest.Stage
+    OUTPUT: GuardrailRequest.Stage
+    COMPONENT_ID_FIELD_NUMBER: _ClassVar[int]
+    SESSION_ID_FIELD_NUMBER: _ClassVar[int]
+    GUARDRAIL_FIELD_NUMBER: _ClassVar[int]
+    STAGE_FIELD_NUMBER: _ClassVar[int]
+    TEXT_FIELD_NUMBER: _ClassVar[int]
+    component_id: str
+    session_id: str
+    guardrail: str
+    stage: GuardrailRequest.Stage
+    text: str
+    def __init__(self, component_id: _Optional[str] = ..., session_id: _Optional[str] = ..., guardrail: _Optional[str] = ..., stage: _Optional[_Union[GuardrailRequest.Stage, str]] = ..., text: _Optional[str] = ...) -> None: ...
+
+class GuardrailResult(_message.Message):
+    __slots__ = ("block",)
+    PASS_FIELD_NUMBER: _ClassVar[int]
+    BLOCK_FIELD_NUMBER: _ClassVar[int]
+    block: str
+    def __init__(self, block: _Optional[str] = ..., **kwargs) -> None: ...

ankka/_proto/ankka/protocol/v1/agent_pb2_grpc.py

@@ -0,0 +1,183 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+from ankka._proto.ankka.protocol.v1 import agent_pb2 as ankka_dot_protocol_dot_v1_dot_agent__pb2
+
+GRPC_GENERATED_VERSION = '1.84.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 ankka/protocol/v1/agent_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
+
+class AgentStub:
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.Plan = channel.unary_unary(
+                '/ankka.protocol.v1.Agent/Plan',
+                request_serializer=ankka_dot_protocol_dot_v1_dot_agent__pb2.PlanRequest.SerializeToString,
+                response_deserializer=ankka_dot_protocol_dot_v1_dot_agent__pb2.PlanReply.FromString,
+                _registered_method=True)
+        self.InvokeTool = channel.unary_unary(
+                '/ankka.protocol.v1.Agent/InvokeTool',
+                request_serializer=ankka_dot_protocol_dot_v1_dot_agent__pb2.ToolRequest.SerializeToString,
+                response_deserializer=ankka_dot_protocol_dot_v1_dot_agent__pb2.ToolResult.FromString,
+                _registered_method=True)
+        self.CheckGuardrail = channel.unary_unary(
+                '/ankka.protocol.v1.Agent/CheckGuardrail',
+                request_serializer=ankka_dot_protocol_dot_v1_dot_agent__pb2.GuardrailRequest.SerializeToString,
+                response_deserializer=ankka_dot_protocol_dot_v1_dot_agent__pb2.GuardrailResult.FromString,
+                _registered_method=True)
+
+
+class AgentServicer:
+    """Missing associated documentation comment in .proto file."""
+
+    def Plan(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def InvokeTool(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def CheckGuardrail(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_AgentServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'Plan': grpc.unary_unary_rpc_method_handler(
+                    servicer.Plan,
+                    request_deserializer=ankka_dot_protocol_dot_v1_dot_agent__pb2.PlanRequest.FromString,
+                    response_serializer=ankka_dot_protocol_dot_v1_dot_agent__pb2.PlanReply.SerializeToString,
+            ),
+            'InvokeTool': grpc.unary_unary_rpc_method_handler(
+                    servicer.InvokeTool,
+                    request_deserializer=ankka_dot_protocol_dot_v1_dot_agent__pb2.ToolRequest.FromString,
+                    response_serializer=ankka_dot_protocol_dot_v1_dot_agent__pb2.ToolResult.SerializeToString,
+            ),
+            'CheckGuardrail': grpc.unary_unary_rpc_method_handler(
+                    servicer.CheckGuardrail,
+                    request_deserializer=ankka_dot_protocol_dot_v1_dot_agent__pb2.GuardrailRequest.FromString,
+                    response_serializer=ankka_dot_protocol_dot_v1_dot_agent__pb2.GuardrailResult.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'ankka.protocol.v1.Agent', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('ankka.protocol.v1.Agent', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class Agent:
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def Plan(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/ankka.protocol.v1.Agent/Plan',
+            ankka_dot_protocol_dot_v1_dot_agent__pb2.PlanRequest.SerializeToString,
+            ankka_dot_protocol_dot_v1_dot_agent__pb2.PlanReply.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def InvokeTool(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/ankka.protocol.v1.Agent/InvokeTool',
+            ankka_dot_protocol_dot_v1_dot_agent__pb2.ToolRequest.SerializeToString,
+            ankka_dot_protocol_dot_v1_dot_agent__pb2.ToolResult.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def CheckGuardrail(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/ankka.protocol.v1.Agent/CheckGuardrail',
+            ankka_dot_protocol_dot_v1_dot_agent__pb2.GuardrailRequest.SerializeToString,
+            ankka_dot_protocol_dot_v1_dot_agent__pb2.GuardrailResult.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

ankka/_proto/ankka/protocol/v1/client_pb2.py

@@ -0,0 +1,52 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: ankka/protocol/v1/client.proto
+# Protobuf Python Version: 7.35.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,
+    7,
+    35,
+    1,
+    '',
+    'ankka/protocol/v1/client.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from ankka._proto.ankka.protocol.v1 import payload_pb2 as ankka_dot_protocol_dot_v1_dot_payload__pb2
+from ankka._proto.ankka.protocol.v1 import discovery_pb2 as ankka_dot_protocol_dot_v1_dot_discovery__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1e\x61nkka/protocol/v1/client.proto\x12\x11\x61nkka.protocol.v1\x1a\x1f\x61nkka/protocol/v1/payload.proto\x1a!ankka/protocol/v1/discovery.proto\"\xc9\x01\n\rInvokeRequest\x12%\n\x04kind\x18\x01 \x01(\x0e\x32\x17.ankka.protocol.v1.Kind\x12\x14\n\x0c\x63omponent_id\x18\x02 \x01(\t\x12\x11\n\tentity_id\x18\x03 \x01(\t\x12\x0c\n\x04name\x18\x04 \x01(\t\x12+\n\x07payload\x18\x05 \x01(\x0b\x32\x1a.ankka.protocol.v1.Payload\x12-\n\x08metadata\x18\x06 \x01(\x0b\x32\x1b.ankka.protocol.v1.Metadata\"u\n\x0bInvokeReply\x12\x31\n\x05reply\x18\x01 \x01(\x0b\x32 .ankka.protocol.v1.Outcome.ReplyH\x00\x12)\n\x05\x65rror\x18\x02 \x01(\x0b\x32\x18.ankka.protocol.v1.ErrorH\x00\x42\x08\n\x06result\"\x81\x01\n\x0bStreamToken\x12\x0e\n\x04text\x18\x01 \x01(\tH\x00\x12-\n\tcompleted\x18\x02 \x01(\x0b\x32\x18.ankka.protocol.v1.EmptyH\x00\x12*\n\x06\x66\x61iled\x18\x03 \x01(\x0b\x32\x18.ankka.protocol.v1.ErrorH\x00\x42\x07\n\x05token\"Z\n\x0cQueryRequest\x12\x0f\n\x07view_id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12+\n\x07payload\x18\x03 \x01(\x0b\x32\x1a.ankka.protocol.v1.Payload\"m\n\nQueryReply\x12*\n\x04rows\x18\x01 \x01(\x0b\x32\x1a.ankka.protocol.v1.PayloadH\x00\x12)\n\x05\x65rror\x18\x02 \x01(\x0b\x32\x18.ankka.protocol.v1.ErrorH\x00\x42\x08\n\x06result\"\xd7\x01\n\x0fScheduleRequest\x12\x10\n\x08timer_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64\x65lay_millis\x18\x02 \x01(\x03\x12%\n\x04kind\x18\x03 \x01(\x0e\x32\x17.ankka.protocol.v1.Kind\x12\x14\n\x0c\x63omponent_id\x18\x04 \x01(\t\x12\x16\n\tentity_id\x18\x05 \x01(\tH\x00\x88\x01\x01\x12\x0c\n\x04name\x18\x06 \x01(\t\x12+\n\x07payload\x18\x07 \x01(\x0b\x32\x1a.ankka.protocol.v1.PayloadB\x0c\n\n_entity_id\"!\n\rCancelRequest\x12\x10\n\x08timer_id\x18\x01 \x01(\t2\x81\x03\n\x06\x43lient\x12J\n\x06Invoke\x12 .ankka.protocol.v1.InvokeRequest\x1a\x1e.ankka.protocol.v1.InvokeReply\x12R\n\x0cInvokeStream\x12 .ankka.protocol.v1.InvokeRequest\x1a\x1e.ankka.protocol.v1.StreamToken0\x01\x12G\n\x05Query\x12\x1f.ankka.protocol.v1.QueryRequest\x1a\x1d.ankka.protocol.v1.QueryReply\x12H\n\x08Schedule\x12\".ankka.protocol.v1.ScheduleRequest\x1a\x18.ankka.protocol.v1.Empty\x12\x44\n\x06\x43\x61ncel\x12 .ankka.protocol.v1.CancelRequest\x1a\x18.ankka.protocol.v1.Emptyb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'ankka.protocol.v1.client_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_INVOKEREQUEST']._serialized_start=122
+  _globals['_INVOKEREQUEST']._serialized_end=323
+  _globals['_INVOKEREPLY']._serialized_start=325
+  _globals['_INVOKEREPLY']._serialized_end=442
+  _globals['_STREAMTOKEN']._serialized_start=445
+  _globals['_STREAMTOKEN']._serialized_end=574
+  _globals['_QUERYREQUEST']._serialized_start=576
+  _globals['_QUERYREQUEST']._serialized_end=666
+  _globals['_QUERYREPLY']._serialized_start=668
+  _globals['_QUERYREPLY']._serialized_end=777
+  _globals['_SCHEDULEREQUEST']._serialized_start=780
+  _globals['_SCHEDULEREQUEST']._serialized_end=995
+  _globals['_CANCELREQUEST']._serialized_start=997
+  _globals['_CANCELREQUEST']._serialized_end=1030
+  _globals['_CLIENT']._serialized_start=1033
+  _globals['_CLIENT']._serialized_end=1418
+# @@protoc_insertion_point(module_scope)

ankka/_proto/ankka/protocol/v1/client_pb2.pyi

@@ -0,0 +1,84 @@
+from ankka._proto.ankka.protocol.v1 import payload_pb2 as _payload_pb2
+from ankka._proto.ankka.protocol.v1 import discovery_pb2 as _discovery_pb2
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class InvokeRequest(_message.Message):
+    __slots__ = ("kind", "component_id", "entity_id", "name", "payload", "metadata")
+    KIND_FIELD_NUMBER: _ClassVar[int]
+    COMPONENT_ID_FIELD_NUMBER: _ClassVar[int]
+    ENTITY_ID_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    PAYLOAD_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    kind: _discovery_pb2.Kind
+    component_id: str
+    entity_id: str
+    name: str
+    payload: _payload_pb2.Payload
+    metadata: _payload_pb2.Metadata
+    def __init__(self, kind: _Optional[_Union[_discovery_pb2.Kind, str]] = ..., component_id: _Optional[str] = ..., entity_id: _Optional[str] = ..., name: _Optional[str] = ..., payload: _Optional[_Union[_payload_pb2.Payload, _Mapping]] = ..., metadata: _Optional[_Union[_payload_pb2.Metadata, _Mapping]] = ...) -> None: ...
+
+class InvokeReply(_message.Message):
+    __slots__ = ("reply", "error")
+    REPLY_FIELD_NUMBER: _ClassVar[int]
+    ERROR_FIELD_NUMBER: _ClassVar[int]
+    reply: _payload_pb2.Outcome.Reply
+    error: _payload_pb2.Error
+    def __init__(self, reply: _Optional[_Union[_payload_pb2.Outcome.Reply, _Mapping]] = ..., error: _Optional[_Union[_payload_pb2.Error, _Mapping]] = ...) -> None: ...
+
+class StreamToken(_message.Message):
+    __slots__ = ("text", "completed", "failed")
+    TEXT_FIELD_NUMBER: _ClassVar[int]
+    COMPLETED_FIELD_NUMBER: _ClassVar[int]
+    FAILED_FIELD_NUMBER: _ClassVar[int]
+    text: str
+    completed: _payload_pb2.Empty
+    failed: _payload_pb2.Error
+    def __init__(self, text: _Optional[str] = ..., completed: _Optional[_Union[_payload_pb2.Empty, _Mapping]] = ..., failed: _Optional[_Union[_payload_pb2.Error, _Mapping]] = ...) -> None: ...
+
+class QueryRequest(_message.Message):
+    __slots__ = ("view_id", "name", "payload")
+    VIEW_ID_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    PAYLOAD_FIELD_NUMBER: _ClassVar[int]
+    view_id: str
+    name: str
+    payload: _payload_pb2.Payload
+    def __init__(self, view_id: _Optional[str] = ..., name: _Optional[str] = ..., payload: _Optional[_Union[_payload_pb2.Payload, _Mapping]] = ...) -> None: ...
+
+class QueryReply(_message.Message):
+    __slots__ = ("rows", "error")
+    ROWS_FIELD_NUMBER: _ClassVar[int]
+    ERROR_FIELD_NUMBER: _ClassVar[int]
+    rows: _payload_pb2.Payload
+    error: _payload_pb2.Error
+    def __init__(self, rows: _Optional[_Union[_payload_pb2.Payload, _Mapping]] = ..., error: _Optional[_Union[_payload_pb2.Error, _Mapping]] = ...) -> None: ...
+
+class ScheduleRequest(_message.Message):
+    __slots__ = ("timer_id", "delay_millis", "kind", "component_id", "entity_id", "name", "payload")
+    TIMER_ID_FIELD_NUMBER: _ClassVar[int]
+    DELAY_MILLIS_FIELD_NUMBER: _ClassVar[int]
+    KIND_FIELD_NUMBER: _ClassVar[int]
+    COMPONENT_ID_FIELD_NUMBER: _ClassVar[int]
+    ENTITY_ID_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    PAYLOAD_FIELD_NUMBER: _ClassVar[int]
+    timer_id: str
+    delay_millis: int
+    kind: _discovery_pb2.Kind
+    component_id: str
+    entity_id: str
+    name: str
+    payload: _payload_pb2.Payload
+    def __init__(self, timer_id: _Optional[str] = ..., delay_millis: _Optional[int] = ..., kind: _Optional[_Union[_discovery_pb2.Kind, str]] = ..., component_id: _Optional[str] = ..., entity_id: _Optional[str] = ..., name: _Optional[str] = ..., payload: _Optional[_Union[_payload_pb2.Payload, _Mapping]] = ...) -> None: ...
+
+class CancelRequest(_message.Message):
+    __slots__ = ("timer_id",)
+    TIMER_ID_FIELD_NUMBER: _ClassVar[int]
+    timer_id: str
+    def __init__(self, timer_id: _Optional[str] = ...) -> None: ...