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

weave/weavesql/llmxdb/capabilities.py

@@ -0,0 +1,487 @@
+# Code generated by sqlc. DO NOT EDIT.
+# versions:
+#   sqlc v1.28.0
+# source: capabilities.sql
+import dataclasses
+import datetime
+from typing import Any, AsyncIterator, Iterator, Optional
+import uuid
+
+import sqlalchemy
+import sqlalchemy.ext.asyncio
+
+from weave.weavesql.llmxdb import models
+
+
+CHECK_MODEL_HAS_CAPABILITY = """-- name: check_model_has_capability \\:one
+SELECT EXISTS (
+  SELECT 1 
+  FROM llm_models 
+  WHERE id = :p1 
+  AND capabilities->'configs' ? :p2
+) as has_capability
+"""
+
+
+GET_CAPABILITY_CONFIG = """-- name: get_capability_config \\:one
+SELECT capabilities->'configs'->@capability_type as config
+FROM llm_models
+WHERE slug = :p1
+  AND is_active = true
+"""
+
+
+GET_MODEL_CAPABILITIES = """-- name: get_model_capabilities \\:one
+SELECT capabilities
+FROM llm_models
+WHERE slug = :p1
+  AND is_active = true
+"""
+
+
+GET_MODEL_CAPABILITY_TYPES = """-- name: get_model_capability_types \\:many
+SELECT jsonb_object_keys(capabilities->'configs') as capability_type
+FROM llm_models
+WHERE id = :p1
+ORDER BY 1
+"""
+
+
+GET_MODELS_WITH_CAPABILITY = """-- name: get_models_with_capability \\:many
+SELECT id, provider_id, model_id, slug, name, display_name, description, version, capabilities, classification, architecture, performance, token_info, pricing, api_details, training, licensing, safety, availability, technical_specs, release_date, training_data_cutoff, deprecation_date, metadata, last_scraped_at, is_active, is_deprecated, created_at, updated_at
+FROM llm_models
+WHERE is_active = true
+  AND capabilities->'configs' ? :p1
+  AND (:p2\\:\\:uuid IS NULL OR provider_id = :p2)
+ORDER BY provider_id, name
+"""
+
+
+GET_MODELS_WITH_STRUCTURED_RESPONSE = """-- name: get_models_with_structured_response \\:many
+SELECT m.id, m.provider_id, m.model_id, m.slug, m.name, m.display_name, m.description, m.version, m.capabilities, m.classification, m.architecture, m.performance, m.token_info, m.pricing, m.api_details, m.training, m.licensing, m.safety, m.availability, m.technical_specs, m.release_date, m.training_data_cutoff, m.deprecation_date, m.metadata, m.last_scraped_at, m.is_active, m.is_deprecated, m.created_at, m.updated_at, 
+       c.config->>'max_schema_depth' as max_schema_depth,
+       c.config->>'requires_json_mode' as requires_json_mode
+FROM llm_models m,
+     LATERAL (
+       SELECT capabilities->'configs'->'CAPABILITY_TYPE_STRUCTURED_RESPONSE' as config
+     ) c
+WHERE m.is_active = true
+  AND m.capabilities->'configs' ? 'CAPABILITY_TYPE_STRUCTURED_RESPONSE'
+  AND (:p1\\:\\:uuid IS NULL OR m.provider_id = :p1)
+ORDER BY m.provider_id, m.name
+"""
+
+
+@dataclasses.dataclass()
+class GetModelsWithStructuredResponseRow:
+    id: uuid.UUID
+    provider_id: uuid.UUID
+    model_id: str
+    slug: str
+    name: str
+    display_name: Optional[str]
+    description: Optional[str]
+    version: Optional[str]
+    capabilities: Any
+    classification: Any
+    architecture: Any
+    performance: Any
+    token_info: Any
+    pricing: Any
+    api_details: Any
+    training: Any
+    licensing: Any
+    safety: Any
+    availability: Any
+    technical_specs: Any
+    release_date: Optional[datetime.datetime]
+    training_data_cutoff: Optional[datetime.datetime]
+    deprecation_date: Optional[datetime.datetime]
+    metadata: Any
+    last_scraped_at: Optional[datetime.datetime]
+    is_active: Optional[bool]
+    is_deprecated: Optional[bool]
+    created_at: Optional[datetime.datetime]
+    updated_at: Optional[datetime.datetime]
+    max_schema_depth: Optional[Any]
+    requires_json_mode: Optional[Any]
+
+
+GET_MODELS_WITH_VISION = """-- name: get_models_with_vision \\:many
+SELECT m.id, m.provider_id, m.model_id, m.slug, m.name, m.display_name, m.description, m.version, m.capabilities, m.classification, m.architecture, m.performance, m.token_info, m.pricing, m.api_details, m.training, m.licensing, m.safety, m.availability, m.technical_specs, m.release_date, m.training_data_cutoff, m.deprecation_date, m.metadata, m.last_scraped_at, m.is_active, m.is_deprecated, m.created_at, m.updated_at,
+       c.config->>'max_images_per_request' as max_images,
+       c.config->>'supported_formats' as formats
+FROM llm_models m,
+     LATERAL (
+       SELECT capabilities->'configs'->'CAPABILITY_TYPE_VISION' as config
+     ) c
+WHERE m.is_active = true
+  AND m.capabilities->'configs' ? 'CAPABILITY_TYPE_VISION'
+  AND (:p1\\:\\:boolean IS NULL OR (c.config->>'supports_ocr')\\:\\:boolean = :p1)
+ORDER BY m.provider_id, m.name
+"""
+
+
+@dataclasses.dataclass()
+class GetModelsWithVisionRow:
+    id: uuid.UUID
+    provider_id: uuid.UUID
+    model_id: str
+    slug: str
+    name: str
+    display_name: Optional[str]
+    description: Optional[str]
+    version: Optional[str]
+    capabilities: Any
+    classification: Any
+    architecture: Any
+    performance: Any
+    token_info: Any
+    pricing: Any
+    api_details: Any
+    training: Any
+    licensing: Any
+    safety: Any
+    availability: Any
+    technical_specs: Any
+    release_date: Optional[datetime.datetime]
+    training_data_cutoff: Optional[datetime.datetime]
+    deprecation_date: Optional[datetime.datetime]
+    metadata: Any
+    last_scraped_at: Optional[datetime.datetime]
+    is_active: Optional[bool]
+    is_deprecated: Optional[bool]
+    created_at: Optional[datetime.datetime]
+    updated_at: Optional[datetime.datetime]
+    max_images: Optional[Any]
+    formats: Optional[Any]
+
+
+class Querier:
+    def __init__(self, conn: sqlalchemy.engine.Connection):
+        self._conn = conn
+
+    def check_model_has_capability(
+        self, *, model_id: uuid.UUID, capability_type: Any
+    ) -> Optional[bool]:
+        row = self._conn.execute(
+            sqlalchemy.text(CHECK_MODEL_HAS_CAPABILITY),
+            {"p1": model_id, "p2": capability_type},
+        ).first()
+        if row is None:
+            return None
+        return row[0]
+
+    def get_capability_config(self, *, slug: str) -> Optional[Optional[Any]]:
+        row = self._conn.execute(
+            sqlalchemy.text(GET_CAPABILITY_CONFIG), {"p1": slug}
+        ).first()
+        if row is None:
+            return None
+        return row[0]
+
+    def get_model_capabilities(self, *, slug: str) -> Optional[Any]:
+        row = self._conn.execute(
+            sqlalchemy.text(GET_MODEL_CAPABILITIES), {"p1": slug}
+        ).first()
+        if row is None:
+            return None
+        return row[0]
+
+    def get_model_capability_types(self, *, model_id: uuid.UUID) -> Iterator[str]:
+        result = self._conn.execute(
+            sqlalchemy.text(GET_MODEL_CAPABILITY_TYPES), {"p1": model_id}
+        )
+        for row in result:
+            yield row[0]
+
+    def get_models_with_capability(
+        self, *, capability_type: Any, provider_id: uuid.UUID
+    ) -> Iterator[models.LlmModel]:
+        result = self._conn.execute(
+            sqlalchemy.text(GET_MODELS_WITH_CAPABILITY),
+            {"p1": capability_type, "p2": provider_id},
+        )
+        for row in result:
+            yield models.LlmModel(
+                id=row[0],
+                provider_id=row[1],
+                model_id=row[2],
+                slug=row[3],
+                name=row[4],
+                display_name=row[5],
+                description=row[6],
+                version=row[7],
+                capabilities=row[8],
+                classification=row[9],
+                architecture=row[10],
+                performance=row[11],
+                token_info=row[12],
+                pricing=row[13],
+                api_details=row[14],
+                training=row[15],
+                licensing=row[16],
+                safety=row[17],
+                availability=row[18],
+                technical_specs=row[19],
+                release_date=row[20],
+                training_data_cutoff=row[21],
+                deprecation_date=row[22],
+                metadata=row[23],
+                last_scraped_at=row[24],
+                is_active=row[25],
+                is_deprecated=row[26],
+                created_at=row[27],
+                updated_at=row[28],
+            )
+
+    def get_models_with_structured_response(
+        self, *, provider_id: uuid.UUID
+    ) -> Iterator[GetModelsWithStructuredResponseRow]:
+        result = self._conn.execute(
+            sqlalchemy.text(GET_MODELS_WITH_STRUCTURED_RESPONSE), {"p1": provider_id}
+        )
+        for row in result:
+            yield GetModelsWithStructuredResponseRow(
+                id=row[0],
+                provider_id=row[1],
+                model_id=row[2],
+                slug=row[3],
+                name=row[4],
+                display_name=row[5],
+                description=row[6],
+                version=row[7],
+                capabilities=row[8],
+                classification=row[9],
+                architecture=row[10],
+                performance=row[11],
+                token_info=row[12],
+                pricing=row[13],
+                api_details=row[14],
+                training=row[15],
+                licensing=row[16],
+                safety=row[17],
+                availability=row[18],
+                technical_specs=row[19],
+                release_date=row[20],
+                training_data_cutoff=row[21],
+                deprecation_date=row[22],
+                metadata=row[23],
+                last_scraped_at=row[24],
+                is_active=row[25],
+                is_deprecated=row[26],
+                created_at=row[27],
+                updated_at=row[28],
+                max_schema_depth=row[29],
+                requires_json_mode=row[30],
+            )
+
+    def get_models_with_vision(
+        self, *, supports_ocr: bool
+    ) -> Iterator[GetModelsWithVisionRow]:
+        result = self._conn.execute(
+            sqlalchemy.text(GET_MODELS_WITH_VISION), {"p1": supports_ocr}
+        )
+        for row in result:
+            yield GetModelsWithVisionRow(
+                id=row[0],
+                provider_id=row[1],
+                model_id=row[2],
+                slug=row[3],
+                name=row[4],
+                display_name=row[5],
+                description=row[6],
+                version=row[7],
+                capabilities=row[8],
+                classification=row[9],
+                architecture=row[10],
+                performance=row[11],
+                token_info=row[12],
+                pricing=row[13],
+                api_details=row[14],
+                training=row[15],
+                licensing=row[16],
+                safety=row[17],
+                availability=row[18],
+                technical_specs=row[19],
+                release_date=row[20],
+                training_data_cutoff=row[21],
+                deprecation_date=row[22],
+                metadata=row[23],
+                last_scraped_at=row[24],
+                is_active=row[25],
+                is_deprecated=row[26],
+                created_at=row[27],
+                updated_at=row[28],
+                max_images=row[29],
+                formats=row[30],
+            )
+
+
+class AsyncQuerier:
+    def __init__(self, conn: sqlalchemy.ext.asyncio.AsyncConnection):
+        self._conn = conn
+
+    async def check_model_has_capability(
+        self, *, model_id: uuid.UUID, capability_type: Any
+    ) -> Optional[bool]:
+        row = (
+            await self._conn.execute(
+                sqlalchemy.text(CHECK_MODEL_HAS_CAPABILITY),
+                {"p1": model_id, "p2": capability_type},
+            )
+        ).first()
+        if row is None:
+            return None
+        return row[0]
+
+    async def get_capability_config(self, *, slug: str) -> Optional[Optional[Any]]:
+        row = (
+            await self._conn.execute(
+                sqlalchemy.text(GET_CAPABILITY_CONFIG), {"p1": slug}
+            )
+        ).first()
+        if row is None:
+            return None
+        return row[0]
+
+    async def get_model_capabilities(self, *, slug: str) -> Optional[Any]:
+        row = (
+            await self._conn.execute(
+                sqlalchemy.text(GET_MODEL_CAPABILITIES), {"p1": slug}
+            )
+        ).first()
+        if row is None:
+            return None
+        return row[0]
+
+    async def get_model_capability_types(
+        self, *, model_id: uuid.UUID
+    ) -> AsyncIterator[str]:
+        result = await self._conn.stream(
+            sqlalchemy.text(GET_MODEL_CAPABILITY_TYPES), {"p1": model_id}
+        )
+        async for row in result:
+            yield row[0]
+
+    async def get_models_with_capability(
+        self, *, capability_type: Any, provider_id: uuid.UUID
+    ) -> AsyncIterator[models.LlmModel]:
+        result = await self._conn.stream(
+            sqlalchemy.text(GET_MODELS_WITH_CAPABILITY),
+            {"p1": capability_type, "p2": provider_id},
+        )
+        async for row in result:
+            yield models.LlmModel(
+                id=row[0],
+                provider_id=row[1],
+                model_id=row[2],
+                slug=row[3],
+                name=row[4],
+                display_name=row[5],
+                description=row[6],
+                version=row[7],
+                capabilities=row[8],
+                classification=row[9],
+                architecture=row[10],
+                performance=row[11],
+                token_info=row[12],
+                pricing=row[13],
+                api_details=row[14],
+                training=row[15],
+                licensing=row[16],
+                safety=row[17],
+                availability=row[18],
+                technical_specs=row[19],
+                release_date=row[20],
+                training_data_cutoff=row[21],
+                deprecation_date=row[22],
+                metadata=row[23],
+                last_scraped_at=row[24],
+                is_active=row[25],
+                is_deprecated=row[26],
+                created_at=row[27],
+                updated_at=row[28],
+            )
+
+    async def get_models_with_structured_response(
+        self, *, provider_id: uuid.UUID
+    ) -> AsyncIterator[GetModelsWithStructuredResponseRow]:
+        result = await self._conn.stream(
+            sqlalchemy.text(GET_MODELS_WITH_STRUCTURED_RESPONSE), {"p1": provider_id}
+        )
+        async for row in result:
+            yield GetModelsWithStructuredResponseRow(
+                id=row[0],
+                provider_id=row[1],
+                model_id=row[2],
+                slug=row[3],
+                name=row[4],
+                display_name=row[5],
+                description=row[6],
+                version=row[7],
+                capabilities=row[8],
+                classification=row[9],
+                architecture=row[10],
+                performance=row[11],
+                token_info=row[12],
+                pricing=row[13],
+                api_details=row[14],
+                training=row[15],
+                licensing=row[16],
+                safety=row[17],
+                availability=row[18],
+                technical_specs=row[19],
+                release_date=row[20],
+                training_data_cutoff=row[21],
+                deprecation_date=row[22],
+                metadata=row[23],
+                last_scraped_at=row[24],
+                is_active=row[25],
+                is_deprecated=row[26],
+                created_at=row[27],
+                updated_at=row[28],
+                max_schema_depth=row[29],
+                requires_json_mode=row[30],
+            )
+
+    async def get_models_with_vision(
+        self, *, supports_ocr: bool
+    ) -> AsyncIterator[GetModelsWithVisionRow]:
+        result = await self._conn.stream(
+            sqlalchemy.text(GET_MODELS_WITH_VISION), {"p1": supports_ocr}
+        )
+        async for row in result:
+            yield GetModelsWithVisionRow(
+                id=row[0],
+                provider_id=row[1],
+                model_id=row[2],
+                slug=row[3],
+                name=row[4],
+                display_name=row[5],
+                description=row[6],
+                version=row[7],
+                capabilities=row[8],
+                classification=row[9],
+                architecture=row[10],
+                performance=row[11],
+                token_info=row[12],
+                pricing=row[13],
+                api_details=row[14],
+                training=row[15],
+                licensing=row[16],
+                safety=row[17],
+                availability=row[18],
+                technical_specs=row[19],
+                release_date=row[20],
+                training_data_cutoff=row[21],
+                deprecation_date=row[22],
+                metadata=row[23],
+                last_scraped_at=row[24],
+                is_active=row[25],
+                is_deprecated=row[26],
+                created_at=row[27],
+                updated_at=row[28],
+                max_images=row[29],
+                formats=row[30],
+            )