Add check if file exists

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_excel_analysis_tool, get_excel_tool, get_csv_analysis_tool, get_csv_tool, _get_file
 
 class BasicAgent:
     def __init__(self, ollama=False, langfuse=False):
@@ -145,7 +145,7 @@ class BasicAgent:
 
     async def __call__(self, question: str, task_id: str = None) -> str:
         file_str = ""
-        if task_id:
+        if file_exists(task_id):
             file_str = f'\nIf you need to load a file, do so by providing the id "{task_id}".'
 
         msg = f"{question}{file_str}"
@@ -193,3 +193,11 @@ class BasicAgent:
             return res
         except:
             return "Error occured. No valid agent response could be determined."
+
+def file_exists(task_id: str) -> bool:
+    try:
+        file = _get_file(task_id)
+    except:
+        return False
+    del file
+    return True