xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-11-05 23:33:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(api): preserve integer knowledge ingestion task IDs

Browse Source
This commit is contained in:
huanghuoguoguo
2026-10-29 15:47:56 +08:00
parent d2176eaa27
commit 8f2f537e34
6 changed files with 116 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit_tests/api/service/test_knowledge_service.py
+2 -2
View File
@@ -265,14 +265,14 @@ async def test_file_listing_checks_parent_knowledge_base_first():
async def test_store_and_delete_file_require_runtime_parent_and_capability():
app = _app()
runtime_kb = SimpleNamespace(
store_file=AsyncMock(return_value='task-a'),
store_file=AsyncMock(return_value=17),
delete_file=AsyncMock(),
)
app.rag_mgr.get_knowledge_base_by_uuid.return_value = runtime_kb
app.rag_mgr.get_knowledge_base_details.return_value = {'knowledge_engine': {'capabilities': ['doc_ingestion']}}
service = KnowledgeService(app)
assert await service.store_file(CONTEXT, 'kb-a', 'upload.pdf', 'author/parser') == 'task-a'
assert await service.store_file(CONTEXT, 'kb-a', 'upload.pdf', 'author/parser') == 17
runtime_kb.store_file.assert_awaited_once_with(CONTEXT, 'upload.pdf', parser_plugin_id='author/parser')
await service.delete_file(CONTEXT, 'kb-a', 'file-a')
runtime_kb.delete_file.assert_awaited_once_with(CONTEXT, 'file-a')