youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-03 17:08:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

fta

Browse Source
This commit is contained in:
jyong
2024-11-05 21:13:43 +08:00
parent a033a53a32
commit 73eb9647a2
6 changed files with 27 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/file/file_manager.py
View File

@ -1,6 +1,6 @@
import base64
from pathlib import Path
import tempfile
from pathlib import Path
from configs import dify_config
from core.file import file_repository

2
api/core/tools/provider/builtin/file_extractor/file_extractor.py
View File

@ -1,7 +1,5 @@
from typing import Any
from core.tools.errors import ToolProviderCredentialValidationError
from core.tools.provider.builtin.dalle.tools.dalle2 import DallE2Tool
from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController

7
api/core/tools/provider/builtin/file_extractor/tools/file_extractor.py
View File

@ -1,12 +1,7 @@
from base64 import b64decode
import json
import tempfile
from typing import Any, Union
from openai import OpenAI
from yarl import URL
from core.file.enums import FileType
from core.file.file_manager import download_to_target_path
from core.rag.extractor.text_extractor import TextExtractor
from core.rag.splitter.fixed_text_splitter import FixedRecursiveCharacterTextSplitter
@ -45,6 +40,6 @@ class FileExtractorTool(BuiltinTool):
content = "\n".join([chunk.page_content for chunk in chunks])
return self.create_text_message(content)
else:
raise ToolParameterValidationError("Please provide either file")