weave-python 0.28.2__py3-none-any.whl → 0.30.3__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 +194 -98
  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 +14 -6
  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 +18 -14
  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 +60 -18
  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 +31 -19
  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 +32 -6
  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 +52 -22
  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 +42 -42
  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 +88 -56
  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 +56 -20
  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 +24 -20
  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 +6 -6
  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 +228 -90
  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 +16 -10
  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 +51 -31
  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 +38 -12
  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 +264 -98
  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 +99 -23
  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 +6 -6
  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 +47 -23
  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 +27 -13
  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 +96 -36
  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 +39 -23
  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 +20 -6
  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 +14 -12
  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 +9 -9
  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 +19 -11
  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 +42 -24
  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 +19 -13
  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.3.dist-info}/METADATA +3 -3
  101. weave_python-0.30.3.dist-info/RECORD +131 -0
  102. {weave_python-0.28.2.dist-info → weave_python-0.30.3.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.3.dist-info}/licenses/LICENSE +0 -0
@@ -3,31 +3,30 @@
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.internal.enum_type_wrapper
11
- import google.protobuf.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.internal import containers as _containers
10
+ from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
11
+ import builtins as _builtins
12
12
  import sys
13
- import typing
13
+ import typing as _typing
14
14
 
15
15
  if sys.version_info >= (3, 10):
16
- import typing as typing_extensions
16
+ from typing import TypeAlias as _TypeAlias
17
17
  else:
18
- import typing_extensions
18
+ from typing_extensions import TypeAlias as _TypeAlias
19
19
 
20
- DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
20
+ DESCRIPTOR: _descriptor.FileDescriptor
21
21
 
22
22
  class _ModelType:
23
- ValueType = typing.NewType("ValueType", builtins.int)
24
- V: typing_extensions.TypeAlias = ValueType
23
+ ValueType = _typing.NewType("ValueType", _builtins.int)
24
+ V: _TypeAlias = ValueType # noqa: Y015
25
25
 
26
26
  class _ModelTypeEnumTypeWrapper(
27
- google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ModelType.ValueType],
28
- builtins.type,
27
+ _enum_type_wrapper._EnumTypeWrapper[_ModelType.ValueType], _builtins.type
29
28
  ):
30
- DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
29
+ DESCRIPTOR: _descriptor.EnumDescriptor
31
30
  MODEL_TYPE_UNSPECIFIED: _ModelType.ValueType # 0
32
31
  MODEL_TYPE_CTGAN: _ModelType.ValueType # 1
33
32
  """Conditional Tabular GAN"""
@@ -70,301 +69,338 @@ MODEL_TYPE_BAYESIAN: ModelType.ValueType # 8
70
69
  """Bayesian Network"""
71
70
  MODEL_TYPE_HMA: ModelType.ValueType # 9
72
71
  """Hierarchical Modeling Algorithm"""
73
- global___ModelType = ModelType
72
+ Global___ModelType: _TypeAlias = ModelType # noqa: Y015
74
73
 
75
- @typing.final
76
- class Configuration(google.protobuf.message.Message):
74
+ @_typing.final
75
+ class Configuration(_message.Message):
77
76
  """SynthesizerConfig defines customization options for synthetic data generation"""
78
77
 
79
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
78
+ DESCRIPTOR: _descriptor.Descriptor
80
79
 
81
- NUM_RECORDS_FIELD_NUMBER: builtins.int
82
- DATA_QUALITY_FIELD_NUMBER: builtins.int
83
- PRIVACY_FIELD_NUMBER: builtins.int
84
- MODEL_FIELD_NUMBER: builtins.int
85
- num_records: builtins.int
80
+ NUM_RECORDS_FIELD_NUMBER: _builtins.int
81
+ DATA_QUALITY_FIELD_NUMBER: _builtins.int
82
+ PRIVACY_FIELD_NUMBER: _builtins.int
83
+ MODEL_FIELD_NUMBER: _builtins.int
84
+ num_records: _builtins.int
86
85
  """Basic configuration
87
86
  Number of synthetic records to generate
88
87
  """
89
- @property
90
- def data_quality(self) -> global___DataQualityConfig:
91
- """Data quality settings
92
- Controls data quality aspects
93
- """
94
-
95
- @property
96
- def privacy(self) -> global___PrivacyConfig:
97
- """Privacy settings
98
- Privacy and anonymization settings
99
- """
100
-
101
- @property
102
- def model(self) -> global___ModelConfig:
103
- """Advanced model configuration
104
- ML model configuration options
105
- """
88
+ @_builtins.property
89
+ def data_quality(self) -> Global___DataQualityConfig:
90
+ """Data quality settings"""
91
+
92
+ @_builtins.property
93
+ def privacy(self) -> Global___PrivacyConfig:
94
+ """Privacy settings"""
95
+
96
+ @_builtins.property
97
+ def model(self) -> Global___ModelConfig:
98
+ """Advanced model configuration"""
106
99
 
107
100
  def __init__(
108
101
  self,
109
102
  *,
110
- num_records: builtins.int = ...,
111
- data_quality: global___DataQualityConfig | None = ...,
112
- privacy: global___PrivacyConfig | None = ...,
113
- model: global___ModelConfig | None = ...,
103
+ num_records: _builtins.int | None = ...,
104
+ data_quality: Global___DataQualityConfig | None = ...,
105
+ privacy: Global___PrivacyConfig | None = ...,
106
+ model: Global___ModelConfig | None = ...,
114
107
  ) -> None: ...
115
- def HasField(
116
- self,
117
- field_name: typing.Literal[
118
- "data_quality", b"data_quality", "model", b"model", "privacy", b"privacy"
119
- ],
120
- ) -> builtins.bool: ...
121
- def ClearField(
122
- self,
123
- field_name: typing.Literal[
124
- "data_quality",
125
- b"data_quality",
126
- "model",
127
- b"model",
128
- "num_records",
129
- b"num_records",
130
- "privacy",
131
- b"privacy",
132
- ],
133
- ) -> None: ...
134
-
135
- global___Configuration = Configuration
136
-
137
- @typing.final
138
- class DataQualityConfig(google.protobuf.message.Message):
108
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
109
+ "data_quality",
110
+ b"data_quality",
111
+ "model",
112
+ b"model",
113
+ "num_records",
114
+ b"num_records",
115
+ "privacy",
116
+ b"privacy",
117
+ ] # noqa: Y015
118
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
119
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
120
+ "data_quality",
121
+ b"data_quality",
122
+ "model",
123
+ b"model",
124
+ "num_records",
125
+ b"num_records",
126
+ "privacy",
127
+ b"privacy",
128
+ ] # noqa: Y015
129
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
130
+
131
+ Global___Configuration: _TypeAlias = Configuration # noqa: Y015
132
+
133
+ @_typing.final
134
+ class DataQualityConfig(_message.Message):
139
135
  """DataQualityConfig controls the quality of generated data"""
140
136
 
141
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
137
+ DESCRIPTOR: _descriptor.Descriptor
142
138
 
143
- CORRELATION_ACCURACY_FIELD_NUMBER: builtins.int
144
- STATISTICAL_SIMILARITY_FIELD_NUMBER: builtins.int
145
- PRESERVE_OUTLIERS_FIELD_NUMBER: builtins.int
146
- ENFORCE_CONSTRAINTS_FIELD_NUMBER: builtins.int
147
- CONSTRAINTS_FIELD_NUMBER: builtins.int
148
- correlation_accuracy: builtins.float
139
+ CORRELATION_ACCURACY_FIELD_NUMBER: _builtins.int
140
+ STATISTICAL_SIMILARITY_FIELD_NUMBER: _builtins.int
141
+ PRESERVE_OUTLIERS_FIELD_NUMBER: _builtins.int
142
+ ENFORCE_CONSTRAINTS_FIELD_NUMBER: _builtins.int
143
+ CONSTRAINTS_FIELD_NUMBER: _builtins.int
144
+ correlation_accuracy: _builtins.float
149
145
  """How closely to maintain correlations (0.0-1.0)"""
150
- statistical_similarity: builtins.float
146
+ statistical_similarity: _builtins.float
151
147
  """Statistical similarity to original data (0.0-1.0)"""
152
- preserve_outliers: builtins.bool
148
+ preserve_outliers: _builtins.bool
153
149
  """Whether to preserve outlier patterns"""
154
- enforce_constraints: builtins.bool
150
+ enforce_constraints: _builtins.bool
155
151
  """Whether to enforce data constraints"""
156
- @property
152
+ @_builtins.property
157
153
  def constraints(
158
154
  self,
159
- ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[
160
- global___ColumnConstraint
161
- ]:
162
- """Column-specific constraints"""
163
-
155
+ ) -> _containers.RepeatedCompositeFieldContainer[Global___ColumnConstraint]: ...
164
156
  def __init__(
165
157
  self,
166
158
  *,
167
- correlation_accuracy: builtins.float = ...,
168
- statistical_similarity: builtins.float = ...,
169
- preserve_outliers: builtins.bool = ...,
170
- enforce_constraints: builtins.bool = ...,
171
- constraints: collections.abc.Iterable[global___ColumnConstraint] | None = ...,
172
- ) -> None: ...
173
- def ClearField(
174
- self,
175
- field_name: typing.Literal[
176
- "constraints",
177
- b"constraints",
178
- "correlation_accuracy",
179
- b"correlation_accuracy",
180
- "enforce_constraints",
181
- b"enforce_constraints",
182
- "preserve_outliers",
183
- b"preserve_outliers",
184
- "statistical_similarity",
185
- b"statistical_similarity",
186
- ],
159
+ correlation_accuracy: _builtins.float | None = ...,
160
+ statistical_similarity: _builtins.float | None = ...,
161
+ preserve_outliers: _builtins.bool | None = ...,
162
+ enforce_constraints: _builtins.bool | None = ...,
163
+ constraints: _abc.Iterable[Global___ColumnConstraint] | None = ...,
187
164
  ) -> None: ...
188
-
189
- global___DataQualityConfig = DataQualityConfig
190
-
191
- @typing.final
192
- class ColumnConstraint(google.protobuf.message.Message):
165
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
166
+ "constraints",
167
+ b"constraints",
168
+ "correlation_accuracy",
169
+ b"correlation_accuracy",
170
+ "enforce_constraints",
171
+ b"enforce_constraints",
172
+ "preserve_outliers",
173
+ b"preserve_outliers",
174
+ "statistical_similarity",
175
+ b"statistical_similarity",
176
+ ] # noqa: Y015
177
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
178
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
179
+ "constraints",
180
+ b"constraints",
181
+ "correlation_accuracy",
182
+ b"correlation_accuracy",
183
+ "enforce_constraints",
184
+ b"enforce_constraints",
185
+ "preserve_outliers",
186
+ b"preserve_outliers",
187
+ "statistical_similarity",
188
+ b"statistical_similarity",
189
+ ] # noqa: Y015
190
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
191
+
192
+ Global___DataQualityConfig: _TypeAlias = DataQualityConfig # noqa: Y015
193
+
194
+ @_typing.final
195
+ class ColumnConstraint(_message.Message):
193
196
  """ColumnConstraint defines constraints for specific columns"""
194
197
 
195
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
198
+ DESCRIPTOR: _descriptor.Descriptor
196
199
 
197
- COLUMN_NAME_FIELD_NUMBER: builtins.int
198
- CONSTRAINT_TYPE_FIELD_NUMBER: builtins.int
199
- MIN_VALUE_FIELD_NUMBER: builtins.int
200
- MAX_VALUE_FIELD_NUMBER: builtins.int
201
- REGEX_PATTERN_FIELD_NUMBER: builtins.int
202
- ALLOWED_VALUES_FIELD_NUMBER: builtins.int
203
- column_name: builtins.str
200
+ COLUMN_NAME_FIELD_NUMBER: _builtins.int
201
+ CONSTRAINT_TYPE_FIELD_NUMBER: _builtins.int
202
+ MIN_VALUE_FIELD_NUMBER: _builtins.int
203
+ MAX_VALUE_FIELD_NUMBER: _builtins.int
204
+ REGEX_PATTERN_FIELD_NUMBER: _builtins.int
205
+ ALLOWED_VALUES_FIELD_NUMBER: _builtins.int
206
+ column_name: _builtins.str
204
207
  """Name of the column"""
205
- constraint_type: builtins.str
208
+ constraint_type: _builtins.str
206
209
  """Type of constraint (e.g., "range", "regex", "categorical")"""
207
- min_value: builtins.str
210
+ min_value: _builtins.str
208
211
  """Minimum value (for numeric constraints)"""
209
- max_value: builtins.str
212
+ max_value: _builtins.str
210
213
  """Maximum value (for numeric constraints)"""
211
- regex_pattern: builtins.str
214
+ regex_pattern: _builtins.str
212
215
  """Regex pattern (for string constraints)"""
213
- @property
214
- def allowed_values(
215
- self,
216
- ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
216
+ @_builtins.property
217
+ def allowed_values(self) -> _containers.RepeatedScalarFieldContainer[_builtins.str]:
217
218
  """List of allowed values (for categorical constraints)"""
218
219
 
219
220
  def __init__(
220
221
  self,
221
222
  *,
222
- column_name: builtins.str = ...,
223
- constraint_type: builtins.str = ...,
224
- min_value: builtins.str = ...,
225
- max_value: builtins.str = ...,
226
- regex_pattern: builtins.str = ...,
227
- allowed_values: collections.abc.Iterable[builtins.str] | None = ...,
223
+ column_name: _builtins.str | None = ...,
224
+ constraint_type: _builtins.str | None = ...,
225
+ min_value: _builtins.str | None = ...,
226
+ max_value: _builtins.str | None = ...,
227
+ regex_pattern: _builtins.str | None = ...,
228
+ allowed_values: _abc.Iterable[_builtins.str] | None = ...,
228
229
  ) -> None: ...
229
- def ClearField(
230
- self,
231
- field_name: typing.Literal[
232
- "allowed_values",
233
- b"allowed_values",
234
- "column_name",
235
- b"column_name",
236
- "constraint_type",
237
- b"constraint_type",
238
- "max_value",
239
- b"max_value",
240
- "min_value",
241
- b"min_value",
242
- "regex_pattern",
243
- b"regex_pattern",
244
- ],
245
- ) -> None: ...
246
-
247
- global___ColumnConstraint = ColumnConstraint
248
-
249
- @typing.final
250
- class PrivacyConfig(google.protobuf.message.Message):
230
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
231
+ "allowed_values",
232
+ b"allowed_values",
233
+ "column_name",
234
+ b"column_name",
235
+ "constraint_type",
236
+ b"constraint_type",
237
+ "max_value",
238
+ b"max_value",
239
+ "min_value",
240
+ b"min_value",
241
+ "regex_pattern",
242
+ b"regex_pattern",
243
+ ] # noqa: Y015
244
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
245
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
246
+ "allowed_values",
247
+ b"allowed_values",
248
+ "column_name",
249
+ b"column_name",
250
+ "constraint_type",
251
+ b"constraint_type",
252
+ "max_value",
253
+ b"max_value",
254
+ "min_value",
255
+ b"min_value",
256
+ "regex_pattern",
257
+ b"regex_pattern",
258
+ ] # noqa: Y015
259
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
260
+
261
+ Global___ColumnConstraint: _TypeAlias = ColumnConstraint # noqa: Y015
262
+
263
+ @_typing.final
264
+ class PrivacyConfig(_message.Message):
251
265
  """PrivacyConfig controls privacy aspects of synthetic data"""
252
266
 
253
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
267
+ DESCRIPTOR: _descriptor.Descriptor
254
268
 
255
- EPSILON_FIELD_NUMBER: builtins.int
256
- K_ANONYMITY_CHECK_FIELD_NUMBER: builtins.int
257
- K_VALUE_FIELD_NUMBER: builtins.int
258
- SENSITIVE_COLUMNS_FIELD_NUMBER: builtins.int
259
- ENABLE_PRIVACY_METRICS_FIELD_NUMBER: builtins.int
260
- epsilon: builtins.float
269
+ EPSILON_FIELD_NUMBER: _builtins.int
270
+ K_ANONYMITY_CHECK_FIELD_NUMBER: _builtins.int
271
+ K_VALUE_FIELD_NUMBER: _builtins.int
272
+ SENSITIVE_COLUMNS_FIELD_NUMBER: _builtins.int
273
+ ENABLE_PRIVACY_METRICS_FIELD_NUMBER: _builtins.int
274
+ epsilon: _builtins.float
261
275
  """Differential privacy epsilon parameter"""
262
- k_anonymity_check: builtins.bool
276
+ k_anonymity_check: _builtins.bool
263
277
  """Whether to enforce k-anonymity"""
264
- k_value: builtins.int
278
+ k_value: _builtins.int
265
279
  """k value for k-anonymity (if enabled)"""
266
- enable_privacy_metrics: builtins.bool
280
+ enable_privacy_metrics: _builtins.bool
267
281
  """Whether to compute privacy metrics"""
268
- @property
282
+ @_builtins.property
269
283
  def sensitive_columns(
270
284
  self,
271
- ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
285
+ ) -> _containers.RepeatedScalarFieldContainer[_builtins.str]:
272
286
  """Columns that contain sensitive information"""
273
287
 
274
288
  def __init__(
275
289
  self,
276
290
  *,
277
- epsilon: builtins.float = ...,
278
- k_anonymity_check: builtins.bool = ...,
279
- k_value: builtins.int = ...,
280
- sensitive_columns: collections.abc.Iterable[builtins.str] | None = ...,
281
- enable_privacy_metrics: builtins.bool = ...,
291
+ epsilon: _builtins.float | None = ...,
292
+ k_anonymity_check: _builtins.bool | None = ...,
293
+ k_value: _builtins.int | None = ...,
294
+ sensitive_columns: _abc.Iterable[_builtins.str] | None = ...,
295
+ enable_privacy_metrics: _builtins.bool | None = ...,
282
296
  ) -> None: ...
283
- def ClearField(
284
- self,
285
- field_name: typing.Literal[
286
- "enable_privacy_metrics",
287
- b"enable_privacy_metrics",
288
- "epsilon",
289
- b"epsilon",
290
- "k_anonymity_check",
291
- b"k_anonymity_check",
292
- "k_value",
293
- b"k_value",
294
- "sensitive_columns",
295
- b"sensitive_columns",
296
- ],
297
- ) -> None: ...
298
-
299
- global___PrivacyConfig = PrivacyConfig
300
-
301
- @typing.final
302
- class ModelConfig(google.protobuf.message.Message):
297
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
298
+ "enable_privacy_metrics",
299
+ b"enable_privacy_metrics",
300
+ "epsilon",
301
+ b"epsilon",
302
+ "k_anonymity_check",
303
+ b"k_anonymity_check",
304
+ "k_value",
305
+ b"k_value",
306
+ "sensitive_columns",
307
+ b"sensitive_columns",
308
+ ] # noqa: Y015
309
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
310
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
311
+ "enable_privacy_metrics",
312
+ b"enable_privacy_metrics",
313
+ "epsilon",
314
+ b"epsilon",
315
+ "k_anonymity_check",
316
+ b"k_anonymity_check",
317
+ "k_value",
318
+ b"k_value",
319
+ "sensitive_columns",
320
+ b"sensitive_columns",
321
+ ] # noqa: Y015
322
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
323
+
324
+ Global___PrivacyConfig: _TypeAlias = PrivacyConfig # noqa: Y015
325
+
326
+ @_typing.final
327
+ class ModelConfig(_message.Message):
303
328
  """ModelConfig controls the ML model used for synthesis"""
304
329
 
305
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
330
+ DESCRIPTOR: _descriptor.Descriptor
306
331
 
307
- @typing.final
308
- class CustomParametersEntry(google.protobuf.message.Message):
309
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
332
+ @_typing.final
333
+ class CustomParametersEntry(_message.Message):
334
+ DESCRIPTOR: _descriptor.Descriptor
310
335
 
311
- KEY_FIELD_NUMBER: builtins.int
312
- VALUE_FIELD_NUMBER: builtins.int
313
- key: builtins.str
314
- value: builtins.str
336
+ KEY_FIELD_NUMBER: _builtins.int
337
+ VALUE_FIELD_NUMBER: _builtins.int
338
+ key: _builtins.str
339
+ value: _builtins.str
315
340
  def __init__(
316
341
  self,
317
342
  *,
318
- key: builtins.str = ...,
319
- value: builtins.str = ...,
343
+ key: _builtins.str | None = ...,
344
+ value: _builtins.str | None = ...,
320
345
  ) -> None: ...
321
- def ClearField(
322
- self, field_name: typing.Literal["key", b"key", "value", b"value"]
323
- ) -> None: ...
324
-
325
- MODEL_TYPE_FIELD_NUMBER: builtins.int
326
- EPOCHS_FIELD_NUMBER: builtins.int
327
- LEARNING_RATE_FIELD_NUMBER: builtins.int
328
- BATCH_SIZE_FIELD_NUMBER: builtins.int
329
- CUSTOM_PARAMETERS_FIELD_NUMBER: builtins.int
330
- model_type: global___ModelType.ValueType
346
+ _HasFieldArgType: _TypeAlias = _typing.Literal["key", b"key", "value", b"value"] # noqa: Y015
347
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
348
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
349
+ "key", b"key", "value", b"value"
350
+ ] # noqa: Y015
351
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
352
+
353
+ MODEL_TYPE_FIELD_NUMBER: _builtins.int
354
+ EPOCHS_FIELD_NUMBER: _builtins.int
355
+ LEARNING_RATE_FIELD_NUMBER: _builtins.int
356
+ BATCH_SIZE_FIELD_NUMBER: _builtins.int
357
+ CUSTOM_PARAMETERS_FIELD_NUMBER: _builtins.int
358
+ model_type: Global___ModelType.ValueType
331
359
  """Type of model to use for synthesis"""
332
- epochs: builtins.int
360
+ epochs: _builtins.int
333
361
  """Training epochs"""
334
- learning_rate: builtins.float
362
+ learning_rate: _builtins.float
335
363
  """Learning rate for training"""
336
- batch_size: builtins.int
364
+ batch_size: _builtins.int
337
365
  """Batch size for training"""
338
- @property
339
- def custom_parameters(
340
- self,
341
- ) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
366
+ @_builtins.property
367
+ def custom_parameters(self) -> _containers.ScalarMap[_builtins.str, _builtins.str]:
342
368
  """Additional model-specific parameters"""
343
369
 
344
370
  def __init__(
345
371
  self,
346
372
  *,
347
- model_type: global___ModelType.ValueType = ...,
348
- epochs: builtins.int = ...,
349
- learning_rate: builtins.float = ...,
350
- batch_size: builtins.int = ...,
351
- custom_parameters: collections.abc.Mapping[builtins.str, builtins.str]
352
- | None = ...,
373
+ model_type: Global___ModelType.ValueType | None = ...,
374
+ epochs: _builtins.int | None = ...,
375
+ learning_rate: _builtins.float | None = ...,
376
+ batch_size: _builtins.int | None = ...,
377
+ custom_parameters: _abc.Mapping[_builtins.str, _builtins.str] | None = ...,
353
378
  ) -> None: ...
354
- def ClearField(
355
- self,
356
- field_name: typing.Literal[
357
- "batch_size",
358
- b"batch_size",
359
- "custom_parameters",
360
- b"custom_parameters",
361
- "epochs",
362
- b"epochs",
363
- "learning_rate",
364
- b"learning_rate",
365
- "model_type",
366
- b"model_type",
367
- ],
368
- ) -> None: ...
369
-
370
- global___ModelConfig = ModelConfig
379
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
380
+ "batch_size",
381
+ b"batch_size",
382
+ "custom_parameters",
383
+ b"custom_parameters",
384
+ "epochs",
385
+ b"epochs",
386
+ "learning_rate",
387
+ b"learning_rate",
388
+ "model_type",
389
+ b"model_type",
390
+ ] # noqa: Y015
391
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
392
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
393
+ "batch_size",
394
+ b"batch_size",
395
+ "custom_parameters",
396
+ b"custom_parameters",
397
+ "epochs",
398
+ b"epochs",
399
+ "learning_rate",
400
+ b"learning_rate",
401
+ "model_type",
402
+ b"model_type",
403
+ ] # noqa: Y015
404
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
405
+
406
+ Global___ModelConfig: _TypeAlias = ModelConfig # noqa: Y015
@@ -3,18 +3,19 @@
3
3
  isort:skip_file
4
4
  """
5
5
 
6
- import abc
7
- import collections.abc
8
- import grpc
9
- import grpc.aio
10
- import typing
6
+ from collections import abc as _abc
7
+ from grpc import aio as _aio
8
+ import abc as _abc_1
9
+ import grpc as _grpc
10
+ import typing as _typing
11
11
 
12
- _T = typing.TypeVar("_T")
12
+ _T = _typing.TypeVar("_T")
13
13
 
14
14
  class _MaybeAsyncIterator(
15
- collections.abc.AsyncIterator[_T],
16
- collections.abc.Iterator[_T],
17
- metaclass=abc.ABCMeta,
15
+ _abc.AsyncIterator[_T], _abc.Iterator[_T], metaclass=_abc_1.ABCMeta
18
16
  ): ...
19
- class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext): # type: ignore[misc, type-arg]
17
+ class _ServicerContext(_grpc.ServicerContext, _aio.ServicerContext): # type: ignore[misc, type-arg]
20
18
  ...
19
+
20
+ GRPC_GENERATED_VERSION: str
21
+ GRPC_VERSION: str