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,235 +3,269 @@
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.struct_pb2
12
- import typing
13
- import weaveapi.synthesize.v1.training_pb2
14
-
15
- DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
16
-
17
- @typing.final
18
- class CreateSynthesizerRequest(google.protobuf.message.Message):
19
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
20
-
21
- ORGANIZATION_ID_FIELD_NUMBER: builtins.int
22
- DATASET_IDS_FIELD_NUMBER: builtins.int
23
- TRAINING_DRIVER_FIELD_NUMBER: builtins.int
24
- DRIVER_CONFIG_FIELD_NUMBER: builtins.int
25
- OUTPUT_STORAGE_CONNECTION_ID_FIELD_NUMBER: builtins.int
26
- organization_id: builtins.str
27
- training_driver: weaveapi.synthesize.v1.training_pb2.TrainingDriver.ValueType
28
- output_storage_connection_id: builtins.str
29
- @property
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 struct_pb2 as _struct_pb2
10
+ from google.protobuf.internal import containers as _containers
11
+ from weaveapi.synthesize.v1 import training_pb2 as _training_pb2
12
+ import builtins as _builtins
13
+ import sys
14
+ import typing as _typing
15
+
16
+ if sys.version_info >= (3, 10):
17
+ from typing import TypeAlias as _TypeAlias
18
+ else:
19
+ from typing_extensions import TypeAlias as _TypeAlias
20
+
21
+ DESCRIPTOR: _descriptor.FileDescriptor
22
+
23
+ @_typing.final
24
+ class CreateSynthesizerRequest(_message.Message):
25
+ DESCRIPTOR: _descriptor.Descriptor
26
+
27
+ ORGANIZATION_ID_FIELD_NUMBER: _builtins.int
28
+ DATASET_IDS_FIELD_NUMBER: _builtins.int
29
+ TRAINING_DRIVER_FIELD_NUMBER: _builtins.int
30
+ DRIVER_CONFIG_FIELD_NUMBER: _builtins.int
31
+ OUTPUT_STORAGE_CONNECTION_ID_FIELD_NUMBER: _builtins.int
32
+ organization_id: _builtins.str
33
+ training_driver: _training_pb2.TrainingDriver.ValueType
34
+ output_storage_connection_id: _builtins.str
35
+ @_builtins.property
30
36
  def dataset_ids(
31
37
  self,
32
- ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[
33
- builtins.str
34
- ]: ...
35
- @property
36
- def driver_config(self) -> google.protobuf.struct_pb2.Struct: ...
38
+ ) -> _containers.RepeatedScalarFieldContainer[_builtins.str]: ...
39
+ @_builtins.property
40
+ def driver_config(self) -> _struct_pb2.Struct: ...
37
41
  def __init__(
38
42
  self,
39
43
  *,
40
- organization_id: builtins.str = ...,
41
- dataset_ids: collections.abc.Iterable[builtins.str] | None = ...,
42
- training_driver: weaveapi.synthesize.v1.training_pb2.TrainingDriver.ValueType = ...,
43
- driver_config: google.protobuf.struct_pb2.Struct | None = ...,
44
- output_storage_connection_id: builtins.str = ...,
45
- ) -> None: ...
46
- def HasField(
47
- self, field_name: typing.Literal["driver_config", b"driver_config"]
48
- ) -> builtins.bool: ...
49
- def ClearField(
50
- self,
51
- field_name: typing.Literal[
52
- "dataset_ids",
53
- b"dataset_ids",
54
- "driver_config",
55
- b"driver_config",
56
- "organization_id",
57
- b"organization_id",
58
- "output_storage_connection_id",
59
- b"output_storage_connection_id",
60
- "training_driver",
61
- b"training_driver",
62
- ],
44
+ organization_id: _builtins.str | None = ...,
45
+ dataset_ids: _abc.Iterable[_builtins.str] | None = ...,
46
+ training_driver: _training_pb2.TrainingDriver.ValueType | None = ...,
47
+ driver_config: _struct_pb2.Struct | None = ...,
48
+ output_storage_connection_id: _builtins.str | None = ...,
63
49
  ) -> None: ...
50
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
51
+ "dataset_ids",
52
+ b"dataset_ids",
53
+ "driver_config",
54
+ b"driver_config",
55
+ "organization_id",
56
+ b"organization_id",
57
+ "output_storage_connection_id",
58
+ b"output_storage_connection_id",
59
+ "training_driver",
60
+ b"training_driver",
61
+ ] # noqa: Y015
62
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
63
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
64
+ "dataset_ids",
65
+ b"dataset_ids",
66
+ "driver_config",
67
+ b"driver_config",
68
+ "organization_id",
69
+ b"organization_id",
70
+ "output_storage_connection_id",
71
+ b"output_storage_connection_id",
72
+ "training_driver",
73
+ b"training_driver",
74
+ ] # noqa: Y015
75
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
64
76
 
65
- global___CreateSynthesizerRequest = CreateSynthesizerRequest
77
+ Global___CreateSynthesizerRequest: _TypeAlias = CreateSynthesizerRequest # noqa: Y015
66
78
 
67
- @typing.final
68
- class CreateSynthesizerResponse(google.protobuf.message.Message):
69
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
79
+ @_typing.final
80
+ class CreateSynthesizerResponse(_message.Message):
81
+ DESCRIPTOR: _descriptor.Descriptor
70
82
 
71
- SYNTHESIZER_ID_FIELD_NUMBER: builtins.int
72
- synthesizer_id: builtins.str
83
+ SYNTHESIZER_ID_FIELD_NUMBER: _builtins.int
84
+ synthesizer_id: _builtins.str
73
85
  def __init__(
74
86
  self,
75
87
  *,
76
- synthesizer_id: builtins.str = ...,
77
- ) -> None: ...
78
- def ClearField(
79
- self, field_name: typing.Literal["synthesizer_id", b"synthesizer_id"]
88
+ synthesizer_id: _builtins.str | None = ...,
80
89
  ) -> None: ...
90
+ _HasFieldArgType: _TypeAlias = _typing.Literal["synthesizer_id", b"synthesizer_id"] # noqa: Y015
91
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
92
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
93
+ "synthesizer_id", b"synthesizer_id"
94
+ ] # noqa: Y015
95
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
81
96
 
82
- global___CreateSynthesizerResponse = CreateSynthesizerResponse
97
+ Global___CreateSynthesizerResponse: _TypeAlias = CreateSynthesizerResponse # noqa: Y015
83
98
 
84
- @typing.final
85
- class GetSynthesizerRequest(google.protobuf.message.Message):
86
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
99
+ @_typing.final
100
+ class GetSynthesizerRequest(_message.Message):
101
+ DESCRIPTOR: _descriptor.Descriptor
87
102
 
88
- SYNTHESIZER_ID_FIELD_NUMBER: builtins.int
89
- synthesizer_id: builtins.str
103
+ SYNTHESIZER_ID_FIELD_NUMBER: _builtins.int
104
+ synthesizer_id: _builtins.str
90
105
  def __init__(
91
106
  self,
92
107
  *,
93
- synthesizer_id: builtins.str = ...,
94
- ) -> None: ...
95
- def ClearField(
96
- self, field_name: typing.Literal["synthesizer_id", b"synthesizer_id"]
108
+ synthesizer_id: _builtins.str | None = ...,
97
109
  ) -> None: ...
110
+ _HasFieldArgType: _TypeAlias = _typing.Literal["synthesizer_id", b"synthesizer_id"] # noqa: Y015
111
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
112
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
113
+ "synthesizer_id", b"synthesizer_id"
114
+ ] # noqa: Y015
115
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
98
116
 
99
- global___GetSynthesizerRequest = GetSynthesizerRequest
117
+ Global___GetSynthesizerRequest: _TypeAlias = GetSynthesizerRequest # noqa: Y015
100
118
 
101
- @typing.final
102
- class GetSynthesizerResponse(google.protobuf.message.Message):
103
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
119
+ @_typing.final
120
+ class GetSynthesizerResponse(_message.Message):
121
+ DESCRIPTOR: _descriptor.Descriptor
104
122
 
105
- SYNTHESIZER_FIELD_NUMBER: builtins.int
106
- @property
107
- def synthesizer(self) -> global___Synthesizer: ...
123
+ SYNTHESIZER_FIELD_NUMBER: _builtins.int
124
+ @_builtins.property
125
+ def synthesizer(self) -> Global___Synthesizer: ...
108
126
  def __init__(
109
127
  self,
110
128
  *,
111
- synthesizer: global___Synthesizer | None = ...,
112
- ) -> None: ...
113
- def HasField(
114
- self, field_name: typing.Literal["synthesizer", b"synthesizer"]
115
- ) -> builtins.bool: ...
116
- def ClearField(
117
- self, field_name: typing.Literal["synthesizer", b"synthesizer"]
129
+ synthesizer: Global___Synthesizer | None = ...,
118
130
  ) -> None: ...
131
+ _HasFieldArgType: _TypeAlias = _typing.Literal["synthesizer", b"synthesizer"] # noqa: Y015
132
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
133
+ _ClearFieldArgType: _TypeAlias = _typing.Literal["synthesizer", b"synthesizer"] # noqa: Y015
134
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
119
135
 
120
- global___GetSynthesizerResponse = GetSynthesizerResponse
136
+ Global___GetSynthesizerResponse: _TypeAlias = GetSynthesizerResponse # noqa: Y015
121
137
 
122
- @typing.final
123
- class ListSynthesizersRequest(google.protobuf.message.Message):
124
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
138
+ @_typing.final
139
+ class ListSynthesizersRequest(_message.Message):
140
+ DESCRIPTOR: _descriptor.Descriptor
125
141
 
126
- ORGANIZATION_ID_FIELD_NUMBER: builtins.int
127
- organization_id: builtins.str
142
+ ORGANIZATION_ID_FIELD_NUMBER: _builtins.int
143
+ organization_id: _builtins.str
128
144
  def __init__(
129
145
  self,
130
146
  *,
131
- organization_id: builtins.str = ...,
132
- ) -> None: ...
133
- def ClearField(
134
- self, field_name: typing.Literal["organization_id", b"organization_id"]
147
+ organization_id: _builtins.str | None = ...,
135
148
  ) -> None: ...
149
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
150
+ "organization_id", b"organization_id"
151
+ ] # noqa: Y015
152
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
153
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
154
+ "organization_id", b"organization_id"
155
+ ] # noqa: Y015
156
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
136
157
 
137
- global___ListSynthesizersRequest = ListSynthesizersRequest
158
+ Global___ListSynthesizersRequest: _TypeAlias = ListSynthesizersRequest # noqa: Y015
138
159
 
139
- @typing.final
140
- class ListSynthesizersResponse(google.protobuf.message.Message):
141
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
160
+ @_typing.final
161
+ class ListSynthesizersResponse(_message.Message):
162
+ DESCRIPTOR: _descriptor.Descriptor
142
163
 
143
- SYNTHESIZERS_FIELD_NUMBER: builtins.int
144
- @property
164
+ SYNTHESIZERS_FIELD_NUMBER: _builtins.int
165
+ @_builtins.property
145
166
  def synthesizers(
146
167
  self,
147
- ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[
148
- global___Synthesizer
149
- ]: ...
168
+ ) -> _containers.RepeatedCompositeFieldContainer[Global___Synthesizer]: ...
150
169
  def __init__(
151
170
  self,
152
171
  *,
153
- synthesizers: collections.abc.Iterable[global___Synthesizer] | None = ...,
154
- ) -> None: ...
155
- def ClearField(
156
- self, field_name: typing.Literal["synthesizers", b"synthesizers"]
172
+ synthesizers: _abc.Iterable[Global___Synthesizer] | None = ...,
157
173
  ) -> None: ...
174
+ _HasFieldArgType: _TypeAlias = _typing.Literal["synthesizers", b"synthesizers"] # noqa: Y015
175
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
176
+ _ClearFieldArgType: _TypeAlias = _typing.Literal["synthesizers", b"synthesizers"] # noqa: Y015
177
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
158
178
 
159
- global___ListSynthesizersResponse = ListSynthesizersResponse
179
+ Global___ListSynthesizersResponse: _TypeAlias = ListSynthesizersResponse # noqa: Y015
160
180
 
161
- @typing.final
162
- class DeleteSynthesizerRequest(google.protobuf.message.Message):
163
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
181
+ @_typing.final
182
+ class DeleteSynthesizerRequest(_message.Message):
183
+ DESCRIPTOR: _descriptor.Descriptor
164
184
 
165
- SYNTHESIZER_ID_FIELD_NUMBER: builtins.int
166
- synthesizer_id: builtins.str
185
+ SYNTHESIZER_ID_FIELD_NUMBER: _builtins.int
186
+ synthesizer_id: _builtins.str
167
187
  def __init__(
168
188
  self,
169
189
  *,
170
- synthesizer_id: builtins.str = ...,
171
- ) -> None: ...
172
- def ClearField(
173
- self, field_name: typing.Literal["synthesizer_id", b"synthesizer_id"]
190
+ synthesizer_id: _builtins.str | None = ...,
174
191
  ) -> None: ...
192
+ _HasFieldArgType: _TypeAlias = _typing.Literal["synthesizer_id", b"synthesizer_id"] # noqa: Y015
193
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
194
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
195
+ "synthesizer_id", b"synthesizer_id"
196
+ ] # noqa: Y015
197
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
175
198
 
176
- global___DeleteSynthesizerRequest = DeleteSynthesizerRequest
199
+ Global___DeleteSynthesizerRequest: _TypeAlias = DeleteSynthesizerRequest # noqa: Y015
177
200
 
178
- @typing.final
179
- class DeleteSynthesizerResponse(google.protobuf.message.Message):
180
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
201
+ @_typing.final
202
+ class DeleteSynthesizerResponse(_message.Message):
203
+ DESCRIPTOR: _descriptor.Descriptor
181
204
 
182
205
  def __init__(
183
206
  self,
184
207
  ) -> None: ...
185
208
 
186
- global___DeleteSynthesizerResponse = DeleteSynthesizerResponse
187
-
188
- @typing.final
189
- class Synthesizer(google.protobuf.message.Message):
190
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
191
-
192
- SYNTHESIZER_ID_FIELD_NUMBER: builtins.int
193
- ORGANIZATION_ID_FIELD_NUMBER: builtins.int
194
- TRAINING_DATASET_IDS_FIELD_NUMBER: builtins.int
195
- TRAINING_DRIVER_FIELD_NUMBER: builtins.int
196
- TRAINING_DRIVER_CONFIG_FIELD_NUMBER: builtins.int
197
- STORAGE_CONNECTION_ID_FIELD_NUMBER: builtins.int
198
- synthesizer_id: builtins.str
199
- organization_id: builtins.str
200
- training_driver: weaveapi.synthesize.v1.training_pb2.TrainingDriver.ValueType
201
- training_driver_config: builtins.bytes
202
- storage_connection_id: builtins.str
203
- @property
209
+ Global___DeleteSynthesizerResponse: _TypeAlias = DeleteSynthesizerResponse # noqa: Y015
210
+
211
+ @_typing.final
212
+ class Synthesizer(_message.Message):
213
+ DESCRIPTOR: _descriptor.Descriptor
214
+
215
+ SYNTHESIZER_ID_FIELD_NUMBER: _builtins.int
216
+ ORGANIZATION_ID_FIELD_NUMBER: _builtins.int
217
+ TRAINING_DATASET_IDS_FIELD_NUMBER: _builtins.int
218
+ TRAINING_DRIVER_FIELD_NUMBER: _builtins.int
219
+ TRAINING_DRIVER_CONFIG_FIELD_NUMBER: _builtins.int
220
+ STORAGE_CONNECTION_ID_FIELD_NUMBER: _builtins.int
221
+ synthesizer_id: _builtins.str
222
+ organization_id: _builtins.str
223
+ training_driver: _training_pb2.TrainingDriver.ValueType
224
+ training_driver_config: _builtins.bytes
225
+ storage_connection_id: _builtins.str
226
+ @_builtins.property
204
227
  def training_dataset_ids(
205
228
  self,
206
- ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[
207
- builtins.str
208
- ]: ...
229
+ ) -> _containers.RepeatedScalarFieldContainer[_builtins.str]: ...
209
230
  def __init__(
210
231
  self,
211
232
  *,
212
- synthesizer_id: builtins.str = ...,
213
- organization_id: builtins.str = ...,
214
- training_dataset_ids: collections.abc.Iterable[builtins.str] | None = ...,
215
- training_driver: weaveapi.synthesize.v1.training_pb2.TrainingDriver.ValueType = ...,
216
- training_driver_config: builtins.bytes = ...,
217
- storage_connection_id: builtins.str = ...,
218
- ) -> None: ...
219
- def ClearField(
220
- self,
221
- field_name: typing.Literal[
222
- "organization_id",
223
- b"organization_id",
224
- "storage_connection_id",
225
- b"storage_connection_id",
226
- "synthesizer_id",
227
- b"synthesizer_id",
228
- "training_dataset_ids",
229
- b"training_dataset_ids",
230
- "training_driver",
231
- b"training_driver",
232
- "training_driver_config",
233
- b"training_driver_config",
234
- ],
233
+ synthesizer_id: _builtins.str | None = ...,
234
+ organization_id: _builtins.str | None = ...,
235
+ training_dataset_ids: _abc.Iterable[_builtins.str] | None = ...,
236
+ training_driver: _training_pb2.TrainingDriver.ValueType | None = ...,
237
+ training_driver_config: _builtins.bytes | None = ...,
238
+ storage_connection_id: _builtins.str | None = ...,
235
239
  ) -> None: ...
240
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
241
+ "organization_id",
242
+ b"organization_id",
243
+ "storage_connection_id",
244
+ b"storage_connection_id",
245
+ "synthesizer_id",
246
+ b"synthesizer_id",
247
+ "training_dataset_ids",
248
+ b"training_dataset_ids",
249
+ "training_driver",
250
+ b"training_driver",
251
+ "training_driver_config",
252
+ b"training_driver_config",
253
+ ] # noqa: Y015
254
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
255
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
256
+ "organization_id",
257
+ b"organization_id",
258
+ "storage_connection_id",
259
+ b"storage_connection_id",
260
+ "synthesizer_id",
261
+ b"synthesizer_id",
262
+ "training_dataset_ids",
263
+ b"training_dataset_ids",
264
+ "training_driver",
265
+ b"training_driver",
266
+ "training_driver_config",
267
+ b"training_driver_config",
268
+ ] # noqa: Y015
269
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
236
270
 
237
- global___Synthesizer = Synthesizer
271
+ Global___Synthesizer: _TypeAlias = Synthesizer # noqa: Y015
@@ -3,114 +3,102 @@
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
11
- import weaveapi.synthesize.v1.service_pb2
6
+ from collections import abc as _abc
7
+ from grpc import aio as _aio
8
+ from weaveapi.synthesize.v1 import service_pb2 as _service_pb2
9
+ import abc as _abc_1
10
+ import grpc as _grpc
11
+ import sys
12
+ import typing as _typing
12
13
 
13
- _T = typing.TypeVar("_T")
14
+ if sys.version_info >= (3, 11):
15
+ from typing import Self as _Self
16
+ else:
17
+ from typing_extensions import Self as _Self
18
+
19
+ _T = _typing.TypeVar("_T")
14
20
 
15
21
  class _MaybeAsyncIterator(
16
- collections.abc.AsyncIterator[_T],
17
- collections.abc.Iterator[_T],
18
- metaclass=abc.ABCMeta,
22
+ _abc.AsyncIterator[_T], _abc.Iterator[_T], metaclass=_abc_1.ABCMeta
19
23
  ): ...
20
- class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext): # type: ignore[misc, type-arg]
24
+ class _ServicerContext(_grpc.ServicerContext, _aio.ServicerContext): # type: ignore[misc, type-arg]
21
25
  ...
22
26
 
23
- class SynthesizeStub:
24
- def __init__(
25
- self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]
26
- ) -> None: ...
27
- CreateSynthesizer: grpc.UnaryUnaryMultiCallable[
28
- weaveapi.synthesize.v1.service_pb2.CreateSynthesizerRequest,
29
- weaveapi.synthesize.v1.service_pb2.CreateSynthesizerResponse,
30
- ]
31
-
32
- GetSynthesizer: grpc.UnaryUnaryMultiCallable[
33
- weaveapi.synthesize.v1.service_pb2.GetSynthesizerRequest,
34
- weaveapi.synthesize.v1.service_pb2.GetSynthesizerResponse,
35
- ]
36
-
37
- ListSynthesizers: grpc.UnaryUnaryMultiCallable[
38
- weaveapi.synthesize.v1.service_pb2.ListSynthesizersRequest,
39
- weaveapi.synthesize.v1.service_pb2.ListSynthesizersResponse,
40
- ]
27
+ GRPC_GENERATED_VERSION: str
28
+ GRPC_VERSION: str
41
29
 
42
- DeleteSynthesizer: grpc.UnaryUnaryMultiCallable[
43
- weaveapi.synthesize.v1.service_pb2.DeleteSynthesizerRequest,
44
- weaveapi.synthesize.v1.service_pb2.DeleteSynthesizerResponse,
30
+ class SynthesizeStub:
31
+ @_typing.overload
32
+ def __new__(cls, channel: _grpc.Channel) -> _Self: ...
33
+ @_typing.overload
34
+ def __new__(cls, channel: _aio.Channel) -> SynthesizeAsyncStub: ...
35
+ CreateSynthesizer: _grpc.UnaryUnaryMultiCallable[
36
+ _service_pb2.CreateSynthesizerRequest, _service_pb2.CreateSynthesizerResponse
45
37
  ]
46
-
47
- class SynthesizeAsyncStub:
48
- CreateSynthesizer: grpc.aio.UnaryUnaryMultiCallable[
49
- weaveapi.synthesize.v1.service_pb2.CreateSynthesizerRequest,
50
- weaveapi.synthesize.v1.service_pb2.CreateSynthesizerResponse,
38
+ GetSynthesizer: _grpc.UnaryUnaryMultiCallable[
39
+ _service_pb2.GetSynthesizerRequest, _service_pb2.GetSynthesizerResponse
51
40
  ]
52
-
53
- GetSynthesizer: grpc.aio.UnaryUnaryMultiCallable[
54
- weaveapi.synthesize.v1.service_pb2.GetSynthesizerRequest,
55
- weaveapi.synthesize.v1.service_pb2.GetSynthesizerResponse,
41
+ ListSynthesizers: _grpc.UnaryUnaryMultiCallable[
42
+ _service_pb2.ListSynthesizersRequest, _service_pb2.ListSynthesizersResponse
56
43
  ]
57
-
58
- ListSynthesizers: grpc.aio.UnaryUnaryMultiCallable[
59
- weaveapi.synthesize.v1.service_pb2.ListSynthesizersRequest,
60
- weaveapi.synthesize.v1.service_pb2.ListSynthesizersResponse,
44
+ DeleteSynthesizer: _grpc.UnaryUnaryMultiCallable[
45
+ _service_pb2.DeleteSynthesizerRequest, _service_pb2.DeleteSynthesizerResponse
61
46
  ]
62
47
 
63
- DeleteSynthesizer: grpc.aio.UnaryUnaryMultiCallable[
64
- weaveapi.synthesize.v1.service_pb2.DeleteSynthesizerRequest,
65
- weaveapi.synthesize.v1.service_pb2.DeleteSynthesizerResponse,
66
- ]
48
+ @_typing.type_check_only
49
+ class SynthesizeAsyncStub(SynthesizeStub):
50
+ def __init__(self, channel: _aio.Channel) -> None: ...
51
+ CreateSynthesizer: _aio.UnaryUnaryMultiCallable[
52
+ _service_pb2.CreateSynthesizerRequest, _service_pb2.CreateSynthesizerResponse
53
+ ] # type: ignore[assignment]
54
+ GetSynthesizer: _aio.UnaryUnaryMultiCallable[
55
+ _service_pb2.GetSynthesizerRequest, _service_pb2.GetSynthesizerResponse
56
+ ] # type: ignore[assignment]
57
+ ListSynthesizers: _aio.UnaryUnaryMultiCallable[
58
+ _service_pb2.ListSynthesizersRequest, _service_pb2.ListSynthesizersResponse
59
+ ] # type: ignore[assignment]
60
+ DeleteSynthesizer: _aio.UnaryUnaryMultiCallable[
61
+ _service_pb2.DeleteSynthesizerRequest, _service_pb2.DeleteSynthesizerResponse
62
+ ] # type: ignore[assignment]
67
63
 
68
- class SynthesizeServicer(metaclass=abc.ABCMeta):
69
- @abc.abstractmethod
64
+ class SynthesizeServicer(metaclass=_abc_1.ABCMeta):
65
+ @_abc_1.abstractmethod
70
66
  def CreateSynthesizer(
71
67
  self,
72
- request: weaveapi.synthesize.v1.service_pb2.CreateSynthesizerRequest,
68
+ request: _service_pb2.CreateSynthesizerRequest,
73
69
  context: _ServicerContext,
74
- ) -> typing.Union[
75
- weaveapi.synthesize.v1.service_pb2.CreateSynthesizerResponse,
76
- collections.abc.Awaitable[
77
- weaveapi.synthesize.v1.service_pb2.CreateSynthesizerResponse
78
- ],
70
+ ) -> _typing.Union[
71
+ _service_pb2.CreateSynthesizerResponse,
72
+ _abc.Awaitable[_service_pb2.CreateSynthesizerResponse],
79
73
  ]: ...
80
- @abc.abstractmethod
74
+ @_abc_1.abstractmethod
81
75
  def GetSynthesizer(
82
76
  self,
83
- request: weaveapi.synthesize.v1.service_pb2.GetSynthesizerRequest,
77
+ request: _service_pb2.GetSynthesizerRequest,
84
78
  context: _ServicerContext,
85
- ) -> typing.Union[
86
- weaveapi.synthesize.v1.service_pb2.GetSynthesizerResponse,
87
- collections.abc.Awaitable[
88
- weaveapi.synthesize.v1.service_pb2.GetSynthesizerResponse
89
- ],
79
+ ) -> _typing.Union[
80
+ _service_pb2.GetSynthesizerResponse,
81
+ _abc.Awaitable[_service_pb2.GetSynthesizerResponse],
90
82
  ]: ...
91
- @abc.abstractmethod
83
+ @_abc_1.abstractmethod
92
84
  def ListSynthesizers(
93
85
  self,
94
- request: weaveapi.synthesize.v1.service_pb2.ListSynthesizersRequest,
86
+ request: _service_pb2.ListSynthesizersRequest,
95
87
  context: _ServicerContext,
96
- ) -> typing.Union[
97
- weaveapi.synthesize.v1.service_pb2.ListSynthesizersResponse,
98
- collections.abc.Awaitable[
99
- weaveapi.synthesize.v1.service_pb2.ListSynthesizersResponse
100
- ],
88
+ ) -> _typing.Union[
89
+ _service_pb2.ListSynthesizersResponse,
90
+ _abc.Awaitable[_service_pb2.ListSynthesizersResponse],
101
91
  ]: ...
102
- @abc.abstractmethod
92
+ @_abc_1.abstractmethod
103
93
  def DeleteSynthesizer(
104
94
  self,
105
- request: weaveapi.synthesize.v1.service_pb2.DeleteSynthesizerRequest,
95
+ request: _service_pb2.DeleteSynthesizerRequest,
106
96
  context: _ServicerContext,
107
- ) -> typing.Union[
108
- weaveapi.synthesize.v1.service_pb2.DeleteSynthesizerResponse,
109
- collections.abc.Awaitable[
110
- weaveapi.synthesize.v1.service_pb2.DeleteSynthesizerResponse
111
- ],
97
+ ) -> _typing.Union[
98
+ _service_pb2.DeleteSynthesizerResponse,
99
+ _abc.Awaitable[_service_pb2.DeleteSynthesizerResponse],
112
100
  ]: ...
113
101
 
114
102
  def add_SynthesizeServicer_to_server(
115
- servicer: SynthesizeServicer, server: typing.Union[grpc.Server, grpc.aio.Server]
103
+ servicer: SynthesizeServicer, server: _typing.Union[_grpc.Server, _aio.Server]
116
104
  ) -> None: ...