frogml-core 0.0.113__py3-none-any.whl → 0.0.115__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 (137) hide show
  1. frogml_core/__init__.py +1 -1
  2. frogml_core/clients/administration/authentication/client.py +2 -2
  3. frogml_core/clients/batch_job_management/client.py +4 -4
  4. frogml_core/clients/build_orchestrator/build_model_request_getter.py +6 -6
  5. frogml_core/clients/build_orchestrator/client.py +12 -12
  6. frogml_core/clients/build_orchestrator/internal_client.py +10 -10
  7. frogml_core/frogml_client/build_api_helpers/build_api_steps.py +3 -3
  8. frogml_core/inner/build_logic/constants/upload_tag.py +7 -7
  9. frogml_core/inner/build_logic/interface/context_interface.py +1 -1
  10. frogml_core/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/strategy.py +4 -4
  11. frogml_core/inner/build_logic/phases/phase_010_fetch_model/set_version_step.py +3 -3
  12. frogml_core/inner/build_logic/phases/phase_020_remote_register_frogml_build/start_remote_build_step.py +3 -3
  13. frogml_core/inner/build_logic/phases/phase_020_remote_register_frogml_build/upload_step.py +11 -9
  14. frogml_core/inner/build_logic/tools/ignore_files.py +3 -3
  15. frogml_core/inner/di_configuration/__init__.py +0 -6
  16. frogml_core/model/adapters/__init__.py +1 -1
  17. frogml_core/model/analytics_logging.py +1 -1
  18. frogml_core/model/tools/adapters/input.py +6 -6
  19. frogml_core/model/tools/adapters/output.py +8 -8
  20. frogml_core/model/tools/run_model_locally.py +2 -2
  21. frogml_core/model/utils/feature_utils.py +1 -1
  22. {frogml_core-0.0.113.dist-info → frogml_core-0.0.115.dist-info}/METADATA +1 -1
  23. {frogml_core-0.0.113.dist-info → frogml_core-0.0.115.dist-info}/RECORD +30 -137
  24. frogml_services_mock/mocks/analytics_api.py +6 -6
  25. frogml_services_mock/mocks/ecosystem_service_api.py +2 -2
  26. frogml_services_mock/mocks/frogml_mocks.py +0 -11
  27. frogml_services_mock/services_mock.py +4 -52
  28. frogml_storage/__init__.py +1 -1
  29. frogml_core/clients/prompt_manager/__init__.py +0 -0
  30. frogml_core/clients/prompt_manager/model_descriptor_mapper.py +0 -196
  31. frogml_core/clients/prompt_manager/prompt_manager_client.py +0 -190
  32. frogml_core/clients/prompt_manager/prompt_proto_mapper.py +0 -264
  33. frogml_core/clients/vector_store/__init__.py +0 -2
  34. frogml_core/clients/vector_store/management_client.py +0 -127
  35. frogml_core/clients/vector_store/serving_client.py +0 -157
  36. frogml_core/clients/workspace_manager/__init__.py +0 -1
  37. frogml_core/clients/workspace_manager/client.py +0 -224
  38. frogml_core/llmops/__init__.py +0 -0
  39. frogml_core/llmops/generation/__init__.py +0 -0
  40. frogml_core/llmops/generation/_steaming.py +0 -78
  41. frogml_core/llmops/generation/base.py +0 -5
  42. frogml_core/llmops/generation/chat/__init__.py +0 -0
  43. frogml_core/llmops/generation/chat/openai/LICENSE.txt +0 -201
  44. frogml_core/llmops/generation/chat/openai/types/__init__.py +0 -0
  45. frogml_core/llmops/generation/chat/openai/types/chat/__init__.py +0 -0
  46. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion.py +0 -88
  47. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_assistant_message_param.py +0 -65
  48. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_chunk.py +0 -153
  49. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_content_part_text_param.py +0 -28
  50. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_function_call_option_param.py +0 -25
  51. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_function_message_param.py +0 -33
  52. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_message.py +0 -56
  53. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_message_param.py +0 -34
  54. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_message_tool_call.py +0 -46
  55. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_message_tool_call_param.py +0 -44
  56. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_named_tool_choice_param.py +0 -32
  57. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_role.py +0 -20
  58. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_system_message_param.py +0 -35
  59. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_token_logprob.py +0 -71
  60. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_tool_choice_option_param.py +0 -28
  61. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_tool_message_param.py +0 -31
  62. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_tool_param.py +0 -29
  63. frogml_core/llmops/generation/chat/openai/types/chat/chat_completion_user_message_param.py +0 -35
  64. frogml_core/llmops/generation/chat/openai/types/chat/completion_create_params.py +0 -279
  65. frogml_core/llmops/generation/chat/openai/types/completion_choice.py +0 -47
  66. frogml_core/llmops/generation/chat/openai/types/completion_create_params.py +0 -209
  67. frogml_core/llmops/generation/chat/openai/types/completion_usage.py +0 -30
  68. frogml_core/llmops/generation/chat/openai/types/model.py +0 -35
  69. frogml_core/llmops/generation/chat/openai/types/shared/__init__.py +0 -3
  70. frogml_core/llmops/generation/chat/openai/types/shared/error_object.py +0 -27
  71. frogml_core/llmops/generation/chat/openai/types/shared/function_definition.py +0 -49
  72. frogml_core/llmops/generation/chat/openai/types/shared/function_parameters.py +0 -20
  73. frogml_core/llmops/generation/chat/openai/types/shared_params/__init__.py +0 -2
  74. frogml_core/llmops/generation/chat/openai/types/shared_params/function_definition.py +0 -49
  75. frogml_core/llmops/generation/chat/openai/types/shared_params/function_parameters.py +0 -22
  76. frogml_core/llmops/generation/streaming.py +0 -26
  77. frogml_core/llmops/model/__init__.py +0 -0
  78. frogml_core/llmops/model/descriptor.py +0 -40
  79. frogml_core/llmops/prompt/__init__.py +0 -0
  80. frogml_core/llmops/prompt/base.py +0 -136
  81. frogml_core/llmops/prompt/chat/__init__.py +0 -0
  82. frogml_core/llmops/prompt/chat/message.py +0 -24
  83. frogml_core/llmops/prompt/chat/template.py +0 -113
  84. frogml_core/llmops/prompt/chat/value.py +0 -10
  85. frogml_core/llmops/prompt/manager.py +0 -138
  86. frogml_core/llmops/prompt/template.py +0 -24
  87. frogml_core/llmops/prompt/value.py +0 -14
  88. frogml_core/llmops/provider/__init__.py +0 -0
  89. frogml_core/llmops/provider/chat.py +0 -44
  90. frogml_core/llmops/provider/openai/__init__.py +0 -0
  91. frogml_core/llmops/provider/openai/client.py +0 -126
  92. frogml_core/llmops/provider/openai/provider.py +0 -93
  93. frogml_core/vector_store/__init__.py +0 -4
  94. frogml_core/vector_store/client.py +0 -151
  95. frogml_core/vector_store/collection.py +0 -429
  96. frogml_core/vector_store/filters.py +0 -359
  97. frogml_core/vector_store/inference_client.py +0 -105
  98. frogml_core/vector_store/rest_helpers.py +0 -81
  99. frogml_core/vector_store/utils/__init__.py +0 -0
  100. frogml_core/vector_store/utils/filter_utils.py +0 -23
  101. frogml_core/vector_store/utils/upsert_utils.py +0 -218
  102. frogml_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2.py +0 -77
  103. frogml_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2.pyi +0 -417
  104. frogml_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2_grpc.py +0 -441
  105. frogml_proto/qwak/prompt/v1/prompt/prompt_pb2.py +0 -69
  106. frogml_proto/qwak/prompt/v1/prompt/prompt_pb2.pyi +0 -415
  107. frogml_proto/qwak/prompt/v1/prompt/prompt_pb2_grpc.py +0 -4
  108. frogml_proto/qwak/vectors/v1/collection/collection_pb2.py +0 -46
  109. frogml_proto/qwak/vectors/v1/collection/collection_pb2.pyi +0 -287
  110. frogml_proto/qwak/vectors/v1/collection/collection_pb2_grpc.py +0 -4
  111. frogml_proto/qwak/vectors/v1/collection/collection_service_pb2.py +0 -60
  112. frogml_proto/qwak/vectors/v1/collection/collection_service_pb2.pyi +0 -258
  113. frogml_proto/qwak/vectors/v1/collection/collection_service_pb2_grpc.py +0 -304
  114. frogml_proto/qwak/vectors/v1/collection/event/collection_event_pb2.py +0 -28
  115. frogml_proto/qwak/vectors/v1/collection/event/collection_event_pb2.pyi +0 -41
  116. frogml_proto/qwak/vectors/v1/collection/event/collection_event_pb2_grpc.py +0 -4
  117. frogml_proto/qwak/vectors/v1/filters_pb2.py +0 -52
  118. frogml_proto/qwak/vectors/v1/filters_pb2.pyi +0 -297
  119. frogml_proto/qwak/vectors/v1/filters_pb2_grpc.py +0 -4
  120. frogml_proto/qwak/vectors/v1/vector_pb2.py +0 -38
  121. frogml_proto/qwak/vectors/v1/vector_pb2.pyi +0 -142
  122. frogml_proto/qwak/vectors/v1/vector_pb2_grpc.py +0 -4
  123. frogml_proto/qwak/vectors/v1/vector_service_pb2.py +0 -53
  124. frogml_proto/qwak/vectors/v1/vector_service_pb2.pyi +0 -243
  125. frogml_proto/qwak/vectors/v1/vector_service_pb2_grpc.py +0 -201
  126. frogml_proto/qwak/workspace/workspace_pb2.py +0 -50
  127. frogml_proto/qwak/workspace/workspace_pb2.pyi +0 -331
  128. frogml_proto/qwak/workspace/workspace_pb2_grpc.py +0 -4
  129. frogml_proto/qwak/workspace/workspace_service_pb2.py +0 -84
  130. frogml_proto/qwak/workspace/workspace_service_pb2.pyi +0 -393
  131. frogml_proto/qwak/workspace/workspace_service_pb2_grpc.py +0 -507
  132. frogml_services_mock/mocks/prompt_manager_service.py +0 -281
  133. frogml_services_mock/mocks/vector_serving_api.py +0 -159
  134. frogml_services_mock/mocks/vectors_management_api.py +0 -97
  135. frogml_services_mock/mocks/workspace_manager_service_mock.py +0 -202
  136. /frogml_core/model/adapters/output_adapters/{qwak_with_default_fallback.py → frogml_with_default_fallback.py} +0 -0
  137. {frogml_core-0.0.113.dist-info → frogml_core-0.0.115.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)