weave-python 0.27.0__py3-none-any.whl → 0.28.0__py3-none-any.whl

weave/weaveapi/modex/v1/service_pb2_grpc.pyi

@@ -1,268 +0,0 @@
-"""
-@generated by mypy-protobuf.  Do not edit manually!
-isort:skip_file
-"""
-
-import abc
-import collections.abc
-import grpc
-import grpc.aio
-import typing
-import weaveapi.modex.v1.service_pb2
-
-_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]
-    ...
-
-class ModexStub:
-    def __init__(
-        self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]
-    ) -> None: ...
-    ListProviders: grpc.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.ListProvidersRequest,
-        weaveapi.modex.v1.service_pb2.ListProvidersResponse,
-    ]
-
-    GetProvider: grpc.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.GetProviderRequest,
-        weaveapi.modex.v1.service_pb2.GetProviderResponse,
-    ]
-
-    ListModels: grpc.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.ListModelsRequest,
-        weaveapi.modex.v1.service_pb2.ListModelsResponse,
-    ]
-
-    GetModel: grpc.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.GetModelRequest,
-        weaveapi.modex.v1.service_pb2.GetModelResponse,
-    ]
-
-    SearchModels: grpc.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.SearchModelsRequest,
-        weaveapi.modex.v1.service_pb2.SearchModelsResponse,
-    ]
-
-    CompareModels: grpc.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.CompareModelsRequest,
-        weaveapi.modex.v1.service_pb2.CompareModelsResponse,
-    ]
-
-    GetModelPriceHistory: grpc.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.GetModelPriceHistoryRequest,
-        weaveapi.modex.v1.service_pb2.GetModelPriceHistoryResponse,
-    ]
-
-    GetModelChanges: grpc.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.GetModelChangesRequest,
-        weaveapi.modex.v1.service_pb2.GetModelChangesResponse,
-    ]
-
-    GetTrendingModels: grpc.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.GetTrendingModelsRequest,
-        weaveapi.modex.v1.service_pb2.GetTrendingModelsResponse,
-    ]
-
-    GetRecentUpdates: grpc.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.GetRecentUpdatesRequest,
-        weaveapi.modex.v1.service_pb2.GetRecentUpdatesResponse,
-    ]
-
-    ExportData: grpc.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.ExportDataRequest,
-        weaveapi.modex.v1.service_pb2.ExportDataResponse,
-    ]
-
-    GetStatistics: grpc.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.GetStatisticsRequest,
-        weaveapi.modex.v1.service_pb2.GetStatisticsResponse,
-    ]
-
-class ModexAsyncStub:
-    ListProviders: grpc.aio.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.ListProvidersRequest,
-        weaveapi.modex.v1.service_pb2.ListProvidersResponse,
-    ]
-
-    GetProvider: grpc.aio.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.GetProviderRequest,
-        weaveapi.modex.v1.service_pb2.GetProviderResponse,
-    ]
-
-    ListModels: grpc.aio.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.ListModelsRequest,
-        weaveapi.modex.v1.service_pb2.ListModelsResponse,
-    ]
-
-    GetModel: grpc.aio.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.GetModelRequest,
-        weaveapi.modex.v1.service_pb2.GetModelResponse,
-    ]
-
-    SearchModels: grpc.aio.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.SearchModelsRequest,
-        weaveapi.modex.v1.service_pb2.SearchModelsResponse,
-    ]
-
-    CompareModels: grpc.aio.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.CompareModelsRequest,
-        weaveapi.modex.v1.service_pb2.CompareModelsResponse,
-    ]
-
-    GetModelPriceHistory: grpc.aio.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.GetModelPriceHistoryRequest,
-        weaveapi.modex.v1.service_pb2.GetModelPriceHistoryResponse,
-    ]
-
-    GetModelChanges: grpc.aio.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.GetModelChangesRequest,
-        weaveapi.modex.v1.service_pb2.GetModelChangesResponse,
-    ]
-
-    GetTrendingModels: grpc.aio.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.GetTrendingModelsRequest,
-        weaveapi.modex.v1.service_pb2.GetTrendingModelsResponse,
-    ]
-
-    GetRecentUpdates: grpc.aio.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.GetRecentUpdatesRequest,
-        weaveapi.modex.v1.service_pb2.GetRecentUpdatesResponse,
-    ]
-
-    ExportData: grpc.aio.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.ExportDataRequest,
-        weaveapi.modex.v1.service_pb2.ExportDataResponse,
-    ]
-
-    GetStatistics: grpc.aio.UnaryUnaryMultiCallable[
-        weaveapi.modex.v1.service_pb2.GetStatisticsRequest,
-        weaveapi.modex.v1.service_pb2.GetStatisticsResponse,
-    ]
-
-class ModexServicer(metaclass=abc.ABCMeta):
-    @abc.abstractmethod
-    def ListProviders(
-        self,
-        request: weaveapi.modex.v1.service_pb2.ListProvidersRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[
-        weaveapi.modex.v1.service_pb2.ListProvidersResponse,
-        collections.abc.Awaitable[weaveapi.modex.v1.service_pb2.ListProvidersResponse],
-    ]: ...
-    @abc.abstractmethod
-    def GetProvider(
-        self,
-        request: weaveapi.modex.v1.service_pb2.GetProviderRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[
-        weaveapi.modex.v1.service_pb2.GetProviderResponse,
-        collections.abc.Awaitable[weaveapi.modex.v1.service_pb2.GetProviderResponse],
-    ]: ...
-    @abc.abstractmethod
-    def ListModels(
-        self,
-        request: weaveapi.modex.v1.service_pb2.ListModelsRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[
-        weaveapi.modex.v1.service_pb2.ListModelsResponse,
-        collections.abc.Awaitable[weaveapi.modex.v1.service_pb2.ListModelsResponse],
-    ]: ...
-    @abc.abstractmethod
-    def GetModel(
-        self,
-        request: weaveapi.modex.v1.service_pb2.GetModelRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[
-        weaveapi.modex.v1.service_pb2.GetModelResponse,
-        collections.abc.Awaitable[weaveapi.modex.v1.service_pb2.GetModelResponse],
-    ]: ...
-    @abc.abstractmethod
-    def SearchModels(
-        self,
-        request: weaveapi.modex.v1.service_pb2.SearchModelsRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[
-        weaveapi.modex.v1.service_pb2.SearchModelsResponse,
-        collections.abc.Awaitable[weaveapi.modex.v1.service_pb2.SearchModelsResponse],
-    ]: ...
-    @abc.abstractmethod
-    def CompareModels(
-        self,
-        request: weaveapi.modex.v1.service_pb2.CompareModelsRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[
-        weaveapi.modex.v1.service_pb2.CompareModelsResponse,
-        collections.abc.Awaitable[weaveapi.modex.v1.service_pb2.CompareModelsResponse],
-    ]: ...
-    @abc.abstractmethod
-    def GetModelPriceHistory(
-        self,
-        request: weaveapi.modex.v1.service_pb2.GetModelPriceHistoryRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[
-        weaveapi.modex.v1.service_pb2.GetModelPriceHistoryResponse,
-        collections.abc.Awaitable[
-            weaveapi.modex.v1.service_pb2.GetModelPriceHistoryResponse
-        ],
-    ]: ...
-    @abc.abstractmethod
-    def GetModelChanges(
-        self,
-        request: weaveapi.modex.v1.service_pb2.GetModelChangesRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[
-        weaveapi.modex.v1.service_pb2.GetModelChangesResponse,
-        collections.abc.Awaitable[
-            weaveapi.modex.v1.service_pb2.GetModelChangesResponse
-        ],
-    ]: ...
-    @abc.abstractmethod
-    def GetTrendingModels(
-        self,
-        request: weaveapi.modex.v1.service_pb2.GetTrendingModelsRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[
-        weaveapi.modex.v1.service_pb2.GetTrendingModelsResponse,
-        collections.abc.Awaitable[
-            weaveapi.modex.v1.service_pb2.GetTrendingModelsResponse
-        ],
-    ]: ...
-    @abc.abstractmethod
-    def GetRecentUpdates(
-        self,
-        request: weaveapi.modex.v1.service_pb2.GetRecentUpdatesRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[
-        weaveapi.modex.v1.service_pb2.GetRecentUpdatesResponse,
-        collections.abc.Awaitable[
-            weaveapi.modex.v1.service_pb2.GetRecentUpdatesResponse
-        ],
-    ]: ...
-    @abc.abstractmethod
-    def ExportData(
-        self,
-        request: weaveapi.modex.v1.service_pb2.ExportDataRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[
-        weaveapi.modex.v1.service_pb2.ExportDataResponse,
-        collections.abc.Awaitable[weaveapi.modex.v1.service_pb2.ExportDataResponse],
-    ]: ...
-    @abc.abstractmethod
-    def GetStatistics(
-        self,
-        request: weaveapi.modex.v1.service_pb2.GetStatisticsRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[
-        weaveapi.modex.v1.service_pb2.GetStatisticsResponse,
-        collections.abc.Awaitable[weaveapi.modex.v1.service_pb2.GetStatisticsResponse],
-    ]: ...
-
-def add_ModexServicer_to_server(
-    servicer: ModexServicer, server: typing.Union[grpc.Server, grpc.aio.Server]
-) -> None: ...

weave/weavesql/weavedb/models.py

@@ -1,124 +0,0 @@
-# Code generated by sqlc. DO NOT EDIT.
-# versions:
-#   sqlc v1.28.0
-import dataclasses
-import enum
-from typing import Any, Optional
-import uuid
-
-
-class StorageCategory(enum.StrEnum):
-    OBJECT_STORE = "object_store"
-    SQL_DATABASE = "sql_database"
-    NOSQL_DATABASE = "nosql_database"
-
-
-class StorageVariant(enum.StrEnum):
-    AWS_S3 = "aws_s3"
-    AZURE_BLOB_STORAGE = "azure_blob_storage"
-    GOOGLE_CLOUD_STORAGE = "google_cloud_storage"
-    MYSQL = "mysql"
-    POSTGRES = "postgres"
-    SQL_SERVER = "sql_server"
-    ORACLE = "oracle"
-    MARIADB = "mariadb"
-    SNOWFLAKE = "snowflake"
-    BIGQUERY = "bigquery"
-    DATABRICKS = "databricks"
-    CASSANDRA = "cassandra"
-    MONGODB = "mongodb"
-    DYNAMODB = "dynamodb"
-    COSMOSDB = "cosmosdb"
-    REDIS = "redis"
-    VALKEY = "valkey"
-    ELASTICSEARCH = "elasticsearch"
-    OPENSEARCH = "opensearch"
-
-
-class SynthesizeV1ColumnDataType(enum.StrEnum):
-    STRING = "string"
-    INTEGER = "integer"
-    FLOAT = "float"
-    BOOLEAN = "boolean"
-    TIMESTAMP = "timestamp"
-    DATE = "date"
-    BINARY = "binary"
-    CATEGORICAL = "categorical"
-    ORDINAL = "ordinal"
-
-
-class SynthesizeV1ColumnRelationshipType(enum.StrEnum):
-    CONTEXT_FOREIGN_KEY = "context_foreign_key"
-    NON_CONTEXT_FOREIGN_KEY = "non_context_foreign_key"
-    ONE_TO_MANY = "one_to_many"
-    MANY_TO_ONE = "many_to_one"
-    MANY_TO_MANY = "many_to_many"
-    SELF_REFERENCE = "self_reference"
-
-
-@dataclasses.dataclass()
-class GenerateV1Dataset:
-    id: uuid.UUID
-    organization_id: uuid.UUID
-    name: str
-    description: Optional[str]
-    storage_connection_id: uuid.UUID
-    schema_id: Optional[uuid.UUID]
-    synthesizer_id: Optional[uuid.UUID]
-    is_synthetic: Optional[bool]
-
-
-@dataclasses.dataclass()
-class Organization:
-    id: uuid.UUID
-    name: str
-    description: Optional[str]
-
-
-@dataclasses.dataclass()
-class StorageConnection:
-    id: uuid.UUID
-    organization_id: uuid.UUID
-    name: str
-    description: Optional[str]
-    category: StorageCategory
-    variant: StorageVariant
-    details: Any
-
-
-@dataclasses.dataclass()
-class SynthesizeV1ColumnRelationship:
-    id: uuid.UUID
-    source_column_id: uuid.UUID
-    source_dataset_id: uuid.UUID
-    target_column_id: uuid.UUID
-    target_dataset_id: uuid.UUID
-    relationship_type: Any
-
-
-@dataclasses.dataclass()
-class SynthesizeV1DataColumn:
-    id: uuid.UUID
-    schema_id: uuid.UUID
-    name: str
-    data_type: Any
-    is_primary_key: Optional[bool]
-    description: Optional[str]
-
-
-@dataclasses.dataclass()
-class SynthesizeV1DataSchema:
-    id: uuid.UUID
-    organization_id: uuid.UUID
-    name: str
-    description: Optional[str]
-
-
-@dataclasses.dataclass()
-class SynthesizeV1Synthesizer:
-    id: uuid.UUID
-    organization_id: uuid.UUID
-    name: str
-    description: Optional[str]
-    storage_connection_id: uuid.UUID
-    metadata: Optional[Any]

weave/weavesql/weavedb/queries.py

@@ -1,306 +0,0 @@
-# Code generated by sqlc. DO NOT EDIT.
-# versions:
-#   sqlc v1.28.0
-# source: queries.sql
-import dataclasses
-from typing import Any, AsyncIterator, Iterator, List, Optional
-import uuid
-
-import sqlalchemy
-import sqlalchemy.ext.asyncio
-
-from weave.weavesql.weavedb 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[])
-"""
-
-
-@dataclasses.dataclass()
-class GetForeignKeyRelationshipsRow:
-    source_dataset_id: uuid.UUID
-    target_dataset_id: uuid.UUID
-    source_column_name: str
-    target_column_name: str
-
-
-GET_STORAGE_CONNECTION = """-- name: get_storage_connection \\:one
-SELECT id, organization_id, name, description, category, variant, details
-FROM storage_connection
-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 generate_v1.dataset d ON d.storage_connection_id = sc.id
-WHERE d.id = :p1
-"""
-
-
-GET_SYNTHESIZER = """-- name: get_synthesizer \\:one
-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 generate_v1.dataset (id, organization_id, name, description, storage_connection_id, schema_id,
-                                 synthesizer_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,
-                               synthesizer_id        = EXCLUDED.synthesizer_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: Optional[uuid.UUID]
-    synthesizer_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
-"""
-
-
-@dataclasses.dataclass()
-class UpsertSynthesizerParams:
-    id: uuid.UUID
-    organization_id: uuid.UUID
-    name: str
-    description: Optional[str]
-    storage_connection_id: uuid.UUID
-    metadata: Optional[Any]
-
-
-class Querier:
-    def __init__(self, conn: sqlalchemy.engine.Connection):
-        self._conn = conn
-
-    def get_foreign_key_relationships(
-        self, *, dataset_ids: List[uuid.UUID]
-    ) -> Iterator[GetForeignKeyRelationshipsRow]:
-        result = self._conn.execute(
-            sqlalchemy.text(GET_FOREIGN_KEY_RELATIONSHIPS), {"p1": dataset_ids}
-        )
-        for row in result:
-            yield GetForeignKeyRelationshipsRow(
-                source_dataset_id=row[0],
-                target_dataset_id=row[1],
-                source_column_name=row[2],
-                target_column_name=row[3],
-            )
-
-    def get_storage_connection(
-        self, *, id: uuid.UUID
-    ) -> Optional[models.StorageConnection]:
-        row = self._conn.execute(
-            sqlalchemy.text(GET_STORAGE_CONNECTION), {"p1": id}
-        ).first()
-        if row is None:
-            return None
-        return models.StorageConnection(
-            id=row[0],
-            organization_id=row[1],
-            name=row[2],
-            description=row[3],
-            category=row[4],
-            variant=row[5],
-            details=row[6],
-        )
-
-    def get_storage_connection_by_dataset_id(
-        self, *, dataset_id: uuid.UUID
-    ) -> Optional[models.StorageConnection]:
-        row = self._conn.execute(
-            sqlalchemy.text(GET_STORAGE_CONNECTION_BY_DATASET_ID), {"p1": dataset_id}
-        ).first()
-        if row is None:
-            return None
-        return models.StorageConnection(
-            id=row[0],
-            organization_id=row[1],
-            name=row[2],
-            description=row[3],
-            category=row[4],
-            variant=row[5],
-            details=row[6],
-        )
-
-    def get_synthesizer(
-        self, *, id: uuid.UUID
-    ) -> Optional[models.SynthesizeV1Synthesizer]:
-        row = self._conn.execute(sqlalchemy.text(GET_SYNTHESIZER), {"p1": id}).first()
-        if row is None:
-            return None
-        return models.SynthesizeV1Synthesizer(
-            id=row[0],
-            organization_id=row[1],
-            name=row[2],
-            description=row[3],
-            storage_connection_id=row[4],
-            metadata=row[5],
-        )
-
-    def upsert_dataset(self, arg: UpsertDatasetParams) -> None:
-        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.synthesizer_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,
-                "p3": arg.name,
-                "p4": arg.description,
-                "p5": arg.storage_connection_id,
-                "p6": arg.metadata,
-            },
-        )
-
-
-class AsyncQuerier:
-    def __init__(self, conn: sqlalchemy.ext.asyncio.AsyncConnection):
-        self._conn = conn
-
-    async def get_foreign_key_relationships(
-        self, *, dataset_ids: List[uuid.UUID]
-    ) -> AsyncIterator[GetForeignKeyRelationshipsRow]:
-        result = await self._conn.stream(
-            sqlalchemy.text(GET_FOREIGN_KEY_RELATIONSHIPS), {"p1": dataset_ids}
-        )
-        async for row in result:
-            yield GetForeignKeyRelationshipsRow(
-                source_dataset_id=row[0],
-                target_dataset_id=row[1],
-                source_column_name=row[2],
-                target_column_name=row[3],
-            )
-
-    async def get_storage_connection(
-        self, *, id: uuid.UUID
-    ) -> Optional[models.StorageConnection]:
-        row = (
-            await self._conn.execute(
-                sqlalchemy.text(GET_STORAGE_CONNECTION), {"p1": id}
-            )
-        ).first()
-        if row is None:
-            return None
-        return models.StorageConnection(
-            id=row[0],
-            organization_id=row[1],
-            name=row[2],
-            description=row[3],
-            category=row[4],
-            variant=row[5],
-            details=row[6],
-        )
-
-    async def get_storage_connection_by_dataset_id(
-        self, *, dataset_id: uuid.UUID
-    ) -> Optional[models.StorageConnection]:
-        row = (
-            await self._conn.execute(
-                sqlalchemy.text(GET_STORAGE_CONNECTION_BY_DATASET_ID),
-                {"p1": dataset_id},
-            )
-        ).first()
-        if row is None:
-            return None
-        return models.StorageConnection(
-            id=row[0],
-            organization_id=row[1],
-            name=row[2],
-            description=row[3],
-            category=row[4],
-            variant=row[5],
-            details=row[6],
-        )
-
-    async def get_synthesizer(
-        self, *, id: uuid.UUID
-    ) -> Optional[models.SynthesizeV1Synthesizer]:
-        row = (
-            await self._conn.execute(sqlalchemy.text(GET_SYNTHESIZER), {"p1": id})
-        ).first()
-        if row is None:
-            return None
-        return models.SynthesizeV1Synthesizer(
-            id=row[0],
-            organization_id=row[1],
-            name=row[2],
-            description=row[3],
-            storage_connection_id=row[4],
-            metadata=row[5],
-        )
-
-    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.synthesizer_id,
-                "p8": arg.is_synthetic,
-            },
-        )
-
-    async def upsert_synthesizer(self, arg: UpsertSynthesizerParams) -> None:
-        await self._conn.execute(
-            sqlalchemy.text(UPSERT_SYNTHESIZER),
-            {
-                "p1": arg.id,
-                "p2": arg.organization_id,
-                "p3": arg.name,
-                "p4": arg.description,
-                "p5": arg.storage_connection_id,
-                "p6": arg.metadata,
-            },
-        )