feat: ops trace add llm model (#7306)

2024-09-04 10:39:00 +08:00
parent 161cc0cda9
commit fee4d3f6ca
3 changed files with 11 additions and 3 deletions
--- a/api/core/workflow/nodes/llm/llm_node.py
+++ b/api/core/workflow/nodes/llm/llm_node.py
@@ -109,7 +109,9 @@ class LLMNode(BaseNode):
                'prompts': PromptMessageUtil.prompt_messages_to_prompt_for_saving(
                    model_mode=model_config.mode,
                    prompt_messages=prompt_messages
-                )
+                ),
+                'model_provider': model_config.provider,
+                'model_name': model_config.model,      
            }

            # handle invoke result