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

Merge pull request #696 from SuanmoSuanyangTechnology/fix/workflow-message-process

Browse Source 
fix(workflow): Fix LLM node, resolve abnormal field reading issue in message caching functionality
This commit is contained in:
Mark
2026-03-26 16:24:08 +08:00
committed by GitHub
parent 5b940e5f1a 6223b80cc4
commit 8dbb067b83
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/app/core/workflow/nodes/base_node.py
View File

@@ -639,8 +639,8 @@ class BaseNode(ABC):
return content return content
elif isinstance(content, FileObject): elif isinstance(content, FileObject):
if content.content_cache.get(f"{provider}_{ModelInfo.is_omni}"): if content.content_cache.get(f"{provider}_{api_config.is_omni}"):
return content.content_cache[f"{provider}_{ModelInfo.is_omni}"] return content.content_cache[f"{provider}_{api_config.is_omni}"]
with get_db_read() as db: with get_db_read() as db:
multimodal_service = MultimodalService(db, api_config=api_config) multimodal_service = MultimodalService(db, api_config=api_config)
file_obj = FileInput( file_obj = FileInput(
@@ -656,7 +656,7 @@ class BaseNode(ABC):
) )
content.set_content(file_obj.get_content()) content.set_content(file_obj.get_content())
if message: if message:
content.content_cache[f"{provider}_{ModelInfo.is_omni}"] = message content.content_cache[f"{provider}_{api_config.is_omni}"] = message
return message return message
return None return None
raise TypeError(f'Unexpected input value type - {type(content)}') raise TypeError(f'Unexpected input value type - {type(content)}')