weave-python 0.28.1__py3-none-any.whl → 0.30.3__py3-none-any.whl

weave/weaveapi/generate/v1/service_pb2.pyi

@@ -3,199 +3,328 @@
 isort:skip_file
 """
 
-import builtins
-import collections.abc
-import google.protobuf.descriptor
-import google.protobuf.internal.containers
-import google.protobuf.message
-import google.protobuf.struct_pb2
-import typing
-
-DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
-
-@typing.final
-class CreateSyntheticDatasetRequest(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    ORGANIZATION_ID_FIELD_NUMBER: builtins.int
-    SYNTHESIZER_ID_FIELD_NUMBER: builtins.int
-    OUTPUT_STORAGE_ID_FIELD_NUMBER: builtins.int
-    CONFIG_FIELD_NUMBER: builtins.int
-    organization_id: builtins.str
-    synthesizer_id: builtins.str
-    output_storage_id: builtins.str
-    @property
-    def config(self) -> google.protobuf.struct_pb2.Struct: ...
+from collections import abc as _abc
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from google.protobuf import struct_pb2 as _struct_pb2
+from google.protobuf.internal import containers as _containers
+import builtins as _builtins
+import sys
+import typing as _typing
+
+if sys.version_info >= (3, 10):
+    from typing import TypeAlias as _TypeAlias
+else:
+    from typing_extensions import TypeAlias as _TypeAlias
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+@_typing.final
+class CreateSyntheticDatasetRequest(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
+
+    ORGANIZATION_ID_FIELD_NUMBER: _builtins.int
+    SYNTHESIZER_ID_FIELD_NUMBER: _builtins.int
+    OUTPUT_STORAGE_ID_FIELD_NUMBER: _builtins.int
+    CONFIG_FIELD_NUMBER: _builtins.int
+    organization_id: _builtins.str
+    synthesizer_id: _builtins.str
+    output_storage_id: _builtins.str
+    @_builtins.property
+    def config(self) -> _struct_pb2.Struct: ...
     def __init__(
         self,
         *,
-        organization_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",
-            "organization_id",
-            b"organization_id",
-            "output_storage_id",
-            b"output_storage_id",
-            "synthesizer_id",
-            b"synthesizer_id",
-        ],
+        organization_id: _builtins.str | None = ...,
+        synthesizer_id: _builtins.str | None = ...,
+        output_storage_id: _builtins.str | None = ...,
+        config: _struct_pb2.Struct | None = ...,
     ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _typing.Literal[
+        "config",
+        b"config",
+        "organization_id",
+        b"organization_id",
+        "output_storage_id",
+        b"output_storage_id",
+        "synthesizer_id",
+        b"synthesizer_id",
+    ]  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal[
+        "config",
+        b"config",
+        "organization_id",
+        b"organization_id",
+        "output_storage_id",
+        b"output_storage_id",
+        "synthesizer_id",
+        b"synthesizer_id",
+    ]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
 
-global___CreateSyntheticDatasetRequest = CreateSyntheticDatasetRequest
+Global___CreateSyntheticDatasetRequest: _TypeAlias = CreateSyntheticDatasetRequest  # noqa: Y015
 
-@typing.final
-class CreateSyntheticDatasetResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+@_typing.final
+class CreateSyntheticDatasetResponse(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
 
-    DATASET_ID_FIELD_NUMBER: builtins.int
-    dataset_id: builtins.str
+    DATASET_ID_FIELD_NUMBER: _builtins.int
+    dataset_id: _builtins.str
     def __init__(
         self,
         *,
-        dataset_id: builtins.str = ...,
-    ) -> None: ...
-    def ClearField(
-        self, field_name: typing.Literal["dataset_id", b"dataset_id"]
+        dataset_id: _builtins.str | None = ...,
     ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _typing.Literal["dataset_id", b"dataset_id"]  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal["dataset_id", b"dataset_id"]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
 
-global___CreateSyntheticDatasetResponse = CreateSyntheticDatasetResponse
+Global___CreateSyntheticDatasetResponse: _TypeAlias = CreateSyntheticDatasetResponse  # noqa: Y015
 
-@typing.final
-class GetSyntheticDatasetRequest(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+@_typing.final
+class CreateSeedDatasetRequest(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
 
-    DATASET_ID_FIELD_NUMBER: builtins.int
-    dataset_id: builtins.str
+    ORGANIZATION_ID_FIELD_NUMBER: _builtins.int
+    OUTPUT_STORAGE_CONNECTION_ID_FIELD_NUMBER: _builtins.int
+    CONFIG_FIELD_NUMBER: _builtins.int
+    NAME_FIELD_NUMBER: _builtins.int
+    DESCRIPTION_FIELD_NUMBER: _builtins.int
+    organization_id: _builtins.str
+    output_storage_connection_id: _builtins.str
+    name: _builtins.str
+    description: _builtins.str
+    @_builtins.property
+    def config(self) -> _struct_pb2.Struct: ...
     def __init__(
         self,
         *,
-        dataset_id: builtins.str = ...,
-    ) -> None: ...
-    def ClearField(
-        self, field_name: typing.Literal["dataset_id", b"dataset_id"]
+        organization_id: _builtins.str | None = ...,
+        output_storage_connection_id: _builtins.str | None = ...,
+        config: _struct_pb2.Struct | None = ...,
+        name: _builtins.str | None = ...,
+        description: _builtins.str | None = ...,
     ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _typing.Literal[
+        "config",
+        b"config",
+        "description",
+        b"description",
+        "name",
+        b"name",
+        "organization_id",
+        b"organization_id",
+        "output_storage_connection_id",
+        b"output_storage_connection_id",
+    ]  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal[
+        "config",
+        b"config",
+        "description",
+        b"description",
+        "name",
+        b"name",
+        "organization_id",
+        b"organization_id",
+        "output_storage_connection_id",
+        b"output_storage_connection_id",
+    ]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
+
+Global___CreateSeedDatasetRequest: _TypeAlias = CreateSeedDatasetRequest  # noqa: Y015
 
-global___GetSyntheticDatasetRequest = GetSyntheticDatasetRequest
+@_typing.final
+class CreateSeedDatasetResponse(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
 
-@typing.final
-class GetSyntheticDatasetResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    @_typing.final
+    class DatasetIdsByTableEntry(_message.Message):
+        DESCRIPTOR: _descriptor.Descriptor
 
-    DATASET_IDS_FIELD_NUMBER: builtins.int
-    @property
+        KEY_FIELD_NUMBER: _builtins.int
+        VALUE_FIELD_NUMBER: _builtins.int
+        key: _builtins.str
+        value: _builtins.str
+        def __init__(
+            self,
+            *,
+            key: _builtins.str | None = ...,
+            value: _builtins.str | None = ...,
+        ) -> None: ...
+        _HasFieldArgType: _TypeAlias = _typing.Literal["key", b"key", "value", b"value"]  # noqa: Y015
+        def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+        _ClearFieldArgType: _TypeAlias = _typing.Literal[
+            "key", b"key", "value", b"value"
+        ]  # noqa: Y015
+        def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
+
+    DATASET_IDS_FIELD_NUMBER: _builtins.int
+    DATASET_IDS_BY_TABLE_FIELD_NUMBER: _builtins.int
+    @_builtins.property
     def dataset_ids(
         self,
-    ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[
-        builtins.str
-    ]: ...
+    ) -> _containers.RepeatedScalarFieldContainer[_builtins.str]: ...
+    @_builtins.property
+    def dataset_ids_by_table(
+        self,
+    ) -> _containers.ScalarMap[_builtins.str, _builtins.str]: ...
     def __init__(
         self,
         *,
-        dataset_ids: collections.abc.Iterable[builtins.str] | None = ...,
-    ) -> None: ...
-    def ClearField(
-        self, field_name: typing.Literal["dataset_ids", b"dataset_ids"]
+        dataset_ids: _abc.Iterable[_builtins.str] | None = ...,
+        dataset_ids_by_table: _abc.Mapping[_builtins.str, _builtins.str] | None = ...,
     ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _typing.Literal[
+        "dataset_ids", b"dataset_ids", "dataset_ids_by_table", b"dataset_ids_by_table"
+    ]  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal[
+        "dataset_ids", b"dataset_ids", "dataset_ids_by_table", b"dataset_ids_by_table"
+    ]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
 
-global___GetSyntheticDatasetResponse = GetSyntheticDatasetResponse
+Global___CreateSeedDatasetResponse: _TypeAlias = CreateSeedDatasetResponse  # noqa: Y015
 
-@typing.final
-class ListSyntheticDatasetsRequest(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+@_typing.final
+class GetSyntheticDatasetRequest(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
 
-    ORGANIZATION_ID_FIELD_NUMBER: builtins.int
-    PAGE_SIZE_FIELD_NUMBER: builtins.int
-    PAGE_TOKEN_FIELD_NUMBER: builtins.int
-    organization_id: builtins.str
-    page_size: builtins.int
-    page_token: builtins.str
+    DATASET_ID_FIELD_NUMBER: _builtins.int
+    dataset_id: _builtins.str
     def __init__(
         self,
         *,
-        organization_id: builtins.str = ...,
-        page_size: builtins.int = ...,
-        page_token: builtins.str = ...,
-    ) -> None: ...
-    def ClearField(
-        self,
-        field_name: typing.Literal[
-            "organization_id",
-            b"organization_id",
-            "page_size",
-            b"page_size",
-            "page_token",
-            b"page_token",
-        ],
+        dataset_id: _builtins.str | None = ...,
     ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _typing.Literal["dataset_id", b"dataset_id"]  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal["dataset_id", b"dataset_id"]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
 
-global___ListSyntheticDatasetsRequest = ListSyntheticDatasetsRequest
+Global___GetSyntheticDatasetRequest: _TypeAlias = GetSyntheticDatasetRequest  # noqa: Y015
 
-@typing.final
-class ListSyntheticDatasetsResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+@_typing.final
+class GetSyntheticDatasetResponse(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
 
-    DATASET_IDS_FIELD_NUMBER: builtins.int
-    NEXT_PAGE_TOKEN_FIELD_NUMBER: builtins.int
-    next_page_token: builtins.str
-    @property
+    DATASET_IDS_FIELD_NUMBER: _builtins.int
+    @_builtins.property
     def dataset_ids(
         self,
-    ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[
-        builtins.str
-    ]: ...
+    ) -> _containers.RepeatedScalarFieldContainer[_builtins.str]: ...
     def __init__(
         self,
         *,
-        dataset_ids: collections.abc.Iterable[builtins.str] | None = ...,
-        next_page_token: builtins.str = ...,
+        dataset_ids: _abc.Iterable[_builtins.str] | None = ...,
     ) -> None: ...
-    def ClearField(
+    _HasFieldArgType: _TypeAlias = _typing.Literal["dataset_ids", b"dataset_ids"]  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal["dataset_ids", b"dataset_ids"]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
+
+Global___GetSyntheticDatasetResponse: _TypeAlias = GetSyntheticDatasetResponse  # noqa: Y015
+
+@_typing.final
+class ListSyntheticDatasetsRequest(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
+
+    ORGANIZATION_ID_FIELD_NUMBER: _builtins.int
+    PAGE_SIZE_FIELD_NUMBER: _builtins.int
+    PAGE_TOKEN_FIELD_NUMBER: _builtins.int
+    organization_id: _builtins.str
+    page_size: _builtins.int
+    page_token: _builtins.str
+    def __init__(
         self,
-        field_name: typing.Literal[
-            "dataset_ids", b"dataset_ids", "next_page_token", b"next_page_token"
-        ],
+        *,
+        organization_id: _builtins.str | None = ...,
+        page_size: _builtins.int | None = ...,
+        page_token: _builtins.str | None = ...,
     ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _typing.Literal[
+        "organization_id",
+        b"organization_id",
+        "page_size",
+        b"page_size",
+        "page_token",
+        b"page_token",
+    ]  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal[
+        "organization_id",
+        b"organization_id",
+        "page_size",
+        b"page_size",
+        "page_token",
+        b"page_token",
+    ]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
 
-global___ListSyntheticDatasetsResponse = ListSyntheticDatasetsResponse
+Global___ListSyntheticDatasetsRequest: _TypeAlias = ListSyntheticDatasetsRequest  # noqa: Y015
 
-@typing.final
-class DeleteSyntheticDatasetRequest(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+@_typing.final
+class ListSyntheticDatasetsResponse(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
 
-    DATASET_ID_FIELD_NUMBER: builtins.int
-    dataset_id: builtins.str
+    DATASET_IDS_FIELD_NUMBER: _builtins.int
+    NEXT_PAGE_TOKEN_FIELD_NUMBER: _builtins.int
+    next_page_token: _builtins.str
+    @_builtins.property
+    def dataset_ids(
+        self,
+    ) -> _containers.RepeatedScalarFieldContainer[_builtins.str]: ...
     def __init__(
         self,
         *,
-        dataset_id: builtins.str = ...,
+        dataset_ids: _abc.Iterable[_builtins.str] | None = ...,
+        next_page_token: _builtins.str | None = ...,
     ) -> None: ...
-    def ClearField(
-        self, field_name: typing.Literal["dataset_id", b"dataset_id"]
+    _HasFieldArgType: _TypeAlias = _typing.Literal[
+        "dataset_ids", b"dataset_ids", "next_page_token", b"next_page_token"
+    ]  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal[
+        "dataset_ids", b"dataset_ids", "next_page_token", b"next_page_token"
+    ]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
+
+Global___ListSyntheticDatasetsResponse: _TypeAlias = ListSyntheticDatasetsResponse  # noqa: Y015
+
+@_typing.final
+class DeleteSyntheticDatasetRequest(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
+
+    DATASET_ID_FIELD_NUMBER: _builtins.int
+    dataset_id: _builtins.str
+    def __init__(
+        self,
+        *,
+        dataset_id: _builtins.str | None = ...,
     ) -> None: ...
+    _HasFieldArgType: _TypeAlias = _typing.Literal["dataset_id", b"dataset_id"]  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal["dataset_id", b"dataset_id"]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
 
-global___DeleteSyntheticDatasetRequest = DeleteSyntheticDatasetRequest
+Global___DeleteSyntheticDatasetRequest: _TypeAlias = DeleteSyntheticDatasetRequest  # noqa: Y015
 
-@typing.final
-class DeleteSyntheticDatasetResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+@_typing.final
+class DeleteSyntheticDatasetResponse(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
 
-    SUCCESS_FIELD_NUMBER: builtins.int
-    success: builtins.bool
+    SUCCESS_FIELD_NUMBER: _builtins.int
+    success: _builtins.bool
     def __init__(
         self,
         *,
-        success: builtins.bool = ...,
+        success: _builtins.bool | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing.Literal["success", b"success"]) -> None: ...
+    _HasFieldArgType: _TypeAlias = _typing.Literal["success", b"success"]  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal["success", b"success"]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
 
-global___DeleteSyntheticDatasetResponse = DeleteSyntheticDatasetResponse
+Global___DeleteSyntheticDatasetResponse: _TypeAlias = DeleteSyntheticDatasetResponse  # noqa: Y015

weave/weaveapi/generate/v1/service_pb2_grpc.py

@@ -23,6 +23,12 @@ class GenerateStub(object):
             response_deserializer=weaveapi_dot_generate_dot_v1_dot_service__pb2.CreateSyntheticDatasetResponse.FromString,
             _registered_method=True,
         )
+        self.CreateSeedDataset = channel.unary_unary(
+            "/weaveapi.generate.v1.Generate/CreateSeedDataset",
+            request_serializer=weaveapi_dot_generate_dot_v1_dot_service__pb2.CreateSeedDatasetRequest.SerializeToString,
+            response_deserializer=weaveapi_dot_generate_dot_v1_dot_service__pb2.CreateSeedDatasetResponse.FromString,
+            _registered_method=True,
+        )
         self.GetSyntheticDataset = channel.unary_unary(
             "/weaveapi.generate.v1.Generate/GetSyntheticDataset",
             request_serializer=weaveapi_dot_generate_dot_v1_dot_service__pb2.GetSyntheticDatasetRequest.SerializeToString,
@@ -52,6 +58,12 @@ class GenerateServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def CreateSeedDataset(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
     def GetSyntheticDataset(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -78,6 +90,11 @@ def add_GenerateServicer_to_server(servicer, server):
             request_deserializer=weaveapi_dot_generate_dot_v1_dot_service__pb2.CreateSyntheticDatasetRequest.FromString,
             response_serializer=weaveapi_dot_generate_dot_v1_dot_service__pb2.CreateSyntheticDatasetResponse.SerializeToString,
         ),
+        "CreateSeedDataset": grpc.unary_unary_rpc_method_handler(
+            servicer.CreateSeedDataset,
+            request_deserializer=weaveapi_dot_generate_dot_v1_dot_service__pb2.CreateSeedDatasetRequest.FromString,
+            response_serializer=weaveapi_dot_generate_dot_v1_dot_service__pb2.CreateSeedDatasetResponse.SerializeToString,
+        ),
         "GetSyntheticDataset": grpc.unary_unary_rpc_method_handler(
             servicer.GetSyntheticDataset,
             request_deserializer=weaveapi_dot_generate_dot_v1_dot_service__pb2.GetSyntheticDatasetRequest.FromString,
@@ -137,6 +154,36 @@ class Generate(object):
             _registered_method=True,
         )
 
+    @staticmethod
+    def CreateSeedDataset(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/weaveapi.generate.v1.Generate/CreateSeedDataset",
+            weaveapi_dot_generate_dot_v1_dot_service__pb2.CreateSeedDatasetRequest.SerializeToString,
+            weaveapi_dot_generate_dot_v1_dot_service__pb2.CreateSeedDatasetResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True,
+        )
+
     @staticmethod
     def GetSyntheticDataset(
         request,