fix: user query be ignored if query_prompt_template is an empty string (#11103)

laipz8200 · web-flow · commit 20c091a5e7ee · 2024-11-26T12:47:59.000+08:00
diff --git a/api/core/workflow/nodes/llm/node.py b/api/core/workflow/nodes/llm/node.py
@@ -137,7 +137,7 @@ def _run(self) -> NodeRunResult | Generator[NodeEvent | InNodeEvent, None, None]
             query = None
             if self.node_data.memory:
                 query = self.node_data.memory.query_prompt_template
-            if query is None and (
+            if not query and (
                 query_variable := self.graph_runtime_state.variable_pool.get(
                     (SYSTEM_VARIABLE_NODE_ID, SystemVariableKey.QUERY)
                 )

Original file line number	Diff line number	Diff line change
`@@ -137,7 +137,7 @@ def _run(self) -> NodeRunResult \| Generator[NodeEvent \| InNodeEvent, None, None]`
`137`	`137`	`query = None`
`138`	`138`	`if self.node_data.memory:`
`139`	`139`	`query = self.node_data.memory.query_prompt_template`
`140`		`- if query is None and (`
	`140`	`+ if not query and (`
`141`	`141`	`query_variable := self.graph_runtime_state.variable_pool.get(`
`142`	`142`	`(SYSTEM_VARIABLE_NODE_ID, SystemVariableKey.QUERY)`
`143`	`143`	`)`