feat: ops trace add llm model (#7306)

2024-09-04 10:39:00 +08:00 · 2024-09-04 10:39:00 +08:00 · fee4d3f6ca
commit fee4d3f6ca
parent 161cc0cda9
3 changed files with 11 additions and 3 deletions
--- a/api/core/ops/langfuse_trace/langfuse_trace.py
+++ b/api/core/ops/langfuse_trace/langfuse_trace.py
@ -204,6 +204,7 @@ class LangFuseDataTrace(BaseTraceInstance):
                node_generation_data = LangfuseGeneration(
                    name="llm",
                    trace_id=trace_id,
+                    model=process_data.get("model_name"),
                    parent_observation_id=node_execution_id,
                    start_time=created_at,
                    end_time=finished_at,