frogml-core 0.0.113__py3-none-any.whl → 0.0.114__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.
Files changed (115) hide show
  1. frogml_core/__init__.py +1 -1
  2. frogml_core/inner/di_configuration/__init__.py +0 -6
  3. {frogml_core-0.0.113.dist-info → frogml_core-0.0.114.dist-info}/METADATA +1 -1
  4. {frogml_core-0.0.113.dist-info → frogml_core-0.0.114.dist-info}/RECORD +8 -115
  5. frogml_services_mock/mocks/frogml_mocks.py +0 -11
  6. frogml_services_mock/services_mock.py +0 -48
  7. frogml_storage/__init__.py +1 -1
  8. frogml_core/clients/prompt_manager/__init__.py +0 -0
  9. frogml_core/clients/prompt_manager/model_descriptor_mapper.py +0 -196
  10. frogml_core/clients/prompt_manager/prompt_manager_client.py +0 -190
  11. frogml_core/clients/prompt_manager/prompt_proto_mapper.py +0 -264
  12. frogml_core/clients/vector_store/__init__.py +0 -2
  13. frogml_core/clients/vector_store/management_client.py +0 -127
  14. frogml_core/clients/vector_store/serving_client.py +0 -157
  15. frogml_core/clients/workspace_manager/__init__.py +0 -1
  16. frogml_core/clients/workspace_manager/client.py +0 -224
  17. frogml_core/llmops/__init__.py +0 -0
  18. frogml_core/llmops/generation/__init__.py +0 -0
  19. frogml_core/llmops/generation/_steaming.py +0 -78
  20. frogml_core/llmops/generation/base.py +0 -5
  21. frogml_core/llmops/generation/chat/__init__.py +0 -0
  22. frogml_core/llmops/generation/chat/openai/LICENSE.txt +0 -201
  23. frogml_core/llmops/generation/chat/openai/types/__init__.py +0 -0
  24. frogml_core/llmops/generation/chat/openai/types/chat/__init__.py +0 -0
  25. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion.py +0 -88
  26. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_assistant_message_param.py +0 -65
  27. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_chunk.py +0 -153
  28. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_content_part_text_param.py +0 -28
  29. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_function_call_option_param.py +0 -25
  30. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_function_message_param.py +0 -33
  31. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_message.py +0 -56
  32. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_message_param.py +0 -34
  33. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_message_tool_call.py +0 -46
  34. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_message_tool_call_param.py +0 -44
  35. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_named_tool_choice_param.py +0 -32
  36. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_role.py +0 -20
  37. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_system_message_param.py +0 -35
  38. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_token_logprob.py +0 -71
  39. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_tool_choice_option_param.py +0 -28
  40. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_tool_message_param.py +0 -31
  41. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_tool_param.py +0 -29
  42. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_user_message_param.py +0 -35
  43. frogml_core/llmops/generation/chat/openai/types/chat/completion_create_params.py +0 -279
  44. frogml_core/llmops/generation/chat/openai/types/completion_choice.py +0 -47
  45. frogml_core/llmops/generation/chat/openai/types/completion_create_params.py +0 -209
  46. frogml_core/llmops/generation/chat/openai/types/completion_usage.py +0 -30
  47. frogml_core/llmops/generation/chat/openai/types/model.py +0 -35
  48. frogml_core/llmops/generation/chat/openai/types/shared/__init__.py +0 -3
  49. frogml_core/llmops/generation/chat/openai/types/shared/error_object.py +0 -27
  50. frogml_core/llmops/generation/chat/openai/types/shared/function_definition.py +0 -49
  51. frogml_core/llmops/generation/chat/openai/types/shared/function_parameters.py +0 -20
  52. frogml_core/llmops/generation/chat/openai/types/shared_params/__init__.py +0 -2
  53. frogml_core/llmops/generation/chat/openai/types/shared_params/function_definition.py +0 -49
  54. frogml_core/llmops/generation/chat/openai/types/shared_params/function_parameters.py +0 -22
  55. frogml_core/llmops/generation/streaming.py +0 -26
  56. frogml_core/llmops/model/__init__.py +0 -0
  57. frogml_core/llmops/model/descriptor.py +0 -40
  58. frogml_core/llmops/prompt/__init__.py +0 -0
  59. frogml_core/llmops/prompt/base.py +0 -136
  60. frogml_core/llmops/prompt/chat/__init__.py +0 -0
  61. frogml_core/llmops/prompt/chat/message.py +0 -24
  62. frogml_core/llmops/prompt/chat/template.py +0 -113
  63. frogml_core/llmops/prompt/chat/value.py +0 -10
  64. frogml_core/llmops/prompt/manager.py +0 -138
  65. frogml_core/llmops/prompt/template.py +0 -24
  66. frogml_core/llmops/prompt/value.py +0 -14
  67. frogml_core/llmops/provider/__init__.py +0 -0
  68. frogml_core/llmops/provider/chat.py +0 -44
  69. frogml_core/llmops/provider/openai/__init__.py +0 -0
  70. frogml_core/llmops/provider/openai/client.py +0 -126
  71. frogml_core/llmops/provider/openai/provider.py +0 -93
  72. frogml_core/vector_store/__init__.py +0 -4
  73. frogml_core/vector_store/client.py +0 -151
  74. frogml_core/vector_store/collection.py +0 -429
  75. frogml_core/vector_store/filters.py +0 -359
  76. frogml_core/vector_store/inference_client.py +0 -105
  77. frogml_core/vector_store/rest_helpers.py +0 -81
  78. frogml_core/vector_store/utils/__init__.py +0 -0
  79. frogml_core/vector_store/utils/filter_utils.py +0 -23
  80. frogml_core/vector_store/utils/upsert_utils.py +0 -218
  81. frogml_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2.py +0 -77
  82. frogml_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2.pyi +0 -417
  83. frogml_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2_grpc.py +0 -441
  84. frogml_proto/qwak/prompt/v1/prompt/prompt_pb2.py +0 -69
  85. frogml_proto/qwak/prompt/v1/prompt/prompt_pb2.pyi +0 -415
  86. frogml_proto/qwak/prompt/v1/prompt/prompt_pb2_grpc.py +0 -4
  87. frogml_proto/qwak/vectors/v1/collection/collection_pb2.py +0 -46
  88. frogml_proto/qwak/vectors/v1/collection/collection_pb2.pyi +0 -287
  89. frogml_proto/qwak/vectors/v1/collection/collection_pb2_grpc.py +0 -4
  90. frogml_proto/qwak/vectors/v1/collection/collection_service_pb2.py +0 -60
  91. frogml_proto/qwak/vectors/v1/collection/collection_service_pb2.pyi +0 -258
  92. frogml_proto/qwak/vectors/v1/collection/collection_service_pb2_grpc.py +0 -304
  93. frogml_proto/qwak/vectors/v1/collection/event/collection_event_pb2.py +0 -28
  94. frogml_proto/qwak/vectors/v1/collection/event/collection_event_pb2.pyi +0 -41
  95. frogml_proto/qwak/vectors/v1/collection/event/collection_event_pb2_grpc.py +0 -4
  96. frogml_proto/qwak/vectors/v1/filters_pb2.py +0 -52
  97. frogml_proto/qwak/vectors/v1/filters_pb2.pyi +0 -297
  98. frogml_proto/qwak/vectors/v1/filters_pb2_grpc.py +0 -4
  99. frogml_proto/qwak/vectors/v1/vector_pb2.py +0 -38
  100. frogml_proto/qwak/vectors/v1/vector_pb2.pyi +0 -142
  101. frogml_proto/qwak/vectors/v1/vector_pb2_grpc.py +0 -4
  102. frogml_proto/qwak/vectors/v1/vector_service_pb2.py +0 -53
  103. frogml_proto/qwak/vectors/v1/vector_service_pb2.pyi +0 -243
  104. frogml_proto/qwak/vectors/v1/vector_service_pb2_grpc.py +0 -201
  105. frogml_proto/qwak/workspace/workspace_pb2.py +0 -50
  106. frogml_proto/qwak/workspace/workspace_pb2.pyi +0 -331
  107. frogml_proto/qwak/workspace/workspace_pb2_grpc.py +0 -4
  108. frogml_proto/qwak/workspace/workspace_service_pb2.py +0 -84
  109. frogml_proto/qwak/workspace/workspace_service_pb2.pyi +0 -393
  110. frogml_proto/qwak/workspace/workspace_service_pb2_grpc.py +0 -507
  111. frogml_services_mock/mocks/prompt_manager_service.py +0 -281
  112. frogml_services_mock/mocks/vector_serving_api.py +0 -159
  113. frogml_services_mock/mocks/vectors_management_api.py +0 -97
  114. frogml_services_mock/mocks/workspace_manager_service_mock.py +0 -202
  115. {frogml_core-0.0.113.dist-info → frogml_core-0.0.114.dist-info}/WHEEL +0 -0
@@ -1,243 +0,0 @@
1
- """
2
- @generated by mypy-protobuf. Do not edit manually!
3
- isort:skip_file
4
- """
5
- import builtins
6
- import collections.abc
7
- import google.protobuf.descriptor
8
- import google.protobuf.internal.containers
9
- import google.protobuf.message
10
- import frogml_proto.qwak.vectors.v1.filters_pb2
11
- import frogml_proto.qwak.vectors.v1.vector_pb2
12
- import sys
13
- import typing
14
-
15
- if sys.version_info >= (3, 8):
16
- import typing as typing_extensions
17
- else:
18
- import typing_extensions
19
-
20
- DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
21
-
22
- class SearchSimilarVectorsRequest(google.protobuf.message.Message):
23
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
24
-
25
- COLLECTION_NAME_FIELD_NUMBER: builtins.int
26
- REFERENCE_VECTOR_FIELD_NUMBER: builtins.int
27
- PROPERTIES_FIELD_NUMBER: builtins.int
28
- FILTER_FIELD_NUMBER: builtins.int
29
- MAX_RESULTS_FIELD_NUMBER: builtins.int
30
- INCLUDE_VECTOR_FIELD_NUMBER: builtins.int
31
- INCLUDE_DISTANCE_FIELD_NUMBER: builtins.int
32
- INCLUDE_ID_FIELD_NUMBER: builtins.int
33
- TENANT_ID_FIELD_NUMBER: builtins.int
34
- collection_name: builtins.str
35
- """Name of the collection to search in"""
36
- @property
37
- def reference_vector(self) -> frogml_proto.qwak.vectors.v1.vector_pb2.DoubleVector:
38
- """Reference vector"""
39
- @property
40
- def properties(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
41
- """properties to retrieve"""
42
- @property
43
- def filter(self) -> frogml_proto.qwak.vectors.v1.filters_pb2.Filter: ...
44
- max_results: builtins.int
45
- """the "k" in top_k"""
46
- include_vector: builtins.bool
47
- """whether to include the actual result vectors in the response"""
48
- include_distance: builtins.bool
49
- """whether to include the distance in the result"""
50
- include_id: builtins.bool
51
- """whether to include ids in the result"""
52
- tenant_id: builtins.str
53
- def __init__(
54
- self,
55
- *,
56
- collection_name: builtins.str = ...,
57
- reference_vector: frogml_proto.qwak.vectors.v1.vector_pb2.DoubleVector | None = ...,
58
- properties: collections.abc.Iterable[builtins.str] | None = ...,
59
- filter: frogml_proto.qwak.vectors.v1.filters_pb2.Filter | None = ...,
60
- max_results: builtins.int = ...,
61
- include_vector: builtins.bool = ...,
62
- include_distance: builtins.bool = ...,
63
- include_id: builtins.bool = ...,
64
- tenant_id: builtins.str = ...,
65
- ) -> None: ...
66
- def HasField(self, field_name: typing_extensions.Literal["filter", b"filter", "filter_opt", b"filter_opt", "optional_tenant", b"optional_tenant", "reference_vector", b"reference_vector", "tenant_id", b"tenant_id"]) -> builtins.bool: ...
67
- def ClearField(self, field_name: typing_extensions.Literal["collection_name", b"collection_name", "filter", b"filter", "filter_opt", b"filter_opt", "include_distance", b"include_distance", "include_id", b"include_id", "include_vector", b"include_vector", "max_results", b"max_results", "optional_tenant", b"optional_tenant", "properties", b"properties", "reference_vector", b"reference_vector", "tenant_id", b"tenant_id"]) -> None: ...
68
- @typing.overload
69
- def WhichOneof(self, oneof_group: typing_extensions.Literal["filter_opt", b"filter_opt"]) -> typing_extensions.Literal["filter"] | None: ...
70
- @typing.overload
71
- def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_tenant", b"optional_tenant"]) -> typing_extensions.Literal["tenant_id"] | None: ...
72
-
73
- global___SearchSimilarVectorsRequest = SearchSimilarVectorsRequest
74
-
75
- class SearchSimilarVectorsResponse(google.protobuf.message.Message):
76
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
77
-
78
- SEARCH_RESULTS_FIELD_NUMBER: builtins.int
79
- @property
80
- def search_results(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.vectors.v1.vector_pb2.SearchResult]: ...
81
- def __init__(
82
- self,
83
- *,
84
- search_results: collections.abc.Iterable[qwak.vectors.v1.vector_pb2.SearchResult] | None = ...,
85
- ) -> None: ...
86
- def ClearField(self, field_name: typing_extensions.Literal["search_results", b"search_results"]) -> None: ...
87
-
88
- global___SearchSimilarVectorsResponse = SearchSimilarVectorsResponse
89
-
90
- class GetVectorSampleRequest(google.protobuf.message.Message):
91
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
92
-
93
- COLLECTION_NAME_FIELD_NUMBER: builtins.int
94
- NUM_SAMPLES_FIELD_NUMBER: builtins.int
95
- INCLUDE_VECTOR_FIELD_NUMBER: builtins.int
96
- TENANT_ID_FIELD_NUMBER: builtins.int
97
- collection_name: builtins.str
98
- """Name of the collection to search in"""
99
- num_samples: builtins.int
100
- """Number of samples to retrieve"""
101
- include_vector: builtins.bool
102
- """Whether to include the actual result vectors in the response"""
103
- tenant_id: builtins.str
104
- def __init__(
105
- self,
106
- *,
107
- collection_name: builtins.str = ...,
108
- num_samples: builtins.int = ...,
109
- include_vector: builtins.bool = ...,
110
- tenant_id: builtins.str = ...,
111
- ) -> None: ...
112
- def HasField(self, field_name: typing_extensions.Literal["optional_tenant", b"optional_tenant", "tenant_id", b"tenant_id"]) -> builtins.bool: ...
113
- def ClearField(self, field_name: typing_extensions.Literal["collection_name", b"collection_name", "include_vector", b"include_vector", "num_samples", b"num_samples", "optional_tenant", b"optional_tenant", "tenant_id", b"tenant_id"]) -> None: ...
114
- def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_tenant", b"optional_tenant"]) -> typing_extensions.Literal["tenant_id"] | None: ...
115
-
116
- global___GetVectorSampleRequest = GetVectorSampleRequest
117
-
118
- class GetVectorSampleResponse(google.protobuf.message.Message):
119
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
120
-
121
- VECTORS_FIELD_NUMBER: builtins.int
122
- @property
123
- def vectors(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.vectors.v1.vector_pb2.StoredVector]: ...
124
- def __init__(
125
- self,
126
- *,
127
- vectors: collections.abc.Iterable[qwak.vectors.v1.vector_pb2.StoredVector] | None = ...,
128
- ) -> None: ...
129
- def ClearField(self, field_name: typing_extensions.Literal["vectors", b"vectors"]) -> None: ...
130
-
131
- global___GetVectorSampleResponse = GetVectorSampleResponse
132
-
133
- class UpsertVectorsRequest(google.protobuf.message.Message):
134
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
135
-
136
- COLLECTION_NAME_FIELD_NUMBER: builtins.int
137
- VECTOR_FIELD_NUMBER: builtins.int
138
- collection_name: builtins.str
139
- @property
140
- def vector(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.vectors.v1.vector_pb2.StoredVector]: ...
141
- def __init__(
142
- self,
143
- *,
144
- collection_name: builtins.str = ...,
145
- vector: collections.abc.Iterable[qwak.vectors.v1.vector_pb2.StoredVector] | None = ...,
146
- ) -> None: ...
147
- def ClearField(self, field_name: typing_extensions.Literal["collection_name", b"collection_name", "vector", b"vector"]) -> None: ...
148
-
149
- global___UpsertVectorsRequest = UpsertVectorsRequest
150
-
151
- class UpsertVectorsResponse(google.protobuf.message.Message):
152
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
153
-
154
- NUM_AFFECTED_FIELD_NUMBER: builtins.int
155
- num_affected: builtins.int
156
- """number of vectors that were affected in the collection"""
157
- def __init__(
158
- self,
159
- *,
160
- num_affected: builtins.int = ...,
161
- ) -> None: ...
162
- def ClearField(self, field_name: typing_extensions.Literal["num_affected", b"num_affected"]) -> None: ...
163
-
164
- global___UpsertVectorsResponse = UpsertVectorsResponse
165
-
166
- class DeleteVectorsRequest(google.protobuf.message.Message):
167
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
168
-
169
- COLLECTION_NAME_FIELD_NUMBER: builtins.int
170
- VECTOR_ID_FIELD_NUMBER: builtins.int
171
- VECTOR_IDENTIFIERS_FIELD_NUMBER: builtins.int
172
- collection_name: builtins.str
173
- @property
174
- def vector_id(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
175
- """passing explicit id is now deprecated, pass a VectorIdentifier"""
176
- @property
177
- def vector_identifiers(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.vectors.v1.vector_pb2.VectorIdentifier]: ...
178
- def __init__(
179
- self,
180
- *,
181
- collection_name: builtins.str = ...,
182
- vector_id: collections.abc.Iterable[builtins.str] | None = ...,
183
- vector_identifiers: collections.abc.Iterable[qwak.vectors.v1.vector_pb2.VectorIdentifier] | None = ...,
184
- ) -> None: ...
185
- def ClearField(self, field_name: typing_extensions.Literal["collection_name", b"collection_name", "vector_id", b"vector_id", "vector_identifiers", b"vector_identifiers"]) -> None: ...
186
-
187
- global___DeleteVectorsRequest = DeleteVectorsRequest
188
-
189
- class DeleteVectorsResponse(google.protobuf.message.Message):
190
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
191
-
192
- NUM_VECTORS_DELETED_FIELD_NUMBER: builtins.int
193
- num_vectors_deleted: builtins.int
194
- def __init__(
195
- self,
196
- *,
197
- num_vectors_deleted: builtins.int = ...,
198
- ) -> None: ...
199
- def ClearField(self, field_name: typing_extensions.Literal["num_vectors_deleted", b"num_vectors_deleted"]) -> None: ...
200
-
201
- global___DeleteVectorsResponse = DeleteVectorsResponse
202
-
203
- class FetchVectorRequest(google.protobuf.message.Message):
204
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
205
-
206
- COLLECTION_NAME_FIELD_NUMBER: builtins.int
207
- VECTOR_ID_FIELD_NUMBER: builtins.int
208
- VECTOR_IDENTIFIER_FIELD_NUMBER: builtins.int
209
- collection_name: builtins.str
210
- """Name of the collection to fetch from"""
211
- vector_id: builtins.str
212
- """Vector id to fetch - this field is deprecated, use VectorIdentifier"""
213
- @property
214
- def vector_identifier(self) -> frogml_proto.qwak.vectors.v1.vector_pb2.VectorIdentifier:
215
- """Vector identifier to fetch"""
216
- def __init__(
217
- self,
218
- *,
219
- collection_name: builtins.str = ...,
220
- vector_id: builtins.str = ...,
221
- vector_identifier: frogml_proto.qwak.vectors.v1.vector_pb2.VectorIdentifier | None = ...,
222
- ) -> None: ...
223
- def HasField(self, field_name: typing_extensions.Literal["vector_identifier", b"vector_identifier"]) -> builtins.bool: ...
224
- def ClearField(self, field_name: typing_extensions.Literal["collection_name", b"collection_name", "vector_id", b"vector_id", "vector_identifier", b"vector_identifier"]) -> None: ...
225
-
226
- global___FetchVectorRequest = FetchVectorRequest
227
-
228
- class FetchVectorResponse(google.protobuf.message.Message):
229
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
230
-
231
- VECTOR_FIELD_NUMBER: builtins.int
232
- @property
233
- def vector(self) -> frogml_proto.qwak.vectors.v1.vector_pb2.StoredVector:
234
- """Fetched vector"""
235
- def __init__(
236
- self,
237
- *,
238
- vector: frogml_proto.qwak.vectors.v1.vector_pb2.StoredVector | None = ...,
239
- ) -> None: ...
240
- def HasField(self, field_name: typing_extensions.Literal["vector", b"vector"]) -> builtins.bool: ...
241
- def ClearField(self, field_name: typing_extensions.Literal["vector", b"vector"]) -> None: ...
242
-
243
- global___FetchVectorResponse = FetchVectorResponse
@@ -1,201 +0,0 @@
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
-
5
- from frogml_proto.qwak.vectors.v1 import vector_service_pb2 as qwak_dot_vectors_dot_v1_dot_vector__service__pb2
6
-
7
-
8
- class VectorServiceStub(object):
9
- """Missing associated documentation comment in .proto file."""
10
-
11
- def __init__(self, channel):
12
- """Constructor.
13
-
14
- Args:
15
- channel: A grpc.Channel.
16
- """
17
- self.SearchSimilarVectors = channel.unary_unary(
18
- '/qwak.vector.store.serving.v1.VectorService/SearchSimilarVectors',
19
- request_serializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.SearchSimilarVectorsRequest.SerializeToString,
20
- response_deserializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.SearchSimilarVectorsResponse.FromString,
21
- )
22
- self.GetVectorSample = channel.unary_unary(
23
- '/qwak.vector.store.serving.v1.VectorService/GetVectorSample',
24
- request_serializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.GetVectorSampleRequest.SerializeToString,
25
- response_deserializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.GetVectorSampleResponse.FromString,
26
- )
27
- self.FetchVector = channel.unary_unary(
28
- '/qwak.vector.store.serving.v1.VectorService/FetchVector',
29
- request_serializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.FetchVectorRequest.SerializeToString,
30
- response_deserializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.FetchVectorResponse.FromString,
31
- )
32
- self.UpsertVectors = channel.unary_unary(
33
- '/qwak.vector.store.serving.v1.VectorService/UpsertVectors',
34
- request_serializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.UpsertVectorsRequest.SerializeToString,
35
- response_deserializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.UpsertVectorsResponse.FromString,
36
- )
37
- self.DeleteVectors = channel.unary_unary(
38
- '/qwak.vector.store.serving.v1.VectorService/DeleteVectors',
39
- request_serializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.DeleteVectorsRequest.SerializeToString,
40
- response_deserializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.DeleteVectorsResponse.FromString,
41
- )
42
-
43
-
44
- class VectorServiceServicer(object):
45
- """Missing associated documentation comment in .proto file."""
46
-
47
- def SearchSimilarVectors(self, request, context):
48
- """Search for similar vectors
49
- """
50
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
51
- context.set_details('Method not implemented!')
52
- raise NotImplementedError('Method not implemented!')
53
-
54
- def GetVectorSample(self, request, context):
55
- """Get vector samples
56
- """
57
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
58
- context.set_details('Method not implemented!')
59
- raise NotImplementedError('Method not implemented!')
60
-
61
- def FetchVector(self, request, context):
62
- """Fetch a specific vector by ID
63
- """
64
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
65
- context.set_details('Method not implemented!')
66
- raise NotImplementedError('Method not implemented!')
67
-
68
- def UpsertVectors(self, request, context):
69
- """Missing associated documentation comment in .proto file."""
70
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
71
- context.set_details('Method not implemented!')
72
- raise NotImplementedError('Method not implemented!')
73
-
74
- def DeleteVectors(self, request, context):
75
- """Missing associated documentation comment in .proto file."""
76
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
77
- context.set_details('Method not implemented!')
78
- raise NotImplementedError('Method not implemented!')
79
-
80
-
81
- def add_VectorServiceServicer_to_server(servicer, server):
82
- rpc_method_handlers = {
83
- 'SearchSimilarVectors': grpc.unary_unary_rpc_method_handler(
84
- servicer.SearchSimilarVectors,
85
- request_deserializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.SearchSimilarVectorsRequest.FromString,
86
- response_serializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.SearchSimilarVectorsResponse.SerializeToString,
87
- ),
88
- 'GetVectorSample': grpc.unary_unary_rpc_method_handler(
89
- servicer.GetVectorSample,
90
- request_deserializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.GetVectorSampleRequest.FromString,
91
- response_serializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.GetVectorSampleResponse.SerializeToString,
92
- ),
93
- 'FetchVector': grpc.unary_unary_rpc_method_handler(
94
- servicer.FetchVector,
95
- request_deserializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.FetchVectorRequest.FromString,
96
- response_serializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.FetchVectorResponse.SerializeToString,
97
- ),
98
- 'UpsertVectors': grpc.unary_unary_rpc_method_handler(
99
- servicer.UpsertVectors,
100
- request_deserializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.UpsertVectorsRequest.FromString,
101
- response_serializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.UpsertVectorsResponse.SerializeToString,
102
- ),
103
- 'DeleteVectors': grpc.unary_unary_rpc_method_handler(
104
- servicer.DeleteVectors,
105
- request_deserializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.DeleteVectorsRequest.FromString,
106
- response_serializer=qwak_dot_vectors_dot_v1_dot_vector__service__pb2.DeleteVectorsResponse.SerializeToString,
107
- ),
108
- }
109
- generic_handler = grpc.method_handlers_generic_handler(
110
- 'qwak.vector.store.serving.v1.VectorService', rpc_method_handlers)
111
- server.add_generic_rpc_handlers((generic_handler,))
112
-
113
-
114
- # This class is part of an EXPERIMENTAL API.
115
- class VectorService(object):
116
- """Missing associated documentation comment in .proto file."""
117
-
118
- @staticmethod
119
- def SearchSimilarVectors(request,
120
- target,
121
- options=(),
122
- channel_credentials=None,
123
- call_credentials=None,
124
- insecure=False,
125
- compression=None,
126
- wait_for_ready=None,
127
- timeout=None,
128
- metadata=None):
129
- return grpc.experimental.unary_unary(request, target, '/qwak.vector.store.serving.v1.VectorService/SearchSimilarVectors',
130
- qwak_dot_vectors_dot_v1_dot_vector__service__pb2.SearchSimilarVectorsRequest.SerializeToString,
131
- qwak_dot_vectors_dot_v1_dot_vector__service__pb2.SearchSimilarVectorsResponse.FromString,
132
- options, channel_credentials,
133
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
134
-
135
- @staticmethod
136
- def GetVectorSample(request,
137
- target,
138
- options=(),
139
- channel_credentials=None,
140
- call_credentials=None,
141
- insecure=False,
142
- compression=None,
143
- wait_for_ready=None,
144
- timeout=None,
145
- metadata=None):
146
- return grpc.experimental.unary_unary(request, target, '/qwak.vector.store.serving.v1.VectorService/GetVectorSample',
147
- qwak_dot_vectors_dot_v1_dot_vector__service__pb2.GetVectorSampleRequest.SerializeToString,
148
- qwak_dot_vectors_dot_v1_dot_vector__service__pb2.GetVectorSampleResponse.FromString,
149
- options, channel_credentials,
150
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
151
-
152
- @staticmethod
153
- def FetchVector(request,
154
- target,
155
- options=(),
156
- channel_credentials=None,
157
- call_credentials=None,
158
- insecure=False,
159
- compression=None,
160
- wait_for_ready=None,
161
- timeout=None,
162
- metadata=None):
163
- return grpc.experimental.unary_unary(request, target, '/qwak.vector.store.serving.v1.VectorService/FetchVector',
164
- qwak_dot_vectors_dot_v1_dot_vector__service__pb2.FetchVectorRequest.SerializeToString,
165
- qwak_dot_vectors_dot_v1_dot_vector__service__pb2.FetchVectorResponse.FromString,
166
- options, channel_credentials,
167
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
168
-
169
- @staticmethod
170
- def UpsertVectors(request,
171
- target,
172
- options=(),
173
- channel_credentials=None,
174
- call_credentials=None,
175
- insecure=False,
176
- compression=None,
177
- wait_for_ready=None,
178
- timeout=None,
179
- metadata=None):
180
- return grpc.experimental.unary_unary(request, target, '/qwak.vector.store.serving.v1.VectorService/UpsertVectors',
181
- qwak_dot_vectors_dot_v1_dot_vector__service__pb2.UpsertVectorsRequest.SerializeToString,
182
- qwak_dot_vectors_dot_v1_dot_vector__service__pb2.UpsertVectorsResponse.FromString,
183
- options, channel_credentials,
184
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
185
-
186
- @staticmethod
187
- def DeleteVectors(request,
188
- target,
189
- options=(),
190
- channel_credentials=None,
191
- call_credentials=None,
192
- insecure=False,
193
- compression=None,
194
- wait_for_ready=None,
195
- timeout=None,
196
- metadata=None):
197
- return grpc.experimental.unary_unary(request, target, '/qwak.vector.store.serving.v1.VectorService/DeleteVectors',
198
- qwak_dot_vectors_dot_v1_dot_vector__service__pb2.DeleteVectorsRequest.SerializeToString,
199
- qwak_dot_vectors_dot_v1_dot_vector__service__pb2.DeleteVectorsResponse.FromString,
200
- options, channel_credentials,
201
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
@@ -1,50 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- # Generated by the protocol buffer compiler. DO NOT EDIT!
3
- # source: frogml_proto.qwak.workspace/workspace.proto
4
- """Generated protocol buffer code."""
5
- from google.protobuf import descriptor as _descriptor
6
- from google.protobuf import descriptor_pool as _descriptor_pool
7
- from google.protobuf import symbol_database as _symbol_database
8
- from google.protobuf.internal import builder as _builder
9
- # @@protoc_insertion_point(imports)
10
-
11
- _sym_db = _symbol_database.Default()
12
-
13
-
14
- from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
15
- from frogml_proto.qwak.user_application.common.v0 import resources_pb2 as qwak_dot_user__application_dot_common_dot_v0_dot_resources__pb2
16
- from frogml_proto.qwak.user_application.v0 import user_application_pb2 as qwak_dot_user__application_dot_v0_dot_user__application__pb2
17
-
18
-
19
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1eqwak/workspace/workspace.proto\x12\x19qwak.workspace.management\x1a\x1fgoogle/protobuf/timestamp.proto\x1a/qwak/user_application/common/v0/resources.proto\x1a/qwak/user_application/v0/user_application.proto\"\xc4\x04\n\tWorkspace\x12\x16\n\x0eworkspace_name\x18\x01 \x01(\t\x12`\n\x1c\x63lient_pod_compute_resources\x18\x02 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\x12\x10\n\x08image_id\x18\x03 \x01(\t\x12\x17\n\x0f\x63reator_user_id\x18\x04 \x01(\t\x12\x1b\n\x13last_update_user_id\x18\x05 \x01(\t\x12/\n\x0b\x63reate_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_update_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x14\n\x0cworkspace_id\x18\t \x01(\t\x12\x0c\n\x04tags\x18\n \x03(\t\x12N\n\x1aworkspace_deployment_state\x18\x0b \x01(\x0e\x32*.qwak.workspace.management.DeploymentState\x12\x44\n\x10workspace_status\x18\x0c \x01(\x0e\x32*.qwak.workspace.management.WorkspaceStatus\x12\x15\n\rerror_message\x18\r \x01(\t\x12\x12\n\nimage_name\x18\x0e \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x0f \x01(\t\x12\x11\n\tbuild_ids\x18\x10 \x03(\t\"\xd3\x01\n\rWorkspaceSpec\x12\x16\n\x0eworkspace_name\x18\x01 \x01(\t\x12`\n\x1c\x63lient_pod_compute_resources\x18\x02 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\x12\x10\n\x08image_id\x18\x03 \x01(\t\x12\x0c\n\x04tags\x18\x04 \x03(\t\x12\x16\n\x0e\x65nvironment_id\x18\x05 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x06 \x01(\t\"\x9d\x02\n\x1bWorkspaceDeploymentSnapshot\x12\x16\n\x0eworkspace_name\x18\x01 \x01(\t\x12\x14\n\x0cworkspace_id\x18\x02 \x01(\t\x12\x36\n\ndeployment\x18\x03 \x01(\x0b\x32\x1e.qwak.user_application.v0.SpecB\x02\x18\x01\x12\x34\n\x08replicas\x18\x04 \x03(\x0b\x32\x1e.qwak.user_application.v0.SpecB\x02\x18\x01\x12,\n\x04pods\x18\x05 \x03(\x0b\x32\x1e.qwak.user_application.v0.Spec\x12\x34\n\x0cstateful_set\x18\x06 \x01(\x0b\x32\x1e.qwak.user_application.v0.Spec\"\xb1\x01\n$UpdateWorkspaceStatusFromEdgeDetails\x12\x12\n\naccount_id\x18\x01 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x02 \x01(\t\x12]\n\x1dworkspace_deployment_snapshot\x18\x03 \x03(\x0b\x32\x36.qwak.workspace.management.WorkspaceDeploymentSnapshot\"Y\n\x0eWorkspaceImage\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12-\n\x04type\x18\x03 \x01(\x0e\x32\x1f.qwak.workspace.management.Type\"\xfb\x01\n\x17\x44\x65\x66\x61ultWorkspaceDetails\x12Y\n\x15\x63pu_compute_resources\x18\x01 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\x12Y\n\x15gpu_compute_resources\x18\x02 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\x12\x14\n\x0c\x63pu_image_id\x18\x03 \x01(\t\x12\x14\n\x0cgpu_image_id\x18\x04 \x01(\t*\x18\n\x04Type\x12\x07\n\x03\x43PU\x10\x00\x12\x07\n\x03GPU\x10\x01*\xaa\x01\n\x0fWorkspaceStatus\x12\x1c\n\x18WORKSPACE_STATUS_UNKNOWN\x10\x00\x12!\n\x1dWORKSPACE_STATUS_INITIALIZING\x10\x01\x12\x1c\n\x18WORKSPACE_STATUS_STARTED\x10\x02\x12\x1c\n\x18WORKSPACE_STATUS_STOPPED\x10\x03\x12\x1a\n\x16WORKSPACE_STATUS_ERROR\x10\x04*\xe0\x07\n\x0f\x44\x65ploymentState\x12\x1c\n\x18\x44\x45PLOYMENT_STATE_UNKNOWN\x10\x00\x12)\n%DEPLOYMENT_STATE_INITIATING_WORKSPACE\x10\x01\x12\x30\n,DEPLOYMENT_STATE_DEPLOYING_REQUEST_WORKSPACE\x10\x02\x12\x33\n/DEPLOYMENT_STATE_FAILED_EVENT_COMMAND_WORKSPACE\x10\x03\x12)\n%DEPLOYMENT_STATE_DELETION_IN_PROGRESS\x10\x04\x12\x33\n/DEPLOYMENT_STATE_RESUME_EVENT_COMMAND_WORKSPACE\x10\x05\x12\x31\n-DEPLOYMENT_STATE_STOP_EVENT_COMMAND_WORKSPACE\x10\x06\x12(\n$DEPLOYMENT_STATE_STOPPING_DEPLOYMENT\x10\x07\x12\'\n#DEPLOYMENT_STATE_STOPPED_DEPLOYMENT\x10\x08\x12 \n\x1c\x44\x45PLOYMENT_STATE_INITIALIZED\x10\t\x12\x1a\n\x16\x44\x45PLOYMENT_STATE_READY\x10\n\x12%\n!DEPLOYMENT_STATE_CONTAINERS_READY\x10\x0b\x12\"\n\x1e\x44\x45PLOYMENT_STATE_POD_SCHEDULED\x10\x0c\x12\"\n\x1e\x44\x45PLOYMENT_STATE_UNSCHEDULABLE\x10\r\x12)\n%DEPLOYMENT_STATE_CONTAINERS_NOT_READY\x10\x0e\x12\x1e\n\x1a\x44\x45PLOYMENT_STATE_AVAILABLE\x10\x0f\x12 \n\x1c\x44\x45PLOYMENT_STATE_PROGRESSING\x10\x10\x12$\n DEPLOYMENT_STATE_REPLICA_FAILURE\x10\x11\x12/\n+DEPLOYMENT_STATE_MINIMUM_REPLICAS_AVAILABLE\x10\x12\x12\x31\n-DEPLOYMENT_STATE_MINIMUM_REPLICAS_UNAVAILABLE\x10\x13\x12\x30\n,DEPLOYMENT_STATE_PROGRESS_DEAD_LINE_EXCEEDED\x10\x14\x12.\n*DEPLOYMENT_STATE_NEW_REPLICA_SET_AVAILABLE\x10\x15\x12\x31\n-DEPLOYMENT_STATE_IDLE_TIME_STOPPED_DEPLOYMENT\x10\x16\x42\x42\n$com.qwak.ai.management.workspace.apiP\x01Z\x18qwak/workspace;workspaceb\x06proto3')
20
-
21
- _globals = globals()
22
- _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
23
- _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.workspace.workspace_pb2', _globals)
24
- if _descriptor._USE_C_DESCRIPTORS == False:
25
-
26
- DESCRIPTOR._options = None
27
- DESCRIPTOR._serialized_options = b'\n$com.qwak.ai.management.workspace.apiP\001Z\030qwak/workspace;workspace'
28
- _WORKSPACEDEPLOYMENTSNAPSHOT.fields_by_name['deployment']._options = None
29
- _WORKSPACEDEPLOYMENTSNAPSHOT.fields_by_name['deployment']._serialized_options = b'\030\001'
30
- _WORKSPACEDEPLOYMENTSNAPSHOT.fields_by_name['replicas']._options = None
31
- _WORKSPACEDEPLOYMENTSNAPSHOT.fields_by_name['replicas']._serialized_options = b'\030\001'
32
- _globals['_TYPE']._serialized_start=1802
33
- _globals['_TYPE']._serialized_end=1826
34
- _globals['_WORKSPACESTATUS']._serialized_start=1829
35
- _globals['_WORKSPACESTATUS']._serialized_end=1999
36
- _globals['_DEPLOYMENTSTATE']._serialized_start=2002
37
- _globals['_DEPLOYMENTSTATE']._serialized_end=2994
38
- _globals['_WORKSPACE']._serialized_start=193
39
- _globals['_WORKSPACE']._serialized_end=773
40
- _globals['_WORKSPACESPEC']._serialized_start=776
41
- _globals['_WORKSPACESPEC']._serialized_end=987
42
- _globals['_WORKSPACEDEPLOYMENTSNAPSHOT']._serialized_start=990
43
- _globals['_WORKSPACEDEPLOYMENTSNAPSHOT']._serialized_end=1275
44
- _globals['_UPDATEWORKSPACESTATUSFROMEDGEDETAILS']._serialized_start=1278
45
- _globals['_UPDATEWORKSPACESTATUSFROMEDGEDETAILS']._serialized_end=1455
46
- _globals['_WORKSPACEIMAGE']._serialized_start=1457
47
- _globals['_WORKSPACEIMAGE']._serialized_end=1546
48
- _globals['_DEFAULTWORKSPACEDETAILS']._serialized_start=1549
49
- _globals['_DEFAULTWORKSPACEDETAILS']._serialized_end=1800
50
- # @@protoc_insertion_point(module_scope)