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

fix(workflow): Fix LLM node, resolve abnormal field reading issue in message caching functionality

Browse Source
This commit is contained in:
Eternity
2026-03-26 16:19:01 +08:00
parent 5b940e5f1a
commit 6223b80cc4
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
elif isinstance(content, FileObject):
if content.content_cache.get(f"{provider}_{ModelInfo.is_omni}"):
return content.content_cache[f"{provider}_{ModelInfo.is_omni}"]
if content.content_cache.get(f"{provider}_{api_config.is_omni}"):
return content.content_cache[f"{provider}_{api_config.is_omni}"]
with get_db_read() as db:
multimodal_service = MultimodalService(db, api_config=api_config)
file_obj = FileInput(
@@ -656,7 +656,7 @@ class BaseNode(ABC):
)
content.set_content(file_obj.get_content())
if message:
content.content_cache[f"{provider}_{ModelInfo.is_omni}"] = message
content.content_cache[f"{provider}_{api_config.is_omni}"] = message
return message
return None
raise TypeError(f'Unexpected input value type - {type(content)}')