cortexdb-client 0.1.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,269 @@
1
+ # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
+ """Client and server classes corresponding to protobuf-defined services."""
3
+ import grpc
4
+ import warnings
5
+
6
+ from cortexdb_client._pb.cortexdb.v1 import knowledge_pb2 as cortexdb_dot_v1_dot_knowledge__pb2
7
+
8
+ GRPC_GENERATED_VERSION = '1.81.0'
9
+ GRPC_VERSION = grpc.__version__
10
+ _version_not_supported = False
11
+
12
+ try:
13
+ from grpc._utilities import first_version_is_lower
14
+ _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
15
+ except ImportError:
16
+ _version_not_supported = True
17
+
18
+ if _version_not_supported:
19
+ raise RuntimeError(
20
+ f'The grpc package installed is at version {GRPC_VERSION},'
21
+ + ' but the generated code in cortexdb/v1/knowledge_pb2_grpc.py depends on'
22
+ + f' grpcio>={GRPC_GENERATED_VERSION}.'
23
+ + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
24
+ + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
25
+ )
26
+
27
+
28
+ class KnowledgeServiceStub:
29
+ """Missing associated documentation comment in .proto file."""
30
+
31
+ def __init__(self, channel):
32
+ """Constructor.
33
+
34
+ Args:
35
+ channel: A grpc.Channel.
36
+ """
37
+ self.SaveKnowledge = channel.unary_unary(
38
+ '/cortexdb.v1.KnowledgeService/SaveKnowledge',
39
+ request_serializer=cortexdb_dot_v1_dot_knowledge__pb2.SaveKnowledgeRequest.SerializeToString,
40
+ response_deserializer=cortexdb_dot_v1_dot_knowledge__pb2.SaveKnowledgeResponse.FromString,
41
+ _registered_method=True)
42
+ self.UpdateKnowledge = channel.unary_unary(
43
+ '/cortexdb.v1.KnowledgeService/UpdateKnowledge',
44
+ request_serializer=cortexdb_dot_v1_dot_knowledge__pb2.UpdateKnowledgeRequest.SerializeToString,
45
+ response_deserializer=cortexdb_dot_v1_dot_knowledge__pb2.SaveKnowledgeResponse.FromString,
46
+ _registered_method=True)
47
+ self.GetKnowledge = channel.unary_unary(
48
+ '/cortexdb.v1.KnowledgeService/GetKnowledge',
49
+ request_serializer=cortexdb_dot_v1_dot_knowledge__pb2.GetKnowledgeRequest.SerializeToString,
50
+ response_deserializer=cortexdb_dot_v1_dot_knowledge__pb2.GetKnowledgeResponse.FromString,
51
+ _registered_method=True)
52
+ self.SearchKnowledge = channel.unary_unary(
53
+ '/cortexdb.v1.KnowledgeService/SearchKnowledge',
54
+ request_serializer=cortexdb_dot_v1_dot_knowledge__pb2.SearchKnowledgeRequest.SerializeToString,
55
+ response_deserializer=cortexdb_dot_v1_dot_knowledge__pb2.SearchKnowledgeResponse.FromString,
56
+ _registered_method=True)
57
+ self.DeleteKnowledge = channel.unary_unary(
58
+ '/cortexdb.v1.KnowledgeService/DeleteKnowledge',
59
+ request_serializer=cortexdb_dot_v1_dot_knowledge__pb2.DeleteKnowledgeRequest.SerializeToString,
60
+ response_deserializer=cortexdb_dot_v1_dot_knowledge__pb2.DeleteKnowledgeResponse.FromString,
61
+ _registered_method=True)
62
+
63
+
64
+ class KnowledgeServiceServicer:
65
+ """Missing associated documentation comment in .proto file."""
66
+
67
+ def SaveKnowledge(self, request, context):
68
+ """Missing associated documentation comment in .proto file."""
69
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
70
+ context.set_details('Method not implemented!')
71
+ raise NotImplementedError('Method not implemented!')
72
+
73
+ def UpdateKnowledge(self, request, context):
74
+ """Missing associated documentation comment in .proto file."""
75
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
76
+ context.set_details('Method not implemented!')
77
+ raise NotImplementedError('Method not implemented!')
78
+
79
+ def GetKnowledge(self, request, context):
80
+ """Missing associated documentation comment in .proto file."""
81
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
82
+ context.set_details('Method not implemented!')
83
+ raise NotImplementedError('Method not implemented!')
84
+
85
+ def SearchKnowledge(self, request, context):
86
+ """Missing associated documentation comment in .proto file."""
87
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
88
+ context.set_details('Method not implemented!')
89
+ raise NotImplementedError('Method not implemented!')
90
+
91
+ def DeleteKnowledge(self, request, context):
92
+ """Missing associated documentation comment in .proto file."""
93
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
94
+ context.set_details('Method not implemented!')
95
+ raise NotImplementedError('Method not implemented!')
96
+
97
+
98
+ def add_KnowledgeServiceServicer_to_server(servicer, server):
99
+ rpc_method_handlers = {
100
+ 'SaveKnowledge': grpc.unary_unary_rpc_method_handler(
101
+ servicer.SaveKnowledge,
102
+ request_deserializer=cortexdb_dot_v1_dot_knowledge__pb2.SaveKnowledgeRequest.FromString,
103
+ response_serializer=cortexdb_dot_v1_dot_knowledge__pb2.SaveKnowledgeResponse.SerializeToString,
104
+ ),
105
+ 'UpdateKnowledge': grpc.unary_unary_rpc_method_handler(
106
+ servicer.UpdateKnowledge,
107
+ request_deserializer=cortexdb_dot_v1_dot_knowledge__pb2.UpdateKnowledgeRequest.FromString,
108
+ response_serializer=cortexdb_dot_v1_dot_knowledge__pb2.SaveKnowledgeResponse.SerializeToString,
109
+ ),
110
+ 'GetKnowledge': grpc.unary_unary_rpc_method_handler(
111
+ servicer.GetKnowledge,
112
+ request_deserializer=cortexdb_dot_v1_dot_knowledge__pb2.GetKnowledgeRequest.FromString,
113
+ response_serializer=cortexdb_dot_v1_dot_knowledge__pb2.GetKnowledgeResponse.SerializeToString,
114
+ ),
115
+ 'SearchKnowledge': grpc.unary_unary_rpc_method_handler(
116
+ servicer.SearchKnowledge,
117
+ request_deserializer=cortexdb_dot_v1_dot_knowledge__pb2.SearchKnowledgeRequest.FromString,
118
+ response_serializer=cortexdb_dot_v1_dot_knowledge__pb2.SearchKnowledgeResponse.SerializeToString,
119
+ ),
120
+ 'DeleteKnowledge': grpc.unary_unary_rpc_method_handler(
121
+ servicer.DeleteKnowledge,
122
+ request_deserializer=cortexdb_dot_v1_dot_knowledge__pb2.DeleteKnowledgeRequest.FromString,
123
+ response_serializer=cortexdb_dot_v1_dot_knowledge__pb2.DeleteKnowledgeResponse.SerializeToString,
124
+ ),
125
+ }
126
+ generic_handler = grpc.method_handlers_generic_handler(
127
+ 'cortexdb.v1.KnowledgeService', rpc_method_handlers)
128
+ server.add_generic_rpc_handlers((generic_handler,))
129
+ server.add_registered_method_handlers('cortexdb.v1.KnowledgeService', rpc_method_handlers)
130
+
131
+
132
+ # This class is part of an EXPERIMENTAL API.
133
+ class KnowledgeService:
134
+ """Missing associated documentation comment in .proto file."""
135
+
136
+ @staticmethod
137
+ def SaveKnowledge(request,
138
+ target,
139
+ options=(),
140
+ channel_credentials=None,
141
+ call_credentials=None,
142
+ insecure=False,
143
+ compression=None,
144
+ wait_for_ready=None,
145
+ timeout=None,
146
+ metadata=None):
147
+ return grpc.experimental.unary_unary(
148
+ request,
149
+ target,
150
+ '/cortexdb.v1.KnowledgeService/SaveKnowledge',
151
+ cortexdb_dot_v1_dot_knowledge__pb2.SaveKnowledgeRequest.SerializeToString,
152
+ cortexdb_dot_v1_dot_knowledge__pb2.SaveKnowledgeResponse.FromString,
153
+ options,
154
+ channel_credentials,
155
+ insecure,
156
+ call_credentials,
157
+ compression,
158
+ wait_for_ready,
159
+ timeout,
160
+ metadata,
161
+ _registered_method=True)
162
+
163
+ @staticmethod
164
+ def UpdateKnowledge(request,
165
+ target,
166
+ options=(),
167
+ channel_credentials=None,
168
+ call_credentials=None,
169
+ insecure=False,
170
+ compression=None,
171
+ wait_for_ready=None,
172
+ timeout=None,
173
+ metadata=None):
174
+ return grpc.experimental.unary_unary(
175
+ request,
176
+ target,
177
+ '/cortexdb.v1.KnowledgeService/UpdateKnowledge',
178
+ cortexdb_dot_v1_dot_knowledge__pb2.UpdateKnowledgeRequest.SerializeToString,
179
+ cortexdb_dot_v1_dot_knowledge__pb2.SaveKnowledgeResponse.FromString,
180
+ options,
181
+ channel_credentials,
182
+ insecure,
183
+ call_credentials,
184
+ compression,
185
+ wait_for_ready,
186
+ timeout,
187
+ metadata,
188
+ _registered_method=True)
189
+
190
+ @staticmethod
191
+ def GetKnowledge(request,
192
+ target,
193
+ options=(),
194
+ channel_credentials=None,
195
+ call_credentials=None,
196
+ insecure=False,
197
+ compression=None,
198
+ wait_for_ready=None,
199
+ timeout=None,
200
+ metadata=None):
201
+ return grpc.experimental.unary_unary(
202
+ request,
203
+ target,
204
+ '/cortexdb.v1.KnowledgeService/GetKnowledge',
205
+ cortexdb_dot_v1_dot_knowledge__pb2.GetKnowledgeRequest.SerializeToString,
206
+ cortexdb_dot_v1_dot_knowledge__pb2.GetKnowledgeResponse.FromString,
207
+ options,
208
+ channel_credentials,
209
+ insecure,
210
+ call_credentials,
211
+ compression,
212
+ wait_for_ready,
213
+ timeout,
214
+ metadata,
215
+ _registered_method=True)
216
+
217
+ @staticmethod
218
+ def SearchKnowledge(request,
219
+ target,
220
+ options=(),
221
+ channel_credentials=None,
222
+ call_credentials=None,
223
+ insecure=False,
224
+ compression=None,
225
+ wait_for_ready=None,
226
+ timeout=None,
227
+ metadata=None):
228
+ return grpc.experimental.unary_unary(
229
+ request,
230
+ target,
231
+ '/cortexdb.v1.KnowledgeService/SearchKnowledge',
232
+ cortexdb_dot_v1_dot_knowledge__pb2.SearchKnowledgeRequest.SerializeToString,
233
+ cortexdb_dot_v1_dot_knowledge__pb2.SearchKnowledgeResponse.FromString,
234
+ options,
235
+ channel_credentials,
236
+ insecure,
237
+ call_credentials,
238
+ compression,
239
+ wait_for_ready,
240
+ timeout,
241
+ metadata,
242
+ _registered_method=True)
243
+
244
+ @staticmethod
245
+ def DeleteKnowledge(request,
246
+ target,
247
+ options=(),
248
+ channel_credentials=None,
249
+ call_credentials=None,
250
+ insecure=False,
251
+ compression=None,
252
+ wait_for_ready=None,
253
+ timeout=None,
254
+ metadata=None):
255
+ return grpc.experimental.unary_unary(
256
+ request,
257
+ target,
258
+ '/cortexdb.v1.KnowledgeService/DeleteKnowledge',
259
+ cortexdb_dot_v1_dot_knowledge__pb2.DeleteKnowledgeRequest.SerializeToString,
260
+ cortexdb_dot_v1_dot_knowledge__pb2.DeleteKnowledgeResponse.FromString,
261
+ options,
262
+ channel_credentials,
263
+ insecure,
264
+ call_credentials,
265
+ compression,
266
+ wait_for_ready,
267
+ timeout,
268
+ metadata,
269
+ _registered_method=True)
@@ -0,0 +1,62 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: cortexdb/v1/memory.proto
5
+ # Protobuf Python Version: 6.33.5
6
+ """Generated protocol buffer code."""
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import runtime_version as _runtime_version
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ from google.protobuf.internal import builder as _builder
12
+ _runtime_version.ValidateProtobufRuntimeVersion(
13
+ _runtime_version.Domain.PUBLIC,
14
+ 6,
15
+ 33,
16
+ 5,
17
+ '',
18
+ 'cortexdb/v1/memory.proto'
19
+ )
20
+ # @@protoc_insertion_point(imports)
21
+
22
+ _sym_db = _symbol_database.Default()
23
+
24
+
25
+ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
26
+ from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
27
+ from cortexdb_client._pb.cortexdb.v1 import common_pb2 as cortexdb_dot_v1_dot_common__pb2
28
+
29
+
30
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x18\x63ortexdb/v1/memory.proto\x12\x0b\x63ortexdb.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x18\x63ortexdb/v1/common.proto\"\xb4\x02\n\x0cMemoryRecord\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0f\n\x07user_id\x18\x02 \x01(\t\x12\x12\n\nsession_id\x18\x03 \x01(\t\x12\r\n\x05scope\x18\x04 \x01(\t\x12\x11\n\tnamespace\x18\x05 \x01(\t\x12\x0c\n\x04role\x18\x06 \x01(\t\x12\x0f\n\x07\x63ontent\x18\x07 \x01(\t\x12)\n\x08metadata\x18\x08 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x12\n\nimportance\x18\t \x01(\x01\x12\x13\n\x0bttl_seconds\x18\n \x01(\x05\x12.\n\nexpires_at\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\ncreated_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\xe0\x01\n\x11SaveMemoryRequest\x12\x11\n\tmemory_id\x18\x01 \x01(\t\x12\x0f\n\x07user_id\x18\x02 \x01(\t\x12\x12\n\nsession_id\x18\x03 \x01(\t\x12\r\n\x05scope\x18\x04 \x01(\t\x12\x11\n\tnamespace\x18\x05 \x01(\t\x12\x0c\n\x04role\x18\x06 \x01(\t\x12\x0f\n\x07\x63ontent\x18\x07 \x01(\t\x12)\n\x08metadata\x18\x08 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x12\n\nimportance\x18\t \x01(\x01\x12\x13\n\x0bttl_seconds\x18\n \x01(\x05\"?\n\x12SaveMemoryResponse\x12)\n\x06memory\x18\x01 \x01(\x0b\x32\x19.cortexdb.v1.MemoryRecord\"\xc7\x01\n\x13UpdateMemoryRequest\x12\x11\n\tmemory_id\x18\x01 \x01(\t\x12\x14\n\x07\x63ontent\x18\x02 \x01(\tH\x00\x88\x01\x01\x12)\n\x08metadata\x18\x03 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x17\n\nimportance\x18\x04 \x01(\x01H\x01\x88\x01\x01\x12\x18\n\x0bttl_seconds\x18\x05 \x01(\x05H\x02\x88\x01\x01\x42\n\n\x08_contentB\r\n\x0b_importanceB\x0e\n\x0c_ttl_seconds\"%\n\x10GetMemoryRequest\x12\x11\n\tmemory_id\x18\x01 \x01(\t\">\n\x11GetMemoryResponse\x12)\n\x06memory\x18\x01 \x01(\x0b\x32\x19.cortexdb.v1.MemoryRecord\"(\n\x13\x44\x65leteMemoryRequest\x12\x11\n\tmemory_id\x18\x01 \x01(\t\":\n\x14\x44\x65leteMemoryResponse\x12\x11\n\tmemory_id\x18\x01 \x01(\t\x12\x0f\n\x07\x64\x65leted\x18\x02 \x01(\x08\"\xf7\x01\n\x13SearchMemoryRequest\x12\r\n\x05query\x18\x01 \x01(\t\x12\x0f\n\x07user_id\x18\x02 \x01(\t\x12\x12\n\nsession_id\x18\x03 \x01(\t\x12\r\n\x05scope\x18\x04 \x01(\t\x12\x11\n\tnamespace\x18\x05 \x01(\t\x12\r\n\x05top_k\x18\x06 \x01(\x05\x12\x10\n\x08keywords\x18\x07 \x03(\t\x12\x19\n\x11\x61lternate_queries\x18\x08 \x03(\t\x12\x16\n\x0eretrieval_mode\x18\t \x01(\t\x12-\n\x04plan\x18\n \x01(\x0b\x32\x1a.cortexdb.v1.RetrievalPlanH\x00\x88\x01\x01\x42\x07\n\x05_plan\"K\n\x0fMemorySearchHit\x12)\n\x06memory\x18\x01 \x01(\x0b\x32\x19.cortexdb.v1.MemoryRecord\x12\r\n\x05score\x18\x02 \x01(\x01\"\xb0\x01\n\x14SearchMemoryResponse\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-\n\x07results\x18\x04 \x03(\x0b\x32\x1c.cortexdb.v1.MemorySearchHit2\xa7\x03\n\rMemoryService\x12M\n\nSaveMemory\x12\x1e.cortexdb.v1.SaveMemoryRequest\x1a\x1f.cortexdb.v1.SaveMemoryResponse\x12Q\n\x0cUpdateMemory\x12 .cortexdb.v1.UpdateMemoryRequest\x1a\x1f.cortexdb.v1.SaveMemoryResponse\x12J\n\tGetMemory\x12\x1d.cortexdb.v1.GetMemoryRequest\x1a\x1e.cortexdb.v1.GetMemoryResponse\x12S\n\x0cSearchMemory\x12 .cortexdb.v1.SearchMemoryRequest\x1a!.cortexdb.v1.SearchMemoryResponse\x12S\n\x0c\x44\x65leteMemory\x12 .cortexdb.v1.DeleteMemoryRequest\x1a!.cortexdb.v1.DeleteMemoryResponseB4Z2github.com/liliang-cn/cortexdb/v2/pkg/rpc/v1;rpcv1b\x06proto3')
31
+
32
+ _globals = globals()
33
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
34
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'cortexdb.v1.memory_pb2', _globals)
35
+ if not _descriptor._USE_C_DESCRIPTORS:
36
+ _globals['DESCRIPTOR']._loaded_options = None
37
+ _globals['DESCRIPTOR']._serialized_options = b'Z2github.com/liliang-cn/cortexdb/v2/pkg/rpc/v1;rpcv1'
38
+ _globals['_MEMORYRECORD']._serialized_start=131
39
+ _globals['_MEMORYRECORD']._serialized_end=439
40
+ _globals['_SAVEMEMORYREQUEST']._serialized_start=442
41
+ _globals['_SAVEMEMORYREQUEST']._serialized_end=666
42
+ _globals['_SAVEMEMORYRESPONSE']._serialized_start=668
43
+ _globals['_SAVEMEMORYRESPONSE']._serialized_end=731
44
+ _globals['_UPDATEMEMORYREQUEST']._serialized_start=734
45
+ _globals['_UPDATEMEMORYREQUEST']._serialized_end=933
46
+ _globals['_GETMEMORYREQUEST']._serialized_start=935
47
+ _globals['_GETMEMORYREQUEST']._serialized_end=972
48
+ _globals['_GETMEMORYRESPONSE']._serialized_start=974
49
+ _globals['_GETMEMORYRESPONSE']._serialized_end=1036
50
+ _globals['_DELETEMEMORYREQUEST']._serialized_start=1038
51
+ _globals['_DELETEMEMORYREQUEST']._serialized_end=1078
52
+ _globals['_DELETEMEMORYRESPONSE']._serialized_start=1080
53
+ _globals['_DELETEMEMORYRESPONSE']._serialized_end=1138
54
+ _globals['_SEARCHMEMORYREQUEST']._serialized_start=1141
55
+ _globals['_SEARCHMEMORYREQUEST']._serialized_end=1388
56
+ _globals['_MEMORYSEARCHHIT']._serialized_start=1390
57
+ _globals['_MEMORYSEARCHHIT']._serialized_end=1465
58
+ _globals['_SEARCHMEMORYRESPONSE']._serialized_start=1468
59
+ _globals['_SEARCHMEMORYRESPONSE']._serialized_end=1644
60
+ _globals['_MEMORYSERVICE']._serialized_start=1647
61
+ _globals['_MEMORYSERVICE']._serialized_end=2070
62
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,269 @@
1
+ # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
+ """Client and server classes corresponding to protobuf-defined services."""
3
+ import grpc
4
+ import warnings
5
+
6
+ from cortexdb_client._pb.cortexdb.v1 import memory_pb2 as cortexdb_dot_v1_dot_memory__pb2
7
+
8
+ GRPC_GENERATED_VERSION = '1.81.0'
9
+ GRPC_VERSION = grpc.__version__
10
+ _version_not_supported = False
11
+
12
+ try:
13
+ from grpc._utilities import first_version_is_lower
14
+ _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
15
+ except ImportError:
16
+ _version_not_supported = True
17
+
18
+ if _version_not_supported:
19
+ raise RuntimeError(
20
+ f'The grpc package installed is at version {GRPC_VERSION},'
21
+ + ' but the generated code in cortexdb/v1/memory_pb2_grpc.py depends on'
22
+ + f' grpcio>={GRPC_GENERATED_VERSION}.'
23
+ + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
24
+ + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
25
+ )
26
+
27
+
28
+ class MemoryServiceStub:
29
+ """Missing associated documentation comment in .proto file."""
30
+
31
+ def __init__(self, channel):
32
+ """Constructor.
33
+
34
+ Args:
35
+ channel: A grpc.Channel.
36
+ """
37
+ self.SaveMemory = channel.unary_unary(
38
+ '/cortexdb.v1.MemoryService/SaveMemory',
39
+ request_serializer=cortexdb_dot_v1_dot_memory__pb2.SaveMemoryRequest.SerializeToString,
40
+ response_deserializer=cortexdb_dot_v1_dot_memory__pb2.SaveMemoryResponse.FromString,
41
+ _registered_method=True)
42
+ self.UpdateMemory = channel.unary_unary(
43
+ '/cortexdb.v1.MemoryService/UpdateMemory',
44
+ request_serializer=cortexdb_dot_v1_dot_memory__pb2.UpdateMemoryRequest.SerializeToString,
45
+ response_deserializer=cortexdb_dot_v1_dot_memory__pb2.SaveMemoryResponse.FromString,
46
+ _registered_method=True)
47
+ self.GetMemory = channel.unary_unary(
48
+ '/cortexdb.v1.MemoryService/GetMemory',
49
+ request_serializer=cortexdb_dot_v1_dot_memory__pb2.GetMemoryRequest.SerializeToString,
50
+ response_deserializer=cortexdb_dot_v1_dot_memory__pb2.GetMemoryResponse.FromString,
51
+ _registered_method=True)
52
+ self.SearchMemory = channel.unary_unary(
53
+ '/cortexdb.v1.MemoryService/SearchMemory',
54
+ request_serializer=cortexdb_dot_v1_dot_memory__pb2.SearchMemoryRequest.SerializeToString,
55
+ response_deserializer=cortexdb_dot_v1_dot_memory__pb2.SearchMemoryResponse.FromString,
56
+ _registered_method=True)
57
+ self.DeleteMemory = channel.unary_unary(
58
+ '/cortexdb.v1.MemoryService/DeleteMemory',
59
+ request_serializer=cortexdb_dot_v1_dot_memory__pb2.DeleteMemoryRequest.SerializeToString,
60
+ response_deserializer=cortexdb_dot_v1_dot_memory__pb2.DeleteMemoryResponse.FromString,
61
+ _registered_method=True)
62
+
63
+
64
+ class MemoryServiceServicer:
65
+ """Missing associated documentation comment in .proto file."""
66
+
67
+ def SaveMemory(self, request, context):
68
+ """Missing associated documentation comment in .proto file."""
69
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
70
+ context.set_details('Method not implemented!')
71
+ raise NotImplementedError('Method not implemented!')
72
+
73
+ def UpdateMemory(self, request, context):
74
+ """Missing associated documentation comment in .proto file."""
75
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
76
+ context.set_details('Method not implemented!')
77
+ raise NotImplementedError('Method not implemented!')
78
+
79
+ def GetMemory(self, request, context):
80
+ """Missing associated documentation comment in .proto file."""
81
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
82
+ context.set_details('Method not implemented!')
83
+ raise NotImplementedError('Method not implemented!')
84
+
85
+ def SearchMemory(self, request, context):
86
+ """Missing associated documentation comment in .proto file."""
87
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
88
+ context.set_details('Method not implemented!')
89
+ raise NotImplementedError('Method not implemented!')
90
+
91
+ def DeleteMemory(self, request, context):
92
+ """Missing associated documentation comment in .proto file."""
93
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
94
+ context.set_details('Method not implemented!')
95
+ raise NotImplementedError('Method not implemented!')
96
+
97
+
98
+ def add_MemoryServiceServicer_to_server(servicer, server):
99
+ rpc_method_handlers = {
100
+ 'SaveMemory': grpc.unary_unary_rpc_method_handler(
101
+ servicer.SaveMemory,
102
+ request_deserializer=cortexdb_dot_v1_dot_memory__pb2.SaveMemoryRequest.FromString,
103
+ response_serializer=cortexdb_dot_v1_dot_memory__pb2.SaveMemoryResponse.SerializeToString,
104
+ ),
105
+ 'UpdateMemory': grpc.unary_unary_rpc_method_handler(
106
+ servicer.UpdateMemory,
107
+ request_deserializer=cortexdb_dot_v1_dot_memory__pb2.UpdateMemoryRequest.FromString,
108
+ response_serializer=cortexdb_dot_v1_dot_memory__pb2.SaveMemoryResponse.SerializeToString,
109
+ ),
110
+ 'GetMemory': grpc.unary_unary_rpc_method_handler(
111
+ servicer.GetMemory,
112
+ request_deserializer=cortexdb_dot_v1_dot_memory__pb2.GetMemoryRequest.FromString,
113
+ response_serializer=cortexdb_dot_v1_dot_memory__pb2.GetMemoryResponse.SerializeToString,
114
+ ),
115
+ 'SearchMemory': grpc.unary_unary_rpc_method_handler(
116
+ servicer.SearchMemory,
117
+ request_deserializer=cortexdb_dot_v1_dot_memory__pb2.SearchMemoryRequest.FromString,
118
+ response_serializer=cortexdb_dot_v1_dot_memory__pb2.SearchMemoryResponse.SerializeToString,
119
+ ),
120
+ 'DeleteMemory': grpc.unary_unary_rpc_method_handler(
121
+ servicer.DeleteMemory,
122
+ request_deserializer=cortexdb_dot_v1_dot_memory__pb2.DeleteMemoryRequest.FromString,
123
+ response_serializer=cortexdb_dot_v1_dot_memory__pb2.DeleteMemoryResponse.SerializeToString,
124
+ ),
125
+ }
126
+ generic_handler = grpc.method_handlers_generic_handler(
127
+ 'cortexdb.v1.MemoryService', rpc_method_handlers)
128
+ server.add_generic_rpc_handlers((generic_handler,))
129
+ server.add_registered_method_handlers('cortexdb.v1.MemoryService', rpc_method_handlers)
130
+
131
+
132
+ # This class is part of an EXPERIMENTAL API.
133
+ class MemoryService:
134
+ """Missing associated documentation comment in .proto file."""
135
+
136
+ @staticmethod
137
+ def SaveMemory(request,
138
+ target,
139
+ options=(),
140
+ channel_credentials=None,
141
+ call_credentials=None,
142
+ insecure=False,
143
+ compression=None,
144
+ wait_for_ready=None,
145
+ timeout=None,
146
+ metadata=None):
147
+ return grpc.experimental.unary_unary(
148
+ request,
149
+ target,
150
+ '/cortexdb.v1.MemoryService/SaveMemory',
151
+ cortexdb_dot_v1_dot_memory__pb2.SaveMemoryRequest.SerializeToString,
152
+ cortexdb_dot_v1_dot_memory__pb2.SaveMemoryResponse.FromString,
153
+ options,
154
+ channel_credentials,
155
+ insecure,
156
+ call_credentials,
157
+ compression,
158
+ wait_for_ready,
159
+ timeout,
160
+ metadata,
161
+ _registered_method=True)
162
+
163
+ @staticmethod
164
+ def UpdateMemory(request,
165
+ target,
166
+ options=(),
167
+ channel_credentials=None,
168
+ call_credentials=None,
169
+ insecure=False,
170
+ compression=None,
171
+ wait_for_ready=None,
172
+ timeout=None,
173
+ metadata=None):
174
+ return grpc.experimental.unary_unary(
175
+ request,
176
+ target,
177
+ '/cortexdb.v1.MemoryService/UpdateMemory',
178
+ cortexdb_dot_v1_dot_memory__pb2.UpdateMemoryRequest.SerializeToString,
179
+ cortexdb_dot_v1_dot_memory__pb2.SaveMemoryResponse.FromString,
180
+ options,
181
+ channel_credentials,
182
+ insecure,
183
+ call_credentials,
184
+ compression,
185
+ wait_for_ready,
186
+ timeout,
187
+ metadata,
188
+ _registered_method=True)
189
+
190
+ @staticmethod
191
+ def GetMemory(request,
192
+ target,
193
+ options=(),
194
+ channel_credentials=None,
195
+ call_credentials=None,
196
+ insecure=False,
197
+ compression=None,
198
+ wait_for_ready=None,
199
+ timeout=None,
200
+ metadata=None):
201
+ return grpc.experimental.unary_unary(
202
+ request,
203
+ target,
204
+ '/cortexdb.v1.MemoryService/GetMemory',
205
+ cortexdb_dot_v1_dot_memory__pb2.GetMemoryRequest.SerializeToString,
206
+ cortexdb_dot_v1_dot_memory__pb2.GetMemoryResponse.FromString,
207
+ options,
208
+ channel_credentials,
209
+ insecure,
210
+ call_credentials,
211
+ compression,
212
+ wait_for_ready,
213
+ timeout,
214
+ metadata,
215
+ _registered_method=True)
216
+
217
+ @staticmethod
218
+ def SearchMemory(request,
219
+ target,
220
+ options=(),
221
+ channel_credentials=None,
222
+ call_credentials=None,
223
+ insecure=False,
224
+ compression=None,
225
+ wait_for_ready=None,
226
+ timeout=None,
227
+ metadata=None):
228
+ return grpc.experimental.unary_unary(
229
+ request,
230
+ target,
231
+ '/cortexdb.v1.MemoryService/SearchMemory',
232
+ cortexdb_dot_v1_dot_memory__pb2.SearchMemoryRequest.SerializeToString,
233
+ cortexdb_dot_v1_dot_memory__pb2.SearchMemoryResponse.FromString,
234
+ options,
235
+ channel_credentials,
236
+ insecure,
237
+ call_credentials,
238
+ compression,
239
+ wait_for_ready,
240
+ timeout,
241
+ metadata,
242
+ _registered_method=True)
243
+
244
+ @staticmethod
245
+ def DeleteMemory(request,
246
+ target,
247
+ options=(),
248
+ channel_credentials=None,
249
+ call_credentials=None,
250
+ insecure=False,
251
+ compression=None,
252
+ wait_for_ready=None,
253
+ timeout=None,
254
+ metadata=None):
255
+ return grpc.experimental.unary_unary(
256
+ request,
257
+ target,
258
+ '/cortexdb.v1.MemoryService/DeleteMemory',
259
+ cortexdb_dot_v1_dot_memory__pb2.DeleteMemoryRequest.SerializeToString,
260
+ cortexdb_dot_v1_dot_memory__pb2.DeleteMemoryResponse.FromString,
261
+ options,
262
+ channel_credentials,
263
+ insecure,
264
+ call_credentials,
265
+ compression,
266
+ wait_for_ready,
267
+ timeout,
268
+ metadata,
269
+ _registered_method=True)
@@ -0,0 +1,47 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: cortexdb/v1/tools.proto
5
+ # Protobuf Python Version: 6.33.5
6
+ """Generated protocol buffer code."""
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import runtime_version as _runtime_version
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ from google.protobuf.internal import builder as _builder
12
+ _runtime_version.ValidateProtobufRuntimeVersion(
13
+ _runtime_version.Domain.PUBLIC,
14
+ 6,
15
+ 33,
16
+ 5,
17
+ '',
18
+ 'cortexdb/v1/tools.proto'
19
+ )
20
+ # @@protoc_insertion_point(imports)
21
+
22
+ _sym_db = _symbol_database.Default()
23
+
24
+
25
+
26
+
27
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x17\x63ortexdb/v1/tools.proto\x12\x0b\x63ortexdb.v1\"N\n\x0eToolDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x19\n\x11input_schema_json\x18\x03 \x01(\t\"\x12\n\x10ListToolsRequest\"?\n\x11ListToolsResponse\x12*\n\x05tools\x18\x01 \x03(\x0b\x32\x1b.cortexdb.v1.ToolDefinition\"2\n\x0f\x43\x61llToolRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x11\n\targs_json\x18\x02 \x01(\t\"\'\n\x10\x43\x61llToolResponse\x12\x13\n\x0bresult_json\x18\x01 \x01(\t2\xa3\x01\n\x0cToolsService\x12J\n\tListTools\x12\x1d.cortexdb.v1.ListToolsRequest\x1a\x1e.cortexdb.v1.ListToolsResponse\x12G\n\x08\x43\x61llTool\x12\x1c.cortexdb.v1.CallToolRequest\x1a\x1d.cortexdb.v1.CallToolResponseB4Z2github.com/liliang-cn/cortexdb/v2/pkg/rpc/v1;rpcv1b\x06proto3')
28
+
29
+ _globals = globals()
30
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
31
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'cortexdb.v1.tools_pb2', _globals)
32
+ if not _descriptor._USE_C_DESCRIPTORS:
33
+ _globals['DESCRIPTOR']._loaded_options = None
34
+ _globals['DESCRIPTOR']._serialized_options = b'Z2github.com/liliang-cn/cortexdb/v2/pkg/rpc/v1;rpcv1'
35
+ _globals['_TOOLDEFINITION']._serialized_start=40
36
+ _globals['_TOOLDEFINITION']._serialized_end=118
37
+ _globals['_LISTTOOLSREQUEST']._serialized_start=120
38
+ _globals['_LISTTOOLSREQUEST']._serialized_end=138
39
+ _globals['_LISTTOOLSRESPONSE']._serialized_start=140
40
+ _globals['_LISTTOOLSRESPONSE']._serialized_end=203
41
+ _globals['_CALLTOOLREQUEST']._serialized_start=205
42
+ _globals['_CALLTOOLREQUEST']._serialized_end=255
43
+ _globals['_CALLTOOLRESPONSE']._serialized_start=257
44
+ _globals['_CALLTOOLRESPONSE']._serialized_end=296
45
+ _globals['_TOOLSSERVICE']._serialized_start=299
46
+ _globals['_TOOLSSERVICE']._serialized_end=462
47
+ # @@protoc_insertion_point(module_scope)