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,469 +3,550 @@
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 typing
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
+ import builtins as _builtins
11
+ import sys
12
+ import typing as _typing
12
13
 
13
- DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
14
+ if sys.version_info >= (3, 10):
15
+ from typing import TypeAlias as _TypeAlias
16
+ else:
17
+ from typing_extensions import TypeAlias as _TypeAlias
14
18
 
15
- @typing.final
16
- class SqlDatabase(google.protobuf.message.Message):
17
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
19
+ DESCRIPTOR: _descriptor.FileDescriptor
18
20
 
19
- TABLES_FIELD_NUMBER: builtins.int
20
- MYSQL_FIELD_NUMBER: builtins.int
21
- POSTGRES_FIELD_NUMBER: builtins.int
22
- SQL_SERVER_FIELD_NUMBER: builtins.int
23
- ORACLE_FIELD_NUMBER: builtins.int
24
- MARIADB_FIELD_NUMBER: builtins.int
25
- SNOWFLAKE_FIELD_NUMBER: builtins.int
26
- BIG_QUERY_FIELD_NUMBER: builtins.int
27
- DATABRICKS_FIELD_NUMBER: builtins.int
28
- @property
29
- def tables(
30
- self,
31
- ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[
32
- builtins.str
33
- ]: ...
34
- @property
35
- def mysql(self) -> global___Mysql: ...
36
- @property
37
- def postgres(self) -> global___Postgres: ...
38
- @property
39
- def sql_server(self) -> global___SqlServer: ...
40
- @property
41
- def oracle(self) -> global___Oracle: ...
42
- @property
43
- def mariadb(self) -> global___MariaDB: ...
44
- @property
45
- def snowflake(self) -> global___Snowflake: ...
46
- @property
47
- def big_query(self) -> global___BigQuery: ...
48
- @property
49
- def databricks(self) -> global___Databricks: ...
21
+ @_typing.final
22
+ class SqlDatabase(_message.Message):
23
+ DESCRIPTOR: _descriptor.Descriptor
24
+
25
+ TABLES_FIELD_NUMBER: _builtins.int
26
+ MYSQL_FIELD_NUMBER: _builtins.int
27
+ POSTGRES_FIELD_NUMBER: _builtins.int
28
+ SQL_SERVER_FIELD_NUMBER: _builtins.int
29
+ ORACLE_FIELD_NUMBER: _builtins.int
30
+ MARIADB_FIELD_NUMBER: _builtins.int
31
+ SNOWFLAKE_FIELD_NUMBER: _builtins.int
32
+ BIG_QUERY_FIELD_NUMBER: _builtins.int
33
+ DATABRICKS_FIELD_NUMBER: _builtins.int
34
+ @_builtins.property
35
+ def tables(self) -> _containers.RepeatedScalarFieldContainer[_builtins.str]: ...
36
+ @_builtins.property
37
+ def mysql(self) -> Global___Mysql: ...
38
+ @_builtins.property
39
+ def postgres(self) -> Global___Postgres: ...
40
+ @_builtins.property
41
+ def sql_server(self) -> Global___SqlServer: ...
42
+ @_builtins.property
43
+ def oracle(self) -> Global___Oracle: ...
44
+ @_builtins.property
45
+ def mariadb(self) -> Global___MariaDB: ...
46
+ @_builtins.property
47
+ def snowflake(self) -> Global___Snowflake: ...
48
+ @_builtins.property
49
+ def big_query(self) -> Global___BigQuery: ...
50
+ @_builtins.property
51
+ def databricks(self) -> Global___Databricks: ...
50
52
  def __init__(
51
53
  self,
52
54
  *,
53
- tables: collections.abc.Iterable[builtins.str] | None = ...,
54
- mysql: global___Mysql | None = ...,
55
- postgres: global___Postgres | None = ...,
56
- sql_server: global___SqlServer | None = ...,
57
- oracle: global___Oracle | None = ...,
58
- mariadb: global___MariaDB | None = ...,
59
- snowflake: global___Snowflake | None = ...,
60
- big_query: global___BigQuery | None = ...,
61
- databricks: global___Databricks | None = ...,
62
- ) -> None: ...
63
- def HasField(
64
- self,
65
- field_name: typing.Literal[
66
- "big_query",
67
- b"big_query",
68
- "databricks",
69
- b"databricks",
70
- "mariadb",
71
- b"mariadb",
72
- "mysql",
73
- b"mysql",
74
- "oracle",
75
- b"oracle",
76
- "postgres",
77
- b"postgres",
78
- "relational_database",
79
- b"relational_database",
80
- "snowflake",
81
- b"snowflake",
82
- "sql_server",
83
- b"sql_server",
84
- ],
85
- ) -> builtins.bool: ...
86
- def ClearField(
87
- self,
88
- field_name: typing.Literal[
89
- "big_query",
90
- b"big_query",
91
- "databricks",
92
- b"databricks",
93
- "mariadb",
94
- b"mariadb",
95
- "mysql",
96
- b"mysql",
97
- "oracle",
98
- b"oracle",
99
- "postgres",
100
- b"postgres",
101
- "relational_database",
102
- b"relational_database",
103
- "snowflake",
104
- b"snowflake",
105
- "sql_server",
106
- b"sql_server",
107
- "tables",
108
- b"tables",
109
- ],
55
+ tables: _abc.Iterable[_builtins.str] | None = ...,
56
+ mysql: Global___Mysql | None = ...,
57
+ postgres: Global___Postgres | None = ...,
58
+ sql_server: Global___SqlServer | None = ...,
59
+ oracle: Global___Oracle | None = ...,
60
+ mariadb: Global___MariaDB | None = ...,
61
+ snowflake: Global___Snowflake | None = ...,
62
+ big_query: Global___BigQuery | None = ...,
63
+ databricks: Global___Databricks | None = ...,
110
64
  ) -> None: ...
65
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
66
+ "big_query",
67
+ b"big_query",
68
+ "databricks",
69
+ b"databricks",
70
+ "mariadb",
71
+ b"mariadb",
72
+ "mysql",
73
+ b"mysql",
74
+ "oracle",
75
+ b"oracle",
76
+ "postgres",
77
+ b"postgres",
78
+ "relational_database",
79
+ b"relational_database",
80
+ "snowflake",
81
+ b"snowflake",
82
+ "sql_server",
83
+ b"sql_server",
84
+ "tables",
85
+ b"tables",
86
+ ] # noqa: Y015
87
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
88
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
89
+ "big_query",
90
+ b"big_query",
91
+ "databricks",
92
+ b"databricks",
93
+ "mariadb",
94
+ b"mariadb",
95
+ "mysql",
96
+ b"mysql",
97
+ "oracle",
98
+ b"oracle",
99
+ "postgres",
100
+ b"postgres",
101
+ "relational_database",
102
+ b"relational_database",
103
+ "snowflake",
104
+ b"snowflake",
105
+ "sql_server",
106
+ b"sql_server",
107
+ "tables",
108
+ b"tables",
109
+ ] # noqa: Y015
110
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
111
+ _WhichOneofReturnType_relational_database: _TypeAlias = _typing.Literal[
112
+ "mysql",
113
+ "postgres",
114
+ "sql_server",
115
+ "oracle",
116
+ "mariadb",
117
+ "snowflake",
118
+ "big_query",
119
+ "databricks",
120
+ ] # noqa: Y015
121
+ _WhichOneofArgType_relational_database: _TypeAlias = _typing.Literal[
122
+ "relational_database", b"relational_database"
123
+ ] # noqa: Y015
111
124
  def WhichOneof(
112
- self, oneof_group: typing.Literal["relational_database", b"relational_database"]
113
- ) -> (
114
- typing.Literal[
115
- "mysql",
116
- "postgres",
117
- "sql_server",
118
- "oracle",
119
- "mariadb",
120
- "snowflake",
121
- "big_query",
122
- "databricks",
123
- ]
124
- | None
125
- ): ...
125
+ self, oneof_group: _WhichOneofArgType_relational_database
126
+ ) -> _WhichOneofReturnType_relational_database | None: ...
126
127
 
127
- global___SqlDatabase = SqlDatabase
128
+ Global___SqlDatabase: _TypeAlias = SqlDatabase # noqa: Y015
128
129
 
129
- @typing.final
130
- class Mysql(google.protobuf.message.Message):
131
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
130
+ @_typing.final
131
+ class Mysql(_message.Message):
132
+ DESCRIPTOR: _descriptor.Descriptor
132
133
 
133
- HOST_FIELD_NUMBER: builtins.int
134
- PORT_FIELD_NUMBER: builtins.int
135
- DATABASE_FIELD_NUMBER: builtins.int
136
- USERNAME_FIELD_NUMBER: builtins.int
137
- PASSWORD_FIELD_NUMBER: builtins.int
138
- SSL_ENABLED_FIELD_NUMBER: builtins.int
139
- host: builtins.str
140
- port: builtins.int
141
- database: builtins.str
142
- username: builtins.str
143
- password: builtins.str
144
- ssl_enabled: builtins.bool
134
+ HOST_FIELD_NUMBER: _builtins.int
135
+ PORT_FIELD_NUMBER: _builtins.int
136
+ DATABASE_FIELD_NUMBER: _builtins.int
137
+ USERNAME_FIELD_NUMBER: _builtins.int
138
+ PASSWORD_FIELD_NUMBER: _builtins.int
139
+ SSL_ENABLED_FIELD_NUMBER: _builtins.int
140
+ host: _builtins.str
141
+ port: _builtins.int
142
+ database: _builtins.str
143
+ username: _builtins.str
144
+ password: _builtins.str
145
+ ssl_enabled: _builtins.bool
145
146
  def __init__(
146
147
  self,
147
148
  *,
148
- host: builtins.str = ...,
149
- port: builtins.int = ...,
150
- database: builtins.str = ...,
151
- username: builtins.str = ...,
152
- password: builtins.str = ...,
153
- ssl_enabled: builtins.bool = ...,
154
- ) -> None: ...
155
- def ClearField(
156
- self,
157
- field_name: typing.Literal[
158
- "database",
159
- b"database",
160
- "host",
161
- b"host",
162
- "password",
163
- b"password",
164
- "port",
165
- b"port",
166
- "ssl_enabled",
167
- b"ssl_enabled",
168
- "username",
169
- b"username",
170
- ],
149
+ host: _builtins.str | None = ...,
150
+ port: _builtins.int | None = ...,
151
+ database: _builtins.str | None = ...,
152
+ username: _builtins.str | None = ...,
153
+ password: _builtins.str | None = ...,
154
+ ssl_enabled: _builtins.bool | None = ...,
171
155
  ) -> None: ...
156
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
157
+ "database",
158
+ b"database",
159
+ "host",
160
+ b"host",
161
+ "password",
162
+ b"password",
163
+ "port",
164
+ b"port",
165
+ "ssl_enabled",
166
+ b"ssl_enabled",
167
+ "username",
168
+ b"username",
169
+ ] # noqa: Y015
170
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
171
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
172
+ "database",
173
+ b"database",
174
+ "host",
175
+ b"host",
176
+ "password",
177
+ b"password",
178
+ "port",
179
+ b"port",
180
+ "ssl_enabled",
181
+ b"ssl_enabled",
182
+ "username",
183
+ b"username",
184
+ ] # noqa: Y015
185
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
172
186
 
173
- global___Mysql = Mysql
187
+ Global___Mysql: _TypeAlias = Mysql # noqa: Y015
174
188
 
175
- @typing.final
176
- class Postgres(google.protobuf.message.Message):
177
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
189
+ @_typing.final
190
+ class Postgres(_message.Message):
191
+ DESCRIPTOR: _descriptor.Descriptor
178
192
 
179
- HOST_FIELD_NUMBER: builtins.int
180
- PORT_FIELD_NUMBER: builtins.int
181
- DATABASE_FIELD_NUMBER: builtins.int
182
- USERNAME_FIELD_NUMBER: builtins.int
183
- PASSWORD_FIELD_NUMBER: builtins.int
184
- SSL_ENABLED_FIELD_NUMBER: builtins.int
185
- host: builtins.str
186
- port: builtins.int
187
- database: builtins.str
188
- username: builtins.str
189
- password: builtins.str
190
- ssl_enabled: builtins.bool
193
+ HOST_FIELD_NUMBER: _builtins.int
194
+ PORT_FIELD_NUMBER: _builtins.int
195
+ DATABASE_FIELD_NUMBER: _builtins.int
196
+ USERNAME_FIELD_NUMBER: _builtins.int
197
+ PASSWORD_FIELD_NUMBER: _builtins.int
198
+ SSL_ENABLED_FIELD_NUMBER: _builtins.int
199
+ host: _builtins.str
200
+ port: _builtins.int
201
+ database: _builtins.str
202
+ username: _builtins.str
203
+ password: _builtins.str
204
+ ssl_enabled: _builtins.bool
191
205
  def __init__(
192
206
  self,
193
207
  *,
194
- host: builtins.str = ...,
195
- port: builtins.int = ...,
196
- database: builtins.str = ...,
197
- username: builtins.str = ...,
198
- password: builtins.str = ...,
199
- ssl_enabled: builtins.bool = ...,
200
- ) -> None: ...
201
- def ClearField(
202
- self,
203
- field_name: typing.Literal[
204
- "database",
205
- b"database",
206
- "host",
207
- b"host",
208
- "password",
209
- b"password",
210
- "port",
211
- b"port",
212
- "ssl_enabled",
213
- b"ssl_enabled",
214
- "username",
215
- b"username",
216
- ],
208
+ host: _builtins.str | None = ...,
209
+ port: _builtins.int | None = ...,
210
+ database: _builtins.str | None = ...,
211
+ username: _builtins.str | None = ...,
212
+ password: _builtins.str | None = ...,
213
+ ssl_enabled: _builtins.bool | None = ...,
217
214
  ) -> None: ...
215
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
216
+ "database",
217
+ b"database",
218
+ "host",
219
+ b"host",
220
+ "password",
221
+ b"password",
222
+ "port",
223
+ b"port",
224
+ "ssl_enabled",
225
+ b"ssl_enabled",
226
+ "username",
227
+ b"username",
228
+ ] # noqa: Y015
229
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
230
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
231
+ "database",
232
+ b"database",
233
+ "host",
234
+ b"host",
235
+ "password",
236
+ b"password",
237
+ "port",
238
+ b"port",
239
+ "ssl_enabled",
240
+ b"ssl_enabled",
241
+ "username",
242
+ b"username",
243
+ ] # noqa: Y015
244
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
218
245
 
219
- global___Postgres = Postgres
246
+ Global___Postgres: _TypeAlias = Postgres # noqa: Y015
220
247
 
221
- @typing.final
222
- class SqlServer(google.protobuf.message.Message):
223
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
248
+ @_typing.final
249
+ class SqlServer(_message.Message):
250
+ DESCRIPTOR: _descriptor.Descriptor
224
251
 
225
- HOST_FIELD_NUMBER: builtins.int
226
- PORT_FIELD_NUMBER: builtins.int
227
- DATABASE_FIELD_NUMBER: builtins.int
228
- USERNAME_FIELD_NUMBER: builtins.int
229
- PASSWORD_FIELD_NUMBER: builtins.int
230
- ENCRYPT_CONNECTION_FIELD_NUMBER: builtins.int
231
- host: builtins.str
232
- port: builtins.int
233
- database: builtins.str
234
- username: builtins.str
235
- password: builtins.str
236
- encrypt_connection: builtins.bool
252
+ HOST_FIELD_NUMBER: _builtins.int
253
+ PORT_FIELD_NUMBER: _builtins.int
254
+ DATABASE_FIELD_NUMBER: _builtins.int
255
+ USERNAME_FIELD_NUMBER: _builtins.int
256
+ PASSWORD_FIELD_NUMBER: _builtins.int
257
+ ENCRYPT_CONNECTION_FIELD_NUMBER: _builtins.int
258
+ host: _builtins.str
259
+ port: _builtins.int
260
+ database: _builtins.str
261
+ username: _builtins.str
262
+ password: _builtins.str
263
+ encrypt_connection: _builtins.bool
237
264
  def __init__(
238
265
  self,
239
266
  *,
240
- host: builtins.str = ...,
241
- port: builtins.int = ...,
242
- database: builtins.str = ...,
243
- username: builtins.str = ...,
244
- password: builtins.str = ...,
245
- encrypt_connection: builtins.bool = ...,
246
- ) -> None: ...
247
- def ClearField(
248
- self,
249
- field_name: typing.Literal[
250
- "database",
251
- b"database",
252
- "encrypt_connection",
253
- b"encrypt_connection",
254
- "host",
255
- b"host",
256
- "password",
257
- b"password",
258
- "port",
259
- b"port",
260
- "username",
261
- b"username",
262
- ],
267
+ host: _builtins.str | None = ...,
268
+ port: _builtins.int | None = ...,
269
+ database: _builtins.str | None = ...,
270
+ username: _builtins.str | None = ...,
271
+ password: _builtins.str | None = ...,
272
+ encrypt_connection: _builtins.bool | None = ...,
263
273
  ) -> None: ...
274
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
275
+ "database",
276
+ b"database",
277
+ "encrypt_connection",
278
+ b"encrypt_connection",
279
+ "host",
280
+ b"host",
281
+ "password",
282
+ b"password",
283
+ "port",
284
+ b"port",
285
+ "username",
286
+ b"username",
287
+ ] # noqa: Y015
288
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
289
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
290
+ "database",
291
+ b"database",
292
+ "encrypt_connection",
293
+ b"encrypt_connection",
294
+ "host",
295
+ b"host",
296
+ "password",
297
+ b"password",
298
+ "port",
299
+ b"port",
300
+ "username",
301
+ b"username",
302
+ ] # noqa: Y015
303
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
264
304
 
265
- global___SqlServer = SqlServer
305
+ Global___SqlServer: _TypeAlias = SqlServer # noqa: Y015
266
306
 
267
- @typing.final
268
- class Oracle(google.protobuf.message.Message):
269
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
307
+ @_typing.final
308
+ class Oracle(_message.Message):
309
+ DESCRIPTOR: _descriptor.Descriptor
270
310
 
271
- CONNECTION_STRING_FIELD_NUMBER: builtins.int
272
- USERNAME_FIELD_NUMBER: builtins.int
273
- PASSWORD_FIELD_NUMBER: builtins.int
274
- connection_string: builtins.str
275
- username: builtins.str
276
- password: builtins.str
311
+ CONNECTION_STRING_FIELD_NUMBER: _builtins.int
312
+ USERNAME_FIELD_NUMBER: _builtins.int
313
+ PASSWORD_FIELD_NUMBER: _builtins.int
314
+ connection_string: _builtins.str
315
+ username: _builtins.str
316
+ password: _builtins.str
277
317
  def __init__(
278
318
  self,
279
319
  *,
280
- connection_string: builtins.str = ...,
281
- username: builtins.str = ...,
282
- password: builtins.str = ...,
283
- ) -> None: ...
284
- def ClearField(
285
- self,
286
- field_name: typing.Literal[
287
- "connection_string",
288
- b"connection_string",
289
- "password",
290
- b"password",
291
- "username",
292
- b"username",
293
- ],
320
+ connection_string: _builtins.str | None = ...,
321
+ username: _builtins.str | None = ...,
322
+ password: _builtins.str | None = ...,
294
323
  ) -> None: ...
324
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
325
+ "connection_string",
326
+ b"connection_string",
327
+ "password",
328
+ b"password",
329
+ "username",
330
+ b"username",
331
+ ] # noqa: Y015
332
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
333
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
334
+ "connection_string",
335
+ b"connection_string",
336
+ "password",
337
+ b"password",
338
+ "username",
339
+ b"username",
340
+ ] # noqa: Y015
341
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
295
342
 
296
- global___Oracle = Oracle
343
+ Global___Oracle: _TypeAlias = Oracle # noqa: Y015
297
344
 
298
- @typing.final
299
- class MariaDB(google.protobuf.message.Message):
300
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
345
+ @_typing.final
346
+ class MariaDB(_message.Message):
347
+ DESCRIPTOR: _descriptor.Descriptor
301
348
 
302
- HOST_FIELD_NUMBER: builtins.int
303
- PORT_FIELD_NUMBER: builtins.int
304
- DATABASE_FIELD_NUMBER: builtins.int
305
- USERNAME_FIELD_NUMBER: builtins.int
306
- PASSWORD_FIELD_NUMBER: builtins.int
307
- SSL_ENABLED_FIELD_NUMBER: builtins.int
308
- host: builtins.str
309
- port: builtins.int
310
- database: builtins.str
311
- username: builtins.str
312
- password: builtins.str
313
- ssl_enabled: builtins.bool
349
+ HOST_FIELD_NUMBER: _builtins.int
350
+ PORT_FIELD_NUMBER: _builtins.int
351
+ DATABASE_FIELD_NUMBER: _builtins.int
352
+ USERNAME_FIELD_NUMBER: _builtins.int
353
+ PASSWORD_FIELD_NUMBER: _builtins.int
354
+ SSL_ENABLED_FIELD_NUMBER: _builtins.int
355
+ host: _builtins.str
356
+ port: _builtins.int
357
+ database: _builtins.str
358
+ username: _builtins.str
359
+ password: _builtins.str
360
+ ssl_enabled: _builtins.bool
314
361
  def __init__(
315
362
  self,
316
363
  *,
317
- host: builtins.str = ...,
318
- port: builtins.int = ...,
319
- database: builtins.str = ...,
320
- username: builtins.str = ...,
321
- password: builtins.str = ...,
322
- ssl_enabled: builtins.bool = ...,
323
- ) -> None: ...
324
- def ClearField(
325
- self,
326
- field_name: typing.Literal[
327
- "database",
328
- b"database",
329
- "host",
330
- b"host",
331
- "password",
332
- b"password",
333
- "port",
334
- b"port",
335
- "ssl_enabled",
336
- b"ssl_enabled",
337
- "username",
338
- b"username",
339
- ],
364
+ host: _builtins.str | None = ...,
365
+ port: _builtins.int | None = ...,
366
+ database: _builtins.str | None = ...,
367
+ username: _builtins.str | None = ...,
368
+ password: _builtins.str | None = ...,
369
+ ssl_enabled: _builtins.bool | None = ...,
340
370
  ) -> None: ...
371
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
372
+ "database",
373
+ b"database",
374
+ "host",
375
+ b"host",
376
+ "password",
377
+ b"password",
378
+ "port",
379
+ b"port",
380
+ "ssl_enabled",
381
+ b"ssl_enabled",
382
+ "username",
383
+ b"username",
384
+ ] # noqa: Y015
385
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
386
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
387
+ "database",
388
+ b"database",
389
+ "host",
390
+ b"host",
391
+ "password",
392
+ b"password",
393
+ "port",
394
+ b"port",
395
+ "ssl_enabled",
396
+ b"ssl_enabled",
397
+ "username",
398
+ b"username",
399
+ ] # noqa: Y015
400
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
341
401
 
342
- global___MariaDB = MariaDB
402
+ Global___MariaDB: _TypeAlias = MariaDB # noqa: Y015
343
403
 
344
- @typing.final
345
- class Snowflake(google.protobuf.message.Message):
346
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
404
+ @_typing.final
405
+ class Snowflake(_message.Message):
406
+ DESCRIPTOR: _descriptor.Descriptor
347
407
 
348
- ACCOUNT_FIELD_NUMBER: builtins.int
349
- WAREHOUSE_FIELD_NUMBER: builtins.int
350
- DATABASE_FIELD_NUMBER: builtins.int
351
- SCHEMA_FIELD_NUMBER: builtins.int
352
- ROLE_FIELD_NUMBER: builtins.int
353
- USERNAME_FIELD_NUMBER: builtins.int
354
- PASSWORD_FIELD_NUMBER: builtins.int
355
- account: builtins.str
356
- warehouse: builtins.str
357
- database: builtins.str
358
- schema: builtins.str
359
- role: builtins.str
360
- username: builtins.str
361
- password: builtins.str
408
+ ACCOUNT_FIELD_NUMBER: _builtins.int
409
+ WAREHOUSE_FIELD_NUMBER: _builtins.int
410
+ DATABASE_FIELD_NUMBER: _builtins.int
411
+ SCHEMA_FIELD_NUMBER: _builtins.int
412
+ ROLE_FIELD_NUMBER: _builtins.int
413
+ USERNAME_FIELD_NUMBER: _builtins.int
414
+ PASSWORD_FIELD_NUMBER: _builtins.int
415
+ account: _builtins.str
416
+ warehouse: _builtins.str
417
+ database: _builtins.str
418
+ schema: _builtins.str
419
+ role: _builtins.str
420
+ username: _builtins.str
421
+ password: _builtins.str
362
422
  def __init__(
363
423
  self,
364
424
  *,
365
- account: builtins.str = ...,
366
- warehouse: builtins.str = ...,
367
- database: builtins.str = ...,
368
- schema: builtins.str = ...,
369
- role: builtins.str = ...,
370
- username: builtins.str = ...,
371
- password: builtins.str = ...,
372
- ) -> None: ...
373
- def ClearField(
374
- self,
375
- field_name: typing.Literal[
376
- "account",
377
- b"account",
378
- "database",
379
- b"database",
380
- "password",
381
- b"password",
382
- "role",
383
- b"role",
384
- "schema",
385
- b"schema",
386
- "username",
387
- b"username",
388
- "warehouse",
389
- b"warehouse",
390
- ],
425
+ account: _builtins.str | None = ...,
426
+ warehouse: _builtins.str | None = ...,
427
+ database: _builtins.str | None = ...,
428
+ schema: _builtins.str | None = ...,
429
+ role: _builtins.str | None = ...,
430
+ username: _builtins.str | None = ...,
431
+ password: _builtins.str | None = ...,
391
432
  ) -> None: ...
433
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
434
+ "account",
435
+ b"account",
436
+ "database",
437
+ b"database",
438
+ "password",
439
+ b"password",
440
+ "role",
441
+ b"role",
442
+ "schema",
443
+ b"schema",
444
+ "username",
445
+ b"username",
446
+ "warehouse",
447
+ b"warehouse",
448
+ ] # noqa: Y015
449
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
450
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
451
+ "account",
452
+ b"account",
453
+ "database",
454
+ b"database",
455
+ "password",
456
+ b"password",
457
+ "role",
458
+ b"role",
459
+ "schema",
460
+ b"schema",
461
+ "username",
462
+ b"username",
463
+ "warehouse",
464
+ b"warehouse",
465
+ ] # noqa: Y015
466
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
392
467
 
393
- global___Snowflake = Snowflake
468
+ Global___Snowflake: _TypeAlias = Snowflake # noqa: Y015
394
469
 
395
- @typing.final
396
- class BigQuery(google.protobuf.message.Message):
397
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
470
+ @_typing.final
471
+ class BigQuery(_message.Message):
472
+ DESCRIPTOR: _descriptor.Descriptor
398
473
 
399
- PROJECT_ID_FIELD_NUMBER: builtins.int
400
- DATASET_FIELD_NUMBER: builtins.int
401
- SERVICE_ACCOUNT_JSON_FIELD_NUMBER: builtins.int
402
- USE_APPLICATION_DEFAULT_FIELD_NUMBER: builtins.int
403
- project_id: builtins.str
404
- dataset: builtins.str
405
- service_account_json: builtins.str
406
- use_application_default: builtins.bool
474
+ PROJECT_ID_FIELD_NUMBER: _builtins.int
475
+ DATASET_FIELD_NUMBER: _builtins.int
476
+ SERVICE_ACCOUNT_JSON_FIELD_NUMBER: _builtins.int
477
+ USE_APPLICATION_DEFAULT_FIELD_NUMBER: _builtins.int
478
+ project_id: _builtins.str
479
+ dataset: _builtins.str
480
+ service_account_json: _builtins.str
481
+ use_application_default: _builtins.bool
407
482
  def __init__(
408
483
  self,
409
484
  *,
410
- project_id: builtins.str = ...,
411
- dataset: builtins.str = ...,
412
- service_account_json: builtins.str = ...,
413
- use_application_default: builtins.bool = ...,
414
- ) -> None: ...
415
- def HasField(
416
- self,
417
- field_name: typing.Literal[
418
- "auth",
419
- b"auth",
420
- "service_account_json",
421
- b"service_account_json",
422
- "use_application_default",
423
- b"use_application_default",
424
- ],
425
- ) -> builtins.bool: ...
426
- def ClearField(
427
- self,
428
- field_name: typing.Literal[
429
- "auth",
430
- b"auth",
431
- "dataset",
432
- b"dataset",
433
- "project_id",
434
- b"project_id",
435
- "service_account_json",
436
- b"service_account_json",
437
- "use_application_default",
438
- b"use_application_default",
439
- ],
485
+ project_id: _builtins.str | None = ...,
486
+ dataset: _builtins.str | None = ...,
487
+ service_account_json: _builtins.str | None = ...,
488
+ use_application_default: _builtins.bool | None = ...,
440
489
  ) -> None: ...
490
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
491
+ "auth",
492
+ b"auth",
493
+ "dataset",
494
+ b"dataset",
495
+ "project_id",
496
+ b"project_id",
497
+ "service_account_json",
498
+ b"service_account_json",
499
+ "use_application_default",
500
+ b"use_application_default",
501
+ ] # noqa: Y015
502
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
503
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
504
+ "auth",
505
+ b"auth",
506
+ "dataset",
507
+ b"dataset",
508
+ "project_id",
509
+ b"project_id",
510
+ "service_account_json",
511
+ b"service_account_json",
512
+ "use_application_default",
513
+ b"use_application_default",
514
+ ] # noqa: Y015
515
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
516
+ _WhichOneofReturnType_auth: _TypeAlias = _typing.Literal[
517
+ "service_account_json", "use_application_default"
518
+ ] # noqa: Y015
519
+ _WhichOneofArgType_auth: _TypeAlias = _typing.Literal["auth", b"auth"] # noqa: Y015
441
520
  def WhichOneof(
442
- self, oneof_group: typing.Literal["auth", b"auth"]
443
- ) -> typing.Literal["service_account_json", "use_application_default"] | None: ...
521
+ self, oneof_group: _WhichOneofArgType_auth
522
+ ) -> _WhichOneofReturnType_auth | None: ...
444
523
 
445
- global___BigQuery = BigQuery
524
+ Global___BigQuery: _TypeAlias = BigQuery # noqa: Y015
446
525
 
447
- @typing.final
448
- class Databricks(google.protobuf.message.Message):
449
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
526
+ @_typing.final
527
+ class Databricks(_message.Message):
528
+ DESCRIPTOR: _descriptor.Descriptor
450
529
 
451
- HOST_FIELD_NUMBER: builtins.int
452
- HTTP_PATH_FIELD_NUMBER: builtins.int
453
- TOKEN_FIELD_NUMBER: builtins.int
454
- host: builtins.str
455
- http_path: builtins.str
456
- token: builtins.str
530
+ HOST_FIELD_NUMBER: _builtins.int
531
+ HTTP_PATH_FIELD_NUMBER: _builtins.int
532
+ TOKEN_FIELD_NUMBER: _builtins.int
533
+ host: _builtins.str
534
+ http_path: _builtins.str
535
+ token: _builtins.str
457
536
  def __init__(
458
537
  self,
459
538
  *,
460
- host: builtins.str = ...,
461
- http_path: builtins.str = ...,
462
- token: builtins.str = ...,
463
- ) -> None: ...
464
- def ClearField(
465
- self,
466
- field_name: typing.Literal[
467
- "host", b"host", "http_path", b"http_path", "token", b"token"
468
- ],
539
+ host: _builtins.str | None = ...,
540
+ http_path: _builtins.str | None = ...,
541
+ token: _builtins.str | None = ...,
469
542
  ) -> None: ...
543
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
544
+ "host", b"host", "http_path", b"http_path", "token", b"token"
545
+ ] # noqa: Y015
546
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
547
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
548
+ "host", b"host", "http_path", b"http_path", "token", b"token"
549
+ ] # noqa: Y015
550
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
470
551
 
471
- global___Databricks = Databricks
552
+ Global___Databricks: _TypeAlias = Databricks # noqa: Y015