This commit is contained in:
jyong
2024-11-05 16:30:49 +08:00
parent 2434b97f84
commit 766d88b29e
5 changed files with 342 additions and 1 deletions

View File

@ -1,4 +1,5 @@
from base64 import b64decode
import json
import tempfile
from typing import Any, Union
@ -41,7 +42,9 @@ class FileExtractorTool(BuiltinTool):
embedding_model_instance=None,
)
chunks = character_splitter.split_documents(documents)
return self.create_json_message(json.dumps([chunk.page_content for chunk in chunks]))
content = "\n".join([chunk.page_content for chunk in chunks])
return self.create_text_message(content)
else:
raise ToolParameterValidationError("Please provide either file")