AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-04-23 06:09:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: ops trace add llm model (#7306)

Browse Source
This commit is contained in:
Joe 2024-09-04 10:39:00 +08:00 committed by GitHub
parent 161cc0cda9
commit fee4d3f6ca
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/core/ops/langfuse_trace/langfuse_trace.py
View File

@ -204,6 +204,7 @@ class LangFuseDataTrace(BaseTraceInstance):
node_generation_data = LangfuseGeneration( node_generation_data = LangfuseGeneration(
name="llm", name="llm",
trace_id=trace_id, trace_id=trace_id,
model=process_data.get("model_name"),
parent_observation_id=node_execution_id, parent_observation_id=node_execution_id,
start_time=created_at, start_time=created_at,
end_time=finished_at, end_time=finished_at,

9
api/core/ops/langsmith_trace/langsmith_trace.py
View File

@ -139,8 +139,7 @@ class LangSmithDataTrace(BaseTraceInstance):
json.loads(node_execution.execution_metadata) if node_execution.execution_metadata else {} json.loads(node_execution.execution_metadata) if node_execution.execution_metadata else {}
) )
node_total_tokens = execution_metadata.get("total_tokens", 0) node_total_tokens = execution_metadata.get("total_tokens", 0)
metadata = execution_metadata.copy()
metadata = json.loads(node_execution.execution_metadata) if node_execution.execution_metadata else {}
metadata.update( metadata.update(
{ {
"workflow_run_id": trace_info.workflow_run_id, "workflow_run_id": trace_info.workflow_run_id,
@ -156,6 +155,12 @@ class LangSmithDataTrace(BaseTraceInstance):
process_data = json.loads(node_execution.process_data) if node_execution.process_data else {} process_data = json.loads(node_execution.process_data) if node_execution.process_data else {}
if process_data and process_data.get("model_mode") == "chat": if process_data and process_data.get("model_mode") == "chat":
run_type = LangSmithRunType.llm run_type = LangSmithRunType.llm
metadata.update(
{
'ls_provider': process_data.get('model_provider', ''),
'ls_model_name': process_data.get('model_name', ''),
}
)
elif node_type == "knowledge-retrieval": elif node_type == "knowledge-retrieval":
run_type = LangSmithRunType.retriever run_type = LangSmithRunType.retriever
else: else:

4
api/core/workflow/nodes/llm/llm_node.py
View File

@ -109,7 +109,9 @@ class LLMNode(BaseNode):
'prompts': PromptMessageUtil.prompt_messages_to_prompt_for_saving( 'prompts': PromptMessageUtil.prompt_messages_to_prompt_for_saving(
model_mode=model_config.mode, model_mode=model_config.mode,
prompt_messages=prompt_messages prompt_messages=prompt_messages
) ),
'model_provider': model_config.provider,
'model_name': model_config.model,
} }
# handle invoke result # handle invoke result