tcsenpai/agenticSeek
1
0
Fork 0
mirror of https://github.com/tcsenpai/agenticSeek.git synced 2025-06-04 18:20:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: better fileFinder read

Browse Source
This commit is contained in:
martin legrand 2025-05-02 14:23:00 +02:00
parent ed76c8415b
commit f17dc0550b
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
sources/tools/fileFinder.py
View File

@ -38,12 +38,13 @@ class FileFinder(Tools):
Returns:
str: The content of the file in markdown format
"""
mime_type, _ = mimetypes.guess_type(file_path)
if mime_type:
if mime_type.startswith(('image/', 'video/', 'audio/')):
return "can't read file type: image, video, or audio files are not supported."
content_raw = self.read_file(file_path)
language = ["python", "bash", "javascript", "html", "css", "json"]
if "text" in file_type:
content = content_raw
elif any(lang in file_type for lang in language):
content = content_raw
elif "pdf" in file_type:
from pypdf import PdfReader
reader = PdfReader(file_path)
@ -51,7 +52,7 @@ class FileFinder(Tools):
elif "binary" in file_type:
content = content_raw.decode('utf-8', errors='replace')
else:
content = f"Can't read file: ![{file_type}]({file_path})"
content = content_raw
return content
def get_file_info(self, file_path: str) -> str:
@ -178,7 +179,7 @@ if __name__ == "__main__":
tool = FileFinder()
result = tool.execute(["""
action=read
name=cover_letter_martin_legrand.pdf
name=tools.py
"""], False)
print("Execution result:")
print(result)