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

weave/weavesql/llmxdb/changes.py

@@ -0,0 +1,297 @@
+# Code generated by sqlc. DO NOT EDIT.
+# versions:
+#   sqlc v1.28.0
+# source: changes.sql
+import dataclasses
+import datetime
+from typing import AsyncIterator, Iterator, List, Optional
+import uuid
+
+import sqlalchemy
+import sqlalchemy.ext.asyncio
+
+from weave.weavesql.llmxdb import models
+
+
+BATCH_INSERT_CHANGES = """-- name: batch_insert_changes \\:copyfrom
+INSERT INTO model_changes (
+    model_id, field_name, old_value, new_value, change_type
+) VALUES (
+    :p1, :p2, :p3, :p4, :p5
+)
+"""
+
+
+@dataclasses.dataclass()
+class BatchInsertChangesParams:
+    model_id: uuid.UUID
+    field_name: str
+    old_value: Optional[str]
+    new_value: Optional[str]
+    change_type: Optional[str]
+
+
+GET_CHANGES_BY_MODEL = """-- name: get_changes_by_model \\:many
+SELECT id, model_id, field_name, old_value, new_value, change_type, detected_at, created_at
+FROM model_changes
+WHERE model_id = :p1
+ORDER BY detected_at DESC
+LIMIT :p2
+"""
+
+
+GET_CHANGES_BY_TYPE = """-- name: get_changes_by_type \\:many
+SELECT mc.id, mc.model_id, mc.field_name, mc.old_value, mc.new_value, mc.change_type, mc.detected_at, mc.created_at,
+       lm.slug as model_slug,
+       lm.name as model_name
+FROM model_changes mc
+         JOIN llm_models lm ON mc.model_id = lm.id
+WHERE mc.change_type = :p1
+  AND mc.detected_at > :p2
+ORDER BY mc.detected_at DESC
+"""
+
+
+@dataclasses.dataclass()
+class GetChangesByTypeRow:
+    id: uuid.UUID
+    model_id: uuid.UUID
+    field_name: str
+    old_value: Optional[str]
+    new_value: Optional[str]
+    change_type: Optional[str]
+    detected_at: Optional[datetime.datetime]
+    created_at: Optional[datetime.datetime]
+    model_slug: str
+    model_name: str
+
+
+GET_RECENT_CHANGES = """-- name: get_recent_changes \\:many
+SELECT id, model_id, field_name, old_value, new_value, change_type, detected_at, created_at
+FROM model_changes
+WHERE detected_at > NOW() - INTERVAL '24 hours'
+ORDER BY detected_at DESC
+LIMIT :p1
+"""
+
+
+INSERT_MODEL_CHANGE = """-- name: insert_model_change \\:one
+INSERT INTO model_changes (model_id, field_name, old_value, new_value, change_type)
+VALUES (:p1, :p2, :p3, :p4, :p5)
+RETURNING id, model_id, field_name, old_value, new_value, change_type, detected_at, created_at
+"""
+
+
+@dataclasses.dataclass()
+class InsertModelChangeParams:
+    model_id: uuid.UUID
+    field_name: str
+    old_value: Optional[str]
+    new_value: Optional[str]
+    change_type: Optional[str]
+
+
+class Querier:
+    def __init__(self, conn: sqlalchemy.engine.Connection):
+        self._conn = conn
+
+    def batch_insert_changes(self, arg_list: List[BatchInsertChangesParams]) -> int:
+        data = list()
+        for item in arg_list:
+            data.append(
+                {
+                    "p1": item.model_id,
+                    "p2": item.field_name,
+                    "p3": item.old_value,
+                    "p4": item.new_value,
+                    "p5": item.change_type,
+                }
+            )
+        result = self._conn.execute(sqlalchemy.text(BATCH_INSERT_CHANGES), data)
+        return result.rowcount
+
+    def get_changes_by_model(
+        self, *, model_id: uuid.UUID, limit_count: int
+    ) -> Iterator[models.ModelChange]:
+        result = self._conn.execute(
+            sqlalchemy.text(GET_CHANGES_BY_MODEL), {"p1": model_id, "p2": limit_count}
+        )
+        for row in result:
+            yield models.ModelChange(
+                id=row[0],
+                model_id=row[1],
+                field_name=row[2],
+                old_value=row[3],
+                new_value=row[4],
+                change_type=row[5],
+                detected_at=row[6],
+                created_at=row[7],
+            )
+
+    def get_changes_by_type(
+        self, *, change_type: Optional[str], since: Optional[datetime.datetime]
+    ) -> Iterator[GetChangesByTypeRow]:
+        result = self._conn.execute(
+            sqlalchemy.text(GET_CHANGES_BY_TYPE), {"p1": change_type, "p2": since}
+        )
+        for row in result:
+            yield GetChangesByTypeRow(
+                id=row[0],
+                model_id=row[1],
+                field_name=row[2],
+                old_value=row[3],
+                new_value=row[4],
+                change_type=row[5],
+                detected_at=row[6],
+                created_at=row[7],
+                model_slug=row[8],
+                model_name=row[9],
+            )
+
+    def get_recent_changes(self, *, limit_count: int) -> Iterator[models.ModelChange]:
+        result = self._conn.execute(
+            sqlalchemy.text(GET_RECENT_CHANGES), {"p1": limit_count}
+        )
+        for row in result:
+            yield models.ModelChange(
+                id=row[0],
+                model_id=row[1],
+                field_name=row[2],
+                old_value=row[3],
+                new_value=row[4],
+                change_type=row[5],
+                detected_at=row[6],
+                created_at=row[7],
+            )
+
+    def insert_model_change(
+        self, arg: InsertModelChangeParams
+    ) -> Optional[models.ModelChange]:
+        row = self._conn.execute(
+            sqlalchemy.text(INSERT_MODEL_CHANGE),
+            {
+                "p1": arg.model_id,
+                "p2": arg.field_name,
+                "p3": arg.old_value,
+                "p4": arg.new_value,
+                "p5": arg.change_type,
+            },
+        ).first()
+        if row is None:
+            return None
+        return models.ModelChange(
+            id=row[0],
+            model_id=row[1],
+            field_name=row[2],
+            old_value=row[3],
+            new_value=row[4],
+            change_type=row[5],
+            detected_at=row[6],
+            created_at=row[7],
+        )
+
+
+class AsyncQuerier:
+    def __init__(self, conn: sqlalchemy.ext.asyncio.AsyncConnection):
+        self._conn = conn
+
+    async def batch_insert_changes(
+        self, arg_list: List[BatchInsertChangesParams]
+    ) -> int:
+        data = list()
+        for item in arg_list:
+            data.append(
+                {
+                    "p1": item.model_id,
+                    "p2": item.field_name,
+                    "p3": item.old_value,
+                    "p4": item.new_value,
+                    "p5": item.change_type,
+                }
+            )
+        result = await self._conn.execute(sqlalchemy.text(BATCH_INSERT_CHANGES), data)
+        return result.rowcount
+
+    async def get_changes_by_model(
+        self, *, model_id: uuid.UUID, limit_count: int
+    ) -> AsyncIterator[models.ModelChange]:
+        result = await self._conn.stream(
+            sqlalchemy.text(GET_CHANGES_BY_MODEL), {"p1": model_id, "p2": limit_count}
+        )
+        async for row in result:
+            yield models.ModelChange(
+                id=row[0],
+                model_id=row[1],
+                field_name=row[2],
+                old_value=row[3],
+                new_value=row[4],
+                change_type=row[5],
+                detected_at=row[6],
+                created_at=row[7],
+            )
+
+    async def get_changes_by_type(
+        self, *, change_type: Optional[str], since: Optional[datetime.datetime]
+    ) -> AsyncIterator[GetChangesByTypeRow]:
+        result = await self._conn.stream(
+            sqlalchemy.text(GET_CHANGES_BY_TYPE), {"p1": change_type, "p2": since}
+        )
+        async for row in result:
+            yield GetChangesByTypeRow(
+                id=row[0],
+                model_id=row[1],
+                field_name=row[2],
+                old_value=row[3],
+                new_value=row[4],
+                change_type=row[5],
+                detected_at=row[6],
+                created_at=row[7],
+                model_slug=row[8],
+                model_name=row[9],
+            )
+
+    async def get_recent_changes(
+        self, *, limit_count: int
+    ) -> AsyncIterator[models.ModelChange]:
+        result = await self._conn.stream(
+            sqlalchemy.text(GET_RECENT_CHANGES), {"p1": limit_count}
+        )
+        async for row in result:
+            yield models.ModelChange(
+                id=row[0],
+                model_id=row[1],
+                field_name=row[2],
+                old_value=row[3],
+                new_value=row[4],
+                change_type=row[5],
+                detected_at=row[6],
+                created_at=row[7],
+            )
+
+    async def insert_model_change(
+        self, arg: InsertModelChangeParams
+    ) -> Optional[models.ModelChange]:
+        row = (
+            await self._conn.execute(
+                sqlalchemy.text(INSERT_MODEL_CHANGE),
+                {
+                    "p1": arg.model_id,
+                    "p2": arg.field_name,
+                    "p3": arg.old_value,
+                    "p4": arg.new_value,
+                    "p5": arg.change_type,
+                },
+            )
+        ).first()
+        if row is None:
+            return None
+        return models.ModelChange(
+            id=row[0],
+            model_id=row[1],
+            field_name=row[2],
+            old_value=row[3],
+            new_value=row[4],
+            change_type=row[5],
+            detected_at=row[6],
+            created_at=row[7],
+        )