From 8ba2ee5ce1ae7fcc4644538a5c9ff4c01ca9e6ec Mon Sep 17 00:00:00 2001
From: 3un01a <3un01a@plasticlabs.ai>
Date: Thu, 13 Nov 2025 12:00:18 -0500
Subject: [PATCH] Add trace config path + additional trace parameters

---
 src/config.py          | 1 +
 src/deriver/deriver.py | 2 ++
 src/utils/logging.py   | 8 +++++++-
 3 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/config.py b/src/config.py
index 6033fc61..787f812a 100644
--- a/src/config.py
+++ b/src/config.py
@@ -362,6 +362,7 @@ class AppSettings(HonchoSettings):
 
     COLLECT_METRICS_LOCAL: bool = False
     LOCAL_METRICS_FILE: str = "metrics.jsonl"
+    LOCAL_TRACE_FILE: str = "trace.jsonl"
 
     NAMESPACE: str = "honcho"  # Top-level namespace for all settings, can be overridden by nested-model settings
 
diff --git a/src/deriver/deriver.py b/src/deriver/deriver.py
index 5a4a6a47..bdd2f9a0 100644
--- a/src/deriver/deriver.py
+++ b/src/deriver/deriver.py
@@ -363,6 +363,8 @@ class CertaintyReasoner:
         from src.utils.logging import save_reasoning_trace
 
         save_reasoning_trace(
+            workspace_id=latest_message.workspace_name,
+            session_name=latest_message.session_name,
             provider=settings.DERIVER.PROVIDER,
             model=settings.DERIVER.MODEL,
             max_tokens=settings.DERIVER.MAX_OUTPUT_TOKENS
diff --git a/src/utils/logging.py b/src/utils/logging.py
index 3d46f208..1d917d3a 100644
--- a/src/utils/logging.py
+++ b/src/utils/logging.py
@@ -252,6 +252,8 @@ def get_route_template(request: Request) -> str:
 
 
 def save_reasoning_trace(
+    workspace_id: str,
+    session_name: str,
     provider: str,
     model: str,
     max_tokens: int,
@@ -274,6 +276,8 @@ def save_reasoning_trace(
 
     Args:
         provider: LLM provider name
+        workspace_id: Workspace ID
+        session_name: Session name
         model: Model name
         max_tokens: Max tokens setting
         peer_id: ID of peer being analyzed
@@ -295,10 +299,12 @@ def save_reasoning_trace(
     import json
     from pathlib import Path
 
-    trace_file = Path("trace.jsonl")
+    trace_file = Path(settings.LOCAL_TRACE_FILE)
 
     trace_data = {
         "timestamp": datetime.datetime.now(datetime.timezone.utc).isoformat(),
+        "workspace_id": workspace_id,
+        "session_name": session_name,
         "provider": provider,
         "model": model,
         "max_tokens": max_tokens,