adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(workflow): fix env timeout configuration and LLM node message role mismatch

Browse Source
This commit is contained in:
Eternity
2026-01-14 16:46:09 +08:00
parent 567624c323
commit b712325399
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/app/core/config.py
View File

@@ -169,7 +169,7 @@ class Settings:
SYSTEM_VERSION: str = os.getenv("SYSTEM_VERSION", "v0.2.0") SYSTEM_VERSION: str = os.getenv("SYSTEM_VERSION", "v0.2.0")
# workflow config # workflow config
WORKFLOW_NODE_TIMEOUT: int = os.getenv("WORKFLOW_NODE_TIMEOUT", 600) WORKFLOW_NODE_TIMEOUT: int = int(os.getenv("WORKFLOW_NODE_TIMEOUT", 600))
def get_memory_output_path(self, filename: str = "") -> str: def get_memory_output_path(self, filename: str = "") -> str:
""" """

2
api/app/core/workflow/nodes/llm/node.py
View File

@@ -102,7 +102,7 @@ class LLMNode(BaseNode):
elif role in ["user", "human"]: elif role in ["user", "human"]:
messages.append({"role": "user", "content": content}) messages.append({"role": "user", "content": content})
elif role in ["ai", "assistant"]: elif role in ["ai", "assistant"]:
messages.append({"role": "user", "content": content}) messages.append({"role": "assistant", "content": content})
else: else:
logger.warning(f"未知的消息角色: {role}，默认使用 user") logger.warning(f"未知的消息角色: {role}，默认使用 user")
messages.append({"role": "user", "content": content}) messages.append({"role": "user", "content": content})