amfs-adapter-postgres 0.2.1__tar.gz → 0.2.2__tar.gz

{amfs_adapter_postgres-0.2.1 → amfs_adapter_postgres-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: amfs-adapter-postgres
-Version: 0.2.1
+Version: 0.2.2
 Summary: AMFS Postgres adapter with back-propagation triggers
 License-Expression: Apache-2.0
 Requires-Python: >=3.11

{amfs_adapter_postgres-0.2.1 → amfs_adapter_postgres-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "amfs-adapter-postgres"
-version = "0.2.1"
+version = "0.2.2"
 description = "AMFS Postgres adapter with back-propagation triggers"
 requires-python = ">=3.11"
 license = "Apache-2.0"

{amfs_adapter_postgres-0.2.1 → amfs_adapter_postgres-0.2.2}/src/amfs_postgres/adapter.py

@@ -60,6 +60,7 @@ from amfs_core.models import (
     QueryEvent,
     SearchQuery,
     SemanticQuery,
+    SessionMetadata,
     Tag,
     ToolCall,
     TraceEntry,
@@ -2595,7 +2596,18 @@ class PostgresAdapter(AdapterABC):
                         trace.session_duration_ms,
                         json.dumps([e.model_dump(mode="json") for e in trace.error_events]),
                         json.dumps(trace.state_diff.model_dump(mode="json")) if trace.state_diff else None,
-                        json.dumps(trace.session_metadata) if trace.session_metadata else "{}",
+                        # Dumped like every other model field on this row. The
+                        # field is typed, so anything arriving over
+                        # ``POST /api/v1/traces`` is a ``SessionMetadata`` by the
+                        # time it reaches here and ``json.dumps`` cannot encode
+                        # it — the trace was lost with a 500. A plain dict still
+                        # gets through ``model_copy(update=...)``, which does not
+                        # validate, so both shapes are accepted.
+                        json.dumps(
+                            trace.session_metadata.model_dump(mode="json")
+                            if isinstance(trace.session_metadata, SessionMetadata)
+                            else trace.session_metadata
+                        ) if trace.session_metadata else "{}",
                         trace.created_at,
                         trace.task_input,
                         trace.response_text,