cortexdb-client 0.1.0__py3-none-any.whl

cortexdb_client/_pb/cortexdb/v1/graphrag_pb2.py

@@ -0,0 +1,84 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: cortexdb/v1/graphrag.proto
+# Protobuf Python Version: 6.33.5
+"""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,
+    33,
+    5,
+    '',
+    'cortexdb/v1/graphrag.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from cortexdb_client._pb.cortexdb.v1 import common_pb2 as cortexdb_dot_v1_dot_common__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1a\x63ortexdb/v1/graphrag.proto\x12\x0b\x63ortexdb.v1\x1a\x18\x63ortexdb/v1/common.proto\"\x81\x02\n\x1aInsertGraphDocumentRequest\x12\n\n\x02id\x18\x01 \x01(\t\x12\r\n\x05title\x18\x02 \x01(\t\x12\x0f\n\x07\x63ontent\x18\x03 \x01(\t\x12G\n\x08metadata\x18\x04 \x03(\x0b\x32\x35.cortexdb.v1.InsertGraphDocumentRequest.MetadataEntry\x12\x12\n\ncollection\x18\x05 \x01(\t\x12\x12\n\nchunk_size\x18\x06 \x01(\x05\x12\x15\n\rchunk_overlap\x18\x07 \x01(\x05\x1a/\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"h\n\x1bInsertGraphDocumentResponse\x12\x18\n\x10\x64ocument_node_id\x18\x01 \x01(\t\x12\x16\n\x0e\x63hunk_node_ids\x18\x02 \x03(\t\x12\x17\n\x0f\x65ntity_node_ids\x18\x03 \x03(\t\"\xe2\x03\n\x15SearchGraphRagRequest\x12\r\n\x05query\x18\x01 \x01(\t\x12\x12\n\ncollection\x18\x02 \x01(\t\x12\r\n\x05top_k\x18\x03 \x01(\x05\x12\x10\n\x08max_hops\x18\x04 \x01(\x05\x12\x1a\n\x12max_related_chunks\x18\x05 \x01(\x05\x12\x1a\n\x12max_context_chunks\x18\x06 \x01(\x05\x12\x19\n\x11max_context_chars\x18\x07 \x01(\x05\x12\x1a\n\x12per_document_limit\x18\x08 \x01(\x05\x12\x0e\n\x06rerank\x18\t \x01(\x08\x12\x16\n\x0e\x64isable_rerank\x18\n \x01(\x08\x12\x18\n\x10\x64iversity_lambda\x18\x0b \x01(\x01\x12\x15\n\rdisable_graph\x18\x0c \x01(\x08\x12\x16\n\x0eretrieval_mode\x18\r \x01(\t\x12\x13\n\x0bgraph_light\x18\x0e \x01(\x08\x12\x1b\n\x13max_expansion_seeds\x18\x0f \x01(\x05\x12\x1b\n\x13max_traversal_nodes\x18\x10 \x01(\x05\x12\x1e\n\x16max_entities_per_chunk\x18\x11 \x01(\x05\x12-\n\x04plan\x18\x12 \x01(\x0b\x32\x1a.cortexdb.v1.RetrievalPlanH\x00\x88\x01\x01\x42\x07\n\x05_plan\"\xd8\x01\n\x16SearchGraphRagResponse\x12\r\n\x05query\x18\x01 \x01(\t\x12(\n\x04plan\x18\x02 \x01(\x0b\x32\x1a.cortexdb.v1.RetrievalPlan\x12\x30\n\x08\x64\x65\x63ision\x18\x03 \x01(\x0b\x32\x1e.cortexdb.v1.RetrievalDecision\x12\x30\n\x06\x63hunks\x18\x04 \x03(\x0b\x32 .cortexdb.v1.GraphRagChunkResult\x12\x10\n\x08\x65ntities\x18\x05 \x03(\t\x12\x0f\n\x07\x63ontext\x18\x06 \x01(\t\"\x9e\x01\n\x11InsertTextRequest\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04text\x18\x02 \x01(\t\x12>\n\x08metadata\x18\x03 \x03(\x0b\x32,.cortexdb.v1.InsertTextRequest.MetadataEntry\x1a/\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x14\n\x12InsertTextResponse\"\xfb\x01\n\x16InsertTextBatchRequest\x12=\n\x05texts\x18\x01 \x03(\x0b\x32..cortexdb.v1.InsertTextBatchRequest.TextsEntry\x12\x43\n\x08metadata\x18\x02 \x03(\x0b\x32\x31.cortexdb.v1.InsertTextBatchRequest.MetadataEntry\x1a,\n\nTextsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a/\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x19\n\x17InsertTextBatchResponse\"\xc6\x01\n\nScoredText\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0f\n\x07\x63ontent\x18\x02 \x01(\t\x12\x0e\n\x06\x64oc_id\x18\x03 \x01(\t\x12\x12\n\ncollection\x18\x04 \x01(\t\x12\x37\n\x08metadata\x18\x05 \x03(\x0b\x32%.cortexdb.v1.ScoredText.MetadataEntry\x12\r\n\x05score\x18\x06 \x01(\x01\x1a/\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"E\n\x11SearchTextRequest\x12\r\n\x05query\x18\x01 \x01(\t\x12\x12\n\ncollection\x18\x02 \x01(\t\x12\r\n\x05top_k\x18\x03 \x01(\x05\">\n\x12SearchTextResponse\x12(\n\x07results\x18\x01 \x03(\x0b\x32\x17.cortexdb.v1.ScoredText\"7\n\x17HybridSearchTextRequest\x12\r\n\x05query\x18\x01 \x01(\t\x12\r\n\x05top_k\x18\x02 \x01(\x05\"D\n\x18HybridSearchTextResponse\x12(\n\x07results\x18\x01 \x03(\x0b\x32\x17.cortexdb.v1.ScoredText2\xb3\x04\n\x0fGraphRagService\x12h\n\x13InsertGraphDocument\x12\'.cortexdb.v1.InsertGraphDocumentRequest\x1a(.cortexdb.v1.InsertGraphDocumentResponse\x12Y\n\x0eSearchGraphRag\x12\".cortexdb.v1.SearchGraphRagRequest\x1a#.cortexdb.v1.SearchGraphRagResponse\x12M\n\nInsertText\x12\x1e.cortexdb.v1.InsertTextRequest\x1a\x1f.cortexdb.v1.InsertTextResponse\x12\\\n\x0fInsertTextBatch\x12#.cortexdb.v1.InsertTextBatchRequest\x1a$.cortexdb.v1.InsertTextBatchResponse\x12M\n\nSearchText\x12\x1e.cortexdb.v1.SearchTextRequest\x1a\x1f.cortexdb.v1.SearchTextResponse\x12_\n\x10HybridSearchText\x12$.cortexdb.v1.HybridSearchTextRequest\x1a%.cortexdb.v1.HybridSearchTextResponseB4Z2github.com/liliang-cn/cortexdb/v2/pkg/rpc/v1;rpcv1b\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'cortexdb.v1.graphrag_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z2github.com/liliang-cn/cortexdb/v2/pkg/rpc/v1;rpcv1'
+  _globals['_INSERTGRAPHDOCUMENTREQUEST_METADATAENTRY']._loaded_options = None
+  _globals['_INSERTGRAPHDOCUMENTREQUEST_METADATAENTRY']._serialized_options = b'8\001'
+  _globals['_INSERTTEXTREQUEST_METADATAENTRY']._loaded_options = None
+  _globals['_INSERTTEXTREQUEST_METADATAENTRY']._serialized_options = b'8\001'
+  _globals['_INSERTTEXTBATCHREQUEST_TEXTSENTRY']._loaded_options = None
+  _globals['_INSERTTEXTBATCHREQUEST_TEXTSENTRY']._serialized_options = b'8\001'
+  _globals['_INSERTTEXTBATCHREQUEST_METADATAENTRY']._loaded_options = None
+  _globals['_INSERTTEXTBATCHREQUEST_METADATAENTRY']._serialized_options = b'8\001'
+  _globals['_SCOREDTEXT_METADATAENTRY']._loaded_options = None
+  _globals['_SCOREDTEXT_METADATAENTRY']._serialized_options = b'8\001'
+  _globals['_INSERTGRAPHDOCUMENTREQUEST']._serialized_start=70
+  _globals['_INSERTGRAPHDOCUMENTREQUEST']._serialized_end=327
+  _globals['_INSERTGRAPHDOCUMENTREQUEST_METADATAENTRY']._serialized_start=280
+  _globals['_INSERTGRAPHDOCUMENTREQUEST_METADATAENTRY']._serialized_end=327
+  _globals['_INSERTGRAPHDOCUMENTRESPONSE']._serialized_start=329
+  _globals['_INSERTGRAPHDOCUMENTRESPONSE']._serialized_end=433
+  _globals['_SEARCHGRAPHRAGREQUEST']._serialized_start=436
+  _globals['_SEARCHGRAPHRAGREQUEST']._serialized_end=918
+  _globals['_SEARCHGRAPHRAGRESPONSE']._serialized_start=921
+  _globals['_SEARCHGRAPHRAGRESPONSE']._serialized_end=1137
+  _globals['_INSERTTEXTREQUEST']._serialized_start=1140
+  _globals['_INSERTTEXTREQUEST']._serialized_end=1298
+  _globals['_INSERTTEXTREQUEST_METADATAENTRY']._serialized_start=280
+  _globals['_INSERTTEXTREQUEST_METADATAENTRY']._serialized_end=327
+  _globals['_INSERTTEXTRESPONSE']._serialized_start=1300
+  _globals['_INSERTTEXTRESPONSE']._serialized_end=1320
+  _globals['_INSERTTEXTBATCHREQUEST']._serialized_start=1323
+  _globals['_INSERTTEXTBATCHREQUEST']._serialized_end=1574
+  _globals['_INSERTTEXTBATCHREQUEST_TEXTSENTRY']._serialized_start=1481
+  _globals['_INSERTTEXTBATCHREQUEST_TEXTSENTRY']._serialized_end=1525
+  _globals['_INSERTTEXTBATCHREQUEST_METADATAENTRY']._serialized_start=280
+  _globals['_INSERTTEXTBATCHREQUEST_METADATAENTRY']._serialized_end=327
+  _globals['_INSERTTEXTBATCHRESPONSE']._serialized_start=1576
+  _globals['_INSERTTEXTBATCHRESPONSE']._serialized_end=1601
+  _globals['_SCOREDTEXT']._serialized_start=1604
+  _globals['_SCOREDTEXT']._serialized_end=1802
+  _globals['_SCOREDTEXT_METADATAENTRY']._serialized_start=280
+  _globals['_SCOREDTEXT_METADATAENTRY']._serialized_end=327
+  _globals['_SEARCHTEXTREQUEST']._serialized_start=1804
+  _globals['_SEARCHTEXTREQUEST']._serialized_end=1873
+  _globals['_SEARCHTEXTRESPONSE']._serialized_start=1875
+  _globals['_SEARCHTEXTRESPONSE']._serialized_end=1937
+  _globals['_HYBRIDSEARCHTEXTREQUEST']._serialized_start=1939
+  _globals['_HYBRIDSEARCHTEXTREQUEST']._serialized_end=1994
+  _globals['_HYBRIDSEARCHTEXTRESPONSE']._serialized_start=1996
+  _globals['_HYBRIDSEARCHTEXTRESPONSE']._serialized_end=2064
+  _globals['_GRAPHRAGSERVICE']._serialized_start=2067
+  _globals['_GRAPHRAGSERVICE']._serialized_end=2630
+# @@protoc_insertion_point(module_scope)

cortexdb_client/_pb/cortexdb/v1/graphrag_pb2_grpc.py

@@ -0,0 +1,312 @@
+# 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 cortexdb_client._pb.cortexdb.v1 import graphrag_pb2 as cortexdb_dot_v1_dot_graphrag__pb2
+
+GRPC_GENERATED_VERSION = '1.81.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 cortexdb/v1/graphrag_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 GraphRagServiceStub:
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.InsertGraphDocument = channel.unary_unary(
+                '/cortexdb.v1.GraphRagService/InsertGraphDocument',
+                request_serializer=cortexdb_dot_v1_dot_graphrag__pb2.InsertGraphDocumentRequest.SerializeToString,
+                response_deserializer=cortexdb_dot_v1_dot_graphrag__pb2.InsertGraphDocumentResponse.FromString,
+                _registered_method=True)
+        self.SearchGraphRag = channel.unary_unary(
+                '/cortexdb.v1.GraphRagService/SearchGraphRag',
+                request_serializer=cortexdb_dot_v1_dot_graphrag__pb2.SearchGraphRagRequest.SerializeToString,
+                response_deserializer=cortexdb_dot_v1_dot_graphrag__pb2.SearchGraphRagResponse.FromString,
+                _registered_method=True)
+        self.InsertText = channel.unary_unary(
+                '/cortexdb.v1.GraphRagService/InsertText',
+                request_serializer=cortexdb_dot_v1_dot_graphrag__pb2.InsertTextRequest.SerializeToString,
+                response_deserializer=cortexdb_dot_v1_dot_graphrag__pb2.InsertTextResponse.FromString,
+                _registered_method=True)
+        self.InsertTextBatch = channel.unary_unary(
+                '/cortexdb.v1.GraphRagService/InsertTextBatch',
+                request_serializer=cortexdb_dot_v1_dot_graphrag__pb2.InsertTextBatchRequest.SerializeToString,
+                response_deserializer=cortexdb_dot_v1_dot_graphrag__pb2.InsertTextBatchResponse.FromString,
+                _registered_method=True)
+        self.SearchText = channel.unary_unary(
+                '/cortexdb.v1.GraphRagService/SearchText',
+                request_serializer=cortexdb_dot_v1_dot_graphrag__pb2.SearchTextRequest.SerializeToString,
+                response_deserializer=cortexdb_dot_v1_dot_graphrag__pb2.SearchTextResponse.FromString,
+                _registered_method=True)
+        self.HybridSearchText = channel.unary_unary(
+                '/cortexdb.v1.GraphRagService/HybridSearchText',
+                request_serializer=cortexdb_dot_v1_dot_graphrag__pb2.HybridSearchTextRequest.SerializeToString,
+                response_deserializer=cortexdb_dot_v1_dot_graphrag__pb2.HybridSearchTextResponse.FromString,
+                _registered_method=True)
+
+
+class GraphRagServiceServicer:
+    """Missing associated documentation comment in .proto file."""
+
+    def InsertGraphDocument(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 SearchGraphRag(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 InsertText(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 InsertTextBatch(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 SearchText(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 HybridSearchText(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_GraphRagServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'InsertGraphDocument': grpc.unary_unary_rpc_method_handler(
+                    servicer.InsertGraphDocument,
+                    request_deserializer=cortexdb_dot_v1_dot_graphrag__pb2.InsertGraphDocumentRequest.FromString,
+                    response_serializer=cortexdb_dot_v1_dot_graphrag__pb2.InsertGraphDocumentResponse.SerializeToString,
+            ),
+            'SearchGraphRag': grpc.unary_unary_rpc_method_handler(
+                    servicer.SearchGraphRag,
+                    request_deserializer=cortexdb_dot_v1_dot_graphrag__pb2.SearchGraphRagRequest.FromString,
+                    response_serializer=cortexdb_dot_v1_dot_graphrag__pb2.SearchGraphRagResponse.SerializeToString,
+            ),
+            'InsertText': grpc.unary_unary_rpc_method_handler(
+                    servicer.InsertText,
+                    request_deserializer=cortexdb_dot_v1_dot_graphrag__pb2.InsertTextRequest.FromString,
+                    response_serializer=cortexdb_dot_v1_dot_graphrag__pb2.InsertTextResponse.SerializeToString,
+            ),
+            'InsertTextBatch': grpc.unary_unary_rpc_method_handler(
+                    servicer.InsertTextBatch,
+                    request_deserializer=cortexdb_dot_v1_dot_graphrag__pb2.InsertTextBatchRequest.FromString,
+                    response_serializer=cortexdb_dot_v1_dot_graphrag__pb2.InsertTextBatchResponse.SerializeToString,
+            ),
+            'SearchText': grpc.unary_unary_rpc_method_handler(
+                    servicer.SearchText,
+                    request_deserializer=cortexdb_dot_v1_dot_graphrag__pb2.SearchTextRequest.FromString,
+                    response_serializer=cortexdb_dot_v1_dot_graphrag__pb2.SearchTextResponse.SerializeToString,
+            ),
+            'HybridSearchText': grpc.unary_unary_rpc_method_handler(
+                    servicer.HybridSearchText,
+                    request_deserializer=cortexdb_dot_v1_dot_graphrag__pb2.HybridSearchTextRequest.FromString,
+                    response_serializer=cortexdb_dot_v1_dot_graphrag__pb2.HybridSearchTextResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'cortexdb.v1.GraphRagService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('cortexdb.v1.GraphRagService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class GraphRagService:
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def InsertGraphDocument(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,
+            '/cortexdb.v1.GraphRagService/InsertGraphDocument',
+            cortexdb_dot_v1_dot_graphrag__pb2.InsertGraphDocumentRequest.SerializeToString,
+            cortexdb_dot_v1_dot_graphrag__pb2.InsertGraphDocumentResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def SearchGraphRag(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,
+            '/cortexdb.v1.GraphRagService/SearchGraphRag',
+            cortexdb_dot_v1_dot_graphrag__pb2.SearchGraphRagRequest.SerializeToString,
+            cortexdb_dot_v1_dot_graphrag__pb2.SearchGraphRagResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def InsertText(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,
+            '/cortexdb.v1.GraphRagService/InsertText',
+            cortexdb_dot_v1_dot_graphrag__pb2.InsertTextRequest.SerializeToString,
+            cortexdb_dot_v1_dot_graphrag__pb2.InsertTextResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def InsertTextBatch(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,
+            '/cortexdb.v1.GraphRagService/InsertTextBatch',
+            cortexdb_dot_v1_dot_graphrag__pb2.InsertTextBatchRequest.SerializeToString,
+            cortexdb_dot_v1_dot_graphrag__pb2.InsertTextBatchResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def SearchText(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,
+            '/cortexdb.v1.GraphRagService/SearchText',
+            cortexdb_dot_v1_dot_graphrag__pb2.SearchTextRequest.SerializeToString,
+            cortexdb_dot_v1_dot_graphrag__pb2.SearchTextResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def HybridSearchText(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,
+            '/cortexdb.v1.GraphRagService/HybridSearchText',
+            cortexdb_dot_v1_dot_graphrag__pb2.HybridSearchTextRequest.SerializeToString,
+            cortexdb_dot_v1_dot_graphrag__pb2.HybridSearchTextResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

cortexdb_client/_pb/cortexdb/v1/knowledge_pb2.py

@@ -0,0 +1,77 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: cortexdb/v1/knowledge.proto
+# Protobuf Python Version: 6.33.5
+"""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,
+    33,
+    5,
+    '',
+    'cortexdb/v1/knowledge.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+from cortexdb_client._pb.cortexdb.v1 import common_pb2 as cortexdb_dot_v1_dot_common__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1b\x63ortexdb/v1/knowledge.proto\x12\x0b\x63ortexdb.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x18\x63ortexdb/v1/common.proto\"\xe9\x02\n\x0fKnowledgeRecord\x12\n\n\x02id\x18\x01 \x01(\t\x12\r\n\x05title\x18\x02 \x01(\t\x12\x0f\n\x07\x63ontent\x18\x03 \x01(\t\x12\x12\n\nsource_url\x18\x04 \x01(\t\x12\x0e\n\x06\x61uthor\x18\x05 \x01(\t\x12\x12\n\ncollection\x18\x06 \x01(\t\x12<\n\x08metadata\x18\x07 \x03(\x0b\x32*.cortexdb.v1.KnowledgeRecord.MetadataEntry\x12\x11\n\tchunk_ids\x18\x08 \x03(\t\x12\x10\n\x08\x65ntities\x18\t \x03(\t\x12.\n\ncreated_at\x18\n \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\nupdated_at\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x1a/\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x86\x03\n\x14SaveKnowledgeRequest\x12\x14\n\x0cknowledge_id\x18\x01 \x01(\t\x12\r\n\x05title\x18\x02 \x01(\t\x12\x0f\n\x07\x63ontent\x18\x03 \x01(\t\x12\x12\n\nsource_url\x18\x04 \x01(\t\x12\x0e\n\x06\x61uthor\x18\x05 \x01(\t\x12\x12\n\ncollection\x18\x06 \x01(\t\x12\x12\n\nchunk_size\x18\x07 \x01(\x05\x12\x15\n\rchunk_overlap\x18\x08 \x01(\x05\x12\x41\n\x08metadata\x18\t \x03(\x0b\x32/.cortexdb.v1.SaveKnowledgeRequest.MetadataEntry\x12.\n\x08\x65ntities\x18\n \x03(\x0b\x32\x1c.cortexdb.v1.ToolEntityInput\x12\x31\n\trelations\x18\x0b \x03(\x0b\x32\x1e.cortexdb.v1.ToolRelationInput\x1a/\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x96\x01\n\x15SaveKnowledgeResponse\x12/\n\tknowledge\x18\x01 \x01(\x0b\x32\x1c.cortexdb.v1.KnowledgeRecord\x12\x18\n\x10\x64ocument_node_id\x18\x02 \x01(\t\x12\x17\n\x0f\x65ntity_node_ids\x18\x03 \x03(\t\x12\x19\n\x11relation_edge_ids\x18\x04 \x03(\t\"\x8d\x04\n\x16UpdateKnowledgeRequest\x12\x14\n\x0cknowledge_id\x18\x01 \x01(\t\x12\x12\n\x05title\x18\x02 \x01(\tH\x00\x88\x01\x01\x12\x14\n\x07\x63ontent\x18\x03 \x01(\tH\x01\x88\x01\x01\x12\x17\n\nsource_url\x18\x04 \x01(\tH\x02\x88\x01\x01\x12\x13\n\x06\x61uthor\x18\x05 \x01(\tH\x03\x88\x01\x01\x12\x17\n\ncollection\x18\x06 \x01(\tH\x04\x88\x01\x01\x12\x17\n\nchunk_size\x18\x07 \x01(\x05H\x05\x88\x01\x01\x12\x1a\n\rchunk_overlap\x18\x08 \x01(\x05H\x06\x88\x01\x01\x12\x43\n\x08metadata\x18\t \x03(\x0b\x32\x31.cortexdb.v1.UpdateKnowledgeRequest.MetadataEntry\x12.\n\x08\x65ntities\x18\n \x03(\x0b\x32\x1c.cortexdb.v1.ToolEntityInput\x12\x31\n\trelations\x18\x0b \x03(\x0b\x32\x1e.cortexdb.v1.ToolRelationInput\x1a/\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x08\n\x06_titleB\n\n\x08_contentB\r\n\x0b_source_urlB\t\n\x07_authorB\r\n\x0b_collectionB\r\n\x0b_chunk_sizeB\x10\n\x0e_chunk_overlap\"+\n\x13GetKnowledgeRequest\x12\x14\n\x0cknowledge_id\x18\x01 \x01(\t\"G\n\x14GetKnowledgeResponse\x12/\n\tknowledge\x18\x01 \x01(\x0b\x32\x1c.cortexdb.v1.KnowledgeRecord\".\n\x16\x44\x65leteKnowledgeRequest\x12\x14\n\x0cknowledge_id\x18\x01 \x01(\t\"@\n\x17\x44\x65leteKnowledgeResponse\x12\x14\n\x0cknowledge_id\x18\x01 \x01(\t\x12\x0f\n\x07\x64\x65leted\x18\x02 \x01(\x08\"\x96\x04\n\x16SearchKnowledgeRequest\x12\r\n\x05query\x18\x01 \x01(\t\x12\x12\n\ncollection\x18\x02 \x01(\t\x12\r\n\x05top_k\x18\x03 \x01(\x05\x12\x10\n\x08max_hops\x18\x04 \x01(\x05\x12\x1a\n\x12max_related_chunks\x18\x05 \x01(\x05\x12\x1a\n\x12max_context_chunks\x18\x06 \x01(\x05\x12\x19\n\x11max_context_chars\x18\x07 \x01(\x05\x12\x1a\n\x12per_document_limit\x18\x08 \x01(\x05\x12\x18\n\x10\x64iversity_lambda\x18\t \x01(\x01\x12\x16\n\x0e\x64isable_rerank\x18\n \x01(\x08\x12\x14\n\x0c\x65ntity_names\x18\x0b \x03(\t\x12\x10\n\x08keywords\x18\x0c \x03(\t\x12\x19\n\x11\x61lternate_queries\x18\r \x03(\t\x12\x16\n\x0eretrieval_mode\x18\x0e \x01(\t\x12\x15\n\rdisable_graph\x18\x0f \x01(\x08\x12\x13\n\x0bgraph_light\x18\x10 \x01(\x08\x12\x1b\n\x13max_expansion_seeds\x18\x11 \x01(\x05\x12\x1b\n\x13max_traversal_nodes\x18\x12 \x01(\x05\x12\x1e\n\x16max_entities_per_chunk\x18\x13 \x01(\x05\x12-\n\x04plan\x18\x14 \x01(\x0b\x32\x1a.cortexdb.v1.RetrievalPlanH\x00\x88\x01\x01\x42\x07\n\x05_plan\"\x94\x02\n\x12KnowledgeSearchHit\x12\x14\n\x0cknowledge_id\x18\x01 \x01(\t\x12\r\n\x05title\x18\x02 \x01(\t\x12\x12\n\nsource_url\x18\x03 \x01(\t\x12\x0e\n\x06\x61uthor\x18\x04 \x01(\t\x12\x0f\n\x07snippet\x18\x05 \x01(\t\x12\r\n\x05score\x18\x06 \x01(\x01\x12\x11\n\tchunk_ids\x18\x07 \x03(\t\x12\x10\n\x08\x65ntities\x18\x08 \x03(\t\x12?\n\x08metadata\x18\t \x03(\x0b\x32-.cortexdb.v1.KnowledgeSearchHit.MetadataEntry\x1a/\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x8b\x02\n\x17SearchKnowledgeResponse\x12\r\n\x05query\x18\x01 \x01(\t\x12(\n\x04plan\x18\x02 \x01(\x0b\x32\x1a.cortexdb.v1.RetrievalPlan\x12\x30\n\x08\x64\x65\x63ision\x18\x03 \x01(\x0b\x32\x1e.cortexdb.v1.RetrievalDecision\x12\x30\n\x07results\x18\x04 \x03(\x0b\x32\x1f.cortexdb.v1.KnowledgeSearchHit\x12\x30\n\x06\x63hunks\x18\x05 \x03(\x0b\x32 .cortexdb.v1.GraphRagChunkResult\x12\x10\n\x08\x65ntities\x18\x06 \x03(\t\x12\x0f\n\x07\x63ontext\x18\x07 \x01(\t2\xd7\x03\n\x10KnowledgeService\x12V\n\rSaveKnowledge\x12!.cortexdb.v1.SaveKnowledgeRequest\x1a\".cortexdb.v1.SaveKnowledgeResponse\x12Z\n\x0fUpdateKnowledge\x12#.cortexdb.v1.UpdateKnowledgeRequest\x1a\".cortexdb.v1.SaveKnowledgeResponse\x12S\n\x0cGetKnowledge\x12 .cortexdb.v1.GetKnowledgeRequest\x1a!.cortexdb.v1.GetKnowledgeResponse\x12\\\n\x0fSearchKnowledge\x12#.cortexdb.v1.SearchKnowledgeRequest\x1a$.cortexdb.v1.SearchKnowledgeResponse\x12\\\n\x0f\x44\x65leteKnowledge\x12#.cortexdb.v1.DeleteKnowledgeRequest\x1a$.cortexdb.v1.DeleteKnowledgeResponseB4Z2github.com/liliang-cn/cortexdb/v2/pkg/rpc/v1;rpcv1b\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'cortexdb.v1.knowledge_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z2github.com/liliang-cn/cortexdb/v2/pkg/rpc/v1;rpcv1'
+  _globals['_KNOWLEDGERECORD_METADATAENTRY']._loaded_options = None
+  _globals['_KNOWLEDGERECORD_METADATAENTRY']._serialized_options = b'8\001'
+  _globals['_SAVEKNOWLEDGEREQUEST_METADATAENTRY']._loaded_options = None
+  _globals['_SAVEKNOWLEDGEREQUEST_METADATAENTRY']._serialized_options = b'8\001'
+  _globals['_UPDATEKNOWLEDGEREQUEST_METADATAENTRY']._loaded_options = None
+  _globals['_UPDATEKNOWLEDGEREQUEST_METADATAENTRY']._serialized_options = b'8\001'
+  _globals['_KNOWLEDGESEARCHHIT_METADATAENTRY']._loaded_options = None
+  _globals['_KNOWLEDGESEARCHHIT_METADATAENTRY']._serialized_options = b'8\001'
+  _globals['_KNOWLEDGERECORD']._serialized_start=104
+  _globals['_KNOWLEDGERECORD']._serialized_end=465
+  _globals['_KNOWLEDGERECORD_METADATAENTRY']._serialized_start=418
+  _globals['_KNOWLEDGERECORD_METADATAENTRY']._serialized_end=465
+  _globals['_SAVEKNOWLEDGEREQUEST']._serialized_start=468
+  _globals['_SAVEKNOWLEDGEREQUEST']._serialized_end=858
+  _globals['_SAVEKNOWLEDGEREQUEST_METADATAENTRY']._serialized_start=418
+  _globals['_SAVEKNOWLEDGEREQUEST_METADATAENTRY']._serialized_end=465
+  _globals['_SAVEKNOWLEDGERESPONSE']._serialized_start=861
+  _globals['_SAVEKNOWLEDGERESPONSE']._serialized_end=1011
+  _globals['_UPDATEKNOWLEDGEREQUEST']._serialized_start=1014
+  _globals['_UPDATEKNOWLEDGEREQUEST']._serialized_end=1539
+  _globals['_UPDATEKNOWLEDGEREQUEST_METADATAENTRY']._serialized_start=418
+  _globals['_UPDATEKNOWLEDGEREQUEST_METADATAENTRY']._serialized_end=465
+  _globals['_GETKNOWLEDGEREQUEST']._serialized_start=1541
+  _globals['_GETKNOWLEDGEREQUEST']._serialized_end=1584
+  _globals['_GETKNOWLEDGERESPONSE']._serialized_start=1586
+  _globals['_GETKNOWLEDGERESPONSE']._serialized_end=1657
+  _globals['_DELETEKNOWLEDGEREQUEST']._serialized_start=1659
+  _globals['_DELETEKNOWLEDGEREQUEST']._serialized_end=1705
+  _globals['_DELETEKNOWLEDGERESPONSE']._serialized_start=1707
+  _globals['_DELETEKNOWLEDGERESPONSE']._serialized_end=1771
+  _globals['_SEARCHKNOWLEDGEREQUEST']._serialized_start=1774
+  _globals['_SEARCHKNOWLEDGEREQUEST']._serialized_end=2308
+  _globals['_KNOWLEDGESEARCHHIT']._serialized_start=2311
+  _globals['_KNOWLEDGESEARCHHIT']._serialized_end=2587
+  _globals['_KNOWLEDGESEARCHHIT_METADATAENTRY']._serialized_start=418
+  _globals['_KNOWLEDGESEARCHHIT_METADATAENTRY']._serialized_end=465
+  _globals['_SEARCHKNOWLEDGERESPONSE']._serialized_start=2590
+  _globals['_SEARCHKNOWLEDGERESPONSE']._serialized_end=2857
+  _globals['_KNOWLEDGESERVICE']._serialized_start=2860
+  _globals['_KNOWLEDGESERVICE']._serialized_end=3331
+# @@protoc_insertion_point(module_scope)