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

Merge pull request #1002 from SuanmoSuanyangTechnology/feature/agent-tool_xjn

Browse Source 
fix(multimodal_service)
This commit is contained in:
山程漫悟
2026-04-24 19:42:38 +08:00
committed by GitHub
parent a9413f57d1 141fd94513
commit 2c14344d3f
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/app/services/multimodal_service.py
View File

@@ -388,6 +388,7 @@ class MultimodalService:
from app.models.workspace_model import Workspace as WorkspaceModel from app.models.workspace_model import Workspace as WorkspaceModel
ws = self.db.query(WorkspaceModel).filter(WorkspaceModel.id == workspace_id).first() ws = self.db.query(WorkspaceModel).filter(WorkspaceModel.id == workspace_id).first()
tenant_id = ws.tenant_id if ws else None tenant_id = ws.tenant_id if ws else None
img_result = []
for img_info in img_infos: for img_info in img_infos:
page = img_info["page"] page = img_info["page"]
index = img_info["index"] index = img_info["index"]
@@ -407,9 +408,10 @@ class MultimodalService:
file_type="image/png", file_type="image/png",
) )
_, img_content = await self._process_image(img_file, strategy_class(img_file)) _, img_content = await self._process_image(img_file, strategy_class(img_file))
result.append(img_content) img_result.append(img_content)
except Exception as img_err: except Exception as img_err:
logger.warning(f"文档图片处理失败: {img_err}") logger.warning(f"文档图片处理失败: {img_err}")
result.extend(img_result)
elif file.type == FileType.AUDIO and "audio" in self.capability: elif file.type == FileType.AUDIO and "audio" in self.capability:
is_support, content = await self._process_audio(file, strategy) is_support, content = await self._process_audio(file, strategy)
result.append(content) result.append(content)