github/owl
1
0
Fork 0
mirror of https://github.com/camel-ai/owl.git synced 2025-12-26 10:07:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix chunkr use bug

Browse Source
This commit is contained in:
Jialong Wu 2025-03-13 15:16:28 +08:00
parent 24d0661faf
commit c45bbbef02
2 changed files with 4 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
owl/utils/document_toolkit.py
View File

@ -144,12 +144,11 @@ class DocumentProcessingToolkit(BaseToolkit):
return True, extracted_text
try:
result = asyncio.run(self._extract_content_with_chunkr(document_path))
raise ValueError("Chunkr is not available.")
return True, result
except Exception as e:
logger.warning(
f"Error occurred while using chunkr to process document: {e}"
f"Error occurred while using Chunkr to process document: {e}"
)
if document_path.endswith(".pdf"):
# try using pypdf to extract text from pdf
@ -226,7 +225,7 @@ class DocumentProcessingToolkit(BaseToolkit):
if result.status == "Failed":
logger.error(
f"Error while processing document {document_path}: {result.message}"
f"Error while processing document {document_path}: {result.message} using Chunkr."
)
return f"Error while processing document: {result.message}"

9
owl/utils/gaia.py
View File

@ -191,15 +191,10 @@ class GAIABenchmark(BaseBenchmark):
except Exception as e:
logger.warning(e)
# raise FileNotFoundError(f"{self.save_to} does not exist.")
datas = [data for data in datas if not self._check_task_completed(data["task_id"])]
logger.info(f"Number of tasks to be processed: {len(datas)}")
# Process tasks
for task in tqdm(datas, desc="Running"):
if self._check_task_completed(task["task_id"]):
logger.info(
f"The following task is already completed:\n task id: {task['task_id']}, question: {task['Question']}"
)
continue
if_prepared_task, info = self._prepare_task(task)
if not if_prepared_task:
_result_info = {