fix(llm_node): Ignore user query when memory is disabled. (#11106)

This commit is contained in:
-LAN- 2024-11-26 13:07:32 +08:00 committed by GitHub
parent 20c091a5e7
commit cbb4e95928
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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,