|
|
|
@@ -159,6 +159,16 @@ class LangFuseDataTrace(BaseTraceInstance): |
|
|
|
"status": status, |
|
|
|
} |
|
|
|
) |
|
|
|
process_data = json.loads(node_execution.process_data) if node_execution.process_data else {} |
|
|
|
model_provider = process_data.get("model_provider", None) |
|
|
|
model_name = process_data.get("model_name", None) |
|
|
|
if model_provider is not None and model_name is not None: |
|
|
|
metadata.update( |
|
|
|
{ |
|
|
|
"model_provider": model_provider, |
|
|
|
"model_name": model_name, |
|
|
|
} |
|
|
|
) |
|
|
|
|
|
|
|
# add span |
|
|
|
if trace_info.message_id: |
|
|
|
@@ -191,7 +201,6 @@ class LangFuseDataTrace(BaseTraceInstance): |
|
|
|
|
|
|
|
self.add_span(langfuse_span_data=span_data) |
|
|
|
|
|
|
|
process_data = json.loads(node_execution.process_data) if node_execution.process_data else {} |
|
|
|
if process_data and process_data.get("model_mode") == "chat": |
|
|
|
total_token = metadata.get("total_tokens", 0) |
|
|
|
# add generation |