Add trace config path + additional trace parameters
This commit is contained in:
parent
6d65191f53
commit
8ba2ee5ce1
|
|
@ -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
|
||||
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
|
|
|
|||
Loading…
Reference in New Issue