flock-core 0.5.0b54__py3-none-any.whl → 0.5.0b56__py3-none-any.whl

flock/logging/telemetry_exporter/duckdb_exporter.py

@@ -87,13 +87,13 @@ class DuckDBSpanExporter(TelemetryExporter):
         try:
             with duckdb.connect(str(self.db_path)) as conn:
                 # Delete spans older than TTL
-                result = conn.execute(
-                    """
+                # Note: DuckDB doesn't support ? placeholders inside INTERVAL expressions
+                # Safe: ttl_days is guaranteed to be an int, no injection risk
+                query = f"""
                     DELETE FROM spans
-                    WHERE created_at < CURRENT_TIMESTAMP - INTERVAL ? DAYS
-                """,
-                    (self.ttl_days,),
-                )
+                    WHERE created_at < CURRENT_TIMESTAMP - INTERVAL '{self.ttl_days} DAYS'
+                """  # nosec B608
+                result = conn.execute(query)
 
                 deleted_count = result.fetchall()[0][0] if result else 0
 

flock/store.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 
 from asyncio import Lock
 from collections import defaultdict
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, TypeVar
 
 from flock.registry import type_registry
 
@@ -17,6 +17,8 @@ if TYPE_CHECKING:
 
     from flock.artifacts import Artifact
 
+T = TypeVar("T")
+
 
 class BlackboardStore:
     async def publish(self, artifact: Artifact) -> None:
@@ -31,6 +33,21 @@ class BlackboardStore:
     async def list_by_type(self, type_name: str) -> builtins.list[Artifact]:
         raise NotImplementedError
 
+    async def get_by_type(self, artifact_type: type[T]) -> builtins.list[T]:
+        """Get artifacts by Pydantic type, returning data already cast.
+
+        Args:
+            artifact_type: The Pydantic model class (e.g., BugAnalysis)
+
+        Returns:
+            List of data objects of the specified type (not Artifact wrappers)
+
+        Example:
+            bug_analyses = await store.get_by_type(BugAnalysis)
+            # Returns list[BugAnalysis] directly, no .data access needed
+        """
+        raise NotImplementedError
+
 
 class InMemoryBlackboardStore(BlackboardStore):
     """Simple in-memory implementation suitable for local dev and tests."""
@@ -58,6 +75,22 @@ class InMemoryBlackboardStore(BlackboardStore):
             canonical = type_registry.resolve_name(type_name)
             return list(self._by_type.get(canonical, []))
 
+    async def get_by_type(self, artifact_type: type[T]) -> builtins.list[T]:
+        """Get artifacts by Pydantic type, returning data already cast.
+
+        Args:
+            artifact_type: The Pydantic model class (e.g., BugAnalysis)
+
+        Returns:
+            List of data objects of the specified type (not Artifact wrappers)
+        """
+        async with self._lock:
+            # Get canonical name from the type
+            canonical = type_registry.resolve_name(artifact_type.__name__)
+            artifacts = self._by_type.get(canonical, [])
+            # Reconstruct Pydantic models from payload dictionaries
+            return [artifact_type(**artifact.payload) for artifact in artifacts]  # type: ignore
+
     async def extend(self, artifacts: Iterable[Artifact]) -> None:  # pragma: no cover - helper
         for artifact in artifacts:
             await self.publish(artifact)