apunkt/PageIndex
1
0
Fork 0
mirror of https://github.com/VectifyAI/PageIndex.git synced 2026-07-24 21:41:04 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix: agent tool get_document_structure now checks document existence

Browse source
This commit is contained in:
Ray 2026-07-24 16:06:22 +08:00
parent b43b56ed04
commit 40693acc85
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
pageindex/backend/local.py
View file

@ -372,9 +372,11 @@ class LocalBackend:
def get_document_structure(doc_name: str) -> str:
"""Get document tree structure (without text). Pass the document's doc_name (a doc_id also works)."""
try:
structure = storage.get_document_structure(col_name, _resolve(doc_name))
structure = backend.get_document_structure(col_name, _resolve(doc_name))
except _DocResolveError as e:
return str(e)
except DocumentNotFoundError:
return json.dumps({"error": f"Document {doc_name!r} not found."})
return json.dumps(remove_fields(structure, fields=["text"]), ensure_ascii=False)
@function_tool