Add read_file tool

basic_agent.py

@@ -18,7 +18,7 @@ from llama_index.core.agent.workflow import (
 )
 
 from multimodality_tools import get_image_qa_tool, get_transcription_tool, \
-    get_excel_analysis_tool, get_excel_tool, get_csv_analysis_tool, get_csv_tool, _get_file
+    get_excel_analysis_tool, get_excel_tool, get_csv_analysis_tool, get_csv_tool, _get_file, get_read_file_tool
 
 class BasicAgent:
     def __init__(self, ollama=False, langfuse=False):
@@ -53,7 +53,7 @@ class BasicAgent:
                 "whether the element to be put in the list is a number or a string."
             ),
             llm=llm,
-            tools=[],
+            tools=[get_read_file_tool()],
             can_handoff_to=["WikiAgent", "WebAgent", "StatsAgent", "AudioAgent", "ImageAgent"],
         )
 

multimodality_tools.py

@@ -163,3 +163,18 @@ def remove_think(output: str) -> str:
     if output:
         return re.sub("<think>.*</think>", "", output).strip()
     return output
+
+def read_txt_or_py_file(file_id: str) -> str:
+    """Read a python or txt file as plain text and return its content."""
+    try:
+        bytes_io = _get_file(file_id)
+    except:
+        return "Error: Invalid file. This file is either not a .py/.txt file or the id does not exist." 
+    bytes_io.seek(0)
+    return bytes_io.read().decode()
+
+def get_read_file_tool():
+    return FunctionTool.from_defaults(
+        fn=read_txt_or_py_file,
+        description="Read a python or txt file as plain text and return its content."
+    )