vrr/agent-zero
2
0
Fork 0
mirror of https://github.com/agent0ai/agent-zero.git synced 2026-04-29 12:10:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

document query QA fix

Browse source
This commit is contained in:
linkliti 2026-01-10 13:17:27 +00:00
parent 3d79fae266
commit 376092c63c
2 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
python/helpers/document_query.py
View file

@ -359,6 +359,7 @@ class DocumentQueryHelper:
self.agent = agent
self.store = DocumentQueryStore.get(agent)
self.progress_callback = progress_callback or (lambda x: None)
self.store_lock = asyncio.Lock()
async def document_qa(
self, document_uris: List[str], questions: Sequence[str]
@ -527,9 +528,10 @@ class DocumentQueryHelper:
if add_to_db:
self.progress_callback(f"Indexing document")
await self.agent.handle_intervention()
success, ids = await self.store.add_document(
document_content, document_uri_norm
)
async with self.store_lock:
success, ids = await self.store.add_document(
document_content, document_uri_norm
)
if not success:
self.progress_callback(f"Failed to index document")
raise ValueError(