LLM/chandra
1
0
Fork 0
mirror of https://github.com/datalab-to/chandra.git synced 2026-01-20 05:50:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Track errors properly

Browse Source
This commit is contained in:
Vik Paruchuri
2025-10-23 16:55:16 -04:00
parent 5acfd8dc6a
commit 528b58c16f
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
chandra/model/__init__.py
View File

@@ -48,6 +48,7 @@ class InferenceManager:
page_box=[0, 0, input_item.image.width, input_item.image.height], page_box=[0, 0, input_item.image.width, input_item.image.height],
token_count=result.token_count, token_count=result.token_count,
images=extract_images(result.raw, chunks, input_item.image), images=extract_images(result.raw, chunks, input_item.image),
error=result.error,
) )
) )
return output return output

1
chandra/model/schema.py
View File

@@ -27,3 +27,4 @@ class BatchOutputItem:
page_box: List[int] page_box: List[int]
token_count: int token_count: int
images: dict images: dict
error: bool