weave-python 0.28.2__py3-none-any.whl → 0.30.4__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 (104) hide show
  1. weave/weaveapi/auth/v1/service_pb2.py +184 -88
  2. weave/weaveapi/auth/v1/service_pb2.pyi +1288 -1134
  3. weave/weaveapi/auth/v1/service_pb2_grpc.pyi +248 -335
  4. weave/weaveapi/auth/v1/session_pb2.py +12 -4
  5. weave/weaveapi/auth/v1/session_pb2.pyi +109 -89
  6. weave/weaveapi/auth/v1/session_pb2_grpc.pyi +11 -10
  7. weave/weaveapi/auth/v1/usage_pb2.py +16 -12
  8. weave/weaveapi/auth/v1/usage_pb2.pyi +251 -197
  9. weave/weaveapi/auth/v1/usage_pb2_grpc.pyi +11 -10
  10. weave/weaveapi/auth/v1/user_pb2.py +58 -16
  11. weave/weaveapi/auth/v1/user_pb2.pyi +588 -505
  12. weave/weaveapi/auth/v1/user_pb2_grpc.pyi +11 -10
  13. weave/weaveapi/generate/v1/configuration_pb2.py +29 -17
  14. weave/weaveapi/generate/v1/configuration_pb2.pyi +277 -241
  15. weave/weaveapi/generate/v1/configuration_pb2_grpc.pyi +11 -10
  16. weave/weaveapi/generate/v1/generate_pb2.py +30 -4
  17. weave/weaveapi/generate/v1/generate_pb2.pyi +138 -40
  18. weave/weaveapi/generate/v1/generate_pb2_grpc.pyi +11 -10
  19. weave/weaveapi/generate/v1/service_pb2.py +50 -20
  20. weave/weaveapi/generate/v1/service_pb2.pyi +264 -135
  21. weave/weaveapi/generate/v1/service_pb2_grpc.py +47 -0
  22. weave/weaveapi/generate/v1/service_pb2_grpc.pyi +89 -78
  23. weave/weaveapi/llmx/v1/architecture_pb2.py +3 -3
  24. weave/weaveapi/llmx/v1/architecture_pb2.pyi +664 -555
  25. weave/weaveapi/llmx/v1/architecture_pb2_grpc.pyi +11 -10
  26. weave/weaveapi/llmx/v1/capabilities_pb2.py +86 -54
  27. weave/weaveapi/llmx/v1/capabilities_pb2.pyi +967 -852
  28. weave/weaveapi/llmx/v1/capabilities_pb2_grpc.pyi +11 -10
  29. weave/weaveapi/llmx/v1/model_pb2.py +54 -18
  30. weave/weaveapi/llmx/v1/model_pb2.pyi +533 -455
  31. weave/weaveapi/llmx/v1/model_pb2_grpc.pyi +11 -10
  32. weave/weaveapi/llmx/v1/pricing_pb2.py +22 -18
  33. weave/weaveapi/llmx/v1/pricing_pb2.pyi +194 -172
  34. weave/weaveapi/llmx/v1/pricing_pb2_grpc.pyi +11 -10
  35. weave/weaveapi/llmx/v1/provider_pb2.py +3 -3
  36. weave/weaveapi/llmx/v1/provider_pb2.pyi +84 -59
  37. weave/weaveapi/llmx/v1/provider_pb2_grpc.pyi +11 -10
  38. weave/weaveapi/llmx/v1/service_pb2.py +224 -86
  39. weave/weaveapi/llmx/v1/service_pb2.pyi +1651 -1403
  40. weave/weaveapi/llmx/v1/service_pb2_grpc.pyi +160 -203
  41. weave/weaveapi/mcpregistry/v1/server_pb2.py +14 -8
  42. weave/weaveapi/mcpregistry/v1/server_pb2.pyi +143 -121
  43. weave/weaveapi/mcpregistry/v1/server_pb2_grpc.pyi +11 -10
  44. weave/weaveapi/mcpregistry/v1/service_pb2.py +47 -27
  45. weave/weaveapi/mcpregistry/v1/service_pb2.pyi +132 -122
  46. weave/weaveapi/mcpregistry/v1/service_pb2_grpc.pyi +87 -112
  47. weave/weaveapi/payment/v1/invoice_pb2.py +36 -10
  48. weave/weaveapi/payment/v1/invoice_pb2.pyi +352 -291
  49. weave/weaveapi/payment/v1/invoice_pb2_grpc.pyi +11 -10
  50. weave/weaveapi/payment/v1/service_pb2.py +256 -90
  51. weave/weaveapi/payment/v1/service_pb2.pyi +1381 -1242
  52. weave/weaveapi/payment/v1/service_pb2_grpc.pyi +229 -319
  53. weave/weaveapi/payment/v1/subscription_pb2.py +97 -21
  54. weave/weaveapi/payment/v1/subscription_pb2.pyi +727 -611
  55. weave/weaveapi/payment/v1/subscription_pb2_grpc.pyi +11 -10
  56. weave/weaveapi/storage/v1/auth_pb2.py +3 -3
  57. weave/weaveapi/storage/v1/auth_pb2.pyi +42 -29
  58. weave/weaveapi/storage/v1/auth_pb2_grpc.pyi +11 -10
  59. weave/weaveapi/storage/v1/nosql_database_pb2.py +45 -21
  60. weave/weaveapi/storage/v1/nosql_database_pb2.pyi +438 -372
  61. weave/weaveapi/storage/v1/nosql_database_pb2_grpc.pyi +11 -10
  62. weave/weaveapi/storage/v1/object_store_pb2.py +25 -11
  63. weave/weaveapi/storage/v1/object_store_pb2.pyi +203 -187
  64. weave/weaveapi/storage/v1/object_store_pb2_grpc.pyi +11 -10
  65. weave/weaveapi/storage/v1/service_pb2.py +94 -34
  66. weave/weaveapi/storage/v1/service_pb2.pyi +414 -357
  67. weave/weaveapi/storage/v1/service_pb2_grpc.pyi +88 -107
  68. weave/weaveapi/storage/v1/sql_database_pb2.py +37 -21
  69. weave/weaveapi/storage/v1/sql_database_pb2.pyi +481 -400
  70. weave/weaveapi/storage/v1/sql_database_pb2_grpc.pyi +11 -10
  71. weave/weaveapi/storage/v1/storage_pb2.py +18 -4
  72. weave/weaveapi/storage/v1/storage_pb2.pyi +79 -70
  73. weave/weaveapi/storage/v1/storage_pb2_grpc.pyi +11 -10
  74. weave/weaveapi/synthesize/v1/dataset_pb2.py +10 -8
  75. weave/weaveapi/synthesize/v1/dataset_pb2.pyi +158 -128
  76. weave/weaveapi/synthesize/v1/dataset_pb2_grpc.pyi +11 -10
  77. weave/weaveapi/synthesize/v1/inline_data_pb2.py +4 -4
  78. weave/weaveapi/synthesize/v1/inline_data_pb2.pyi +31 -27
  79. weave/weaveapi/synthesize/v1/inline_data_pb2_grpc.pyi +11 -10
  80. weave/weaveapi/synthesize/v1/relationship_pb2.py +17 -9
  81. weave/weaveapi/synthesize/v1/relationship_pb2.pyi +67 -64
  82. weave/weaveapi/synthesize/v1/relationship_pb2_grpc.pyi +11 -10
  83. weave/weaveapi/synthesize/v1/service_pb2.py +40 -22
  84. weave/weaveapi/synthesize/v1/service_pb2.pyi +202 -168
  85. weave/weaveapi/synthesize/v1/service_pb2_grpc.pyi +67 -79
  86. weave/weaveapi/synthesize/v1/training_pb2.py +17 -11
  87. weave/weaveapi/synthesize/v1/training_pb2.pyi +119 -106
  88. weave/weaveapi/synthesize/v1/training_pb2_grpc.pyi +11 -10
  89. weave/weavesql/llmxdb/capabilities.py +487 -0
  90. weave/weavesql/llmxdb/changes.py +297 -0
  91. weave/weavesql/llmxdb/models.py +594 -0
  92. weave/weavesql/llmxdb/providers.py +348 -0
  93. weave/weavesql/llmxdb/scraper_runs.py +287 -0
  94. weave/weavesql/llmxdb/search.py +721 -0
  95. weave/weavesql/weavedb/dataset.py +75 -0
  96. weave/weavesql/weavedb/models.py +135 -0
  97. weave/weavesql/weavedb/relationships.py +72 -0
  98. weave/weavesql/weavedb/storage.py +113 -0
  99. weave/weavesql/weavedb/synthesizer.py +107 -0
  100. {weave_python-0.28.2.dist-info → weave_python-0.30.4.dist-info}/METADATA +3 -3
  101. weave_python-0.30.4.dist-info/RECORD +131 -0
  102. {weave_python-0.28.2.dist-info → weave_python-0.30.4.dist-info}/WHEEL +1 -1
  103. weave_python-0.28.2.dist-info/RECORD +0 -120
  104. {weave_python-0.28.2.dist-info → weave_python-0.30.4.dist-info}/licenses/LICENSE +0 -0
@@ -3,149 +3,153 @@
3
3
  isort:skip_file
4
4
  """
5
5
 
6
- import builtins
7
- import collections.abc
8
- import google.protobuf.descriptor
9
- import google.protobuf.internal.containers
10
- import google.protobuf.message
11
- import google.protobuf.timestamp_pb2
12
- import typing
13
- import weaveapi.llmx.v1.architecture_pb2
14
- import weaveapi.llmx.v1.capabilities_pb2
15
- import weaveapi.llmx.v1.pricing_pb2
16
-
17
- DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
18
-
19
- @typing.final
20
- class Model(google.protobuf.message.Message):
6
+ from collections import abc as _abc
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import message as _message
9
+ from google.protobuf import timestamp_pb2 as _timestamp_pb2
10
+ from google.protobuf.internal import containers as _containers
11
+ from weaveapi.llmx.v1 import architecture_pb2 as _architecture_pb2
12
+ from weaveapi.llmx.v1 import capabilities_pb2 as _capabilities_pb2
13
+ from weaveapi.llmx.v1 import pricing_pb2 as _pricing_pb2
14
+ import builtins as _builtins
15
+ import sys
16
+ import typing as _typing
17
+
18
+ if sys.version_info >= (3, 10):
19
+ from typing import TypeAlias as _TypeAlias
20
+ else:
21
+ from typing_extensions import TypeAlias as _TypeAlias
22
+
23
+ DESCRIPTOR: _descriptor.FileDescriptor
24
+
25
+ @_typing.final
26
+ class Model(_message.Message):
21
27
  """Model represents a complete AI model with all its metadata, capabilities, and pricing.
22
28
  This is the primary entity for model discovery and comparison.
23
29
  """
24
30
 
25
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
26
-
27
- ID_FIELD_NUMBER: builtins.int
28
- PROVIDER_ID_FIELD_NUMBER: builtins.int
29
- PROVIDER_SLUG_FIELD_NUMBER: builtins.int
30
- PROVIDER_NAME_FIELD_NUMBER: builtins.int
31
- MODEL_ID_FIELD_NUMBER: builtins.int
32
- SLUG_FIELD_NUMBER: builtins.int
33
- NAME_FIELD_NUMBER: builtins.int
34
- DISPLAY_NAME_FIELD_NUMBER: builtins.int
35
- DESCRIPTION_FIELD_NUMBER: builtins.int
36
- VERSION_FIELD_NUMBER: builtins.int
37
- RELEASE_DATE_FIELD_NUMBER: builtins.int
38
- TRAINING_DATA_CUTOFF_FIELD_NUMBER: builtins.int
39
- DEPRECATION_DATE_FIELD_NUMBER: builtins.int
40
- CAPABILITIES_FIELD_NUMBER: builtins.int
41
- CLASSIFICATION_FIELD_NUMBER: builtins.int
42
- PERFORMANCE_FIELD_NUMBER: builtins.int
43
- TOKENS_FIELD_NUMBER: builtins.int
44
- PRICING_FIELD_NUMBER: builtins.int
45
- CONFIGURATION_FIELD_NUMBER: builtins.int
46
- API_DETAILS_FIELD_NUMBER: builtins.int
47
- AVAILABILITY_FIELD_NUMBER: builtins.int
48
- ARCHITECTURE_FIELD_NUMBER: builtins.int
49
- TRAINING_FIELD_NUMBER: builtins.int
50
- SAFETY_FIELD_NUMBER: builtins.int
51
- LICENSING_FIELD_NUMBER: builtins.int
52
- TECHNICAL_SPECS_FIELD_NUMBER: builtins.int
53
- LAST_SCRAPED_AT_FIELD_NUMBER: builtins.int
54
- DATA_SOURCES_FIELD_NUMBER: builtins.int
55
- IS_ACTIVE_FIELD_NUMBER: builtins.int
56
- IS_DEPRECATED_FIELD_NUMBER: builtins.int
57
- REPLACEMENT_MODEL_ID_FIELD_NUMBER: builtins.int
58
- id: builtins.str
31
+ DESCRIPTOR: _descriptor.Descriptor
32
+
33
+ ID_FIELD_NUMBER: _builtins.int
34
+ PROVIDER_ID_FIELD_NUMBER: _builtins.int
35
+ PROVIDER_SLUG_FIELD_NUMBER: _builtins.int
36
+ PROVIDER_NAME_FIELD_NUMBER: _builtins.int
37
+ MODEL_ID_FIELD_NUMBER: _builtins.int
38
+ SLUG_FIELD_NUMBER: _builtins.int
39
+ NAME_FIELD_NUMBER: _builtins.int
40
+ DISPLAY_NAME_FIELD_NUMBER: _builtins.int
41
+ DESCRIPTION_FIELD_NUMBER: _builtins.int
42
+ VERSION_FIELD_NUMBER: _builtins.int
43
+ RELEASE_DATE_FIELD_NUMBER: _builtins.int
44
+ TRAINING_DATA_CUTOFF_FIELD_NUMBER: _builtins.int
45
+ DEPRECATION_DATE_FIELD_NUMBER: _builtins.int
46
+ CAPABILITIES_FIELD_NUMBER: _builtins.int
47
+ CLASSIFICATION_FIELD_NUMBER: _builtins.int
48
+ PERFORMANCE_FIELD_NUMBER: _builtins.int
49
+ TOKENS_FIELD_NUMBER: _builtins.int
50
+ PRICING_FIELD_NUMBER: _builtins.int
51
+ CONFIGURATION_FIELD_NUMBER: _builtins.int
52
+ API_DETAILS_FIELD_NUMBER: _builtins.int
53
+ AVAILABILITY_FIELD_NUMBER: _builtins.int
54
+ ARCHITECTURE_FIELD_NUMBER: _builtins.int
55
+ TRAINING_FIELD_NUMBER: _builtins.int
56
+ SAFETY_FIELD_NUMBER: _builtins.int
57
+ LICENSING_FIELD_NUMBER: _builtins.int
58
+ TECHNICAL_SPECS_FIELD_NUMBER: _builtins.int
59
+ LAST_SCRAPED_AT_FIELD_NUMBER: _builtins.int
60
+ DATA_SOURCES_FIELD_NUMBER: _builtins.int
61
+ IS_ACTIVE_FIELD_NUMBER: _builtins.int
62
+ IS_DEPRECATED_FIELD_NUMBER: _builtins.int
63
+ REPLACEMENT_MODEL_ID_FIELD_NUMBER: _builtins.int
64
+ id: _builtins.str
59
65
  """Unique internal identifier for this model record in our database.
60
66
  Example: "mdl_01234567-89ab-cdef-0123-456789abcdef"
61
67
  Generated by our system, not provider-specific
62
68
  """
63
- provider_id: builtins.str
69
+ provider_id: _builtins.str
64
70
  """Provider's unique identifier in our system.
65
71
  Example: "pvd_openai_2024", "pvd_anthropic_2024"
66
72
  Links to Provider entity
67
73
  """
68
- provider_slug: builtins.str
74
+ provider_slug: _builtins.str
69
75
  """URL-friendly provider identifier for routing.
70
76
  Examples: "openai", "anthropic", "google", "meta"
71
77
  Used in API paths like /models/openai/gpt-4
72
78
  """
73
- provider_name: builtins.str
79
+ provider_name: _builtins.str
74
80
  """Human-readable provider name for display.
75
81
  Examples: "OpenAI", "Anthropic", "Google AI", "Meta AI"
76
82
  """
77
- model_id: builtins.str
83
+ model_id: _builtins.str
78
84
  """Provider's specific model identifier as they define it.
79
85
  Examples: "gpt-4-turbo-2024-04-09", "claude-3-opus-20240229"
80
86
  This is what you use in their API calls
81
87
  """
82
- slug: builtins.str
88
+ slug: _builtins.str
83
89
  """Combined provider/model identifier for unique referencing.
84
90
  Format: "{provider_slug}/{model_id}"
85
91
  Examples: "openai/gpt-4", "anthropic/claude-3-opus"
86
92
  """
87
- name: builtins.str
93
+ name: _builtins.str
88
94
  """Model's full technical name from the provider.
89
95
  Examples: "gpt-4-turbo-2024-04-09", "claude-3-opus-20240229"
90
96
  May include version dates or technical details
91
97
  """
92
- display_name: builtins.str
98
+ display_name: _builtins.str
93
99
  """User-friendly display name for UI presentation.
94
100
  Examples: "GPT-4 Turbo", "Claude 3 Opus", "Gemini Pro"
95
101
  Simplified version of technical name
96
102
  """
97
- description: builtins.str
103
+ description: _builtins.str
98
104
  """Human-readable description of the model's purpose and capabilities.
99
105
  Example: "Most capable GPT-4 model, optimized for complex tasks requiring
100
106
  advanced reasoning, coding, and creative writing. Supports vision, function
101
107
  calling, and JSON mode."
102
108
  """
103
- version: builtins.str
109
+ version: _builtins.str
104
110
  """Model version string from provider.
105
111
  Examples: "2024-04-09", "v1.0", "preview-0125"
106
112
  May be date-based or semantic versioning
107
113
  """
108
- is_active: builtins.bool
114
+ is_active: _builtins.bool
109
115
  """Whether the model is currently available for use.
110
116
  false if model is down, in maintenance, or discontinued
111
117
  """
112
- is_deprecated: builtins.bool
118
+ is_deprecated: _builtins.bool
113
119
  """Whether the model has been officially deprecated.
114
120
  true once deprecation_date has passed or provider announces EOL
115
121
  """
116
- replacement_model_id: builtins.str
122
+ replacement_model_id: _builtins.str
117
123
  """If deprecated, the recommended replacement model's ID.
118
124
  Example: "mdl_gpt4_turbo_2024" replacing "mdl_gpt4_2023"
119
125
  Helps with migration paths
120
126
  """
121
- @property
122
- def release_date(self) -> google.protobuf.timestamp_pb2.Timestamp:
127
+ @_builtins.property
128
+ def release_date(self) -> _timestamp_pb2.Timestamp:
123
129
  """=== Temporal Information ===
124
130
 
125
131
  When the model was publicly released by the provider.
126
132
  Example: 2024-04-09 for GPT-4 Turbo April release
127
133
  """
128
134
 
129
- @property
130
- def training_data_cutoff(self) -> google.protobuf.timestamp_pb2.Timestamp:
135
+ @_builtins.property
136
+ def training_data_cutoff(self) -> _timestamp_pb2.Timestamp:
131
137
  """Last date of data used in training (knowledge cutoff).
132
138
  Example: 2023-12-01 means model knows events up to this date
133
139
  Important for factual queries and current events
134
140
  """
135
141
 
136
- @property
137
- def deprecation_date(self) -> google.protobuf.timestamp_pb2.Timestamp:
142
+ @_builtins.property
143
+ def deprecation_date(self) -> _timestamp_pb2.Timestamp:
138
144
  """When the model will be/was deprecated (if scheduled).
139
145
  Example: 2025-01-15 for planned sunset
140
146
  Null if no deprecation planned
141
147
  """
142
148
 
143
- @property
149
+ @_builtins.property
144
150
  def capabilities(
145
151
  self,
146
- ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[
147
- weaveapi.llmx.v1.capabilities_pb2.Capability
148
- ]:
152
+ ) -> _containers.RepeatedCompositeFieldContainer[_capabilities_pb2.Capability]:
149
153
  """=== Core Properties ===
150
154
 
151
155
  List of specific capabilities this model supports.
@@ -153,84 +157,84 @@ class Model(google.protobuf.message.Message):
153
157
  See capabilities.proto for full enumeration
154
158
  """
155
159
 
156
- @property
157
- def classification(self) -> global___ModelClassification:
160
+ @_builtins.property
161
+ def classification(self) -> Global___ModelClassification:
158
162
  """Model type, architecture, and licensing classification.
159
163
  Categorizes the model for filtering and comparison
160
164
  """
161
165
 
162
- @property
163
- def performance(self) -> global___ModelPerformance:
166
+ @_builtins.property
167
+ def performance(self) -> Global___ModelPerformance:
164
168
  """Benchmark scores and performance metrics.
165
169
  Standardized scores for comparing model capabilities
166
170
  """
167
171
 
168
- @property
169
- def tokens(self) -> global___TokenInfo:
172
+ @_builtins.property
173
+ def tokens(self) -> Global___TokenInfo:
170
174
  """Token limits, processing speed, and tokenizer information.
171
175
  Critical for understanding model constraints
172
176
  """
173
177
 
174
- @property
175
- def pricing(self) -> weaveapi.llmx.v1.pricing_pb2.Pricing:
178
+ @_builtins.property
179
+ def pricing(self) -> _pricing_pb2.Pricing:
176
180
  """Complete pricing structure for all operations.
177
181
  Includes standard, batch, cached, and tool pricing
178
182
  """
179
183
 
180
- @property
181
- def configuration(self) -> global___Configuration:
184
+ @_builtins.property
185
+ def configuration(self) -> Global___Configuration:
182
186
  """Supported parameter ranges for generation.
183
187
  Temperature, top_p, and other sampling parameters
184
188
  """
185
189
 
186
- @property
187
- def api_details(self) -> global___APIDetails:
190
+ @_builtins.property
191
+ def api_details(self) -> Global___APIDetails:
188
192
  """=== Integration Details ===
189
193
 
190
194
  API endpoint information and rate limits.
191
195
  Everything needed to call the model programmatically
192
196
  """
193
197
 
194
- @property
195
- def availability(self) -> global___Availability:
198
+ @_builtins.property
199
+ def availability(self) -> Global___Availability:
196
200
  """Available regions and platforms.
197
201
  Where and how the model can be accessed
198
202
  """
199
203
 
200
- @property
201
- def architecture(self) -> weaveapi.llmx.v1.architecture_pb2.Architecture:
204
+ @_builtins.property
205
+ def architecture(self) -> _architecture_pb2.Architecture:
202
206
  """=== Technical Details ===
203
207
 
204
208
  Technical architecture information.
205
209
  Transformer type, layer count, attention mechanism, etc.
206
210
  """
207
211
 
208
- @property
209
- def training(self) -> weaveapi.llmx.v1.architecture_pb2.Training:
212
+ @_builtins.property
213
+ def training(self) -> _architecture_pb2.Training:
210
214
  """Training dataset and methodology information.
211
215
  How the model was trained and on what data
212
216
  """
213
217
 
214
- @property
215
- def safety(self) -> weaveapi.llmx.v1.architecture_pb2.Safety:
218
+ @_builtins.property
219
+ def safety(self) -> _architecture_pb2.Safety:
216
220
  """Built-in safety features and content filtering.
217
221
  Moderation capabilities and safety guardrails
218
222
  """
219
223
 
220
- @property
221
- def licensing(self) -> weaveapi.llmx.v1.architecture_pb2.Licensing:
224
+ @_builtins.property
225
+ def licensing(self) -> _architecture_pb2.Licensing:
222
226
  """License type and usage restrictions.
223
227
  Legal terms for using the model
224
228
  """
225
229
 
226
- @property
227
- def technical_specs(self) -> weaveapi.llmx.v1.architecture_pb2.TechnicalSpecs:
230
+ @_builtins.property
231
+ def technical_specs(self) -> _architecture_pb2.TechnicalSpecs:
228
232
  """Hardware requirements and technical details.
229
233
  Memory, compute requirements for self-hosting
230
234
  """
231
235
 
232
- @property
233
- def last_scraped_at(self) -> google.protobuf.timestamp_pb2.Timestamp:
236
+ @_builtins.property
237
+ def last_scraped_at(self) -> _timestamp_pb2.Timestamp:
234
238
  """=== Metadata ===
235
239
 
236
240
  When this model information was last updated in our system.
@@ -238,10 +242,8 @@ class Model(google.protobuf.message.Message):
238
242
  For tracking data freshness
239
243
  """
240
244
 
241
- @property
242
- def data_sources(
243
- self,
244
- ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
245
+ @_builtins.property
246
+ def data_sources(self) -> _containers.RepeatedScalarFieldContainer[_builtins.str]:
245
247
  """Sources where this model information was collected from.
246
248
  Examples: ["https://openai.com/pricing", "https://platform.openai.com/docs"]
247
249
  For data provenance and verification
@@ -250,266 +252,301 @@ class Model(google.protobuf.message.Message):
250
252
  def __init__(
251
253
  self,
252
254
  *,
253
- id: builtins.str = ...,
254
- provider_id: builtins.str = ...,
255
- provider_slug: builtins.str = ...,
256
- provider_name: builtins.str = ...,
257
- model_id: builtins.str = ...,
258
- slug: builtins.str = ...,
259
- name: builtins.str = ...,
260
- display_name: builtins.str = ...,
261
- description: builtins.str = ...,
262
- version: builtins.str = ...,
263
- release_date: google.protobuf.timestamp_pb2.Timestamp | None = ...,
264
- training_data_cutoff: google.protobuf.timestamp_pb2.Timestamp | None = ...,
265
- deprecation_date: google.protobuf.timestamp_pb2.Timestamp | None = ...,
266
- capabilities: collections.abc.Iterable[
267
- weaveapi.llmx.v1.capabilities_pb2.Capability
268
- ]
269
- | None = ...,
270
- classification: global___ModelClassification | None = ...,
271
- performance: global___ModelPerformance | None = ...,
272
- tokens: global___TokenInfo | None = ...,
273
- pricing: weaveapi.llmx.v1.pricing_pb2.Pricing | None = ...,
274
- configuration: global___Configuration | None = ...,
275
- api_details: global___APIDetails | None = ...,
276
- availability: global___Availability | None = ...,
277
- architecture: weaveapi.llmx.v1.architecture_pb2.Architecture | None = ...,
278
- training: weaveapi.llmx.v1.architecture_pb2.Training | None = ...,
279
- safety: weaveapi.llmx.v1.architecture_pb2.Safety | None = ...,
280
- licensing: weaveapi.llmx.v1.architecture_pb2.Licensing | None = ...,
281
- technical_specs: weaveapi.llmx.v1.architecture_pb2.TechnicalSpecs | None = ...,
282
- last_scraped_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
283
- data_sources: collections.abc.Iterable[builtins.str] | None = ...,
284
- is_active: builtins.bool = ...,
285
- is_deprecated: builtins.bool = ...,
286
- replacement_model_id: builtins.str = ...,
255
+ id: _builtins.str | None = ...,
256
+ provider_id: _builtins.str | None = ...,
257
+ provider_slug: _builtins.str | None = ...,
258
+ provider_name: _builtins.str | None = ...,
259
+ model_id: _builtins.str | None = ...,
260
+ slug: _builtins.str | None = ...,
261
+ name: _builtins.str | None = ...,
262
+ display_name: _builtins.str | None = ...,
263
+ description: _builtins.str | None = ...,
264
+ version: _builtins.str | None = ...,
265
+ release_date: _timestamp_pb2.Timestamp | None = ...,
266
+ training_data_cutoff: _timestamp_pb2.Timestamp | None = ...,
267
+ deprecation_date: _timestamp_pb2.Timestamp | None = ...,
268
+ capabilities: _abc.Iterable[_capabilities_pb2.Capability] | None = ...,
269
+ classification: Global___ModelClassification | None = ...,
270
+ performance: Global___ModelPerformance | None = ...,
271
+ tokens: Global___TokenInfo | None = ...,
272
+ pricing: _pricing_pb2.Pricing | None = ...,
273
+ configuration: Global___Configuration | None = ...,
274
+ api_details: Global___APIDetails | None = ...,
275
+ availability: Global___Availability | None = ...,
276
+ architecture: _architecture_pb2.Architecture | None = ...,
277
+ training: _architecture_pb2.Training | None = ...,
278
+ safety: _architecture_pb2.Safety | None = ...,
279
+ licensing: _architecture_pb2.Licensing | None = ...,
280
+ technical_specs: _architecture_pb2.TechnicalSpecs | None = ...,
281
+ last_scraped_at: _timestamp_pb2.Timestamp | None = ...,
282
+ data_sources: _abc.Iterable[_builtins.str] | None = ...,
283
+ is_active: _builtins.bool | None = ...,
284
+ is_deprecated: _builtins.bool | None = ...,
285
+ replacement_model_id: _builtins.str | None = ...,
287
286
  ) -> None: ...
288
- def HasField(
289
- self,
290
- field_name: typing.Literal[
291
- "api_details",
292
- b"api_details",
293
- "architecture",
294
- b"architecture",
295
- "availability",
296
- b"availability",
297
- "classification",
298
- b"classification",
299
- "configuration",
300
- b"configuration",
301
- "deprecation_date",
302
- b"deprecation_date",
303
- "last_scraped_at",
304
- b"last_scraped_at",
305
- "licensing",
306
- b"licensing",
307
- "performance",
308
- b"performance",
309
- "pricing",
310
- b"pricing",
311
- "release_date",
312
- b"release_date",
313
- "safety",
314
- b"safety",
315
- "technical_specs",
316
- b"technical_specs",
317
- "tokens",
318
- b"tokens",
319
- "training",
320
- b"training",
321
- "training_data_cutoff",
322
- b"training_data_cutoff",
323
- ],
324
- ) -> builtins.bool: ...
325
- def ClearField(
326
- self,
327
- field_name: typing.Literal[
328
- "api_details",
329
- b"api_details",
330
- "architecture",
331
- b"architecture",
332
- "availability",
333
- b"availability",
334
- "capabilities",
335
- b"capabilities",
336
- "classification",
337
- b"classification",
338
- "configuration",
339
- b"configuration",
340
- "data_sources",
341
- b"data_sources",
342
- "deprecation_date",
343
- b"deprecation_date",
344
- "description",
345
- b"description",
346
- "display_name",
347
- b"display_name",
348
- "id",
349
- b"id",
350
- "is_active",
351
- b"is_active",
352
- "is_deprecated",
353
- b"is_deprecated",
354
- "last_scraped_at",
355
- b"last_scraped_at",
356
- "licensing",
357
- b"licensing",
358
- "model_id",
359
- b"model_id",
360
- "name",
361
- b"name",
362
- "performance",
363
- b"performance",
364
- "pricing",
365
- b"pricing",
366
- "provider_id",
367
- b"provider_id",
368
- "provider_name",
369
- b"provider_name",
370
- "provider_slug",
371
- b"provider_slug",
372
- "release_date",
373
- b"release_date",
374
- "replacement_model_id",
375
- b"replacement_model_id",
376
- "safety",
377
- b"safety",
378
- "slug",
379
- b"slug",
380
- "technical_specs",
381
- b"technical_specs",
382
- "tokens",
383
- b"tokens",
384
- "training",
385
- b"training",
386
- "training_data_cutoff",
387
- b"training_data_cutoff",
388
- "version",
389
- b"version",
390
- ],
391
- ) -> None: ...
392
-
393
- global___Model = Model
394
-
395
- @typing.final
396
- class ModelClassification(google.protobuf.message.Message):
287
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
288
+ "api_details",
289
+ b"api_details",
290
+ "architecture",
291
+ b"architecture",
292
+ "availability",
293
+ b"availability",
294
+ "capabilities",
295
+ b"capabilities",
296
+ "classification",
297
+ b"classification",
298
+ "configuration",
299
+ b"configuration",
300
+ "data_sources",
301
+ b"data_sources",
302
+ "deprecation_date",
303
+ b"deprecation_date",
304
+ "description",
305
+ b"description",
306
+ "display_name",
307
+ b"display_name",
308
+ "id",
309
+ b"id",
310
+ "is_active",
311
+ b"is_active",
312
+ "is_deprecated",
313
+ b"is_deprecated",
314
+ "last_scraped_at",
315
+ b"last_scraped_at",
316
+ "licensing",
317
+ b"licensing",
318
+ "model_id",
319
+ b"model_id",
320
+ "name",
321
+ b"name",
322
+ "performance",
323
+ b"performance",
324
+ "pricing",
325
+ b"pricing",
326
+ "provider_id",
327
+ b"provider_id",
328
+ "provider_name",
329
+ b"provider_name",
330
+ "provider_slug",
331
+ b"provider_slug",
332
+ "release_date",
333
+ b"release_date",
334
+ "replacement_model_id",
335
+ b"replacement_model_id",
336
+ "safety",
337
+ b"safety",
338
+ "slug",
339
+ b"slug",
340
+ "technical_specs",
341
+ b"technical_specs",
342
+ "tokens",
343
+ b"tokens",
344
+ "training",
345
+ b"training",
346
+ "training_data_cutoff",
347
+ b"training_data_cutoff",
348
+ "version",
349
+ b"version",
350
+ ] # noqa: Y015
351
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
352
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
353
+ "api_details",
354
+ b"api_details",
355
+ "architecture",
356
+ b"architecture",
357
+ "availability",
358
+ b"availability",
359
+ "capabilities",
360
+ b"capabilities",
361
+ "classification",
362
+ b"classification",
363
+ "configuration",
364
+ b"configuration",
365
+ "data_sources",
366
+ b"data_sources",
367
+ "deprecation_date",
368
+ b"deprecation_date",
369
+ "description",
370
+ b"description",
371
+ "display_name",
372
+ b"display_name",
373
+ "id",
374
+ b"id",
375
+ "is_active",
376
+ b"is_active",
377
+ "is_deprecated",
378
+ b"is_deprecated",
379
+ "last_scraped_at",
380
+ b"last_scraped_at",
381
+ "licensing",
382
+ b"licensing",
383
+ "model_id",
384
+ b"model_id",
385
+ "name",
386
+ b"name",
387
+ "performance",
388
+ b"performance",
389
+ "pricing",
390
+ b"pricing",
391
+ "provider_id",
392
+ b"provider_id",
393
+ "provider_name",
394
+ b"provider_name",
395
+ "provider_slug",
396
+ b"provider_slug",
397
+ "release_date",
398
+ b"release_date",
399
+ "replacement_model_id",
400
+ b"replacement_model_id",
401
+ "safety",
402
+ b"safety",
403
+ "slug",
404
+ b"slug",
405
+ "technical_specs",
406
+ b"technical_specs",
407
+ "tokens",
408
+ b"tokens",
409
+ "training",
410
+ b"training",
411
+ "training_data_cutoff",
412
+ b"training_data_cutoff",
413
+ "version",
414
+ b"version",
415
+ ] # noqa: Y015
416
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
417
+
418
+ Global___Model: _TypeAlias = Model # noqa: Y015
419
+
420
+ @_typing.final
421
+ class ModelClassification(_message.Message):
397
422
  """ModelClassification categorizes models for filtering and comparison."""
398
423
 
399
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
424
+ DESCRIPTOR: _descriptor.Descriptor
400
425
 
401
- MODEL_TYPE_FIELD_NUMBER: builtins.int
402
- ARCHITECTURE_FIELD_NUMBER: builtins.int
403
- PARAMETER_COUNT_FIELD_NUMBER: builtins.int
404
- IS_OPEN_SOURCE_FIELD_NUMBER: builtins.int
405
- LICENSE_TYPE_FIELD_NUMBER: builtins.int
406
- model_type: builtins.str
426
+ MODEL_TYPE_FIELD_NUMBER: _builtins.int
427
+ ARCHITECTURE_FIELD_NUMBER: _builtins.int
428
+ PARAMETER_COUNT_FIELD_NUMBER: _builtins.int
429
+ IS_OPEN_SOURCE_FIELD_NUMBER: _builtins.int
430
+ LICENSE_TYPE_FIELD_NUMBER: _builtins.int
431
+ model_type: _builtins.str
407
432
  """Primary model type defining its training approach.
408
433
  Examples: "foundation", "fine-tuned", "instruct", "chat", "reasoning"
409
434
  "foundation" = base model, "instruct" = instruction-following
410
435
  """
411
- architecture: builtins.str
436
+ architecture: _builtins.str
412
437
  """Underlying model architecture family.
413
438
  Examples: "GPT", "LLaMA", "BERT", "T5", "Transformer", "Mamba"
414
439
  """
415
- parameter_count: builtins.int
440
+ parameter_count: _builtins.int
416
441
  """Approximate number of parameters in billions.
417
442
  Examples: 7 (7B), 70 (70B), 175 (175B for GPT-3)
418
443
  """
419
- is_open_source: builtins.bool
444
+ is_open_source: _builtins.bool
420
445
  """Whether model weights are publicly available.
421
446
  true for LLaMA, Mistral; false for GPT-4, Claude
422
447
  """
423
- license_type: builtins.str
448
+ license_type: _builtins.str
424
449
  """Software license if open source.
425
450
  Examples: "MIT", "Apache-2.0", "Custom", "Proprietary"
426
451
  """
427
452
  def __init__(
428
453
  self,
429
454
  *,
430
- model_type: builtins.str = ...,
431
- architecture: builtins.str = ...,
432
- parameter_count: builtins.int = ...,
433
- is_open_source: builtins.bool = ...,
434
- license_type: builtins.str = ...,
435
- ) -> None: ...
436
- def ClearField(
437
- self,
438
- field_name: typing.Literal[
439
- "architecture",
440
- b"architecture",
441
- "is_open_source",
442
- b"is_open_source",
443
- "license_type",
444
- b"license_type",
445
- "model_type",
446
- b"model_type",
447
- "parameter_count",
448
- b"parameter_count",
449
- ],
455
+ model_type: _builtins.str | None = ...,
456
+ architecture: _builtins.str | None = ...,
457
+ parameter_count: _builtins.int | None = ...,
458
+ is_open_source: _builtins.bool | None = ...,
459
+ license_type: _builtins.str | None = ...,
450
460
  ) -> None: ...
451
-
452
- global___ModelClassification = ModelClassification
453
-
454
- @typing.final
455
- class ModelPerformance(google.protobuf.message.Message):
461
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
462
+ "architecture",
463
+ b"architecture",
464
+ "is_open_source",
465
+ b"is_open_source",
466
+ "license_type",
467
+ b"license_type",
468
+ "model_type",
469
+ b"model_type",
470
+ "parameter_count",
471
+ b"parameter_count",
472
+ ] # noqa: Y015
473
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
474
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
475
+ "architecture",
476
+ b"architecture",
477
+ "is_open_source",
478
+ b"is_open_source",
479
+ "license_type",
480
+ b"license_type",
481
+ "model_type",
482
+ b"model_type",
483
+ "parameter_count",
484
+ b"parameter_count",
485
+ ] # noqa: Y015
486
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
487
+
488
+ Global___ModelClassification: _TypeAlias = ModelClassification # noqa: Y015
489
+
490
+ @_typing.final
491
+ class ModelPerformance(_message.Message):
456
492
  """ModelPerformance captures standardized benchmark scores."""
457
493
 
458
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
494
+ DESCRIPTOR: _descriptor.Descriptor
459
495
 
460
- @typing.final
461
- class BenchmarkScoresEntry(google.protobuf.message.Message):
462
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
496
+ @_typing.final
497
+ class BenchmarkScoresEntry(_message.Message):
498
+ DESCRIPTOR: _descriptor.Descriptor
463
499
 
464
- KEY_FIELD_NUMBER: builtins.int
465
- VALUE_FIELD_NUMBER: builtins.int
466
- key: builtins.str
467
- value: builtins.float
500
+ KEY_FIELD_NUMBER: _builtins.int
501
+ VALUE_FIELD_NUMBER: _builtins.int
502
+ key: _builtins.str
503
+ value: _builtins.float
468
504
  def __init__(
469
505
  self,
470
506
  *,
471
- key: builtins.str = ...,
472
- value: builtins.float = ...,
473
- ) -> None: ...
474
- def ClearField(
475
- self, field_name: typing.Literal["key", b"key", "value", b"value"]
507
+ key: _builtins.str | None = ...,
508
+ value: _builtins.float | None = ...,
476
509
  ) -> None: ...
477
-
478
- REASONING_SCORE_FIELD_NUMBER: builtins.int
479
- CODING_SCORE_FIELD_NUMBER: builtins.int
480
- CREATIVE_SCORE_FIELD_NUMBER: builtins.int
481
- FACTUAL_SCORE_FIELD_NUMBER: builtins.int
482
- MATH_SCORE_FIELD_NUMBER: builtins.int
483
- BENCHMARK_SCORES_FIELD_NUMBER: builtins.int
484
- reasoning_score: builtins.float
510
+ _HasFieldArgType: _TypeAlias = _typing.Literal["key", b"key", "value", b"value"] # noqa: Y015
511
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
512
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
513
+ "key", b"key", "value", b"value"
514
+ ] # noqa: Y015
515
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
516
+
517
+ REASONING_SCORE_FIELD_NUMBER: _builtins.int
518
+ CODING_SCORE_FIELD_NUMBER: _builtins.int
519
+ CREATIVE_SCORE_FIELD_NUMBER: _builtins.int
520
+ FACTUAL_SCORE_FIELD_NUMBER: _builtins.int
521
+ MATH_SCORE_FIELD_NUMBER: _builtins.int
522
+ BENCHMARK_SCORES_FIELD_NUMBER: _builtins.int
523
+ reasoning_score: _builtins.float
485
524
  """General reasoning ability score (0-10 scale).
486
525
  Based on benchmarks like ARC, HellaSwag, MMLU
487
526
  Example: 8.5 for strong reasoning models
488
527
  """
489
- coding_score: builtins.float
528
+ coding_score: _builtins.float
490
529
  """Programming and code generation score (0-10).
491
530
  Based on HumanEval, MBPP benchmarks
492
531
  Example: 9.2 for specialized coding models
493
532
  """
494
- creative_score: builtins.float
533
+ creative_score: _builtins.float
495
534
  """Creative writing and generation score (0-10).
496
535
  Based on human evaluations and creative benchmarks
497
536
  Example: 7.8 for models good at storytelling
498
537
  """
499
- factual_score: builtins.float
538
+ factual_score: _builtins.float
500
539
  """Factual accuracy and knowledge score (0-10).
501
540
  Based on TruthfulQA, factual benchmarks
502
541
  Example: 8.0 for models with good factual recall
503
542
  """
504
- math_score: builtins.float
543
+ math_score: _builtins.float
505
544
  """Mathematical reasoning score (0-10).
506
545
  Based on GSM8K, MATH benchmarks
507
546
  Example: 9.5 for models excelling at math
508
547
  """
509
- @property
510
- def benchmark_scores(
511
- self,
512
- ) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.float]:
548
+ @_builtins.property
549
+ def benchmark_scores(self) -> _containers.ScalarMap[_builtins.str, _builtins.float]:
513
550
  """Raw benchmark scores by name.
514
551
  Keys: "mmlu", "humaneval", "gsm8k", "truthfulqa", "arc", "hellaswag"
515
552
  Values: Actual benchmark scores (usually 0-100)
@@ -519,59 +556,71 @@ class ModelPerformance(google.protobuf.message.Message):
519
556
  def __init__(
520
557
  self,
521
558
  *,
522
- reasoning_score: builtins.float = ...,
523
- coding_score: builtins.float = ...,
524
- creative_score: builtins.float = ...,
525
- factual_score: builtins.float = ...,
526
- math_score: builtins.float = ...,
527
- benchmark_scores: collections.abc.Mapping[builtins.str, builtins.float]
528
- | None = ...,
559
+ reasoning_score: _builtins.float | None = ...,
560
+ coding_score: _builtins.float | None = ...,
561
+ creative_score: _builtins.float | None = ...,
562
+ factual_score: _builtins.float | None = ...,
563
+ math_score: _builtins.float | None = ...,
564
+ benchmark_scores: _abc.Mapping[_builtins.str, _builtins.float] | None = ...,
529
565
  ) -> None: ...
530
- def ClearField(
531
- self,
532
- field_name: typing.Literal[
533
- "benchmark_scores",
534
- b"benchmark_scores",
535
- "coding_score",
536
- b"coding_score",
537
- "creative_score",
538
- b"creative_score",
539
- "factual_score",
540
- b"factual_score",
541
- "math_score",
542
- b"math_score",
543
- "reasoning_score",
544
- b"reasoning_score",
545
- ],
546
- ) -> None: ...
547
-
548
- global___ModelPerformance = ModelPerformance
549
-
550
- @typing.final
551
- class TokenInfo(google.protobuf.message.Message):
566
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
567
+ "benchmark_scores",
568
+ b"benchmark_scores",
569
+ "coding_score",
570
+ b"coding_score",
571
+ "creative_score",
572
+ b"creative_score",
573
+ "factual_score",
574
+ b"factual_score",
575
+ "math_score",
576
+ b"math_score",
577
+ "reasoning_score",
578
+ b"reasoning_score",
579
+ ] # noqa: Y015
580
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
581
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
582
+ "benchmark_scores",
583
+ b"benchmark_scores",
584
+ "coding_score",
585
+ b"coding_score",
586
+ "creative_score",
587
+ b"creative_score",
588
+ "factual_score",
589
+ b"factual_score",
590
+ "math_score",
591
+ b"math_score",
592
+ "reasoning_score",
593
+ b"reasoning_score",
594
+ ] # noqa: Y015
595
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
596
+
597
+ Global___ModelPerformance: _TypeAlias = ModelPerformance # noqa: Y015
598
+
599
+ @_typing.final
600
+ class TokenInfo(_message.Message):
552
601
  """TokenInfo describes token limits and processing capabilities."""
553
602
 
554
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
603
+ DESCRIPTOR: _descriptor.Descriptor
555
604
 
556
- CONTEXT_WINDOW_FIELD_NUMBER: builtins.int
557
- MAX_OUTPUT_TOKENS_FIELD_NUMBER: builtins.int
558
- TOKENIZER_FIELD_NUMBER: builtins.int
559
- TOKENS_PER_SECOND_FIELD_NUMBER: builtins.int
560
- context_window: builtins.int
605
+ CONTEXT_WINDOW_FIELD_NUMBER: _builtins.int
606
+ MAX_OUTPUT_TOKENS_FIELD_NUMBER: _builtins.int
607
+ TOKENIZER_FIELD_NUMBER: _builtins.int
608
+ TOKENS_PER_SECOND_FIELD_NUMBER: _builtins.int
609
+ context_window: _builtins.int
561
610
  """Maximum tokens the model can process in a single request.
562
611
  Examples: 4096, 8192, 32768, 128000, 200000
563
612
  Includes both input and output tokens
564
613
  """
565
- max_output_tokens: builtins.int
614
+ max_output_tokens: _builtins.int
566
615
  """Maximum tokens that can be generated in response.
567
616
  Examples: 4096 for most models, 16384 for some
568
617
  May be less than context_window
569
618
  """
570
- tokenizer: builtins.str
619
+ tokenizer: _builtins.str
571
620
  """Tokenizer algorithm used.
572
621
  Examples: "cl100k_base" (GPT-4), "claude", "sentencepiece"
573
622
  """
574
- tokens_per_second: builtins.int
623
+ tokens_per_second: _builtins.int
575
624
  """Average generation speed in tokens per second.
576
625
  Example: 50 for typical API, 100+ for optimized inference
577
626
  May vary based on load and tier
@@ -579,156 +628,181 @@ class TokenInfo(google.protobuf.message.Message):
579
628
  def __init__(
580
629
  self,
581
630
  *,
582
- context_window: builtins.int = ...,
583
- max_output_tokens: builtins.int = ...,
584
- tokenizer: builtins.str = ...,
585
- tokens_per_second: builtins.int = ...,
631
+ context_window: _builtins.int | None = ...,
632
+ max_output_tokens: _builtins.int | None = ...,
633
+ tokenizer: _builtins.str | None = ...,
634
+ tokens_per_second: _builtins.int | None = ...,
586
635
  ) -> None: ...
587
- def ClearField(
588
- self,
589
- field_name: typing.Literal[
590
- "context_window",
591
- b"context_window",
592
- "max_output_tokens",
593
- b"max_output_tokens",
594
- "tokenizer",
595
- b"tokenizer",
596
- "tokens_per_second",
597
- b"tokens_per_second",
598
- ],
599
- ) -> None: ...
600
-
601
- global___TokenInfo = TokenInfo
602
-
603
- @typing.final
604
- class Configuration(google.protobuf.message.Message):
636
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
637
+ "context_window",
638
+ b"context_window",
639
+ "max_output_tokens",
640
+ b"max_output_tokens",
641
+ "tokenizer",
642
+ b"tokenizer",
643
+ "tokens_per_second",
644
+ b"tokens_per_second",
645
+ ] # noqa: Y015
646
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
647
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
648
+ "context_window",
649
+ b"context_window",
650
+ "max_output_tokens",
651
+ b"max_output_tokens",
652
+ "tokenizer",
653
+ b"tokenizer",
654
+ "tokens_per_second",
655
+ b"tokens_per_second",
656
+ ] # noqa: Y015
657
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
658
+
659
+ Global___TokenInfo: _TypeAlias = TokenInfo # noqa: Y015
660
+
661
+ @_typing.final
662
+ class Configuration(_message.Message):
605
663
  """Configuration defines supported generation parameters."""
606
664
 
607
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
665
+ DESCRIPTOR: _descriptor.Descriptor
608
666
 
609
- TEMPERATURE_MIN_FIELD_NUMBER: builtins.int
610
- TEMPERATURE_MAX_FIELD_NUMBER: builtins.int
611
- TEMPERATURE_DEFAULT_FIELD_NUMBER: builtins.int
612
- TOP_P_MIN_FIELD_NUMBER: builtins.int
613
- TOP_P_MAX_FIELD_NUMBER: builtins.int
614
- temperature_min: builtins.float
667
+ TEMPERATURE_MIN_FIELD_NUMBER: _builtins.int
668
+ TEMPERATURE_MAX_FIELD_NUMBER: _builtins.int
669
+ TEMPERATURE_DEFAULT_FIELD_NUMBER: _builtins.int
670
+ TOP_P_MIN_FIELD_NUMBER: _builtins.int
671
+ TOP_P_MAX_FIELD_NUMBER: _builtins.int
672
+ temperature_min: _builtins.float
615
673
  """Minimum allowed temperature value.
616
674
  Usually 0.0 for deterministic output
617
675
  """
618
- temperature_max: builtins.float
676
+ temperature_max: _builtins.float
619
677
  """Maximum allowed temperature value.
620
678
  Usually 1.0 or 2.0 for high randomness
621
679
  """
622
- temperature_default: builtins.float
680
+ temperature_default: _builtins.float
623
681
  """Default temperature if not specified.
624
682
  Typically 0.7 or 1.0
625
683
  """
626
- top_p_min: builtins.float
684
+ top_p_min: _builtins.float
627
685
  """Minimum allowed top_p (nucleus sampling).
628
686
  Usually 0.0 to disable
629
687
  """
630
- top_p_max: builtins.float
688
+ top_p_max: _builtins.float
631
689
  """Maximum allowed top_p value.
632
690
  Usually 1.0 for full vocabulary
633
691
  """
634
692
  def __init__(
635
693
  self,
636
694
  *,
637
- temperature_min: builtins.float = ...,
638
- temperature_max: builtins.float = ...,
639
- temperature_default: builtins.float = ...,
640
- top_p_min: builtins.float = ...,
641
- top_p_max: builtins.float = ...,
695
+ temperature_min: _builtins.float | None = ...,
696
+ temperature_max: _builtins.float | None = ...,
697
+ temperature_default: _builtins.float | None = ...,
698
+ top_p_min: _builtins.float | None = ...,
699
+ top_p_max: _builtins.float | None = ...,
642
700
  ) -> None: ...
643
- def ClearField(
644
- self,
645
- field_name: typing.Literal[
646
- "temperature_default",
647
- b"temperature_default",
648
- "temperature_max",
649
- b"temperature_max",
650
- "temperature_min",
651
- b"temperature_min",
652
- "top_p_max",
653
- b"top_p_max",
654
- "top_p_min",
655
- b"top_p_min",
656
- ],
657
- ) -> None: ...
658
-
659
- global___Configuration = Configuration
660
-
661
- @typing.final
662
- class APIDetails(google.protobuf.message.Message):
701
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
702
+ "temperature_default",
703
+ b"temperature_default",
704
+ "temperature_max",
705
+ b"temperature_max",
706
+ "temperature_min",
707
+ b"temperature_min",
708
+ "top_p_max",
709
+ b"top_p_max",
710
+ "top_p_min",
711
+ b"top_p_min",
712
+ ] # noqa: Y015
713
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
714
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
715
+ "temperature_default",
716
+ b"temperature_default",
717
+ "temperature_max",
718
+ b"temperature_max",
719
+ "temperature_min",
720
+ b"temperature_min",
721
+ "top_p_max",
722
+ b"top_p_max",
723
+ "top_p_min",
724
+ b"top_p_min",
725
+ ] # noqa: Y015
726
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
727
+
728
+ Global___Configuration: _TypeAlias = Configuration # noqa: Y015
729
+
730
+ @_typing.final
731
+ class APIDetails(_message.Message):
663
732
  """APIDetails provides integration information."""
664
733
 
665
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
734
+ DESCRIPTOR: _descriptor.Descriptor
666
735
 
667
- ENDPOINT_FIELD_NUMBER: builtins.int
668
- VERSION_FIELD_NUMBER: builtins.int
669
- RATE_LIMIT_RPM_FIELD_NUMBER: builtins.int
670
- RATE_LIMIT_TPM_FIELD_NUMBER: builtins.int
671
- endpoint: builtins.str
736
+ ENDPOINT_FIELD_NUMBER: _builtins.int
737
+ VERSION_FIELD_NUMBER: _builtins.int
738
+ RATE_LIMIT_RPM_FIELD_NUMBER: _builtins.int
739
+ RATE_LIMIT_TPM_FIELD_NUMBER: _builtins.int
740
+ endpoint: _builtins.str
672
741
  """Base API endpoint URL.
673
742
  Example: "https://api.openai.com/v1/chat/completions"
674
743
  """
675
- version: builtins.str
744
+ version: _builtins.str
676
745
  """API version identifier.
677
746
  Examples: "v1", "2024-02-01", "beta"
678
747
  """
679
- rate_limit_rpm: builtins.int
748
+ rate_limit_rpm: _builtins.int
680
749
  """Rate limit in requests per minute.
681
750
  Example: 500 for standard tier, 10000 for enterprise
682
751
  """
683
- rate_limit_tpm: builtins.int
752
+ rate_limit_tpm: _builtins.int
684
753
  """Rate limit in tokens per minute.
685
754
  Example: 90000 for GPT-4, 200000 for GPT-3.5
686
755
  """
687
756
  def __init__(
688
757
  self,
689
758
  *,
690
- endpoint: builtins.str = ...,
691
- version: builtins.str = ...,
692
- rate_limit_rpm: builtins.int = ...,
693
- rate_limit_tpm: builtins.int = ...,
759
+ endpoint: _builtins.str | None = ...,
760
+ version: _builtins.str | None = ...,
761
+ rate_limit_rpm: _builtins.int | None = ...,
762
+ rate_limit_tpm: _builtins.int | None = ...,
694
763
  ) -> None: ...
695
- def ClearField(
696
- self,
697
- field_name: typing.Literal[
698
- "endpoint",
699
- b"endpoint",
700
- "rate_limit_rpm",
701
- b"rate_limit_rpm",
702
- "rate_limit_tpm",
703
- b"rate_limit_tpm",
704
- "version",
705
- b"version",
706
- ],
707
- ) -> None: ...
708
-
709
- global___APIDetails = APIDetails
710
-
711
- @typing.final
712
- class Availability(google.protobuf.message.Message):
764
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
765
+ "endpoint",
766
+ b"endpoint",
767
+ "rate_limit_rpm",
768
+ b"rate_limit_rpm",
769
+ "rate_limit_tpm",
770
+ b"rate_limit_tpm",
771
+ "version",
772
+ b"version",
773
+ ] # noqa: Y015
774
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
775
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
776
+ "endpoint",
777
+ b"endpoint",
778
+ "rate_limit_rpm",
779
+ b"rate_limit_rpm",
780
+ "rate_limit_tpm",
781
+ b"rate_limit_tpm",
782
+ "version",
783
+ b"version",
784
+ ] # noqa: Y015
785
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
786
+
787
+ Global___APIDetails: _TypeAlias = APIDetails # noqa: Y015
788
+
789
+ @_typing.final
790
+ class Availability(_message.Message):
713
791
  """Availability describes where the model can be accessed."""
714
792
 
715
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
793
+ DESCRIPTOR: _descriptor.Descriptor
716
794
 
717
- REGIONS_FIELD_NUMBER: builtins.int
718
- PLATFORMS_FIELD_NUMBER: builtins.int
719
- @property
720
- def regions(
721
- self,
722
- ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
795
+ REGIONS_FIELD_NUMBER: _builtins.int
796
+ PLATFORMS_FIELD_NUMBER: _builtins.int
797
+ @_builtins.property
798
+ def regions(self) -> _containers.RepeatedScalarFieldContainer[_builtins.str]:
723
799
  """Geographic regions where available.
724
800
  Examples: ["us-east-1", "eu-west-1", "asia-pacific"]
725
801
  May use provider-specific region codes
726
802
  """
727
803
 
728
- @property
729
- def platforms(
730
- self,
731
- ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
804
+ @_builtins.property
805
+ def platforms(self) -> _containers.RepeatedScalarFieldContainer[_builtins.str]:
732
806
  """Platforms/services offering the model.
733
807
  Examples: ["api", "playground", "azure", "vertex-ai", "bedrock"]
734
808
  """
@@ -736,12 +810,16 @@ class Availability(google.protobuf.message.Message):
736
810
  def __init__(
737
811
  self,
738
812
  *,
739
- regions: collections.abc.Iterable[builtins.str] | None = ...,
740
- platforms: collections.abc.Iterable[builtins.str] | None = ...,
741
- ) -> None: ...
742
- def ClearField(
743
- self,
744
- field_name: typing.Literal["platforms", b"platforms", "regions", b"regions"],
813
+ regions: _abc.Iterable[_builtins.str] | None = ...,
814
+ platforms: _abc.Iterable[_builtins.str] | None = ...,
745
815
  ) -> None: ...
746
-
747
- global___Availability = Availability
816
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
817
+ "platforms", b"platforms", "regions", b"regions"
818
+ ] # noqa: Y015
819
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
820
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
821
+ "platforms", b"platforms", "regions", b"regions"
822
+ ] # noqa: Y015
823
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
824
+
825
+ Global___Availability: _TypeAlias = Availability # noqa: Y015