otdf-python 0.1.10__py3-none-any.whl → 0.3.5__py3-none-any.whl

otdf_python_proto/entityresolution/entity_resolution_pb2_connect.py

@@ -0,0 +1,149 @@
+# Generated Connect client code
+
+from __future__ import annotations
+from collections.abc import AsyncIterator
+from collections.abc import Iterator
+from collections.abc import Iterable
+import aiohttp
+import urllib3
+import typing
+import sys
+
+from connectrpc.client_async import AsyncConnectClient
+from connectrpc.client_sync import ConnectClient
+from connectrpc.client_protocol import ConnectProtocol
+from connectrpc.client_connect import ConnectProtocolError
+from connectrpc.headers import HeaderInput
+from connectrpc.server import ClientRequest
+from connectrpc.server import ClientStream
+from connectrpc.server import ServerResponse
+from connectrpc.server import ServerStream
+from connectrpc.server_sync import ConnectWSGI
+from connectrpc.streams import StreamInput
+from connectrpc.streams import AsyncStreamOutput
+from connectrpc.streams import StreamOutput
+from connectrpc.unary import UnaryOutput
+from connectrpc.unary import ClientStreamingOutput
+
+if typing.TYPE_CHECKING:
+    # wsgiref.types was added in Python 3.11.
+    if sys.version_info >= (3, 11):
+        from wsgiref.types import WSGIApplication
+    else:
+        from _typeshed.wsgi import WSGIApplication
+
+import entityresolution.entity_resolution_pb2
+
+class EntityResolutionServiceClient:
+    def __init__(
+        self,
+        base_url: str,
+        http_client: urllib3.PoolManager | None = None,
+        protocol: ConnectProtocol = ConnectProtocol.CONNECT_PROTOBUF,
+    ):
+        self.base_url = base_url
+        self._connect_client = ConnectClient(http_client, protocol)
+    def call_resolve_entities(
+        self, req: entityresolution.entity_resolution_pb2.ResolveEntitiesRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[entityresolution.entity_resolution_pb2.ResolveEntitiesResponse]:
+        """Low-level method to call ResolveEntities, granting access to errors and metadata"""
+        url = self.base_url + "/entityresolution.EntityResolutionService/ResolveEntities"
+        return self._connect_client.call_unary(url, req, entityresolution.entity_resolution_pb2.ResolveEntitiesResponse,extra_headers, timeout_seconds)
+
+
+    def resolve_entities(
+        self, req: entityresolution.entity_resolution_pb2.ResolveEntitiesRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> entityresolution.entity_resolution_pb2.ResolveEntitiesResponse:
+        response = self.call_resolve_entities(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+    def call_create_entity_chain_from_jwt(
+        self, req: entityresolution.entity_resolution_pb2.CreateEntityChainFromJwtRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[entityresolution.entity_resolution_pb2.CreateEntityChainFromJwtResponse]:
+        """Low-level method to call CreateEntityChainFromJwt, granting access to errors and metadata"""
+        url = self.base_url + "/entityresolution.EntityResolutionService/CreateEntityChainFromJwt"
+        return self._connect_client.call_unary(url, req, entityresolution.entity_resolution_pb2.CreateEntityChainFromJwtResponse,extra_headers, timeout_seconds)
+
+
+    def create_entity_chain_from_jwt(
+        self, req: entityresolution.entity_resolution_pb2.CreateEntityChainFromJwtRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> entityresolution.entity_resolution_pb2.CreateEntityChainFromJwtResponse:
+        response = self.call_create_entity_chain_from_jwt(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+
+class AsyncEntityResolutionServiceClient:
+    def __init__(
+        self,
+        base_url: str,
+        http_client: aiohttp.ClientSession,
+        protocol: ConnectProtocol = ConnectProtocol.CONNECT_PROTOBUF,
+    ):
+        self.base_url = base_url
+        self._connect_client = AsyncConnectClient(http_client, protocol)
+
+    async def call_resolve_entities(
+        self, req: entityresolution.entity_resolution_pb2.ResolveEntitiesRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[entityresolution.entity_resolution_pb2.ResolveEntitiesResponse]:
+        """Low-level method to call ResolveEntities, granting access to errors and metadata"""
+        url = self.base_url + "/entityresolution.EntityResolutionService/ResolveEntities"
+        return await self._connect_client.call_unary(url, req, entityresolution.entity_resolution_pb2.ResolveEntitiesResponse,extra_headers, timeout_seconds)
+
+    async def resolve_entities(
+        self, req: entityresolution.entity_resolution_pb2.ResolveEntitiesRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> entityresolution.entity_resolution_pb2.ResolveEntitiesResponse:
+        response = await self.call_resolve_entities(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+    async def call_create_entity_chain_from_jwt(
+        self, req: entityresolution.entity_resolution_pb2.CreateEntityChainFromJwtRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[entityresolution.entity_resolution_pb2.CreateEntityChainFromJwtResponse]:
+        """Low-level method to call CreateEntityChainFromJwt, granting access to errors and metadata"""
+        url = self.base_url + "/entityresolution.EntityResolutionService/CreateEntityChainFromJwt"
+        return await self._connect_client.call_unary(url, req, entityresolution.entity_resolution_pb2.CreateEntityChainFromJwtResponse,extra_headers, timeout_seconds)
+
+    async def create_entity_chain_from_jwt(
+        self, req: entityresolution.entity_resolution_pb2.CreateEntityChainFromJwtRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> entityresolution.entity_resolution_pb2.CreateEntityChainFromJwtResponse:
+        response = await self.call_create_entity_chain_from_jwt(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+
+@typing.runtime_checkable
+class EntityResolutionServiceProtocol(typing.Protocol):
+    def resolve_entities(self, req: ClientRequest[entityresolution.entity_resolution_pb2.ResolveEntitiesRequest]) -> ServerResponse[entityresolution.entity_resolution_pb2.ResolveEntitiesResponse]:
+        ...
+    def create_entity_chain_from_jwt(self, req: ClientRequest[entityresolution.entity_resolution_pb2.CreateEntityChainFromJwtRequest]) -> ServerResponse[entityresolution.entity_resolution_pb2.CreateEntityChainFromJwtResponse]:
+        ...
+
+ENTITY_RESOLUTION_SERVICE_PATH_PREFIX = "/entityresolution.EntityResolutionService"
+
+def wsgi_entity_resolution_service(implementation: EntityResolutionServiceProtocol) -> WSGIApplication:
+    app = ConnectWSGI()
+    app.register_unary_rpc("/entityresolution.EntityResolutionService/ResolveEntities", implementation.resolve_entities, entityresolution.entity_resolution_pb2.ResolveEntitiesRequest)
+    app.register_unary_rpc("/entityresolution.EntityResolutionService/CreateEntityChainFromJwt", implementation.create_entity_chain_from_jwt, entityresolution.entity_resolution_pb2.CreateEntityChainFromJwtRequest)
+    return app

otdf_python_proto/entityresolution/v2/entity_resolution_pb2.py

@@ -0,0 +1,55 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: entityresolution/v2/entity_resolution.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,
+    '',
+    'entityresolution/v2/entity_resolution.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from buf.validate import validate_pb2 as buf_dot_validate_dot_validate__pb2
+from entity import entity_pb2 as entity_dot_entity__pb2
+from google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2
+from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+entityresolution/v2/entity_resolution.proto\x12\x13\x65ntityresolution.v2\x1a\x1b\x62uf/validate/validate.proto\x1a\x13\x65ntity/entity.proto\x1a\x19google/protobuf/any.proto\x1a\x1cgoogle/protobuf/struct.proto\"{\n\x14\x45ntityRepresentation\x12\x1f\n\x0boriginal_id\x18\x01 \x01(\tR\noriginalId\x12\x42\n\x10\x61\x64\x64itional_props\x18\x02 \x03(\x0b\x32\x17.google.protobuf.StructR\x0f\x61\x64\x64itionalProps\"Q\n\x16ResolveEntitiesRequest\x12\x37\n\x08\x65ntities\x18\x01 \x03(\x0b\x32\x0e.entity.EntityB\x0b\xbaH\x08\x92\x01\x02\x08\x01\xc8\x01\x01R\x08\x65ntities\"{\n\x17ResolveEntitiesResponse\x12`\n\x16\x65ntity_representations\x18\x01 \x03(\x0b\x32).entityresolution.v2.EntityRepresentationR\x15\x65ntityRepresentations\"\x8b\x01\n\x13\x45ntityNotFoundError\x12\x12\n\x04\x63ode\x18\x01 \x01(\x05R\x04\x63ode\x12\x18\n\x07message\x18\x02 \x01(\tR\x07message\x12.\n\x07\x64\x65tails\x18\x03 \x03(\x0b\x32\x14.google.protobuf.AnyR\x07\x64\x65tails\x12\x16\n\x06\x65ntity\x18\x04 \x01(\tR\x06\x65ntity\"L\n#CreateEntityChainsFromTokensRequest\x12%\n\x06tokens\x18\x01 \x03(\x0b\x32\r.entity.TokenR\x06tokens\"`\n$CreateEntityChainsFromTokensResponse\x12\x38\n\rentity_chains\x18\x01 \x03(\x0b\x32\x13.entity.EntityChainR\x0c\x65ntityChains2\xa1\x02\n\x17\x45ntityResolutionService\x12n\n\x0fResolveEntities\x12+.entityresolution.v2.ResolveEntitiesRequest\x1a,.entityresolution.v2.ResolveEntitiesResponse\"\x00\x12\x95\x01\n\x1c\x43reateEntityChainsFromTokens\x12\x38.entityresolution.v2.CreateEntityChainsFromTokensRequest\x1a\x39.entityresolution.v2.CreateEntityChainsFromTokensResponse\"\x00\x42\x9d\x01\n\x17\x63om.entityresolution.v2B\x15\x45ntityResolutionProtoP\x01\xa2\x02\x03\x45XX\xaa\x02\x13\x45ntityresolution.V2\xca\x02\x13\x45ntityresolution\\V2\xe2\x02\x1f\x45ntityresolution\\V2\\GPBMetadata\xea\x02\x14\x45ntityresolution::V2b\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'entityresolution.v2.entity_resolution_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\027com.entityresolution.v2B\025EntityResolutionProtoP\001\242\002\003EXX\252\002\023Entityresolution.V2\312\002\023Entityresolution\\V2\342\002\037Entityresolution\\V2\\GPBMetadata\352\002\024Entityresolution::V2'
+  _globals['_RESOLVEENTITIESREQUEST'].fields_by_name['entities']._loaded_options = None
+  _globals['_RESOLVEENTITIESREQUEST'].fields_by_name['entities']._serialized_options = b'\272H\010\222\001\002\010\001\310\001\001'
+  _globals['_ENTITYREPRESENTATION']._serialized_start=175
+  _globals['_ENTITYREPRESENTATION']._serialized_end=298
+  _globals['_RESOLVEENTITIESREQUEST']._serialized_start=300
+  _globals['_RESOLVEENTITIESREQUEST']._serialized_end=381
+  _globals['_RESOLVEENTITIESRESPONSE']._serialized_start=383
+  _globals['_RESOLVEENTITIESRESPONSE']._serialized_end=506
+  _globals['_ENTITYNOTFOUNDERROR']._serialized_start=509
+  _globals['_ENTITYNOTFOUNDERROR']._serialized_end=648
+  _globals['_CREATEENTITYCHAINSFROMTOKENSREQUEST']._serialized_start=650
+  _globals['_CREATEENTITYCHAINSFROMTOKENSREQUEST']._serialized_end=726
+  _globals['_CREATEENTITYCHAINSFROMTOKENSRESPONSE']._serialized_start=728
+  _globals['_CREATEENTITYCHAINSFROMTOKENSRESPONSE']._serialized_end=824
+  _globals['_ENTITYRESOLUTIONSERVICE']._serialized_start=827
+  _globals['_ENTITYRESOLUTIONSERVICE']._serialized_end=1116
+# @@protoc_insertion_point(module_scope)

otdf_python_proto/entityresolution/v2/entity_resolution_pb2.pyi

@@ -0,0 +1,55 @@
+from buf.validate import validate_pb2 as _validate_pb2
+from entity import entity_pb2 as _entity_pb2
+from google.protobuf import any_pb2 as _any_pb2
+from google.protobuf import struct_pb2 as _struct_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 EntityRepresentation(_message.Message):
+    __slots__ = ("original_id", "additional_props")
+    ORIGINAL_ID_FIELD_NUMBER: _ClassVar[int]
+    ADDITIONAL_PROPS_FIELD_NUMBER: _ClassVar[int]
+    original_id: str
+    additional_props: _containers.RepeatedCompositeFieldContainer[_struct_pb2.Struct]
+    def __init__(self, original_id: _Optional[str] = ..., additional_props: _Optional[_Iterable[_Union[_struct_pb2.Struct, _Mapping]]] = ...) -> None: ...
+
+class ResolveEntitiesRequest(_message.Message):
+    __slots__ = ("entities",)
+    ENTITIES_FIELD_NUMBER: _ClassVar[int]
+    entities: _containers.RepeatedCompositeFieldContainer[_entity_pb2.Entity]
+    def __init__(self, entities: _Optional[_Iterable[_Union[_entity_pb2.Entity, _Mapping]]] = ...) -> None: ...
+
+class ResolveEntitiesResponse(_message.Message):
+    __slots__ = ("entity_representations",)
+    ENTITY_REPRESENTATIONS_FIELD_NUMBER: _ClassVar[int]
+    entity_representations: _containers.RepeatedCompositeFieldContainer[EntityRepresentation]
+    def __init__(self, entity_representations: _Optional[_Iterable[_Union[EntityRepresentation, _Mapping]]] = ...) -> None: ...
+
+class EntityNotFoundError(_message.Message):
+    __slots__ = ("code", "message", "details", "entity")
+    CODE_FIELD_NUMBER: _ClassVar[int]
+    MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    DETAILS_FIELD_NUMBER: _ClassVar[int]
+    ENTITY_FIELD_NUMBER: _ClassVar[int]
+    code: int
+    message: str
+    details: _containers.RepeatedCompositeFieldContainer[_any_pb2.Any]
+    entity: str
+    def __init__(self, code: _Optional[int] = ..., message: _Optional[str] = ..., details: _Optional[_Iterable[_Union[_any_pb2.Any, _Mapping]]] = ..., entity: _Optional[str] = ...) -> None: ...
+
+class CreateEntityChainsFromTokensRequest(_message.Message):
+    __slots__ = ("tokens",)
+    TOKENS_FIELD_NUMBER: _ClassVar[int]
+    tokens: _containers.RepeatedCompositeFieldContainer[_entity_pb2.Token]
+    def __init__(self, tokens: _Optional[_Iterable[_Union[_entity_pb2.Token, _Mapping]]] = ...) -> None: ...
+
+class CreateEntityChainsFromTokensResponse(_message.Message):
+    __slots__ = ("entity_chains",)
+    ENTITY_CHAINS_FIELD_NUMBER: _ClassVar[int]
+    entity_chains: _containers.RepeatedCompositeFieldContainer[_entity_pb2.EntityChain]
+    def __init__(self, entity_chains: _Optional[_Iterable[_Union[_entity_pb2.EntityChain, _Mapping]]] = ...) -> None: ...

otdf_python_proto/entityresolution/v2/entity_resolution_pb2_connect.py

@@ -0,0 +1,149 @@
+# Generated Connect client code
+
+from __future__ import annotations
+from collections.abc import AsyncIterator
+from collections.abc import Iterator
+from collections.abc import Iterable
+import aiohttp
+import urllib3
+import typing
+import sys
+
+from connectrpc.client_async import AsyncConnectClient
+from connectrpc.client_sync import ConnectClient
+from connectrpc.client_protocol import ConnectProtocol
+from connectrpc.client_connect import ConnectProtocolError
+from connectrpc.headers import HeaderInput
+from connectrpc.server import ClientRequest
+from connectrpc.server import ClientStream
+from connectrpc.server import ServerResponse
+from connectrpc.server import ServerStream
+from connectrpc.server_sync import ConnectWSGI
+from connectrpc.streams import StreamInput
+from connectrpc.streams import AsyncStreamOutput
+from connectrpc.streams import StreamOutput
+from connectrpc.unary import UnaryOutput
+from connectrpc.unary import ClientStreamingOutput
+
+if typing.TYPE_CHECKING:
+    # wsgiref.types was added in Python 3.11.
+    if sys.version_info >= (3, 11):
+        from wsgiref.types import WSGIApplication
+    else:
+        from _typeshed.wsgi import WSGIApplication
+
+import entityresolution.v2.entity_resolution_pb2
+
+class EntityResolutionServiceClient:
+    def __init__(
+        self,
+        base_url: str,
+        http_client: urllib3.PoolManager | None = None,
+        protocol: ConnectProtocol = ConnectProtocol.CONNECT_PROTOBUF,
+    ):
+        self.base_url = base_url
+        self._connect_client = ConnectClient(http_client, protocol)
+    def call_resolve_entities(
+        self, req: entityresolution.v2.entity_resolution_pb2.ResolveEntitiesRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[entityresolution.v2.entity_resolution_pb2.ResolveEntitiesResponse]:
+        """Low-level method to call ResolveEntities, granting access to errors and metadata"""
+        url = self.base_url + "/entityresolution.v2.EntityResolutionService/ResolveEntities"
+        return self._connect_client.call_unary(url, req, entityresolution.v2.entity_resolution_pb2.ResolveEntitiesResponse,extra_headers, timeout_seconds)
+
+
+    def resolve_entities(
+        self, req: entityresolution.v2.entity_resolution_pb2.ResolveEntitiesRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> entityresolution.v2.entity_resolution_pb2.ResolveEntitiesResponse:
+        response = self.call_resolve_entities(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+    def call_create_entity_chains_from_tokens(
+        self, req: entityresolution.v2.entity_resolution_pb2.CreateEntityChainsFromTokensRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[entityresolution.v2.entity_resolution_pb2.CreateEntityChainsFromTokensResponse]:
+        """Low-level method to call CreateEntityChainsFromTokens, granting access to errors and metadata"""
+        url = self.base_url + "/entityresolution.v2.EntityResolutionService/CreateEntityChainsFromTokens"
+        return self._connect_client.call_unary(url, req, entityresolution.v2.entity_resolution_pb2.CreateEntityChainsFromTokensResponse,extra_headers, timeout_seconds)
+
+
+    def create_entity_chains_from_tokens(
+        self, req: entityresolution.v2.entity_resolution_pb2.CreateEntityChainsFromTokensRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> entityresolution.v2.entity_resolution_pb2.CreateEntityChainsFromTokensResponse:
+        response = self.call_create_entity_chains_from_tokens(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+
+class AsyncEntityResolutionServiceClient:
+    def __init__(
+        self,
+        base_url: str,
+        http_client: aiohttp.ClientSession,
+        protocol: ConnectProtocol = ConnectProtocol.CONNECT_PROTOBUF,
+    ):
+        self.base_url = base_url
+        self._connect_client = AsyncConnectClient(http_client, protocol)
+
+    async def call_resolve_entities(
+        self, req: entityresolution.v2.entity_resolution_pb2.ResolveEntitiesRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[entityresolution.v2.entity_resolution_pb2.ResolveEntitiesResponse]:
+        """Low-level method to call ResolveEntities, granting access to errors and metadata"""
+        url = self.base_url + "/entityresolution.v2.EntityResolutionService/ResolveEntities"
+        return await self._connect_client.call_unary(url, req, entityresolution.v2.entity_resolution_pb2.ResolveEntitiesResponse,extra_headers, timeout_seconds)
+
+    async def resolve_entities(
+        self, req: entityresolution.v2.entity_resolution_pb2.ResolveEntitiesRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> entityresolution.v2.entity_resolution_pb2.ResolveEntitiesResponse:
+        response = await self.call_resolve_entities(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+    async def call_create_entity_chains_from_tokens(
+        self, req: entityresolution.v2.entity_resolution_pb2.CreateEntityChainsFromTokensRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[entityresolution.v2.entity_resolution_pb2.CreateEntityChainsFromTokensResponse]:
+        """Low-level method to call CreateEntityChainsFromTokens, granting access to errors and metadata"""
+        url = self.base_url + "/entityresolution.v2.EntityResolutionService/CreateEntityChainsFromTokens"
+        return await self._connect_client.call_unary(url, req, entityresolution.v2.entity_resolution_pb2.CreateEntityChainsFromTokensResponse,extra_headers, timeout_seconds)
+
+    async def create_entity_chains_from_tokens(
+        self, req: entityresolution.v2.entity_resolution_pb2.CreateEntityChainsFromTokensRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> entityresolution.v2.entity_resolution_pb2.CreateEntityChainsFromTokensResponse:
+        response = await self.call_create_entity_chains_from_tokens(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+
+@typing.runtime_checkable
+class EntityResolutionServiceProtocol(typing.Protocol):
+    def resolve_entities(self, req: ClientRequest[entityresolution.v2.entity_resolution_pb2.ResolveEntitiesRequest]) -> ServerResponse[entityresolution.v2.entity_resolution_pb2.ResolveEntitiesResponse]:
+        ...
+    def create_entity_chains_from_tokens(self, req: ClientRequest[entityresolution.v2.entity_resolution_pb2.CreateEntityChainsFromTokensRequest]) -> ServerResponse[entityresolution.v2.entity_resolution_pb2.CreateEntityChainsFromTokensResponse]:
+        ...
+
+ENTITY_RESOLUTION_SERVICE_PATH_PREFIX = "/entityresolution.v2.EntityResolutionService"
+
+def wsgi_entity_resolution_service(implementation: EntityResolutionServiceProtocol) -> WSGIApplication:
+    app = ConnectWSGI()
+    app.register_unary_rpc("/entityresolution.v2.EntityResolutionService/ResolveEntities", implementation.resolve_entities, entityresolution.v2.entity_resolution_pb2.ResolveEntitiesRequest)
+    app.register_unary_rpc("/entityresolution.v2.EntityResolutionService/CreateEntityChainsFromTokens", implementation.create_entity_chains_from_tokens, entityresolution.v2.entity_resolution_pb2.CreateEntityChainsFromTokensRequest)
+    return app

otdf_python_proto/kas/__init__.py

@@ -0,0 +1,9 @@
+"""KAS (Key Access Service) protobuf definitions."""
+
+from .kas_pb2 import *
+from .kas_pb2_connect import AccessServiceClient, AsyncAccessServiceClient
+
+__all__ = [
+    "AccessServiceClient",
+    "AsyncAccessServiceClient",
+]

otdf_python_proto/kas/kas_pb2.py

@@ -0,0 +1,103 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: kas/kas.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,
+    '',
+    'kas/kas.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2
+from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
+from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
+from protoc_gen_openapiv2.options import annotations_pb2 as protoc__gen__openapiv2_dot_options_dot_annotations__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\rkas/kas.proto\x12\x03kas\x1a\x1cgoogle/api/annotations.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1egoogle/protobuf/wrappers.proto\x1a.protoc-gen-openapiv2/options/annotations.proto\"\r\n\x0bInfoRequest\"(\n\x0cInfoResponse\x12\x18\n\x07version\x18\x01 \x01(\tR\x07version\"6\n\x16LegacyPublicKeyRequest\x12\x1c\n\talgorithm\x18\x01 \x01(\tR\talgorithm\";\n\rPolicyBinding\x12\x16\n\talgorithm\x18\x01 \x01(\tR\x03\x61lg\x12\x12\n\x04hash\x18\x02 \x01(\tR\x04hash\"\xd3\x02\n\tKeyAccess\x12-\n\x12\x65ncrypted_metadata\x18\x01 \x01(\tR\x11\x65ncryptedMetadata\x12\x39\n\x0epolicy_binding\x18\x02 \x01(\x0b\x32\x12.kas.PolicyBindingR\rpolicyBinding\x12\x1a\n\x08protocol\x18\x03 \x01(\tR\x08protocol\x12\x16\n\x08key_type\x18\x04 \x01(\tR\x04type\x12\x14\n\x07kas_url\x18\x05 \x01(\tR\x03url\x12\x10\n\x03kid\x18\x06 \x01(\tR\x03kid\x12\x15\n\x08split_id\x18\x07 \x01(\tR\x03sid\x12\x1f\n\x0bwrapped_key\x18\x08 \x01(\x0cR\nwrappedKey\x12\x16\n\x06header\x18\t \x01(\x0cR\x06header\x12\x30\n\x14\x65phemeral_public_key\x18\n \x01(\tR\x12\x65phemeralPublicKey\"\x86\x05\n\x15UnsignedRewrapRequest\x12*\n\x11\x63lient_public_key\x18\x01 \x01(\tR\x0f\x63lientPublicKey\x12H\n\x08requests\x18\x02 \x03(\x0b\x32,.kas.UnsignedRewrapRequest.WithPolicyRequestR\x08requests\x12\x31\n\nkey_access\x18\x03 \x01(\x0b\x32\x0e.kas.KeyAccessB\x02\x18\x01R\tkeyAccess\x12\x1a\n\x06policy\x18\x04 \x01(\tB\x02\x18\x01R\x06policy\x12 \n\talgorithm\x18\x05 \x01(\tB\x02\x18\x01R\talgorithm\x1a\x30\n\nWithPolicy\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04\x62ody\x18\x02 \x01(\tR\x04\x62ody\x1a\x82\x01\n\x13WithKeyAccessObject\x12/\n\x14key_access_object_id\x18\x01 \x01(\tR\x11keyAccessObjectId\x12:\n\x11key_access_object\x18\x02 \x01(\x0b\x32\x0e.kas.KeyAccessR\x0fkeyAccessObject\x1a\xce\x01\n\x11WithPolicyRequest\x12\\\n\x12key_access_objects\x18\x01 \x03(\x0b\x32..kas.UnsignedRewrapRequest.WithKeyAccessObjectR\x10keyAccessObjects\x12=\n\x06policy\x18\x02 \x01(\x0b\x32%.kas.UnsignedRewrapRequest.WithPolicyR\x06policy\x12\x1c\n\talgorithm\x18\x03 \x01(\tR\talgorithm\"\xb1\x01\n\x10PublicKeyRequest\x12Q\n\talgorithm\x18\x01 \x01(\tB3\x92\x41\x30\x32.algorithm type rsa:<keysize> or ec:<curvename>R\talgorithm\x12&\n\x03\x66mt\x18\x02 \x01(\tB\x14\x92\x41\x11\x32\x0fresponse formatR\x03\x66mt\x12\"\n\x01v\x18\x03 \x01(\tB\x14\x92\x41\x11\x32\x0frequest versionR\x01v\"D\n\x11PublicKeyResponse\x12\x1d\n\npublic_key\x18\x01 \x01(\tR\tpublicKey\x12\x10\n\x03kid\x18\x02 \x01(\tR\x03kid\"O\n\rRewrapRequest\x12\x30\n\x14signed_request_token\x18\x01 \x01(\tR\x12signedRequestTokenJ\x04\x08\x02\x10\x03R\x06\x62\x65\x61rer\"\xc7\x02\n\x15KeyAccessRewrapResult\x12\x44\n\x08metadata\x18\x01 \x03(\x0b\x32(.kas.KeyAccessRewrapResult.MetadataEntryR\x08metadata\x12/\n\x14key_access_object_id\x18\x02 \x01(\tR\x11keyAccessObjectId\x12\x16\n\x06status\x18\x03 \x01(\tR\x06status\x12(\n\x0fkas_wrapped_key\x18\x04 \x01(\x0cH\x00R\rkasWrappedKey\x12\x16\n\x05\x65rror\x18\x05 \x01(\tH\x00R\x05\x65rror\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\x08\n\x06result\"g\n\x12PolicyRewrapResult\x12\x1b\n\tpolicy_id\x18\x01 \x01(\tR\x08policyId\x12\x34\n\x07results\x18\x02 \x03(\x0b\x32\x1a.kas.KeyAccessRewrapResultR\x07results\"\xea\x02\n\x0eRewrapResponse\x12\x41\n\x08metadata\x18\x01 \x03(\x0b\x32!.kas.RewrapResponse.MetadataEntryB\x02\x18\x01R\x08metadata\x12\x30\n\x12\x65ntity_wrapped_key\x18\x02 \x01(\x0c\x42\x02\x18\x01R\x10\x65ntityWrappedKey\x12,\n\x12session_public_key\x18\x03 \x01(\tR\x10sessionPublicKey\x12)\n\x0eschema_version\x18\x04 \x01(\tB\x02\x18\x01R\rschemaVersion\x12\x35\n\tresponses\x18\x05 \x03(\x0b\x32\x17.kas.PolicyRewrapResultR\tresponses\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x32\xd1\x02\n\rAccessService\x12i\n\tPublicKey\x12\x15.kas.PublicKeyRequest\x1a\x16.kas.PublicKeyResponse\"-\x90\x02\x01\x92\x41\tJ\x07\n\x03\x32\x30\x30\x12\x00\x82\xd3\xe4\x93\x02\x18\x12\x16/kas/v2/kas_public_key\x12{\n\x0fLegacyPublicKey\x12\x1b.kas.LegacyPublicKeyRequest\x1a\x1c.google.protobuf.StringValue\"-\x88\x02\x01\x90\x02\x01\x92\x41\tJ\x07\n\x03\x32\x30\x30\x12\x00\x82\xd3\xe4\x93\x02\x15\x12\x13/kas/kas_public_key\x12X\n\x06Rewrap\x12\x12.kas.RewrapRequest\x1a\x13.kas.RewrapResponse\"%\x92\x41\tJ\x07\n\x03\x32\x30\x30\x12\x00\x82\xd3\xe4\x93\x02\x13\"\x0e/kas/v2/rewrap:\x01*B\xb5\x01\n\x07\x63om.kasB\x08KasProtoP\x01\xa2\x02\x03KXX\xaa\x02\x03Kas\xca\x02\x03Kas\xe2\x02\x0fKas\\GPBMetadata\xea\x02\x03Kas\x92\x41s\x12q\n\x1aOpenTDF Key Access Service*L\n\x12\x42SD 3-Clause Clear\x12\x36https://github.com/opentdf/backend/blob/master/LICENSE2\x05\x31.5.0b\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'kas.kas_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\007com.kasB\010KasProtoP\001\242\002\003KXX\252\002\003Kas\312\002\003Kas\342\002\017Kas\\GPBMetadata\352\002\003Kas\222As\022q\n\032OpenTDF Key Access Service*L\n\022BSD 3-Clause Clear\0226https://github.com/opentdf/backend/blob/master/LICENSE2\0051.5.0'
+  _globals['_UNSIGNEDREWRAPREQUEST'].fields_by_name['key_access']._loaded_options = None
+  _globals['_UNSIGNEDREWRAPREQUEST'].fields_by_name['key_access']._serialized_options = b'\030\001'
+  _globals['_UNSIGNEDREWRAPREQUEST'].fields_by_name['policy']._loaded_options = None
+  _globals['_UNSIGNEDREWRAPREQUEST'].fields_by_name['policy']._serialized_options = b'\030\001'
+  _globals['_UNSIGNEDREWRAPREQUEST'].fields_by_name['algorithm']._loaded_options = None
+  _globals['_UNSIGNEDREWRAPREQUEST'].fields_by_name['algorithm']._serialized_options = b'\030\001'
+  _globals['_PUBLICKEYREQUEST'].fields_by_name['algorithm']._loaded_options = None
+  _globals['_PUBLICKEYREQUEST'].fields_by_name['algorithm']._serialized_options = b'\222A02.algorithm type rsa:<keysize> or ec:<curvename>'
+  _globals['_PUBLICKEYREQUEST'].fields_by_name['fmt']._loaded_options = None
+  _globals['_PUBLICKEYREQUEST'].fields_by_name['fmt']._serialized_options = b'\222A\0212\017response format'
+  _globals['_PUBLICKEYREQUEST'].fields_by_name['v']._loaded_options = None
+  _globals['_PUBLICKEYREQUEST'].fields_by_name['v']._serialized_options = b'\222A\0212\017request version'
+  _globals['_KEYACCESSREWRAPRESULT_METADATAENTRY']._loaded_options = None
+  _globals['_KEYACCESSREWRAPRESULT_METADATAENTRY']._serialized_options = b'8\001'
+  _globals['_REWRAPRESPONSE_METADATAENTRY']._loaded_options = None
+  _globals['_REWRAPRESPONSE_METADATAENTRY']._serialized_options = b'8\001'
+  _globals['_REWRAPRESPONSE'].fields_by_name['metadata']._loaded_options = None
+  _globals['_REWRAPRESPONSE'].fields_by_name['metadata']._serialized_options = b'\030\001'
+  _globals['_REWRAPRESPONSE'].fields_by_name['entity_wrapped_key']._loaded_options = None
+  _globals['_REWRAPRESPONSE'].fields_by_name['entity_wrapped_key']._serialized_options = b'\030\001'
+  _globals['_REWRAPRESPONSE'].fields_by_name['schema_version']._loaded_options = None
+  _globals['_REWRAPRESPONSE'].fields_by_name['schema_version']._serialized_options = b'\030\001'
+  _globals['_ACCESSSERVICE'].methods_by_name['PublicKey']._loaded_options = None
+  _globals['_ACCESSSERVICE'].methods_by_name['PublicKey']._serialized_options = b'\220\002\001\222A\tJ\007\n\003200\022\000\202\323\344\223\002\030\022\026/kas/v2/kas_public_key'
+  _globals['_ACCESSSERVICE'].methods_by_name['LegacyPublicKey']._loaded_options = None
+  _globals['_ACCESSSERVICE'].methods_by_name['LegacyPublicKey']._serialized_options = b'\210\002\001\220\002\001\222A\tJ\007\n\003200\022\000\202\323\344\223\002\025\022\023/kas/kas_public_key'
+  _globals['_ACCESSSERVICE'].methods_by_name['Rewrap']._loaded_options = None
+  _globals['_ACCESSSERVICE'].methods_by_name['Rewrap']._serialized_options = b'\222A\tJ\007\n\003200\022\000\202\323\344\223\002\023\"\016/kas/v2/rewrap:\001*'
+  _globals['_INFOREQUEST']._serialized_start=162
+  _globals['_INFOREQUEST']._serialized_end=175
+  _globals['_INFORESPONSE']._serialized_start=177
+  _globals['_INFORESPONSE']._serialized_end=217
+  _globals['_LEGACYPUBLICKEYREQUEST']._serialized_start=219
+  _globals['_LEGACYPUBLICKEYREQUEST']._serialized_end=273
+  _globals['_POLICYBINDING']._serialized_start=275
+  _globals['_POLICYBINDING']._serialized_end=334
+  _globals['_KEYACCESS']._serialized_start=337
+  _globals['_KEYACCESS']._serialized_end=676
+  _globals['_UNSIGNEDREWRAPREQUEST']._serialized_start=679
+  _globals['_UNSIGNEDREWRAPREQUEST']._serialized_end=1325
+  _globals['_UNSIGNEDREWRAPREQUEST_WITHPOLICY']._serialized_start=935
+  _globals['_UNSIGNEDREWRAPREQUEST_WITHPOLICY']._serialized_end=983
+  _globals['_UNSIGNEDREWRAPREQUEST_WITHKEYACCESSOBJECT']._serialized_start=986
+  _globals['_UNSIGNEDREWRAPREQUEST_WITHKEYACCESSOBJECT']._serialized_end=1116
+  _globals['_UNSIGNEDREWRAPREQUEST_WITHPOLICYREQUEST']._serialized_start=1119
+  _globals['_UNSIGNEDREWRAPREQUEST_WITHPOLICYREQUEST']._serialized_end=1325
+  _globals['_PUBLICKEYREQUEST']._serialized_start=1328
+  _globals['_PUBLICKEYREQUEST']._serialized_end=1505
+  _globals['_PUBLICKEYRESPONSE']._serialized_start=1507
+  _globals['_PUBLICKEYRESPONSE']._serialized_end=1575
+  _globals['_REWRAPREQUEST']._serialized_start=1577
+  _globals['_REWRAPREQUEST']._serialized_end=1656
+  _globals['_KEYACCESSREWRAPRESULT']._serialized_start=1659
+  _globals['_KEYACCESSREWRAPRESULT']._serialized_end=1986
+  _globals['_KEYACCESSREWRAPRESULT_METADATAENTRY']._serialized_start=1893
+  _globals['_KEYACCESSREWRAPRESULT_METADATAENTRY']._serialized_end=1976
+  _globals['_POLICYREWRAPRESULT']._serialized_start=1988
+  _globals['_POLICYREWRAPRESULT']._serialized_end=2091
+  _globals['_REWRAPRESPONSE']._serialized_start=2094
+  _globals['_REWRAPRESPONSE']._serialized_end=2456
+  _globals['_REWRAPRESPONSE_METADATAENTRY']._serialized_start=1893
+  _globals['_REWRAPRESPONSE_METADATAENTRY']._serialized_end=1976
+  _globals['_ACCESSSERVICE']._serialized_start=2459
+  _globals['_ACCESSSERVICE']._serialized_end=2796
+# @@protoc_insertion_point(module_scope)