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

rag tool progress and optimization

Browse source
This commit is contained in:
frdel 2025-06-17 23:05:28 +02:00
parent 602d60cc8f
commit 11f7c6009f
2 changed files with 43 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

10
python/tools/document_query.py
View file

@ -10,7 +10,15 @@ class DocumentQueryTool(Tool):
if not isinstance(document_uri, str) or not document_uri:
return Response(message="Error: no document provided", break_loop=False)
try:
helper = DocumentQueryHelper(self.agent)
progress = []
# logging callback
def progress_callback(msg):
progress.append(msg)
self.log.update(progress="\n".join(progress))
helper = DocumentQueryHelper(self.agent, progress_callback)
if not queries:
content = await helper.document_get_content(document_uri)
else: