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,415 +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 google.protobuf.timestamp_pb2
11
- import frogml_proto.qwak.model_descriptor.open_ai_descriptor_pb2
12
- import sys
13
-
14
- if sys.version_info >= (3, 8):
15
- import typing as typing_extensions
16
- else:
17
- import typing_extensions
18
-
19
- DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
20
-
21
- class Prompt(google.protobuf.message.Message):
22
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
23
-
24
- NAME_FIELD_NUMBER: builtins.int
25
- DEFAULT_VERSION_DEFINITION_FIELD_NUMBER: builtins.int
26
- METADATA_FIELD_NUMBER: builtins.int
27
- PROMPT_SPEC_FIELD_NUMBER: builtins.int
28
- name: builtins.str
29
- """Prompt name"""
30
- @property
31
- def default_version_definition(self) -> global___PromptVersionDefinition:
32
- """Prompt default version definition"""
33
- @property
34
- def metadata(self) -> global___PromptMetadata:
35
- """Prompt metadata"""
36
- @property
37
- def prompt_spec(self) -> global___PromptSpec:
38
- """Prompt spec"""
39
- def __init__(
40
- self,
41
- *,
42
- name: builtins.str = ...,
43
- default_version_definition: global___PromptVersionDefinition | None = ...,
44
- metadata: global___PromptMetadata | None = ...,
45
- prompt_spec: global___PromptSpec | None = ...,
46
- ) -> None: ...
47
- def HasField(self, field_name: typing_extensions.Literal["default_version_definition", b"default_version_definition", "metadata", b"metadata", "prompt_spec", b"prompt_spec"]) -> builtins.bool: ...
48
- def ClearField(self, field_name: typing_extensions.Literal["default_version_definition", b"default_version_definition", "metadata", b"metadata", "name", b"name", "prompt_spec", b"prompt_spec"]) -> None: ...
49
-
50
- global___Prompt = Prompt
51
-
52
- class PromptVersion(google.protobuf.message.Message):
53
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
54
-
55
- PROMPT_NAME_FIELD_NUMBER: builtins.int
56
- PROMPT_VERSION_DEFINITION_FIELD_NUMBER: builtins.int
57
- prompt_name: builtins.str
58
- """Prompt name"""
59
- @property
60
- def prompt_version_definition(self) -> global___PromptVersionDefinition:
61
- """Prompt version definition"""
62
- def __init__(
63
- self,
64
- *,
65
- prompt_name: builtins.str = ...,
66
- prompt_version_definition: global___PromptVersionDefinition | None = ...,
67
- ) -> None: ...
68
- def HasField(self, field_name: typing_extensions.Literal["prompt_version_definition", b"prompt_version_definition"]) -> builtins.bool: ...
69
- def ClearField(self, field_name: typing_extensions.Literal["prompt_name", b"prompt_name", "prompt_version_definition", b"prompt_version_definition"]) -> None: ...
70
-
71
- global___PromptVersion = PromptVersion
72
-
73
- class PromptVersionDefinition(google.protobuf.message.Message):
74
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
75
-
76
- VERSION_NUMBER_FIELD_NUMBER: builtins.int
77
- VERSION_SPEC_FIELD_NUMBER: builtins.int
78
- VERSION_METADATA_FIELD_NUMBER: builtins.int
79
- version_number: builtins.int
80
- """Prompt version number"""
81
- @property
82
- def version_spec(self) -> global___PromptVersionSpec:
83
- """Prompt spec"""
84
- @property
85
- def version_metadata(self) -> global___PromptVersionMetadata:
86
- """Prompt version metadata"""
87
- def __init__(
88
- self,
89
- *,
90
- version_number: builtins.int = ...,
91
- version_spec: global___PromptVersionSpec | None = ...,
92
- version_metadata: global___PromptVersionMetadata | None = ...,
93
- ) -> None: ...
94
- def HasField(self, field_name: typing_extensions.Literal["version_metadata", b"version_metadata", "version_spec", b"version_spec"]) -> builtins.bool: ...
95
- def ClearField(self, field_name: typing_extensions.Literal["version_metadata", b"version_metadata", "version_number", b"version_number", "version_spec", b"version_spec"]) -> None: ...
96
-
97
- global___PromptVersionDefinition = PromptVersionDefinition
98
-
99
- class PromptBrief(google.protobuf.message.Message):
100
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
101
-
102
- NAME_FIELD_NUMBER: builtins.int
103
- VERSION_BRIEFS_FIELD_NUMBER: builtins.int
104
- name: builtins.str
105
- """Prompt name"""
106
- @property
107
- def version_briefs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___PromptVersionBrief]:
108
- """Prompt version briefs"""
109
- def __init__(
110
- self,
111
- *,
112
- name: builtins.str = ...,
113
- version_briefs: collections.abc.Iterable[global___PromptVersionBrief] | None = ...,
114
- ) -> None: ...
115
- def ClearField(self, field_name: typing_extensions.Literal["name", b"name", "version_briefs", b"version_briefs"]) -> None: ...
116
-
117
- global___PromptBrief = PromptBrief
118
-
119
- class PromptVersionBrief(google.protobuf.message.Message):
120
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
121
-
122
- VERSION_NUMBER_FIELD_NUMBER: builtins.int
123
- MODEL_PROVIDER_BRIEF_FIELD_NUMBER: builtins.int
124
- version_number: builtins.int
125
- """Prompt version number"""
126
- @property
127
- def model_provider_brief(self) -> global___ModelProviderBrief: ...
128
- def __init__(
129
- self,
130
- *,
131
- version_number: builtins.int = ...,
132
- model_provider_brief: global___ModelProviderBrief | None = ...,
133
- ) -> None: ...
134
- def HasField(self, field_name: typing_extensions.Literal["model_provider_brief", b"model_provider_brief", "optional_model_provider_brief", b"optional_model_provider_brief"]) -> builtins.bool: ...
135
- def ClearField(self, field_name: typing_extensions.Literal["model_provider_brief", b"model_provider_brief", "optional_model_provider_brief", b"optional_model_provider_brief", "version_number", b"version_number"]) -> None: ...
136
- def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_model_provider_brief", b"optional_model_provider_brief"]) -> typing_extensions.Literal["model_provider_brief"] | None: ...
137
-
138
- global___PromptVersionBrief = PromptVersionBrief
139
-
140
- class ModelProviderBrief(google.protobuf.message.Message):
141
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
142
-
143
- MODEL_PROVIDER_CASE_NUMBER_FIELD_NUMBER: builtins.int
144
- model_provider_case_number: builtins.int
145
- """model provider case number as derived from `PromptModelDescriptor.ModelProviderCase`"""
146
- def __init__(
147
- self,
148
- *,
149
- model_provider_case_number: builtins.int = ...,
150
- ) -> None: ...
151
- def ClearField(self, field_name: typing_extensions.Literal["model_provider_case_number", b"model_provider_case_number"]) -> None: ...
152
-
153
- global___ModelProviderBrief = ModelProviderBrief
154
-
155
- class PromptSpec(google.protobuf.message.Message):
156
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
157
-
158
- DESCRIPTION_FIELD_NUMBER: builtins.int
159
- description: builtins.str
160
- """Prompt description"""
161
- def __init__(
162
- self,
163
- *,
164
- description: builtins.str = ...,
165
- ) -> None: ...
166
- def ClearField(self, field_name: typing_extensions.Literal["description", b"description"]) -> None: ...
167
-
168
- global___PromptSpec = PromptSpec
169
-
170
- class PromptVersionSpec(google.protobuf.message.Message):
171
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
172
-
173
- DESCRIPTION_FIELD_NUMBER: builtins.int
174
- PROMPT_TEMPLATE_FIELD_NUMBER: builtins.int
175
- MODEL_DESCRIPTOR_FIELD_NUMBER: builtins.int
176
- description: builtins.str
177
- """Prompt version description"""
178
- @property
179
- def prompt_template(self) -> global___PromptTemplate:
180
- """The template of the prompt"""
181
- @property
182
- def model_descriptor(self) -> global___PromptModelDescriptor: ...
183
- def __init__(
184
- self,
185
- *,
186
- description: builtins.str = ...,
187
- prompt_template: global___PromptTemplate | None = ...,
188
- model_descriptor: global___PromptModelDescriptor | None = ...,
189
- ) -> None: ...
190
- def HasField(self, field_name: typing_extensions.Literal["model_descriptor", b"model_descriptor", "optional_model_descriptor", b"optional_model_descriptor", "prompt_template", b"prompt_template"]) -> builtins.bool: ...
191
- def ClearField(self, field_name: typing_extensions.Literal["description", b"description", "model_descriptor", b"model_descriptor", "optional_model_descriptor", b"optional_model_descriptor", "prompt_template", b"prompt_template"]) -> None: ...
192
- def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_model_descriptor", b"optional_model_descriptor"]) -> typing_extensions.Literal["model_descriptor"] | None: ...
193
-
194
- global___PromptVersionSpec = PromptVersionSpec
195
-
196
- class PromptTemplate(google.protobuf.message.Message):
197
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
198
-
199
- CHAT_PROMPT_TEMPLATE_FIELD_NUMBER: builtins.int
200
- @property
201
- def chat_prompt_template(self) -> global___ChatPromptTemplate:
202
- """Chat prompt template"""
203
- def __init__(
204
- self,
205
- *,
206
- chat_prompt_template: global___ChatPromptTemplate | None = ...,
207
- ) -> None: ...
208
- def HasField(self, field_name: typing_extensions.Literal["chat_prompt_template", b"chat_prompt_template", "type", b"type"]) -> builtins.bool: ...
209
- def ClearField(self, field_name: typing_extensions.Literal["chat_prompt_template", b"chat_prompt_template", "type", b"type"]) -> None: ...
210
- def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["chat_prompt_template"] | None: ...
211
-
212
- global___PromptTemplate = PromptTemplate
213
-
214
- class ChatPromptTemplate(google.protobuf.message.Message):
215
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
216
-
217
- CHAT_MESSAGES_FIELD_NUMBER: builtins.int
218
- @property
219
- def chat_messages(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ChatMessage]:
220
- """Templated chat message"""
221
- def __init__(
222
- self,
223
- *,
224
- chat_messages: collections.abc.Iterable[global___ChatMessage] | None = ...,
225
- ) -> None: ...
226
- def ClearField(self, field_name: typing_extensions.Literal["chat_messages", b"chat_messages"]) -> None: ...
227
-
228
- global___ChatPromptTemplate = ChatPromptTemplate
229
-
230
- class ChatMessage(google.protobuf.message.Message):
231
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
232
-
233
- ROLE_FIELD_NUMBER: builtins.int
234
- TEMPLATE_FIELD_NUMBER: builtins.int
235
- @property
236
- def role(self) -> global___PromptMessageRole:
237
- """Role"""
238
- @property
239
- def template(self) -> global___ChatMessageTemplate:
240
- """Chat message template"""
241
- def __init__(
242
- self,
243
- *,
244
- role: global___PromptMessageRole | None = ...,
245
- template: global___ChatMessageTemplate | None = ...,
246
- ) -> None: ...
247
- def HasField(self, field_name: typing_extensions.Literal["role", b"role", "template", b"template"]) -> builtins.bool: ...
248
- def ClearField(self, field_name: typing_extensions.Literal["role", b"role", "template", b"template"]) -> None: ...
249
-
250
- global___ChatMessage = ChatMessage
251
-
252
- class ChatMessageTemplate(google.protobuf.message.Message):
253
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
254
-
255
- TEXT_TEMPLATE_FIELD_NUMBER: builtins.int
256
- @property
257
- def text_template(self) -> global___TextTemplate:
258
- """Text message template"""
259
- def __init__(
260
- self,
261
- *,
262
- text_template: global___TextTemplate | None = ...,
263
- ) -> None: ...
264
- def HasField(self, field_name: typing_extensions.Literal["text_template", b"text_template", "type", b"type"]) -> builtins.bool: ...
265
- def ClearField(self, field_name: typing_extensions.Literal["text_template", b"text_template", "type", b"type"]) -> None: ...
266
- def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["text_template"] | None: ...
267
-
268
- global___ChatMessageTemplate = ChatMessageTemplate
269
-
270
- class TextTemplate(google.protobuf.message.Message):
271
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
272
-
273
- TEMPLATE_FIELD_NUMBER: builtins.int
274
- template: builtins.str
275
- """Text template"""
276
- def __init__(
277
- self,
278
- *,
279
- template: builtins.str = ...,
280
- ) -> None: ...
281
- def ClearField(self, field_name: typing_extensions.Literal["template", b"template"]) -> None: ...
282
-
283
- global___TextTemplate = TextTemplate
284
-
285
- class PromptMessageRole(google.protobuf.message.Message):
286
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
287
-
288
- HUMAN_ROLE_FIELD_NUMBER: builtins.int
289
- SYSTEM_ROLE_FIELD_NUMBER: builtins.int
290
- AI_ROLE_FIELD_NUMBER: builtins.int
291
- @property
292
- def human_role(self) -> global___HumanPromptMessageRole:
293
- """Human role"""
294
- @property
295
- def system_role(self) -> global___SystemPromptMessageRole:
296
- """System role"""
297
- @property
298
- def ai_role(self) -> global___AIPromptMessageRole:
299
- """AI role"""
300
- def __init__(
301
- self,
302
- *,
303
- human_role: global___HumanPromptMessageRole | None = ...,
304
- system_role: global___SystemPromptMessageRole | None = ...,
305
- ai_role: global___AIPromptMessageRole | None = ...,
306
- ) -> None: ...
307
- def HasField(self, field_name: typing_extensions.Literal["ai_role", b"ai_role", "human_role", b"human_role", "role", b"role", "system_role", b"system_role"]) -> builtins.bool: ...
308
- def ClearField(self, field_name: typing_extensions.Literal["ai_role", b"ai_role", "human_role", b"human_role", "role", b"role", "system_role", b"system_role"]) -> None: ...
309
- def WhichOneof(self, oneof_group: typing_extensions.Literal["role", b"role"]) -> typing_extensions.Literal["human_role", "system_role", "ai_role"] | None: ...
310
-
311
- global___PromptMessageRole = PromptMessageRole
312
-
313
- class HumanPromptMessageRole(google.protobuf.message.Message):
314
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
315
-
316
- def __init__(
317
- self,
318
- ) -> None: ...
319
-
320
- global___HumanPromptMessageRole = HumanPromptMessageRole
321
-
322
- class SystemPromptMessageRole(google.protobuf.message.Message):
323
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
324
-
325
- def __init__(
326
- self,
327
- ) -> None: ...
328
-
329
- global___SystemPromptMessageRole = SystemPromptMessageRole
330
-
331
- class AIPromptMessageRole(google.protobuf.message.Message):
332
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
333
-
334
- def __init__(
335
- self,
336
- ) -> None: ...
337
-
338
- global___AIPromptMessageRole = AIPromptMessageRole
339
-
340
- class PromptModelDescriptor(google.protobuf.message.Message):
341
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
342
-
343
- OPEN_AI_PROVIDER_FIELD_NUMBER: builtins.int
344
- @property
345
- def open_ai_provider(self) -> global___PromptOpenAIProvider:
346
- """OpenAI provider"""
347
- def __init__(
348
- self,
349
- *,
350
- open_ai_provider: global___PromptOpenAIProvider | None = ...,
351
- ) -> None: ...
352
- def HasField(self, field_name: typing_extensions.Literal["model_provider", b"model_provider", "open_ai_provider", b"open_ai_provider"]) -> builtins.bool: ...
353
- def ClearField(self, field_name: typing_extensions.Literal["model_provider", b"model_provider", "open_ai_provider", b"open_ai_provider"]) -> None: ...
354
- def WhichOneof(self, oneof_group: typing_extensions.Literal["model_provider", b"model_provider"]) -> typing_extensions.Literal["open_ai_provider"] | None: ...
355
-
356
- global___PromptModelDescriptor = PromptModelDescriptor
357
-
358
- class PromptOpenAIProvider(google.protobuf.message.Message):
359
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
360
-
361
- OPEN_AI_MODEL_DESCRIPTOR_FIELD_NUMBER: builtins.int
362
- @property
363
- def open_ai_model_descriptor(self) -> frogml_proto.qwak.model_descriptor.open_ai_descriptor_pb2.OpenAIModelDescriptor:
364
- """OpenAI model descriptor"""
365
- def __init__(
366
- self,
367
- *,
368
- open_ai_model_descriptor: frogml_proto.qwak.model_descriptor.open_ai_descriptor_pb2.OpenAIModelDescriptor | None = ...,
369
- ) -> None: ...
370
- def HasField(self, field_name: typing_extensions.Literal["open_ai_model_descriptor", b"open_ai_model_descriptor"]) -> builtins.bool: ...
371
- def ClearField(self, field_name: typing_extensions.Literal["open_ai_model_descriptor", b"open_ai_model_descriptor"]) -> None: ...
372
-
373
- global___PromptOpenAIProvider = PromptOpenAIProvider
374
-
375
- class PromptMetadata(google.protobuf.message.Message):
376
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
377
-
378
- CREATED_AT_FIELD_NUMBER: builtins.int
379
- CREATED_BY_FIELD_NUMBER: builtins.int
380
- @property
381
- def created_at(self) -> google.protobuf.timestamp_pb2.Timestamp:
382
- """When the prompt was created"""
383
- created_by: builtins.str
384
- """User who created the prompt"""
385
- def __init__(
386
- self,
387
- *,
388
- created_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
389
- created_by: builtins.str = ...,
390
- ) -> None: ...
391
- def HasField(self, field_name: typing_extensions.Literal["created_at", b"created_at"]) -> builtins.bool: ...
392
- def ClearField(self, field_name: typing_extensions.Literal["created_at", b"created_at", "created_by", b"created_by"]) -> None: ...
393
-
394
- global___PromptMetadata = PromptMetadata
395
-
396
- class PromptVersionMetadata(google.protobuf.message.Message):
397
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
398
-
399
- CREATED_AT_FIELD_NUMBER: builtins.int
400
- CREATED_BY_FIELD_NUMBER: builtins.int
401
- @property
402
- def created_at(self) -> google.protobuf.timestamp_pb2.Timestamp:
403
- """When the prompt version was created"""
404
- created_by: builtins.str
405
- """User who created the prompt version"""
406
- def __init__(
407
- self,
408
- *,
409
- created_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
410
- created_by: builtins.str = ...,
411
- ) -> None: ...
412
- def HasField(self, field_name: typing_extensions.Literal["created_at", b"created_at"]) -> builtins.bool: ...
413
- def ClearField(self, field_name: typing_extensions.Literal["created_at", b"created_at", "created_by", b"created_by"]) -> None: ...
414
-
415
- global___PromptVersionMetadata = PromptVersionMetadata
@@ -1,4 +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
-
@@ -1,46 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- # Generated by the protocol buffer compiler. DO NOT EDIT!
3
- # source: frogml_proto.qwak.vectors/v1/collection/collection.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
-
16
-
17
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+qwak/vectors/v1/collection/collection.proto\x12\x1fqwak.vector.store.v1.collection\x1a\x1fgoogle/protobuf/timestamp.proto\"\xa2\x01\n\x0cQwakMetadata\x12.\n\ncreated_at\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ncreated_by\x18\x02 \x01(\t\x12\x34\n\x10last_modified_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x04 \x01(\t\"?\n\x1aVectorCollectionVectorizer\x12\x19\n\x0fqwak_model_name\x18\x01 \x01(\tH\x00\x42\x06\n\x04type\"\x85\x02\n\x14VectorCollectionSpec\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12O\n\nvectorizer\x18\x03 \x01(\x0b\x32;.qwak.vector.store.v1.collection.VectorCollectionVectorizer\x12G\n\x06metric\x18\x04 \x01(\x0e\x32\x37.qwak.vector.store.v1.collection.VectorCollectionMetric\x12\x11\n\tdimension\x18\x05 \x01(\x05\x12\x1d\n\x15multi_tenancy_enabled\x18\x06 \x01(\x08\"x\n\x1aVectorCollectionDefinition\x12\n\n\x02id\x18\x01 \x01(\t\x12N\n\x0f\x63ollection_spec\x18\x02 \x01(\x0b\x32\x35.qwak.vector.store.v1.collection.VectorCollectionSpec\"\xed\x01\n\x10VectorCollection\x12?\n\x08metadata\x18\x01 \x01(\x0b\x32-.qwak.vector.store.v1.collection.QwakMetadata\x12O\n\ndefinition\x18\x02 \x01(\x0b\x32;.qwak.vector.store.v1.collection.VectorCollectionDefinition\x12G\n\x06status\x18\x03 \x01(\x0e\x32\x37.qwak.vector.store.v1.collection.VectorCollectionStatus\"Z\n\x08Property\x12\x0c\n\x04name\x18\x01 \x01(\t\x12@\n\tdata_type\x18\x02 \x01(\x0e\x32-.qwak.vector.store.v1.collection.PropertyType\"h\n\x12\x43ollectionMetadata\x12\x13\n\x0bnum_vectors\x18\x01 \x01(\x03\x12=\n\nproperties\x18\x02 \x03(\x0b\x32).qwak.vector.store.v1.collection.Property*\x8f\x02\n\x16VectorCollectionStatus\x12\x1d\n\x19\x43OLLECTION_STATUS_INVALID\x10\x00\x12&\n\"COLLECTION_STATUS_CREATE_REQUESTED\x10\x01\x12\x1d\n\x19\x43OLLECTION_STATUS_CREATED\x10\x02\x12#\n\x1f\x43OLLECTION_STATUS_CREATE_FAILED\x10\x03\x12&\n\"COLLECTION_STATUS_DELETE_REQUESTED\x10\x04\x12#\n\x1f\x43OLLECTION_STATUS_DELETE_FAILED\x10\x05\x12\x1d\n\x19\x43OLLECTION_STATUS_DELETED\x10\x06*\xd3\x01\n\x16VectorCollectionMetric\x12\x1d\n\x19\x43OLLECTION_METRIC_INVALID\x10\x00\x12 \n\x1c\x43OLLECTION_METRIC_L2_SQUARED\x10\x01\x12\x1c\n\x18\x43OLLECTION_METRIC_COSINE\x10\x02\x12!\n\x1d\x43OLLECTION_METRIC_DOT_PRODUCT\x10\x03\x12\x18\n\x14\x43OLLECTION_METRIC_L1\x10\x04\x12\x1d\n\x19\x43OLLECTION_METRIC_HAMMING\x10\x05*\xac\x01\n\x0cPropertyType\x12\x19\n\x15PROPERTY_TYPE_INVALID\x10\x00\x12\x18\n\x14PROPERTY_TYPE_STRING\x10\x01\x12\x15\n\x11PROPERTY_TYPE_INT\x10\x02\x12\x18\n\x14PROPERTY_TYPE_DOUBLE\x10\x03\x12\x19\n\x15PROPERTY_TYPE_BOOLEAN\x10\x04\x12\x1b\n\x17PROPERTY_TYPE_TIMESTAMP\x10\x05\x42K\n!com.qwak.ai.vectors.v1.collectionP\x01Z$qwak/vector_management/v1;collectionb\x06proto3')
18
-
19
- _globals = globals()
20
- _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
21
- _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.vectors.v1.collection.collection_pb2', _globals)
22
- if _descriptor._USE_C_DESCRIPTORS == False:
23
-
24
- DESCRIPTOR._options = None
25
- DESCRIPTOR._serialized_options = b'\n!com.qwak.ai.vectors.v1.collectionP\001Z$qwak/vector_management/v1;collection'
26
- _globals['_VECTORCOLLECTIONSTATUS']._serialized_start=1168
27
- _globals['_VECTORCOLLECTIONSTATUS']._serialized_end=1439
28
- _globals['_VECTORCOLLECTIONMETRIC']._serialized_start=1442
29
- _globals['_VECTORCOLLECTIONMETRIC']._serialized_end=1653
30
- _globals['_PROPERTYTYPE']._serialized_start=1656
31
- _globals['_PROPERTYTYPE']._serialized_end=1828
32
- _globals['_QWAKMETADATA']._serialized_start=114
33
- _globals['_QWAKMETADATA']._serialized_end=276
34
- _globals['_VECTORCOLLECTIONVECTORIZER']._serialized_start=278
35
- _globals['_VECTORCOLLECTIONVECTORIZER']._serialized_end=341
36
- _globals['_VECTORCOLLECTIONSPEC']._serialized_start=344
37
- _globals['_VECTORCOLLECTIONSPEC']._serialized_end=605
38
- _globals['_VECTORCOLLECTIONDEFINITION']._serialized_start=607
39
- _globals['_VECTORCOLLECTIONDEFINITION']._serialized_end=727
40
- _globals['_VECTORCOLLECTION']._serialized_start=730
41
- _globals['_VECTORCOLLECTION']._serialized_end=967
42
- _globals['_PROPERTY']._serialized_start=969
43
- _globals['_PROPERTY']._serialized_end=1059
44
- _globals['_COLLECTIONMETADATA']._serialized_start=1061
45
- _globals['_COLLECTIONMETADATA']._serialized_end=1165
46
- # @@protoc_insertion_point(module_scope)