weave-python 0.15.0__tar.gz → 0.17.0__tar.gz

{weave_python-0.15.0 → weave_python-0.17.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weave-python
-Version: 0.15.0
+Version: 0.17.0
 Requires-Python: >=3.10
 Requires-Dist: protobuf>=6.30.2
 Requires-Dist: sqlalchemy>=2.0.40

{weave_python-0.15.0 → weave_python-0.17.0}/weave/datamanagement/db/models.py

@@ -115,7 +115,7 @@ class SynthesizeV1Dataset:
 
 
 @dataclasses.dataclass()
-class SynthesizeV1Model:
+class SynthesizeV1Synthesizer:
     id: uuid.UUID
     organization_id: uuid.UUID
     name: str

{weave_python-0.15.0 → weave_python-0.17.0}/weave/datamanagement/db/queries.py

@@ -13,21 +13,18 @@ from weave.datamanagement.db import models
 
 
 GET_FOREIGN_KEY_RELATIONSHIPS = """-- name: get_foreign_key_relationships \\:many
-SELECT 
-    cr.source_dataset_id,
-    cr.target_dataset_id,
-    sc.name AS source_column_name,
-    tc.name AS target_column_name
-FROM 
-    synthesize_v1.column_relationship cr
-JOIN 
-    synthesize_v1.data_column sc ON cr.source_column_id = sc.id
-JOIN 
-    synthesize_v1.data_column tc ON cr.target_column_id = tc.id
-WHERE 
-    cr.relationship_type IN ('context_foreign_key', 'non_context_foreign_key')
-    AND cr.source_dataset_id = ANY(:p1\\:\\:UUID[])
-    AND cr.target_dataset_id = ANY(:p1\\:\\:UUID[])
+SELECT cr.source_dataset_id,
+       cr.target_dataset_id,
+       sc.name AS source_column_name,
+       tc.name AS target_column_name
+FROM synthesize_v1.column_relationship cr
+         JOIN
+     synthesize_v1.data_column sc ON cr.source_column_id = sc.id
+         JOIN
+     synthesize_v1.data_column tc ON cr.target_column_id = tc.id
+WHERE cr.relationship_type IN ('context_foreign_key', 'non_context_foreign_key')
+  AND cr.source_dataset_id = ANY (:p1\\:\\:UUID[])
+  AND cr.target_dataset_id = ANY (:p1\\:\\:UUID[])
 """
 
 
@@ -49,24 +46,55 @@ WHERE id = :p1
 GET_STORAGE_CONNECTION_BY_DATASET_ID = """-- name: get_storage_connection_by_dataset_id \\:one
 SELECT sc.id, sc.organization_id, sc.name, sc.description, sc.category, sc.variant, sc.details
 FROM storage_connection sc
-JOIN synthesize_v1.dataset d ON d.storage_connection_id = sc.id
+         JOIN synthesize_v1.dataset d ON d.storage_connection_id = sc.id
 WHERE d.id = :p1
 """
 
 
-UPSERT_MODEL = """-- name: upsert_model \\:exec
-INSERT INTO synthesize_v1.model (id, organization_id, name, description, storage_connection_id, metadata)
+GET_SYNTHESIZER = """-- name: get_synthesizer \\:exec
+SELECT id, organization_id, name, description, storage_connection_id, metadata
+FROM synthesize_v1.synthesizer
+WHERE id = :p1
+"""
+
+
+UPSERT_DATASET = """-- name: upsert_dataset \\:exec
+INSERT INTO synthesize_v1.dataset (id, organization_id, name, description, storage_connection_id, schema_id, model_id,
+                                   is_synthetic)
+VALUES (:p1, :p2, :p3, :p4, :p5, :p6, :p7, :p8)
+ON CONFLICT (id) DO UPDATE SET name                  = EXCLUDED.name,
+                               description           = EXCLUDED.description,
+                               storage_connection_id = EXCLUDED.storage_connection_id,
+                               schema_id             = EXCLUDED.schema_id,
+                               model_id              = EXCLUDED.model_id,
+                               is_synthetic          = EXCLUDED.is_synthetic
+"""
+
+
+@dataclasses.dataclass()
+class UpsertDatasetParams:
+    id: uuid.UUID
+    organization_id: uuid.UUID
+    name: str
+    description: Optional[str]
+    storage_connection_id: uuid.UUID
+    schema_id: uuid.UUID
+    model_id: Optional[uuid.UUID]
+    is_synthetic: Optional[bool]
+
+
+UPSERT_SYNTHESIZER = """-- name: upsert_synthesizer \\:exec
+INSERT INTO synthesize_v1.synthesizer (id, organization_id, name, description, storage_connection_id, metadata)
 VALUES (:p1, :p2, :p3, :p4, :p5, :p6)
-ON CONFLICT (id) DO UPDATE SET
-    name = EXCLUDED.name,
-    description = EXCLUDED.description,
-    storage_connection_id = EXCLUDED.storage_connection_id,
-    metadata = EXCLUDED.metadata
+ON CONFLICT (id) DO UPDATE SET name                  = EXCLUDED.name,
+                               description           = EXCLUDED.description,
+                               storage_connection_id = EXCLUDED.storage_connection_id,
+                               metadata              = EXCLUDED.metadata
 """
 
 
 @dataclasses.dataclass()
-class UpsertModelParams:
+class UpsertSynthesizerParams:
     id: uuid.UUID
     organization_id: uuid.UUID
     name: str
@@ -129,9 +157,27 @@ class Querier:
             details=row[6],
         )
 
-    def upsert_model(self, arg: UpsertModelParams) -> None:
+    def get_synthesizer(self, *, id: uuid.UUID) -> None:
+        self._conn.execute(sqlalchemy.text(GET_SYNTHESIZER), {"p1": id})
+
+    def upsert_dataset(self, arg: UpsertDatasetParams) -> None:
         self._conn.execute(
-            sqlalchemy.text(UPSERT_MODEL),
+            sqlalchemy.text(UPSERT_DATASET),
+            {
+                "p1": arg.id,
+                "p2": arg.organization_id,
+                "p3": arg.name,
+                "p4": arg.description,
+                "p5": arg.storage_connection_id,
+                "p6": arg.schema_id,
+                "p7": arg.model_id,
+                "p8": arg.is_synthetic,
+            },
+        )
+
+    def upsert_synthesizer(self, arg: UpsertSynthesizerParams) -> None:
+        self._conn.execute(
+            sqlalchemy.text(UPSERT_SYNTHESIZER),
             {
                 "p1": arg.id,
                 "p2": arg.organization_id,
@@ -202,9 +248,27 @@ class AsyncQuerier:
             details=row[6],
         )
 
-    async def upsert_model(self, arg: UpsertModelParams) -> None:
+    async def get_synthesizer(self, *, id: uuid.UUID) -> None:
+        await self._conn.execute(sqlalchemy.text(GET_SYNTHESIZER), {"p1": id})
+
+    async def upsert_dataset(self, arg: UpsertDatasetParams) -> None:
+        await self._conn.execute(
+            sqlalchemy.text(UPSERT_DATASET),
+            {
+                "p1": arg.id,
+                "p2": arg.organization_id,
+                "p3": arg.name,
+                "p4": arg.description,
+                "p5": arg.storage_connection_id,
+                "p6": arg.schema_id,
+                "p7": arg.model_id,
+                "p8": arg.is_synthetic,
+            },
+        )
+
+    async def upsert_synthesizer(self, arg: UpsertSynthesizerParams) -> None:
         await self._conn.execute(
-            sqlalchemy.text(UPSERT_MODEL),
+            sqlalchemy.text(UPSERT_SYNTHESIZER),
             {
                 "p1": arg.id,
                 "p2": arg.organization_id,

weave_python-0.17.0/weave/datamanagement/generate/v1/configuration_pb2.py

@@ -0,0 +1,57 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: weave/datamanagement/generate/v1/configuration.proto
+# Protobuf Python Version: 6.30.2
+"""Generated protocol buffer code."""
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    30,
+    2,
+    "",
+    "weave/datamanagement/generate/v1/configuration.proto",
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n4weave/datamanagement/generate/v1/configuration.proto\x12 weave.datamanagement.generate.v1"\x98\x02\n\rConfiguration\x12\x1f\n\x0bnum_records\x18\x01 \x01(\x03R\nnumRecords\x12V\n\x0c\x64\x61ta_quality\x18\x03 \x01(\x0b\x32\x33.weave.datamanagement.generate.v1.DataQualityConfigR\x0b\x64\x61taQuality\x12I\n\x07privacy\x18\x04 \x01(\x0b\x32/.weave.datamanagement.generate.v1.PrivacyConfigR\x07privacy\x12\x43\n\x05model\x18\x05 \x01(\x0b\x32-.weave.datamanagement.generate.v1.ModelConfigR\x05model"\xb1\x02\n\x11\x44\x61taQualityConfig\x12\x31\n\x14\x63orrelation_accuracy\x18\x01 \x01(\x02R\x13\x63orrelationAccuracy\x12\x35\n\x16statistical_similarity\x18\x02 \x01(\x02R\x15statisticalSimilarity\x12+\n\x11preserve_outliers\x18\x03 \x01(\x08R\x10preserveOutliers\x12/\n\x13\x65nforce_constraints\x18\x04 \x01(\x08R\x12\x65nforceConstraints\x12T\n\x0b\x63onstraints\x18\x05 \x03(\x0b\x32\x32.weave.datamanagement.generate.v1.ColumnConstraintR\x0b\x63onstraints"\xe2\x01\n\x10\x43olumnConstraint\x12\x1f\n\x0b\x63olumn_name\x18\x01 \x01(\tR\ncolumnName\x12\'\n\x0f\x63onstraint_type\x18\x02 \x01(\tR\x0e\x63onstraintType\x12\x1b\n\tmin_value\x18\x03 \x01(\tR\x08minValue\x12\x1b\n\tmax_value\x18\x04 \x01(\tR\x08maxValue\x12#\n\rregex_pattern\x18\x05 \x01(\tR\x0cregexPattern\x12%\n\x0e\x61llowed_values\x18\x06 \x03(\tR\rallowedValues"\xd1\x01\n\rPrivacyConfig\x12\x18\n\x07\x65psilon\x18\x01 \x01(\x02R\x07\x65psilon\x12*\n\x11k_anonymity_check\x18\x02 \x01(\x08R\x0fkAnonymityCheck\x12\x17\n\x07k_value\x18\x03 \x01(\x05R\x06kValue\x12+\n\x11sensitive_columns\x18\x04 \x03(\tR\x10sensitiveColumns\x12\x34\n\x16\x65nable_privacy_metrics\x18\x05 \x01(\x08R\x14\x65nablePrivacyMetrics"\xec\x02\n\x0bModelConfig\x12J\n\nmodel_type\x18\x01 \x01(\x0e\x32+.weave.datamanagement.generate.v1.ModelTypeR\tmodelType\x12\x16\n\x06\x65pochs\x18\x02 \x01(\x05R\x06\x65pochs\x12#\n\rlearning_rate\x18\x03 \x01(\x02R\x0clearningRate\x12\x1d\n\nbatch_size\x18\x04 \x01(\x05R\tbatchSize\x12p\n\x11\x63ustom_parameters\x18\x05 \x03(\x0b\x32\x43.weave.datamanagement.generate.v1.ModelConfig.CustomParametersEntryR\x10\x63ustomParameters\x1a\x43\n\x15\x43ustomParametersEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01*\xea\x01\n\tModelType\x12\x1a\n\x16MODEL_TYPE_UNSPECIFIED\x10\x00\x12\x14\n\x10MODEL_TYPE_CTGAN\x10\x01\x12\x13\n\x0fMODEL_TYPE_TVAE\x10\x02\x12\x15\n\x11MODEL_TYPE_COPULA\x10\x03\x12\x13\n\x0fMODEL_TYPE_FAST\x10\x04\x12\x12\n\x0eMODEL_TYPE_GAN\x10\x05\x12\x13\n\x0fMODEL_TYPE_WGAN\x10\x06\x12\x14\n\x10MODEL_TYPE_SMOTE\x10\x07\x12\x17\n\x13MODEL_TYPE_BAYESIAN\x10\x08\x12\x12\n\x0eMODEL_TYPE_HMA\x10\tB\xab\x02\n$com.weave.datamanagement.generate.v1B\x12\x43onfigurationProtoP\x01ZLgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/generate/v1;generate\xa2\x02\x03WDG\xaa\x02 Weave.Datamanagement.Generate.V1\xca\x02 Weave\\Datamanagement\\Generate\\V1\xe2\x02,Weave\\Datamanagement\\Generate\\V1\\GPBMetadata\xea\x02#Weave::Datamanagement::Generate::V1b\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(
+    DESCRIPTOR, "weave.datamanagement.generate.v1.configuration_pb2", _globals
+)
+if not _descriptor._USE_C_DESCRIPTORS:
+    _globals["DESCRIPTOR"]._loaded_options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n$com.weave.datamanagement.generate.v1B\022ConfigurationProtoP\001ZLgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/generate/v1;generate\242\002\003WDG\252\002 Weave.Datamanagement.Generate.V1\312\002 Weave\\Datamanagement\\Generate\\V1\342\002,Weave\\Datamanagement\\Generate\\V1\\GPBMetadata\352\002#Weave::Datamanagement::Generate::V1"
+    _globals["_MODELCONFIG_CUSTOMPARAMETERSENTRY"]._loaded_options = None
+    _globals["_MODELCONFIG_CUSTOMPARAMETERSENTRY"]._serialized_options = b"8\001"
+    _globals["_MODELTYPE"]._serialized_start = 1490
+    _globals["_MODELTYPE"]._serialized_end = 1724
+    _globals["_CONFIGURATION"]._serialized_start = 91
+    _globals["_CONFIGURATION"]._serialized_end = 371
+    _globals["_DATAQUALITYCONFIG"]._serialized_start = 374
+    _globals["_DATAQUALITYCONFIG"]._serialized_end = 679
+    _globals["_COLUMNCONSTRAINT"]._serialized_start = 682
+    _globals["_COLUMNCONSTRAINT"]._serialized_end = 908
+    _globals["_PRIVACYCONFIG"]._serialized_start = 911
+    _globals["_PRIVACYCONFIG"]._serialized_end = 1120
+    _globals["_MODELCONFIG"]._serialized_start = 1123
+    _globals["_MODELCONFIG"]._serialized_end = 1487
+    _globals["_MODELCONFIG_CUSTOMPARAMETERSENTRY"]._serialized_start = 1420
+    _globals["_MODELCONFIG_CUSTOMPARAMETERSENTRY"]._serialized_end = 1487
+# @@protoc_insertion_point(module_scope)

{weave_python-0.15.0 → weave_python-0.17.0}/weave/datamanagement/generate/v1/configuration_pb2.pyi

@@ -79,7 +79,6 @@ class Configuration(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     NUM_RECORDS_FIELD_NUMBER: builtins.int
-    DATASET_NAME_FIELD_NUMBER: builtins.int
     DATA_QUALITY_FIELD_NUMBER: builtins.int
     PRIVACY_FIELD_NUMBER: builtins.int
     MODEL_FIELD_NUMBER: builtins.int
@@ -87,8 +86,6 @@ class Configuration(google.protobuf.message.Message):
     """Basic configuration
     Number of synthetic records to generate
     """
-    dataset_name: builtins.str
-    """Name for the generated dataset"""
     @property
     def data_quality(self) -> global___DataQualityConfig:
         """Data quality settings
@@ -111,7 +108,6 @@ class Configuration(google.protobuf.message.Message):
         self,
         *,
         num_records: builtins.int = ...,
-        dataset_name: builtins.str = ...,
         data_quality: global___DataQualityConfig | None = ...,
         privacy: global___PrivacyConfig | None = ...,
         model: global___ModelConfig | None = ...,
@@ -127,8 +123,6 @@ class Configuration(google.protobuf.message.Message):
         field_name: typing.Literal[
             "data_quality",
             b"data_quality",
-            "dataset_name",
-            b"dataset_name",
             "model",
             b"model",
             "num_records",

weave_python-0.17.0/weave/datamanagement/generate/v1/generate_pb2.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: weave/datamanagement/generate/v1/generate.proto
+# Protobuf Python Version: 6.30.2
+"""Generated protocol buffer code."""
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    30,
+    2,
+    "",
+    "weave/datamanagement/generate/v1/generate.proto",
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n/weave/datamanagement/generate/v1/generate.proto\x12 weave.datamanagement.generate.v1\x1a\x1cgoogle/protobuf/struct.proto"\xb1\x01\n\x1bStartDatasetGenerationEvent\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12%\n\x0esynthesizer_id\x18\x02 \x01(\tR\rsynthesizerId\x12*\n\x11output_storage_id\x18\x03 \x01(\tR\x0foutputStorageId\x12/\n\x06\x63onfig\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\x06\x63onfigB\xa6\x02\n$com.weave.datamanagement.generate.v1B\rGenerateProtoP\x01ZLgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/generate/v1;generate\xa2\x02\x03WDG\xaa\x02 Weave.Datamanagement.Generate.V1\xca\x02 Weave\\Datamanagement\\Generate\\V1\xe2\x02,Weave\\Datamanagement\\Generate\\V1\\GPBMetadata\xea\x02#Weave::Datamanagement::Generate::V1b\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(
+    DESCRIPTOR, "weave.datamanagement.generate.v1.generate_pb2", _globals
+)
+if not _descriptor._USE_C_DESCRIPTORS:
+    _globals["DESCRIPTOR"]._loaded_options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n$com.weave.datamanagement.generate.v1B\rGenerateProtoP\001ZLgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/generate/v1;generate\242\002\003WDG\252\002 Weave.Datamanagement.Generate.V1\312\002 Weave\\Datamanagement\\Generate\\V1\342\002,Weave\\Datamanagement\\Generate\\V1\\GPBMetadata\352\002#Weave::Datamanagement::Generate::V1"
+    _globals["_STARTDATASETGENERATIONEVENT"]._serialized_start = 116
+    _globals["_STARTDATASETGENERATIONEVENT"]._serialized_end = 293
+# @@protoc_insertion_point(module_scope)

weave_python-0.17.0/weave/datamanagement/generate/v1/generate_pb2.pyi

@@ -0,0 +1,52 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import google.protobuf.struct_pb2
+import typing
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class StartDatasetGenerationEvent(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ID_FIELD_NUMBER: builtins.int
+    SYNTHESIZER_ID_FIELD_NUMBER: builtins.int
+    OUTPUT_STORAGE_ID_FIELD_NUMBER: builtins.int
+    CONFIG_FIELD_NUMBER: builtins.int
+    id: builtins.str
+    synthesizer_id: builtins.str
+    output_storage_id: builtins.str
+    @property
+    def config(self) -> google.protobuf.struct_pb2.Struct: ...
+    def __init__(
+        self,
+        *,
+        id: builtins.str = ...,
+        synthesizer_id: builtins.str = ...,
+        output_storage_id: builtins.str = ...,
+        config: google.protobuf.struct_pb2.Struct | None = ...,
+    ) -> None: ...
+    def HasField(
+        self, field_name: typing.Literal["config", b"config"]
+    ) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "config",
+            b"config",
+            "id",
+            b"id",
+            "output_storage_id",
+            b"output_storage_id",
+            "synthesizer_id",
+            b"synthesizer_id",
+        ],
+    ) -> None: ...
+
+global___StartDatasetGenerationEvent = StartDatasetGenerationEvent

{weave_python-0.15.0 → weave_python-0.17.0}/weave/datamanagement/generate/v1/service_pb2.py

@@ -25,7 +25,7 @@ _sym_db = _symbol_database.Default()
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n.weave/datamanagement/generate/v1/service.proto\x12 weave.datamanagement.generate.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x34weave/datamanagement/generate/v1/configuration.proto"\xe4\x01\n\x1d\x43reateSyntheticDatasetRequest\x12\'\n\x0forganization_id\x18\x01 \x01(\tR\x0eorganizationId\x12%\n\x0esynthesizer_id\x18\x02 \x01(\tR\rsynthesizerId\x12*\n\x11output_storage_id\x18\x03 \x01(\tR\x0foutputStorageId\x12G\n\x06\x63onfig\x18\x04 \x01(\x0b\x32/.weave.datamanagement.generate.v1.ConfigurationR\x06\x63onfig"V\n\x1e\x43reateSyntheticDatasetResponse\x12\x1d\n\ndataset_id\x18\x01 \x01(\tR\tdatasetId\x12\x15\n\x06job_id\x18\x02 \x01(\tR\x05jobId";\n\x1aGetSyntheticDatasetRequest\x12\x1d\n\ndataset_id\x18\x01 \x01(\tR\tdatasetId">\n\x1bGetSyntheticDatasetResponse\x12\x1f\n\x0b\x64\x61taset_ids\x18\x01 \x03(\tR\ndatasetIds"\x83\x01\n\x1cListSyntheticDatasetsRequest\x12\'\n\x0forganization_id\x18\x01 \x01(\tR\x0eorganizationId\x12\x1b\n\tpage_size\x18\x02 \x01(\x05R\x08pageSize\x12\x1d\n\npage_token\x18\x03 \x01(\tR\tpageToken"h\n\x1dListSyntheticDatasetsResponse\x12\x1f\n\x0b\x64\x61taset_ids\x18\x01 \x03(\tR\ndatasetIds\x12&\n\x0fnext_page_token\x18\x02 \x01(\tR\rnextPageToken">\n\x1d\x44\x65leteSyntheticDatasetRequest\x12\x1d\n\ndataset_id\x18\x01 \x01(\tR\tdatasetId":\n\x1e\x44\x65leteSyntheticDatasetResponse\x12\x18\n\x07success\x18\x01 \x01(\x08R\x07success2\xbb\x06\n\x08Generate\x12\xcd\x01\n\x16\x43reateSyntheticDataset\x12?.weave.datamanagement.generate.v1.CreateSyntheticDatasetRequest\x1a@.weave.datamanagement.generate.v1.CreateSyntheticDatasetResponse"0\x82\xd3\xe4\x93\x02*"%/v1/generate/synthetic-dataset/create:\x01*\x12\xc7\x01\n\x13GetSyntheticDataset\x12<.weave.datamanagement.generate.v1.GetSyntheticDatasetRequest\x1a=.weave.datamanagement.generate.v1.GetSyntheticDatasetResponse"3\x82\xd3\xe4\x93\x02-\x12+/v1/generate/synthetic-dataset/{dataset_id}\x12\xc1\x01\n\x15ListSyntheticDatasets\x12>.weave.datamanagement.generate.v1.ListSyntheticDatasetsRequest\x1a?.weave.datamanagement.generate.v1.ListSyntheticDatasetsResponse"\'\x82\xd3\xe4\x93\x02!\x12\x1f/v1/generate/synthetic-datasets\x12\xd0\x01\n\x16\x44\x65leteSyntheticDataset\x12?.weave.datamanagement.generate.v1.DeleteSyntheticDatasetRequest\x1a@.weave.datamanagement.generate.v1.DeleteSyntheticDatasetResponse"3\x82\xd3\xe4\x93\x02-*+/v1/generate/synthetic-dataset/{dataset_id}B\xa5\x02\n$com.weave.datamanagement.generate.v1B\x0cServiceProtoP\x01ZLgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/generate/v1;generate\xa2\x02\x03WDG\xaa\x02 Weave.Datamanagement.Generate.V1\xca\x02 Weave\\Datamanagement\\Generate\\V1\xe2\x02,Weave\\Datamanagement\\Generate\\V1\\GPBMetadata\xea\x02#Weave::Datamanagement::Generate::V1b\x06proto3'
+    b'\n.weave/datamanagement/generate/v1/service.proto\x12 weave.datamanagement.generate.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x1cgoogle/protobuf/struct.proto"\xcc\x01\n\x1d\x43reateSyntheticDatasetRequest\x12\'\n\x0forganization_id\x18\x01 \x01(\tR\x0eorganizationId\x12%\n\x0esynthesizer_id\x18\x02 \x01(\tR\rsynthesizerId\x12*\n\x11output_storage_id\x18\x03 \x01(\tR\x0foutputStorageId\x12/\n\x06\x63onfig\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\x06\x63onfig"?\n\x1e\x43reateSyntheticDatasetResponse\x12\x1d\n\ndataset_id\x18\x01 \x01(\tR\tdatasetId";\n\x1aGetSyntheticDatasetRequest\x12\x1d\n\ndataset_id\x18\x01 \x01(\tR\tdatasetId">\n\x1bGetSyntheticDatasetResponse\x12\x1f\n\x0b\x64\x61taset_ids\x18\x01 \x03(\tR\ndatasetIds"\x83\x01\n\x1cListSyntheticDatasetsRequest\x12\'\n\x0forganization_id\x18\x01 \x01(\tR\x0eorganizationId\x12\x1b\n\tpage_size\x18\x02 \x01(\x05R\x08pageSize\x12\x1d\n\npage_token\x18\x03 \x01(\tR\tpageToken"h\n\x1dListSyntheticDatasetsResponse\x12\x1f\n\x0b\x64\x61taset_ids\x18\x01 \x03(\tR\ndatasetIds\x12&\n\x0fnext_page_token\x18\x02 \x01(\tR\rnextPageToken">\n\x1d\x44\x65leteSyntheticDatasetRequest\x12\x1d\n\ndataset_id\x18\x01 \x01(\tR\tdatasetId":\n\x1e\x44\x65leteSyntheticDatasetResponse\x12\x18\n\x07success\x18\x01 \x01(\x08R\x07success2\xbb\x06\n\x08Generate\x12\xcd\x01\n\x16\x43reateSyntheticDataset\x12?.weave.datamanagement.generate.v1.CreateSyntheticDatasetRequest\x1a@.weave.datamanagement.generate.v1.CreateSyntheticDatasetResponse"0\x82\xd3\xe4\x93\x02*"%/v1/generate/synthetic-dataset/create:\x01*\x12\xc7\x01\n\x13GetSyntheticDataset\x12<.weave.datamanagement.generate.v1.GetSyntheticDatasetRequest\x1a=.weave.datamanagement.generate.v1.GetSyntheticDatasetResponse"3\x82\xd3\xe4\x93\x02-\x12+/v1/generate/synthetic-dataset/{dataset_id}\x12\xc1\x01\n\x15ListSyntheticDatasets\x12>.weave.datamanagement.generate.v1.ListSyntheticDatasetsRequest\x1a?.weave.datamanagement.generate.v1.ListSyntheticDatasetsResponse"\'\x82\xd3\xe4\x93\x02!\x12\x1f/v1/generate/synthetic-datasets\x12\xd0\x01\n\x16\x44\x65leteSyntheticDataset\x12?.weave.datamanagement.generate.v1.DeleteSyntheticDatasetRequest\x1a@.weave.datamanagement.generate.v1.DeleteSyntheticDatasetResponse"3\x82\xd3\xe4\x93\x02-*+/v1/generate/synthetic-dataset/{dataset_id}B\xa5\x02\n$com.weave.datamanagement.generate.v1B\x0cServiceProtoP\x01ZLgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/generate/v1;generate\xa2\x02\x03WDG\xaa\x02 Weave.Datamanagement.Generate.V1\xca\x02 Weave\\Datamanagement\\Generate\\V1\xe2\x02,Weave\\Datamanagement\\Generate\\V1\\GPBMetadata\xea\x02#Weave::Datamanagement::Generate::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -68,22 +68,22 @@ if not _descriptor._USE_C_DESCRIPTORS:
     ]._serialized_options = (
         b"\202\323\344\223\002-*+/v1/generate/synthetic-dataset/{dataset_id}"
     )
-    _globals["_CREATESYNTHETICDATASETREQUEST"]._serialized_start = 169
-    _globals["_CREATESYNTHETICDATASETREQUEST"]._serialized_end = 397
-    _globals["_CREATESYNTHETICDATASETRESPONSE"]._serialized_start = 399
-    _globals["_CREATESYNTHETICDATASETRESPONSE"]._serialized_end = 485
-    _globals["_GETSYNTHETICDATASETREQUEST"]._serialized_start = 487
-    _globals["_GETSYNTHETICDATASETREQUEST"]._serialized_end = 546
-    _globals["_GETSYNTHETICDATASETRESPONSE"]._serialized_start = 548
-    _globals["_GETSYNTHETICDATASETRESPONSE"]._serialized_end = 610
-    _globals["_LISTSYNTHETICDATASETSREQUEST"]._serialized_start = 613
-    _globals["_LISTSYNTHETICDATASETSREQUEST"]._serialized_end = 744
-    _globals["_LISTSYNTHETICDATASETSRESPONSE"]._serialized_start = 746
-    _globals["_LISTSYNTHETICDATASETSRESPONSE"]._serialized_end = 850
-    _globals["_DELETESYNTHETICDATASETREQUEST"]._serialized_start = 852
-    _globals["_DELETESYNTHETICDATASETREQUEST"]._serialized_end = 914
-    _globals["_DELETESYNTHETICDATASETRESPONSE"]._serialized_start = 916
-    _globals["_DELETESYNTHETICDATASETRESPONSE"]._serialized_end = 974
-    _globals["_GENERATE"]._serialized_start = 977
-    _globals["_GENERATE"]._serialized_end = 1804
+    _globals["_CREATESYNTHETICDATASETREQUEST"]._serialized_start = 145
+    _globals["_CREATESYNTHETICDATASETREQUEST"]._serialized_end = 349
+    _globals["_CREATESYNTHETICDATASETRESPONSE"]._serialized_start = 351
+    _globals["_CREATESYNTHETICDATASETRESPONSE"]._serialized_end = 414
+    _globals["_GETSYNTHETICDATASETREQUEST"]._serialized_start = 416
+    _globals["_GETSYNTHETICDATASETREQUEST"]._serialized_end = 475
+    _globals["_GETSYNTHETICDATASETRESPONSE"]._serialized_start = 477
+    _globals["_GETSYNTHETICDATASETRESPONSE"]._serialized_end = 539
+    _globals["_LISTSYNTHETICDATASETSREQUEST"]._serialized_start = 542
+    _globals["_LISTSYNTHETICDATASETSREQUEST"]._serialized_end = 673
+    _globals["_LISTSYNTHETICDATASETSRESPONSE"]._serialized_start = 675
+    _globals["_LISTSYNTHETICDATASETSRESPONSE"]._serialized_end = 779
+    _globals["_DELETESYNTHETICDATASETREQUEST"]._serialized_start = 781
+    _globals["_DELETESYNTHETICDATASETREQUEST"]._serialized_end = 843
+    _globals["_DELETESYNTHETICDATASETRESPONSE"]._serialized_start = 845
+    _globals["_DELETESYNTHETICDATASETRESPONSE"]._serialized_end = 903
+    _globals["_GENERATE"]._serialized_start = 906
+    _globals["_GENERATE"]._serialized_end = 1733
 # @@protoc_insertion_point(module_scope)

{weave_python-0.15.0 → weave_python-0.17.0}/weave/datamanagement/generate/v1/service_pb2.pyi

@@ -8,8 +8,8 @@ import collections.abc
 import google.protobuf.descriptor
 import google.protobuf.internal.containers
 import google.protobuf.message
+import google.protobuf.struct_pb2
 import typing
-import weave.datamanagement.generate.v1.configuration_pb2
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
@@ -25,17 +25,14 @@ class CreateSyntheticDatasetRequest(google.protobuf.message.Message):
     synthesizer_id: builtins.str
     output_storage_id: builtins.str
     @property
-    def config(
-        self,
-    ) -> weave.datamanagement.generate.v1.configuration_pb2.Configuration: ...
+    def config(self) -> google.protobuf.struct_pb2.Struct: ...
     def __init__(
         self,
         *,
         organization_id: builtins.str = ...,
         synthesizer_id: builtins.str = ...,
         output_storage_id: builtins.str = ...,
-        config: weave.datamanagement.generate.v1.configuration_pb2.Configuration
-        | None = ...,
+        config: google.protobuf.struct_pb2.Struct | None = ...,
     ) -> None: ...
     def HasField(
         self, field_name: typing.Literal["config", b"config"]
@@ -61,18 +58,14 @@ class CreateSyntheticDatasetResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     DATASET_ID_FIELD_NUMBER: builtins.int
-    JOB_ID_FIELD_NUMBER: builtins.int
     dataset_id: builtins.str
-    job_id: builtins.str
     def __init__(
         self,
         *,
         dataset_id: builtins.str = ...,
-        job_id: builtins.str = ...,
     ) -> None: ...
     def ClearField(
-        self,
-        field_name: typing.Literal["dataset_id", b"dataset_id", "job_id", b"job_id"],
+        self, field_name: typing.Literal["dataset_id", b"dataset_id"]
     ) -> None: ...
 
 global___CreateSyntheticDatasetResponse = CreateSyntheticDatasetResponse

weave_python-0.17.0/weave/datamanagement/synthesize/v1/service_pb2.py

@@ -0,0 +1,79 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: weave/datamanagement/synthesize/v1/service.proto
+# Protobuf Python Version: 6.30.2
+"""Generated protocol buffer code."""
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    30,
+    2,
+    "",
+    "weave/datamanagement/synthesize/v1/service.proto",
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n0weave/datamanagement/synthesize/v1/service.proto\x12"weave.datamanagement.synthesize.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x31weave/datamanagement/synthesize/v1/training.proto"\xc0\x02\n\x18\x43reateSynthesizerRequest\x12\'\n\x0forganization_id\x18\x01 \x01(\tR\x0eorganizationId\x12\x1f\n\x0b\x64\x61taset_ids\x18\x02 \x03(\tR\ndatasetIds\x12[\n\x0ftraining_driver\x18\x03 \x01(\x0e\x32\x32.weave.datamanagement.synthesize.v1.TrainingDriverR\x0etrainingDriver\x12<\n\rdriver_config\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\x0c\x64riverConfig\x12?\n\x1coutput_storage_connection_id\x18\x05 \x01(\tR\x19outputStorageConnectionId"B\n\x19\x43reateSynthesizerResponse\x12%\n\x0esynthesizer_id\x18\x01 \x01(\tR\rsynthesizerId">\n\x15GetSynthesizerRequest\x12%\n\x0esynthesizer_id\x18\x01 \x01(\tR\rsynthesizerId"k\n\x16GetSynthesizerResponse\x12Q\n\x0bsynthesizer\x18\x01 \x01(\x0b\x32/.weave.datamanagement.synthesize.v1.SynthesizerR\x0bsynthesizer"B\n\x17ListSynthesizersRequest\x12\'\n\x0forganization_id\x18\x01 \x01(\tR\x0eorganizationId"o\n\x18ListSynthesizersResponse\x12S\n\x0csynthesizers\x18\x01 \x03(\x0b\x32/.weave.datamanagement.synthesize.v1.SynthesizerR\x0csynthesizers"A\n\x18\x44\x65leteSynthesizerRequest\x12%\n\x0esynthesizer_id\x18\x01 \x01(\tR\rsynthesizerId"\x1b\n\x19\x44\x65leteSynthesizerResponse"\xd6\x02\n\x0bSynthesizer\x12%\n\x0esynthesizer_id\x18\x01 \x01(\tR\rsynthesizerId\x12\'\n\x0forganization_id\x18\x02 \x01(\tR\x0eorganizationId\x12\x30\n\x14training_dataset_ids\x18\x03 \x03(\tR\x12trainingDatasetIds\x12[\n\x0ftraining_driver\x18\x04 \x01(\x0e\x32\x32.weave.datamanagement.synthesize.v1.TrainingDriverR\x0etrainingDriver\x12\x34\n\x16training_driver_config\x18\x05 \x01(\x0cR\x14trainingDriverConfig\x12\x32\n\x15storage_connection_id\x18\x06 \x01(\tR\x13storageConnectionId2\xd8\x05\n\nSynthesize\x12\xb2\x01\n\x11\x43reateSynthesizer\x12<.weave.datamanagement.synthesize.v1.CreateSynthesizerRequest\x1a=.weave.datamanagement.synthesize.v1.CreateSynthesizerResponse" \x82\xd3\xe4\x93\x02\x1a"\x15/v1/synthesize/create:\x01*\x12\xb0\x01\n\x0eGetSynthesizer\x12\x39.weave.datamanagement.synthesize.v1.GetSynthesizerRequest\x1a:.weave.datamanagement.synthesize.v1.GetSynthesizerResponse"\'\x82\xd3\xe4\x93\x02!\x12\x1f/v1/synthesize/{synthesizer_id}\x12\xa5\x01\n\x10ListSynthesizers\x12;.weave.datamanagement.synthesize.v1.ListSynthesizersRequest\x1a<.weave.datamanagement.synthesize.v1.ListSynthesizersResponse"\x16\x82\xd3\xe4\x93\x02\x10\x12\x0e/v1/synthesize\x12\xb9\x01\n\x11\x44\x65leteSynthesizer\x12<.weave.datamanagement.synthesize.v1.DeleteSynthesizerRequest\x1a=.weave.datamanagement.synthesize.v1.DeleteSynthesizerResponse"\'\x82\xd3\xe4\x93\x02!*\x1f/v1/synthesize/{synthesizer_id}B\xb3\x02\n&com.weave.datamanagement.synthesize.v1B\x0cServiceProtoP\x01ZPgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/synthesize/v1;synthesize\xa2\x02\x03WDS\xaa\x02"Weave.Datamanagement.Synthesize.V1\xca\x02"Weave\\Datamanagement\\Synthesize\\V1\xe2\x02.Weave\\Datamanagement\\Synthesize\\V1\\GPBMetadata\xea\x02%Weave::Datamanagement::Synthesize::V1b\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(
+    DESCRIPTOR, "weave.datamanagement.synthesize.v1.service_pb2", _globals
+)
+if not _descriptor._USE_C_DESCRIPTORS:
+    _globals["DESCRIPTOR"]._loaded_options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b'\n&com.weave.datamanagement.synthesize.v1B\014ServiceProtoP\001ZPgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/synthesize/v1;synthesize\242\002\003WDS\252\002"Weave.Datamanagement.Synthesize.V1\312\002"Weave\\Datamanagement\\Synthesize\\V1\342\002.Weave\\Datamanagement\\Synthesize\\V1\\GPBMetadata\352\002%Weave::Datamanagement::Synthesize::V1'
+    _globals["_SYNTHESIZE"].methods_by_name["CreateSynthesizer"]._loaded_options = None
+    _globals["_SYNTHESIZE"].methods_by_name[
+        "CreateSynthesizer"
+    ]._serialized_options = b'\202\323\344\223\002\032"\025/v1/synthesize/create:\001*'
+    _globals["_SYNTHESIZE"].methods_by_name["GetSynthesizer"]._loaded_options = None
+    _globals["_SYNTHESIZE"].methods_by_name[
+        "GetSynthesizer"
+    ]._serialized_options = (
+        b"\202\323\344\223\002!\022\037/v1/synthesize/{synthesizer_id}"
+    )
+    _globals["_SYNTHESIZE"].methods_by_name["ListSynthesizers"]._loaded_options = None
+    _globals["_SYNTHESIZE"].methods_by_name[
+        "ListSynthesizers"
+    ]._serialized_options = b"\202\323\344\223\002\020\022\016/v1/synthesize"
+    _globals["_SYNTHESIZE"].methods_by_name["DeleteSynthesizer"]._loaded_options = None
+    _globals["_SYNTHESIZE"].methods_by_name[
+        "DeleteSynthesizer"
+    ]._serialized_options = b"\202\323\344\223\002!*\037/v1/synthesize/{synthesizer_id}"
+    _globals["_CREATESYNTHESIZERREQUEST"]._serialized_start = 200
+    _globals["_CREATESYNTHESIZERREQUEST"]._serialized_end = 520
+    _globals["_CREATESYNTHESIZERRESPONSE"]._serialized_start = 522
+    _globals["_CREATESYNTHESIZERRESPONSE"]._serialized_end = 588
+    _globals["_GETSYNTHESIZERREQUEST"]._serialized_start = 590
+    _globals["_GETSYNTHESIZERREQUEST"]._serialized_end = 652
+    _globals["_GETSYNTHESIZERRESPONSE"]._serialized_start = 654
+    _globals["_GETSYNTHESIZERRESPONSE"]._serialized_end = 761
+    _globals["_LISTSYNTHESIZERSREQUEST"]._serialized_start = 763
+    _globals["_LISTSYNTHESIZERSREQUEST"]._serialized_end = 829
+    _globals["_LISTSYNTHESIZERSRESPONSE"]._serialized_start = 831
+    _globals["_LISTSYNTHESIZERSRESPONSE"]._serialized_end = 942
+    _globals["_DELETESYNTHESIZERREQUEST"]._serialized_start = 944
+    _globals["_DELETESYNTHESIZERREQUEST"]._serialized_end = 1009
+    _globals["_DELETESYNTHESIZERRESPONSE"]._serialized_start = 1011
+    _globals["_DELETESYNTHESIZERRESPONSE"]._serialized_end = 1038
+    _globals["_SYNTHESIZER"]._serialized_start = 1041
+    _globals["_SYNTHESIZER"]._serialized_end = 1383
+    _globals["_SYNTHESIZE"]._serialized_start = 1386
+    _globals["_SYNTHESIZE"]._serialized_end = 2114
+# @@protoc_insertion_point(module_scope)

{weave_python-0.15.0 → weave_python-0.17.0}/weave/datamanagement/synthesize/v1/service_pb2.pyi

@@ -8,6 +8,7 @@ import collections.abc
 import google.protobuf.descriptor
 import google.protobuf.internal.containers
 import google.protobuf.message
+import google.protobuf.struct_pb2
 import typing
 import weave.datamanagement.synthesize.v1.training_pb2
 
@@ -26,7 +27,6 @@ class CreateSynthesizerRequest(google.protobuf.message.Message):
     training_driver: (
         weave.datamanagement.synthesize.v1.training_pb2.TrainingDriver.ValueType
     )
-    driver_config: builtins.bytes
     output_storage_connection_id: builtins.str
     @property
     def dataset_ids(
@@ -34,15 +34,20 @@ class CreateSynthesizerRequest(google.protobuf.message.Message):
     ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[
         builtins.str
     ]: ...
+    @property
+    def driver_config(self) -> google.protobuf.struct_pb2.Struct: ...
     def __init__(
         self,
         *,
         organization_id: builtins.str = ...,
         dataset_ids: collections.abc.Iterable[builtins.str] | None = ...,
         training_driver: weave.datamanagement.synthesize.v1.training_pb2.TrainingDriver.ValueType = ...,
-        driver_config: builtins.bytes = ...,
+        driver_config: google.protobuf.struct_pb2.Struct | None = ...,
         output_storage_connection_id: builtins.str = ...,
     ) -> None: ...
+    def HasField(
+        self, field_name: typing.Literal["driver_config", b"driver_config"]
+    ) -> builtins.bool: ...
     def ClearField(
         self,
         field_name: typing.Literal[
@@ -65,15 +70,15 @@ global___CreateSynthesizerRequest = CreateSynthesizerRequest
 class CreateSynthesizerResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
-    TRAINING_ID_FIELD_NUMBER: builtins.int
-    training_id: builtins.str
+    SYNTHESIZER_ID_FIELD_NUMBER: builtins.int
+    synthesizer_id: builtins.str
     def __init__(
         self,
         *,
-        training_id: builtins.str = ...,
+        synthesizer_id: builtins.str = ...,
     ) -> None: ...
     def ClearField(
-        self, field_name: typing.Literal["training_id", b"training_id"]
+        self, field_name: typing.Literal["synthesizer_id", b"synthesizer_id"]
     ) -> None: ...
 
 global___CreateSynthesizerResponse = CreateSynthesizerResponse

weave_python-0.17.0/weave/datamanagement/synthesize/v1/training_pb2.py

@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: weave/datamanagement/synthesize/v1/training.proto
+# Protobuf Python Version: 6.30.2
+"""Generated protocol buffer code."""
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    30,
+    2,
+    "",
+    "weave/datamanagement/synthesize/v1/training.proto",
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n1weave/datamanagement/synthesize/v1/training.proto\x12"weave.datamanagement.synthesize.v1\x1a\x1cgoogle/protobuf/struct.proto"\xb8\x02\n\x12StartTrainingEvent\x12%\n\x0esynthesizer_id\x18\x01 \x01(\tR\rsynthesizerId\x12\x1f\n\x0b\x64\x61taset_ids\x18\x02 \x03(\tR\ndatasetIds\x12[\n\x0ftraining_driver\x18\x03 \x01(\x0e\x32\x32.weave.datamanagement.synthesize.v1.TrainingDriverR\x0etrainingDriver\x12<\n\rdriver_config\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\x0c\x64riverConfig\x12?\n\x1coutput_storage_connection_id\x18\x05 \x01(\tR\x19outputStorageConnectionId"\x8d\x02\n\x15TrainingProgressEvent\x12L\n\njob_status\x18\x01 \x01(\x0e\x32-.weave.datamanagement.synthesize.v1.JobStatusR\tjobStatus\x12\x1b\n\tstep_name\x18\x02 \x01(\tR\x08stepName\x12N\n\x0bstep_status\x18\x03 \x01(\x0e\x32-.weave.datamanagement.synthesize.v1.JobStatusR\nstepStatus\x12#\n\rstep_progress\x18\x04 \x01(\x02R\x0cstepProgress\x12\x14\n\x05\x65rror\x18\x05 \x01(\tR\x05\x65rror*\x88\x01\n\x0eTrainingDriver\x12\x1f\n\x1bTRAINING_DRIVER_UNSPECIFIED\x10\x00\x12\x1c\n\x18TRAINING_DRIVER_MOSTLYAI\x10\x01\x12\x17\n\x13TRAINING_DRIVER_SDV\x10\x02\x12\x1e\n\x1aTRAINING_DRIVER_SMARTNOISE\x10\x03*\x88\x01\n\tJobStatus\x12\x1a\n\x16JOB_STATUS_UNSPECIFIED\x10\x00\x12\x16\n\x12JOB_STATUS_PENDING\x10\x01\x12\x16\n\x12JOB_STATUS_RUNNING\x10\x02\x12\x18\n\x14JOB_STATUS_COMPLETED\x10\x03\x12\x15\n\x11JOB_STATUS_FAILED\x10\x04\x42\xb4\x02\n&com.weave.datamanagement.synthesize.v1B\rTrainingProtoP\x01ZPgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/synthesize/v1;synthesize\xa2\x02\x03WDS\xaa\x02"Weave.Datamanagement.Synthesize.V1\xca\x02"Weave\\Datamanagement\\Synthesize\\V1\xe2\x02.Weave\\Datamanagement\\Synthesize\\V1\\GPBMetadata\xea\x02%Weave::Datamanagement::Synthesize::V1b\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(
+    DESCRIPTOR, "weave.datamanagement.synthesize.v1.training_pb2", _globals
+)
+if not _descriptor._USE_C_DESCRIPTORS:
+    _globals["DESCRIPTOR"]._loaded_options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b'\n&com.weave.datamanagement.synthesize.v1B\rTrainingProtoP\001ZPgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/synthesize/v1;synthesize\242\002\003WDS\252\002"Weave.Datamanagement.Synthesize.V1\312\002"Weave\\Datamanagement\\Synthesize\\V1\342\002.Weave\\Datamanagement\\Synthesize\\V1\\GPBMetadata\352\002%Weave::Datamanagement::Synthesize::V1'
+    _globals["_TRAININGDRIVER"]._serialized_start = 707
+    _globals["_TRAININGDRIVER"]._serialized_end = 843
+    _globals["_JOBSTATUS"]._serialized_start = 846
+    _globals["_JOBSTATUS"]._serialized_end = 982
+    _globals["_STARTTRAININGEVENT"]._serialized_start = 120
+    _globals["_STARTTRAININGEVENT"]._serialized_end = 432
+    _globals["_TRAININGPROGRESSEVENT"]._serialized_start = 435
+    _globals["_TRAININGPROGRESSEVENT"]._serialized_end = 704
+# @@protoc_insertion_point(module_scope)

{weave_python-0.15.0 → weave_python-0.17.0}/weave/datamanagement/synthesize/v1/training_pb2.pyi

@@ -9,6 +9,7 @@ import google.protobuf.descriptor
 import google.protobuf.internal.containers
 import google.protobuf.internal.enum_type_wrapper
 import google.protobuf.message
+import google.protobuf.struct_pb2
 import sys
 import typing
 
@@ -71,14 +72,13 @@ global___JobStatus = JobStatus
 class StartTrainingEvent(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
-    ID_FIELD_NUMBER: builtins.int
+    SYNTHESIZER_ID_FIELD_NUMBER: builtins.int
     DATASET_IDS_FIELD_NUMBER: builtins.int
     TRAINING_DRIVER_FIELD_NUMBER: builtins.int
     DRIVER_CONFIG_FIELD_NUMBER: builtins.int
     OUTPUT_STORAGE_CONNECTION_ID_FIELD_NUMBER: builtins.int
-    id: builtins.str
+    synthesizer_id: builtins.str
     training_driver: global___TrainingDriver.ValueType
-    driver_config: builtins.bytes
     output_storage_connection_id: builtins.str
     @property
     def dataset_ids(
@@ -86,15 +86,20 @@ class StartTrainingEvent(google.protobuf.message.Message):
     ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[
         builtins.str
     ]: ...
+    @property
+    def driver_config(self) -> google.protobuf.struct_pb2.Struct: ...
     def __init__(
         self,
         *,
-        id: builtins.str = ...,
+        synthesizer_id: builtins.str = ...,
         dataset_ids: collections.abc.Iterable[builtins.str] | None = ...,
         training_driver: global___TrainingDriver.ValueType = ...,
-        driver_config: builtins.bytes = ...,
+        driver_config: google.protobuf.struct_pb2.Struct | None = ...,
         output_storage_connection_id: builtins.str = ...,
     ) -> None: ...
+    def HasField(
+        self, field_name: typing.Literal["driver_config", b"driver_config"]
+    ) -> builtins.bool: ...
     def ClearField(
         self,
         field_name: typing.Literal[
@@ -102,10 +107,10 @@ class StartTrainingEvent(google.protobuf.message.Message):
             b"dataset_ids",
             "driver_config",
             b"driver_config",
-            "id",
-            b"id",
             "output_storage_connection_id",
             b"output_storage_connection_id",
+            "synthesizer_id",
+            b"synthesizer_id",
             "training_driver",
             b"training_driver",
         ],

weave_python-0.17.0/weave/datamanagement/synthesize/v1/training_pb2_grpc.py

@@ -0,0 +1,2 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""

weave_python-0.17.0/weave/datamanagement/synthesize/v1/training_pb2_grpc.pyi

@@ -0,0 +1,20 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import abc
+import collections.abc
+import grpc
+import grpc.aio
+import typing
+
+_T = typing.TypeVar("_T")
+
+class _MaybeAsyncIterator(
+    collections.abc.AsyncIterator[_T],
+    collections.abc.Iterator[_T],
+    metaclass=abc.ABCMeta,
+): ...
+class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext):  # type: ignore[misc, type-arg]
+    ...

weave_python-0.15.0/weave/datamanagement/generate/v1/configuration_pb2.py

@@ -1,57 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# NO CHECKED-IN PROTOBUF GENCODE
-# source: weave/datamanagement/generate/v1/configuration.proto
-# Protobuf Python Version: 6.30.2
-"""Generated protocol buffer code."""
-
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import runtime_version as _runtime_version
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-
-_runtime_version.ValidateProtobufRuntimeVersion(
-    _runtime_version.Domain.PUBLIC,
-    6,
-    30,
-    2,
-    "",
-    "weave/datamanagement/generate/v1/configuration.proto",
-)
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n4weave/datamanagement/generate/v1/configuration.proto\x12 weave.datamanagement.generate.v1"\xbb\x02\n\rConfiguration\x12\x1f\n\x0bnum_records\x18\x01 \x01(\x03R\nnumRecords\x12!\n\x0c\x64\x61taset_name\x18\x02 \x01(\tR\x0b\x64\x61tasetName\x12V\n\x0c\x64\x61ta_quality\x18\x03 \x01(\x0b\x32\x33.weave.datamanagement.generate.v1.DataQualityConfigR\x0b\x64\x61taQuality\x12I\n\x07privacy\x18\x04 \x01(\x0b\x32/.weave.datamanagement.generate.v1.PrivacyConfigR\x07privacy\x12\x43\n\x05model\x18\x05 \x01(\x0b\x32-.weave.datamanagement.generate.v1.ModelConfigR\x05model"\xb1\x02\n\x11\x44\x61taQualityConfig\x12\x31\n\x14\x63orrelation_accuracy\x18\x01 \x01(\x02R\x13\x63orrelationAccuracy\x12\x35\n\x16statistical_similarity\x18\x02 \x01(\x02R\x15statisticalSimilarity\x12+\n\x11preserve_outliers\x18\x03 \x01(\x08R\x10preserveOutliers\x12/\n\x13\x65nforce_constraints\x18\x04 \x01(\x08R\x12\x65nforceConstraints\x12T\n\x0b\x63onstraints\x18\x05 \x03(\x0b\x32\x32.weave.datamanagement.generate.v1.ColumnConstraintR\x0b\x63onstraints"\xe2\x01\n\x10\x43olumnConstraint\x12\x1f\n\x0b\x63olumn_name\x18\x01 \x01(\tR\ncolumnName\x12\'\n\x0f\x63onstraint_type\x18\x02 \x01(\tR\x0e\x63onstraintType\x12\x1b\n\tmin_value\x18\x03 \x01(\tR\x08minValue\x12\x1b\n\tmax_value\x18\x04 \x01(\tR\x08maxValue\x12#\n\rregex_pattern\x18\x05 \x01(\tR\x0cregexPattern\x12%\n\x0e\x61llowed_values\x18\x06 \x03(\tR\rallowedValues"\xd1\x01\n\rPrivacyConfig\x12\x18\n\x07\x65psilon\x18\x01 \x01(\x02R\x07\x65psilon\x12*\n\x11k_anonymity_check\x18\x02 \x01(\x08R\x0fkAnonymityCheck\x12\x17\n\x07k_value\x18\x03 \x01(\x05R\x06kValue\x12+\n\x11sensitive_columns\x18\x04 \x03(\tR\x10sensitiveColumns\x12\x34\n\x16\x65nable_privacy_metrics\x18\x05 \x01(\x08R\x14\x65nablePrivacyMetrics"\xec\x02\n\x0bModelConfig\x12J\n\nmodel_type\x18\x01 \x01(\x0e\x32+.weave.datamanagement.generate.v1.ModelTypeR\tmodelType\x12\x16\n\x06\x65pochs\x18\x02 \x01(\x05R\x06\x65pochs\x12#\n\rlearning_rate\x18\x03 \x01(\x02R\x0clearningRate\x12\x1d\n\nbatch_size\x18\x04 \x01(\x05R\tbatchSize\x12p\n\x11\x63ustom_parameters\x18\x05 \x03(\x0b\x32\x43.weave.datamanagement.generate.v1.ModelConfig.CustomParametersEntryR\x10\x63ustomParameters\x1a\x43\n\x15\x43ustomParametersEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01*\xea\x01\n\tModelType\x12\x1a\n\x16MODEL_TYPE_UNSPECIFIED\x10\x00\x12\x14\n\x10MODEL_TYPE_CTGAN\x10\x01\x12\x13\n\x0fMODEL_TYPE_TVAE\x10\x02\x12\x15\n\x11MODEL_TYPE_COPULA\x10\x03\x12\x13\n\x0fMODEL_TYPE_FAST\x10\x04\x12\x12\n\x0eMODEL_TYPE_GAN\x10\x05\x12\x13\n\x0fMODEL_TYPE_WGAN\x10\x06\x12\x14\n\x10MODEL_TYPE_SMOTE\x10\x07\x12\x17\n\x13MODEL_TYPE_BAYESIAN\x10\x08\x12\x12\n\x0eMODEL_TYPE_HMA\x10\tB\xab\x02\n$com.weave.datamanagement.generate.v1B\x12\x43onfigurationProtoP\x01ZLgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/generate/v1;generate\xa2\x02\x03WDG\xaa\x02 Weave.Datamanagement.Generate.V1\xca\x02 Weave\\Datamanagement\\Generate\\V1\xe2\x02,Weave\\Datamanagement\\Generate\\V1\\GPBMetadata\xea\x02#Weave::Datamanagement::Generate::V1b\x06proto3'
-)
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(
-    DESCRIPTOR, "weave.datamanagement.generate.v1.configuration_pb2", _globals
-)
-if not _descriptor._USE_C_DESCRIPTORS:
-    _globals["DESCRIPTOR"]._loaded_options = None
-    _globals[
-        "DESCRIPTOR"
-    ]._serialized_options = b"\n$com.weave.datamanagement.generate.v1B\022ConfigurationProtoP\001ZLgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/generate/v1;generate\242\002\003WDG\252\002 Weave.Datamanagement.Generate.V1\312\002 Weave\\Datamanagement\\Generate\\V1\342\002,Weave\\Datamanagement\\Generate\\V1\\GPBMetadata\352\002#Weave::Datamanagement::Generate::V1"
-    _globals["_MODELCONFIG_CUSTOMPARAMETERSENTRY"]._loaded_options = None
-    _globals["_MODELCONFIG_CUSTOMPARAMETERSENTRY"]._serialized_options = b"8\001"
-    _globals["_MODELTYPE"]._serialized_start = 1525
-    _globals["_MODELTYPE"]._serialized_end = 1759
-    _globals["_CONFIGURATION"]._serialized_start = 91
-    _globals["_CONFIGURATION"]._serialized_end = 406
-    _globals["_DATAQUALITYCONFIG"]._serialized_start = 409
-    _globals["_DATAQUALITYCONFIG"]._serialized_end = 714
-    _globals["_COLUMNCONSTRAINT"]._serialized_start = 717
-    _globals["_COLUMNCONSTRAINT"]._serialized_end = 943
-    _globals["_PRIVACYCONFIG"]._serialized_start = 946
-    _globals["_PRIVACYCONFIG"]._serialized_end = 1155
-    _globals["_MODELCONFIG"]._serialized_start = 1158
-    _globals["_MODELCONFIG"]._serialized_end = 1522
-    _globals["_MODELCONFIG_CUSTOMPARAMETERSENTRY"]._serialized_start = 1455
-    _globals["_MODELCONFIG_CUSTOMPARAMETERSENTRY"]._serialized_end = 1522
-# @@protoc_insertion_point(module_scope)

weave_python-0.15.0/weave/datamanagement/synthesize/v1/service_pb2.py

@@ -1,79 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# NO CHECKED-IN PROTOBUF GENCODE
-# source: weave/datamanagement/synthesize/v1/service.proto
-# Protobuf Python Version: 6.30.2
-"""Generated protocol buffer code."""
-
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import runtime_version as _runtime_version
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-
-_runtime_version.ValidateProtobufRuntimeVersion(
-    _runtime_version.Domain.PUBLIC,
-    6,
-    30,
-    2,
-    "",
-    "weave/datamanagement/synthesize/v1/service.proto",
-)
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n0weave/datamanagement/synthesize/v1/service.proto\x12"weave.datamanagement.synthesize.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x31weave/datamanagement/synthesize/v1/training.proto"\xa7\x02\n\x18\x43reateSynthesizerRequest\x12\'\n\x0forganization_id\x18\x01 \x01(\tR\x0eorganizationId\x12\x1f\n\x0b\x64\x61taset_ids\x18\x02 \x03(\tR\ndatasetIds\x12[\n\x0ftraining_driver\x18\x03 \x01(\x0e\x32\x32.weave.datamanagement.synthesize.v1.TrainingDriverR\x0etrainingDriver\x12#\n\rdriver_config\x18\x04 \x01(\x0cR\x0c\x64riverConfig\x12?\n\x1coutput_storage_connection_id\x18\x05 \x01(\tR\x19outputStorageConnectionId"<\n\x19\x43reateSynthesizerResponse\x12\x1f\n\x0btraining_id\x18\x01 \x01(\tR\ntrainingId">\n\x15GetSynthesizerRequest\x12%\n\x0esynthesizer_id\x18\x01 \x01(\tR\rsynthesizerId"k\n\x16GetSynthesizerResponse\x12Q\n\x0bsynthesizer\x18\x01 \x01(\x0b\x32/.weave.datamanagement.synthesize.v1.SynthesizerR\x0bsynthesizer"B\n\x17ListSynthesizersRequest\x12\'\n\x0forganization_id\x18\x01 \x01(\tR\x0eorganizationId"o\n\x18ListSynthesizersResponse\x12S\n\x0csynthesizers\x18\x01 \x03(\x0b\x32/.weave.datamanagement.synthesize.v1.SynthesizerR\x0csynthesizers"A\n\x18\x44\x65leteSynthesizerRequest\x12%\n\x0esynthesizer_id\x18\x01 \x01(\tR\rsynthesizerId"\x1b\n\x19\x44\x65leteSynthesizerResponse"\xd6\x02\n\x0bSynthesizer\x12%\n\x0esynthesizer_id\x18\x01 \x01(\tR\rsynthesizerId\x12\'\n\x0forganization_id\x18\x02 \x01(\tR\x0eorganizationId\x12\x30\n\x14training_dataset_ids\x18\x03 \x03(\tR\x12trainingDatasetIds\x12[\n\x0ftraining_driver\x18\x04 \x01(\x0e\x32\x32.weave.datamanagement.synthesize.v1.TrainingDriverR\x0etrainingDriver\x12\x34\n\x16training_driver_config\x18\x05 \x01(\x0cR\x14trainingDriverConfig\x12\x32\n\x15storage_connection_id\x18\x06 \x01(\tR\x13storageConnectionId2\xd8\x05\n\nSynthesize\x12\xb2\x01\n\x11\x43reateSynthesizer\x12<.weave.datamanagement.synthesize.v1.CreateSynthesizerRequest\x1a=.weave.datamanagement.synthesize.v1.CreateSynthesizerResponse" \x82\xd3\xe4\x93\x02\x1a"\x15/v1/synthesize/create:\x01*\x12\xb0\x01\n\x0eGetSynthesizer\x12\x39.weave.datamanagement.synthesize.v1.GetSynthesizerRequest\x1a:.weave.datamanagement.synthesize.v1.GetSynthesizerResponse"\'\x82\xd3\xe4\x93\x02!\x12\x1f/v1/synthesize/{synthesizer_id}\x12\xa5\x01\n\x10ListSynthesizers\x12;.weave.datamanagement.synthesize.v1.ListSynthesizersRequest\x1a<.weave.datamanagement.synthesize.v1.ListSynthesizersResponse"\x16\x82\xd3\xe4\x93\x02\x10\x12\x0e/v1/synthesize\x12\xb9\x01\n\x11\x44\x65leteSynthesizer\x12<.weave.datamanagement.synthesize.v1.DeleteSynthesizerRequest\x1a=.weave.datamanagement.synthesize.v1.DeleteSynthesizerResponse"\'\x82\xd3\xe4\x93\x02!*\x1f/v1/synthesize/{synthesizer_id}B\xb3\x02\n&com.weave.datamanagement.synthesize.v1B\x0cServiceProtoP\x01ZPgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/synthesize/v1;synthesize\xa2\x02\x03WDS\xaa\x02"Weave.Datamanagement.Synthesize.V1\xca\x02"Weave\\Datamanagement\\Synthesize\\V1\xe2\x02.Weave\\Datamanagement\\Synthesize\\V1\\GPBMetadata\xea\x02%Weave::Datamanagement::Synthesize::V1b\x06proto3'
-)
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(
-    DESCRIPTOR, "weave.datamanagement.synthesize.v1.service_pb2", _globals
-)
-if not _descriptor._USE_C_DESCRIPTORS:
-    _globals["DESCRIPTOR"]._loaded_options = None
-    _globals[
-        "DESCRIPTOR"
-    ]._serialized_options = b'\n&com.weave.datamanagement.synthesize.v1B\014ServiceProtoP\001ZPgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/synthesize/v1;synthesize\242\002\003WDS\252\002"Weave.Datamanagement.Synthesize.V1\312\002"Weave\\Datamanagement\\Synthesize\\V1\342\002.Weave\\Datamanagement\\Synthesize\\V1\\GPBMetadata\352\002%Weave::Datamanagement::Synthesize::V1'
-    _globals["_SYNTHESIZE"].methods_by_name["CreateSynthesizer"]._loaded_options = None
-    _globals["_SYNTHESIZE"].methods_by_name[
-        "CreateSynthesizer"
-    ]._serialized_options = b'\202\323\344\223\002\032"\025/v1/synthesize/create:\001*'
-    _globals["_SYNTHESIZE"].methods_by_name["GetSynthesizer"]._loaded_options = None
-    _globals["_SYNTHESIZE"].methods_by_name[
-        "GetSynthesizer"
-    ]._serialized_options = (
-        b"\202\323\344\223\002!\022\037/v1/synthesize/{synthesizer_id}"
-    )
-    _globals["_SYNTHESIZE"].methods_by_name["ListSynthesizers"]._loaded_options = None
-    _globals["_SYNTHESIZE"].methods_by_name[
-        "ListSynthesizers"
-    ]._serialized_options = b"\202\323\344\223\002\020\022\016/v1/synthesize"
-    _globals["_SYNTHESIZE"].methods_by_name["DeleteSynthesizer"]._loaded_options = None
-    _globals["_SYNTHESIZE"].methods_by_name[
-        "DeleteSynthesizer"
-    ]._serialized_options = b"\202\323\344\223\002!*\037/v1/synthesize/{synthesizer_id}"
-    _globals["_CREATESYNTHESIZERREQUEST"]._serialized_start = 170
-    _globals["_CREATESYNTHESIZERREQUEST"]._serialized_end = 465
-    _globals["_CREATESYNTHESIZERRESPONSE"]._serialized_start = 467
-    _globals["_CREATESYNTHESIZERRESPONSE"]._serialized_end = 527
-    _globals["_GETSYNTHESIZERREQUEST"]._serialized_start = 529
-    _globals["_GETSYNTHESIZERREQUEST"]._serialized_end = 591
-    _globals["_GETSYNTHESIZERRESPONSE"]._serialized_start = 593
-    _globals["_GETSYNTHESIZERRESPONSE"]._serialized_end = 700
-    _globals["_LISTSYNTHESIZERSREQUEST"]._serialized_start = 702
-    _globals["_LISTSYNTHESIZERSREQUEST"]._serialized_end = 768
-    _globals["_LISTSYNTHESIZERSRESPONSE"]._serialized_start = 770
-    _globals["_LISTSYNTHESIZERSRESPONSE"]._serialized_end = 881
-    _globals["_DELETESYNTHESIZERREQUEST"]._serialized_start = 883
-    _globals["_DELETESYNTHESIZERREQUEST"]._serialized_end = 948
-    _globals["_DELETESYNTHESIZERRESPONSE"]._serialized_start = 950
-    _globals["_DELETESYNTHESIZERRESPONSE"]._serialized_end = 977
-    _globals["_SYNTHESIZER"]._serialized_start = 980
-    _globals["_SYNTHESIZER"]._serialized_end = 1322
-    _globals["_SYNTHESIZE"]._serialized_start = 1325
-    _globals["_SYNTHESIZE"]._serialized_end = 2053
-# @@protoc_insertion_point(module_scope)

weave_python-0.15.0/weave/datamanagement/synthesize/v1/training_pb2.py

@@ -1,49 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# NO CHECKED-IN PROTOBUF GENCODE
-# source: weave/datamanagement/synthesize/v1/training.proto
-# Protobuf Python Version: 6.30.2
-"""Generated protocol buffer code."""
-
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import runtime_version as _runtime_version
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-
-_runtime_version.ValidateProtobufRuntimeVersion(
-    _runtime_version.Domain.PUBLIC,
-    6,
-    30,
-    2,
-    "",
-    "weave/datamanagement/synthesize/v1/training.proto",
-)
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n1weave/datamanagement/synthesize/v1/training.proto\x12"weave.datamanagement.synthesize.v1"\x88\x02\n\x12StartTrainingEvent\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1f\n\x0b\x64\x61taset_ids\x18\x02 \x03(\tR\ndatasetIds\x12[\n\x0ftraining_driver\x18\x03 \x01(\x0e\x32\x32.weave.datamanagement.synthesize.v1.TrainingDriverR\x0etrainingDriver\x12#\n\rdriver_config\x18\x04 \x01(\x0cR\x0c\x64riverConfig\x12?\n\x1coutput_storage_connection_id\x18\x05 \x01(\tR\x19outputStorageConnectionId"\x8d\x02\n\x15TrainingProgressEvent\x12L\n\njob_status\x18\x01 \x01(\x0e\x32-.weave.datamanagement.synthesize.v1.JobStatusR\tjobStatus\x12\x1b\n\tstep_name\x18\x02 \x01(\tR\x08stepName\x12N\n\x0bstep_status\x18\x03 \x01(\x0e\x32-.weave.datamanagement.synthesize.v1.JobStatusR\nstepStatus\x12#\n\rstep_progress\x18\x04 \x01(\x02R\x0cstepProgress\x12\x14\n\x05\x65rror\x18\x05 \x01(\tR\x05\x65rror*\x88\x01\n\x0eTrainingDriver\x12\x1f\n\x1bTRAINING_DRIVER_UNSPECIFIED\x10\x00\x12\x1c\n\x18TRAINING_DRIVER_MOSTLYAI\x10\x01\x12\x17\n\x13TRAINING_DRIVER_SDV\x10\x02\x12\x1e\n\x1aTRAINING_DRIVER_SMARTNOISE\x10\x03*\x88\x01\n\tJobStatus\x12\x1a\n\x16JOB_STATUS_UNSPECIFIED\x10\x00\x12\x16\n\x12JOB_STATUS_PENDING\x10\x01\x12\x16\n\x12JOB_STATUS_RUNNING\x10\x02\x12\x18\n\x14JOB_STATUS_COMPLETED\x10\x03\x12\x15\n\x11JOB_STATUS_FAILED\x10\x04\x42\xb4\x02\n&com.weave.datamanagement.synthesize.v1B\rTrainingProtoP\x01ZPgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/synthesize/v1;synthesize\xa2\x02\x03WDS\xaa\x02"Weave.Datamanagement.Synthesize.V1\xca\x02"Weave\\Datamanagement\\Synthesize\\V1\xe2\x02.Weave\\Datamanagement\\Synthesize\\V1\\GPBMetadata\xea\x02%Weave::Datamanagement::Synthesize::V1b\x06proto3'
-)
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(
-    DESCRIPTOR, "weave.datamanagement.synthesize.v1.training_pb2", _globals
-)
-if not _descriptor._USE_C_DESCRIPTORS:
-    _globals["DESCRIPTOR"]._loaded_options = None
-    _globals[
-        "DESCRIPTOR"
-    ]._serialized_options = b'\n&com.weave.datamanagement.synthesize.v1B\rTrainingProtoP\001ZPgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/synthesize/v1;synthesize\242\002\003WDS\252\002"Weave.Datamanagement.Synthesize.V1\312\002"Weave\\Datamanagement\\Synthesize\\V1\342\002.Weave\\Datamanagement\\Synthesize\\V1\\GPBMetadata\352\002%Weave::Datamanagement::Synthesize::V1'
-    _globals["_TRAININGDRIVER"]._serialized_start = 629
-    _globals["_TRAININGDRIVER"]._serialized_end = 765
-    _globals["_JOBSTATUS"]._serialized_start = 768
-    _globals["_JOBSTATUS"]._serialized_end = 904
-    _globals["_STARTTRAININGEVENT"]._serialized_start = 90
-    _globals["_STARTTRAININGEVENT"]._serialized_end = 354
-    _globals["_TRAININGPROGRESSEVENT"]._serialized_start = 357
-    _globals["_TRAININGPROGRESSEVENT"]._serialized_end = 626
-# @@protoc_insertion_point(module_scope)