opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-15 01:38:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix llm memory

Browse Source
This commit is contained in:
takatost 2024-03-21 15:36:25 +08:00
parent d71eae8f93
commit 72818e946d
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/workflow/nodes/llm/llm_node.py
View File

@ -73,6 +73,8 @@ class LLMNode(BaseNode):
# fetch prompt messages
prompt_messages, stop = self._fetch_prompt_messages(
node_data=node_data,
query=variable_pool.get_variable_value(['sys', SystemVariable.QUERY.value])
if node_data.memory else None,
inputs=inputs,
files=files,
context=context,
@ -391,6 +393,7 @@ class LLMNode(BaseNode):
return memory
def _fetch_prompt_messages(self, node_data: LLMNodeData,
query: Optional[str],
inputs: dict[str, str],
files: list[FileVar],
context: Optional[str],
@ -400,6 +403,7 @@ class LLMNode(BaseNode):
"""
Fetch prompt messages
:param node_data: node data
:param query: query
:param inputs: inputs
:param files: files
:param context: context
@ -411,7 +415,7 @@ class LLMNode(BaseNode):
prompt_messages = prompt_transform.get_prompt(
prompt_template=node_data.prompt_template,
inputs=inputs,
query='',
query=query if query else '',
files=files,
context=context,
memory_config=node_data.memory,