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

[add] bedrock claude support

Browse Source
This commit is contained in:
Mark
2026-02-03 12:05:39 +08:00
parent e2f047d035
commit 3f42ea2c61
4 changed files with 224 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
api/app/services/draft_run_service.py
View File

@@ -413,9 +413,11 @@ class DraftRunService:
# 6. 处理多模态文件
processed_files = None
if files:
multimodal_service = MultimodalService(self.db)
# 获取 provider 信息
provider = api_key_config.get("provider", "openai")
multimodal_service = MultimodalService(self.db, provider=provider)
processed_files = await multimodal_service.process_files(files)
logger.info(f"处理了 {len(processed_files)} 个文件")
logger.info(f"处理了 {len(processed_files)} 个文件，provider={provider}")
# 7. 知识库检索
context = None
@@ -659,9 +661,11 @@ class DraftRunService:
# 6. 处理多模态文件
processed_files = None
if files:
multimodal_service = MultimodalService(self.db)
# 获取 provider 信息
provider = api_key_config.get("provider", "openai")
multimodal_service = MultimodalService(self.db, provider=provider)
processed_files = await multimodal_service.process_files(files)
logger.info(f"处理了 {len(processed_files)} 个文件")
logger.info(f"处理了 {len(processed_files)} 个文件，provider={provider}")
# 7. 知识库检索
context = None