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

weave/weavesql/weavedb/dataset.py

@@ -0,0 +1,75 @@
+# Code generated by sqlc. DO NOT EDIT.
+# versions:
+#   sqlc v1.28.0
+# source: dataset.sql
+import dataclasses
+from typing import Optional
+import uuid
+
+import sqlalchemy
+import sqlalchemy.ext.asyncio
+
+
+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]
+
+
+class Querier:
+    def __init__(self, conn: sqlalchemy.engine.Connection):
+        self._conn = conn
+
+    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,
+            },
+        )
+
+
+class AsyncQuerier:
+    def __init__(self, conn: sqlalchemy.ext.asyncio.AsyncConnection):
+        self._conn = conn
+
+    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,
+            },
+        )

weave/weavesql/weavedb/models.py

@@ -0,0 +1,135 @@
+# Code generated by sqlc. DO NOT EDIT.
+# versions:
+#   sqlc v1.28.0
+import dataclasses
+import datetime
+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 LlmProviderCredential:
+    id: uuid.UUID
+    organization_id: uuid.UUID
+    provider_slug: str
+    credentials: Any
+    created_at: Optional[datetime.datetime]
+    updated_at: Optional[datetime.datetime]
+
+
+@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/relationships.py

@@ -0,0 +1,72 @@
+# Code generated by sqlc. DO NOT EDIT.
+# versions:
+#   sqlc v1.28.0
+# source: relationships.sql
+import dataclasses
+from typing import AsyncIterator, Iterator, List
+import uuid
+
+import sqlalchemy
+import sqlalchemy.ext.asyncio
+
+
+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
+
+
+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],
+            )
+
+
+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],
+            )

weave/weavesql/weavedb/storage.py

@@ -0,0 +1,113 @@
+# Code generated by sqlc. DO NOT EDIT.
+# versions:
+#   sqlc v1.28.0
+# source: storage.sql
+from typing import Optional
+import uuid
+
+import sqlalchemy
+import sqlalchemy.ext.asyncio
+
+from weave.weavesql.weavedb import models
+
+
+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
+"""
+
+
+class Querier:
+    def __init__(self, conn: sqlalchemy.engine.Connection):
+        self._conn = conn
+
+    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],
+        )
+
+
+class AsyncQuerier:
+    def __init__(self, conn: sqlalchemy.ext.asyncio.AsyncConnection):
+        self._conn = conn
+
+    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],
+        )

weave/weavesql/weavedb/synthesizer.py

@@ -0,0 +1,107 @@
+# Code generated by sqlc. DO NOT EDIT.
+# versions:
+#   sqlc v1.28.0
+# source: synthesizer.sql
+import dataclasses
+from typing import Any, Optional
+import uuid
+
+import sqlalchemy
+import sqlalchemy.ext.asyncio
+
+from weave.weavesql.weavedb import models
+
+
+GET_SYNTHESIZER = """-- name: get_synthesizer \\:one
+SELECT id, organization_id, name, description, storage_connection_id, metadata
+FROM synthesize_v1.synthesizer
+WHERE id = :p1
+"""
+
+
+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_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_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_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_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,
+            },
+        )

weave_python-0.30.3.dist-info/METADATA

@@ -0,0 +1,52 @@
+Metadata-Version: 2.4
+Name: weave-python
+Version: 0.30.3
+Author: Weave Labs
+License: FCL-1.0-ALv2
+License-File: LICENSE
+Requires-Python: >=3.10
+Requires-Dist: protobuf>=6.30.2
+Requires-Dist: sqlalchemy>=2.0.40
+Description-Content-Type: text/markdown
+
+# Weave Python SDK
+
+Weave Python provides client utilities for integrating with Weave platform services,
+covering authentication, generation, synthesis, storage, and more. It is intended
+for Python applications that need to orchestrate Weave workflows without managing
+low-level HTTP plumbing.
+
+## Installation
+
+```bash
+pip install weave-python
+```
+
+The package requires Python 3.10 or newer.
+
+## Quick Start
+
+```python
+from weave.weaveapi import auth, generate
+
+# Authenticate with your Weave credentials
+session = auth.login_with_api_key("your-api-key")
+
+# Send a generation request
+result = generate.text(session=session, prompt="Hello Weave!")
+print(result.text)
+```
+
+Refer to the individual subpackages under `weave.weaveapi` (for example `auth`,
+`synthesize`, `storage`, and `payment`) for domain-specific helpers.
+
+## Contributing
+
+1. Create a virtual environment and install dev dependencies with `uv pip sync uv.lock`.
+2. Run formatting and linting via `task format` and `task lint`.
+3. Open a pull request with context on the workflow or service you changed.
+
+## License
+
+Distributed under the Apache Software License and MIT License. See `LICENSE` for
+additional licensing details.