Verify exact production traces and record rollout

apps/ai-service/adapters/postgres.py

@@ -18,6 +18,8 @@ class RetrievalTrace:
     reason: str
     resolved_drug_id: str | None
     citations: tuple[dict[str, Any], ...]
+    correlation_id: str | None = None
+    otel_trace_id: str | None = None
     created_at: datetime | None = None
 
 
@@ -55,6 +57,8 @@ class PostgresTraceRepository:
         reason: str,
         resolved_drug_id: str | None,
         citations: tuple[dict[str, Any], ...],
+        correlation_id: str | None = None,
+        otel_trace_id: str | None = None,
     ) -> str:
         import psycopg
 
@@ -64,12 +68,14 @@ class PostgresTraceRepository:
                 """
                 INSERT INTO rag_retrieval_trace (
                     trace_id, query_text, subject_scope, query_intent,
-                    decision, reason, resolved_drug_id, citations
-                ) VALUES (%s, %s, %s, %s, %s, %s, %s, %s::jsonb)
+                    decision, reason, resolved_drug_id, citations,
+                    correlation_id, otel_trace_id
+                ) VALUES (%s, %s, %s, %s, %s, %s, %s, %s::jsonb, %s, %s)
                 """,
                 (
                     trace_id, query, subject_scope, intent, decision, reason,
                     resolved_drug_id, json.dumps(citations, ensure_ascii=False),
+                    correlation_id, otel_trace_id,
                 ),
             )
         return trace_id
@@ -81,7 +87,8 @@ class PostgresTraceRepository:
             row = connection.execute(
                 """
                 SELECT trace_id::text, query_text, subject_scope, query_intent,
-                       decision, reason, resolved_drug_id, citations, created_at
+                       decision, reason, resolved_drug_id, citations,
+                       correlation_id, otel_trace_id, created_at
                 FROM rag_retrieval_trace WHERE trace_id = %s
                 """,
                 (trace_id,),
@@ -91,7 +98,8 @@ class PostgresTraceRepository:
         return RetrievalTrace(
             trace_id=row[0], query=row[1], subject_scope=row[2], intent=row[3],
             decision=row[4], reason=row[5], resolved_drug_id=row[6],
-            citations=tuple(row[7]), created_at=row[8],
+            citations=tuple(row[7]), correlation_id=row[8], otel_trace_id=row[9],
+            created_at=row[10],
         )