增加 grpc 生成命令

rpc/ai_pb2.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: rpc/ai.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0crpc/ai.proto\"Y\n\x12\x41\x64\x64\x44ocumentRequest\x12\x0c\n\x04text\x18\x01 \x01(\t\x12\x0f\n\x07user_id\x18\x02 \x01(\x04\x12\x10\n\x08\x64\x61tabase\x18\x03 \x01(\t\x12\x12\n\ncollection\x18\x04 \x01(\t\"\x1e\n\x10\x41\x64\x64\x44ocumentReply\x12\n\n\x02id\x18\x01 \x01(\t2C\n\x08LLMQuery\x12\x37\n\x0b\x41\x64\x64\x44ocument\x12\x13.AddDocumentRequest\x1a\x11.AddDocumentReply\"\x00\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'rpc.ai_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  _globals['_ADDDOCUMENTREQUEST']._serialized_start=16
+  _globals['_ADDDOCUMENTREQUEST']._serialized_end=105
+  _globals['_ADDDOCUMENTREPLY']._serialized_start=107
+  _globals['_ADDDOCUMENTREPLY']._serialized_end=137
+  _globals['_LLMQUERY']._serialized_start=139
+  _globals['_LLMQUERY']._serialized_end=206
+# @@protoc_insertion_point(module_scope)

rpc/ai_pb2_grpc.py

@@ -0,0 +1,66 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from rpc import ai_pb2 as rpc_dot_ai__pb2
+
+
+class LLMQueryStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.AddDocument = channel.unary_unary(
+                '/LLMQuery/AddDocument',
+                request_serializer=rpc_dot_ai__pb2.AddDocumentRequest.SerializeToString,
+                response_deserializer=rpc_dot_ai__pb2.AddDocumentReply.FromString,
+                )
+
+
+class LLMQueryServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def AddDocument(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_LLMQueryServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'AddDocument': grpc.unary_unary_rpc_method_handler(
+                    servicer.AddDocument,
+                    request_deserializer=rpc_dot_ai__pb2.AddDocumentRequest.FromString,
+                    response_serializer=rpc_dot_ai__pb2.AddDocumentReply.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'LLMQuery', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class LLMQuery(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def AddDocument(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, '/LLMQuery/AddDocument',
+            rpc_dot_ai__pb2.AddDocumentRequest.SerializeToString,
+            rpc_dot_ai__pb2.AddDocumentReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)