| @@ -137,12 +137,12 @@ class LLMNode(BaseNode[LLMNodeData]): | |||
| query = None | |||
| if self.node_data.memory: | |||
| query = self.node_data.memory.query_prompt_template | |||
| if not query and ( | |||
| query_variable := self.graph_runtime_state.variable_pool.get( | |||
| (SYSTEM_VARIABLE_NODE_ID, SystemVariableKey.QUERY) | |||
| ) | |||
| ): | |||
| query = query_variable.text | |||
| if not query and ( | |||
| query_variable := self.graph_runtime_state.variable_pool.get( | |||
| (SYSTEM_VARIABLE_NODE_ID, SystemVariableKey.QUERY) | |||
| ) | |||
| ): | |||
| query = query_variable.text | |||
| prompt_messages, stop = self._fetch_prompt_messages( | |||
| user_query=query, | |||