hermes-client-python 1.4.11__py3-none-any.whl

hermes_client_python/hermes_pb2_grpc.py

@@ -0,0 +1,485 @@
+# 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 . import hermes_pb2 as hermes__pb2
+
+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 hermes_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 SearchServiceStub(object):
+    """Search service
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.Search = channel.unary_unary(
+                '/hermes.SearchService/Search',
+                request_serializer=hermes__pb2.SearchRequest.SerializeToString,
+                response_deserializer=hermes__pb2.SearchResponse.FromString,
+                _registered_method=True)
+        self.GetDocument = channel.unary_unary(
+                '/hermes.SearchService/GetDocument',
+                request_serializer=hermes__pb2.GetDocumentRequest.SerializeToString,
+                response_deserializer=hermes__pb2.GetDocumentResponse.FromString,
+                _registered_method=True)
+        self.GetIndexInfo = channel.unary_unary(
+                '/hermes.SearchService/GetIndexInfo',
+                request_serializer=hermes__pb2.GetIndexInfoRequest.SerializeToString,
+                response_deserializer=hermes__pb2.GetIndexInfoResponse.FromString,
+                _registered_method=True)
+
+
+class SearchServiceServicer(object):
+    """Search service
+    """
+
+    def Search(self, request, context):
+        """Search for documents
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetDocument(self, request, context):
+        """Get document by ID
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetIndexInfo(self, request, context):
+        """Get index info
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_SearchServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'Search': grpc.unary_unary_rpc_method_handler(
+                    servicer.Search,
+                    request_deserializer=hermes__pb2.SearchRequest.FromString,
+                    response_serializer=hermes__pb2.SearchResponse.SerializeToString,
+            ),
+            'GetDocument': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetDocument,
+                    request_deserializer=hermes__pb2.GetDocumentRequest.FromString,
+                    response_serializer=hermes__pb2.GetDocumentResponse.SerializeToString,
+            ),
+            'GetIndexInfo': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetIndexInfo,
+                    request_deserializer=hermes__pb2.GetIndexInfoRequest.FromString,
+                    response_serializer=hermes__pb2.GetIndexInfoResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'hermes.SearchService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('hermes.SearchService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class SearchService(object):
+    """Search service
+    """
+
+    @staticmethod
+    def Search(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,
+            '/hermes.SearchService/Search',
+            hermes__pb2.SearchRequest.SerializeToString,
+            hermes__pb2.SearchResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def GetDocument(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,
+            '/hermes.SearchService/GetDocument',
+            hermes__pb2.GetDocumentRequest.SerializeToString,
+            hermes__pb2.GetDocumentResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def GetIndexInfo(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,
+            '/hermes.SearchService/GetIndexInfo',
+            hermes__pb2.GetIndexInfoRequest.SerializeToString,
+            hermes__pb2.GetIndexInfoResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+
+class IndexServiceStub(object):
+    """Index service
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.CreateIndex = channel.unary_unary(
+                '/hermes.IndexService/CreateIndex',
+                request_serializer=hermes__pb2.CreateIndexRequest.SerializeToString,
+                response_deserializer=hermes__pb2.CreateIndexResponse.FromString,
+                _registered_method=True)
+        self.IndexDocuments = channel.stream_unary(
+                '/hermes.IndexService/IndexDocuments',
+                request_serializer=hermes__pb2.IndexDocumentRequest.SerializeToString,
+                response_deserializer=hermes__pb2.IndexDocumentsResponse.FromString,
+                _registered_method=True)
+        self.BatchIndexDocuments = channel.unary_unary(
+                '/hermes.IndexService/BatchIndexDocuments',
+                request_serializer=hermes__pb2.BatchIndexDocumentsRequest.SerializeToString,
+                response_deserializer=hermes__pb2.BatchIndexDocumentsResponse.FromString,
+                _registered_method=True)
+        self.Commit = channel.unary_unary(
+                '/hermes.IndexService/Commit',
+                request_serializer=hermes__pb2.CommitRequest.SerializeToString,
+                response_deserializer=hermes__pb2.CommitResponse.FromString,
+                _registered_method=True)
+        self.ForceMerge = channel.unary_unary(
+                '/hermes.IndexService/ForceMerge',
+                request_serializer=hermes__pb2.ForceMergeRequest.SerializeToString,
+                response_deserializer=hermes__pb2.ForceMergeResponse.FromString,
+                _registered_method=True)
+        self.DeleteIndex = channel.unary_unary(
+                '/hermes.IndexService/DeleteIndex',
+                request_serializer=hermes__pb2.DeleteIndexRequest.SerializeToString,
+                response_deserializer=hermes__pb2.DeleteIndexResponse.FromString,
+                _registered_method=True)
+
+
+class IndexServiceServicer(object):
+    """Index service
+    """
+
+    def CreateIndex(self, request, context):
+        """Create a new index (supports both structured schema and SDL string)
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def IndexDocuments(self, request_iterator, context):
+        """Add documents to index (streaming)
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def BatchIndexDocuments(self, request, context):
+        """Add documents in batch
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Commit(self, request, context):
+        """Commit pending changes
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ForceMerge(self, request, context):
+        """Force merge segments
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def DeleteIndex(self, request, context):
+        """Delete an index
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_IndexServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'CreateIndex': grpc.unary_unary_rpc_method_handler(
+                    servicer.CreateIndex,
+                    request_deserializer=hermes__pb2.CreateIndexRequest.FromString,
+                    response_serializer=hermes__pb2.CreateIndexResponse.SerializeToString,
+            ),
+            'IndexDocuments': grpc.stream_unary_rpc_method_handler(
+                    servicer.IndexDocuments,
+                    request_deserializer=hermes__pb2.IndexDocumentRequest.FromString,
+                    response_serializer=hermes__pb2.IndexDocumentsResponse.SerializeToString,
+            ),
+            'BatchIndexDocuments': grpc.unary_unary_rpc_method_handler(
+                    servicer.BatchIndexDocuments,
+                    request_deserializer=hermes__pb2.BatchIndexDocumentsRequest.FromString,
+                    response_serializer=hermes__pb2.BatchIndexDocumentsResponse.SerializeToString,
+            ),
+            'Commit': grpc.unary_unary_rpc_method_handler(
+                    servicer.Commit,
+                    request_deserializer=hermes__pb2.CommitRequest.FromString,
+                    response_serializer=hermes__pb2.CommitResponse.SerializeToString,
+            ),
+            'ForceMerge': grpc.unary_unary_rpc_method_handler(
+                    servicer.ForceMerge,
+                    request_deserializer=hermes__pb2.ForceMergeRequest.FromString,
+                    response_serializer=hermes__pb2.ForceMergeResponse.SerializeToString,
+            ),
+            'DeleteIndex': grpc.unary_unary_rpc_method_handler(
+                    servicer.DeleteIndex,
+                    request_deserializer=hermes__pb2.DeleteIndexRequest.FromString,
+                    response_serializer=hermes__pb2.DeleteIndexResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'hermes.IndexService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('hermes.IndexService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class IndexService(object):
+    """Index service
+    """
+
+    @staticmethod
+    def CreateIndex(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,
+            '/hermes.IndexService/CreateIndex',
+            hermes__pb2.CreateIndexRequest.SerializeToString,
+            hermes__pb2.CreateIndexResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def IndexDocuments(request_iterator,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.stream_unary(
+            request_iterator,
+            target,
+            '/hermes.IndexService/IndexDocuments',
+            hermes__pb2.IndexDocumentRequest.SerializeToString,
+            hermes__pb2.IndexDocumentsResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def BatchIndexDocuments(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,
+            '/hermes.IndexService/BatchIndexDocuments',
+            hermes__pb2.BatchIndexDocumentsRequest.SerializeToString,
+            hermes__pb2.BatchIndexDocumentsResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def Commit(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,
+            '/hermes.IndexService/Commit',
+            hermes__pb2.CommitRequest.SerializeToString,
+            hermes__pb2.CommitResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def ForceMerge(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,
+            '/hermes.IndexService/ForceMerge',
+            hermes__pb2.ForceMergeRequest.SerializeToString,
+            hermes__pb2.ForceMergeResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def DeleteIndex(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,
+            '/hermes.IndexService/DeleteIndex',
+            hermes__pb2.DeleteIndexRequest.SerializeToString,
+            hermes__pb2.DeleteIndexResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

hermes_client_python/types.py

@@ -0,0 +1,48 @@
+"""Type definitions for Hermes client."""
+
+from dataclasses import dataclass, field
+from typing import Any
+
+
+@dataclass
+class Document:
+    """A document with field values."""
+
+    fields: dict[str, Any] = field(default_factory=dict)
+
+    def __getitem__(self, key: str) -> Any:
+        return self.fields[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.fields[key] = value
+
+    def get(self, key: str, default: Any = None) -> Any:
+        return self.fields.get(key, default)
+
+
+@dataclass
+class SearchHit:
+    """A single search result."""
+
+    doc_id: int
+    score: float
+    fields: dict[str, Any] = field(default_factory=dict)
+
+
+@dataclass
+class SearchResponse:
+    """Search response with hits and metadata."""
+
+    hits: list[SearchHit]
+    total_hits: int
+    took_ms: int
+
+
+@dataclass
+class IndexInfo:
+    """Information about an index."""
+
+    index_name: str
+    num_docs: int
+    num_segments: int
+    schema: str