Prechádzať zdrojové kódy

fix: avoid llm node result var not init issue while do retry. (#14286)

Jacky Wu 1 mesiac pred
rodič
commit
ed5596a8f4
1 zmenil súbory, kde vykonal 3 pridanie a 3 odobranie
  1. 3 3
      api/core/workflow/nodes/llm/node.py

+ 3 - 3
api/core/workflow/nodes/llm/node.py

@@ -94,6 +94,9 @@ class LLMNode(BaseNode[LLMNodeData]):
     def _run(self) -> Generator[NodeEvent | InNodeEvent, None, None]:
         node_inputs: Optional[dict[str, Any]] = None
         process_data = None
+        result_text = ""
+        usage = LLMUsage.empty_usage()
+        finish_reason = None
 
         try:
             # init messages template
@@ -178,9 +181,6 @@ class LLMNode(BaseNode[LLMNodeData]):
                 stop=stop,
             )
 
-            result_text = ""
-            usage = LLMUsage.empty_usage()
-            finish_reason = None
             for event in generator:
                 if isinstance(event, RunStreamChunkEvent):
                     yield event