Correct prompts

agents/agent.py

@@ -1,20 +1,7 @@
 """LangGraph Agent for GAIA Benchmark"""
-import os
-from dotenv import load_dotenv
-
-from langgraph.graph import START, StateGraph, MessagesState
-from langgraph.prebuilt import tools_condition, ToolNode
-from langchain_core.messages import SystemMessage, HumanMessage
-from langchain_groq import ChatGroq
-from tools.CodeToolkit import execute_code_multilang
-from tools.DocumentsToolkit import (
-    save_and_read_file,
-    download_file_from_url,
-    extract_text_from_image,
-    analyze_csv_file,
-    analyze_excel_file,
-    analyze_word_file,
-    analyze_pdf_file
+from tools.SearchToolkit import wiki_search, web_search, arxiv_search, vector_store
+from tools.MathsToolkit import (
+    multiply, add, subtract, divide, modulus, power, square_root
 )
 from tools.ImagesToolkit import (
     analyze_image,
@@ -23,16 +10,30 @@ from tools.ImagesToolkit import (
     generate_simple_image,
     combine_images
 )
-from tools.MathsToolkit import (
-    multiply, add, subtract, divide, modulus, power, square_root
+from tools.DocumentsToolkit import (
+    save_and_read_file,
+    download_file_from_url,
+    extract_text_from_image,
+    analyze_csv_file,
+    analyze_excel_file,
+    analyze_word_file,
+    analyze_pdf_file
 )
-from tools.SearchToolkit import wiki_search, web_search, arxiv_search, vector_store
-
+from tools.CodeToolkit import execute_code_multilang
+from langchain_groq import ChatGroq
+from langchain_core.messages import SystemMessage, HumanMessage
+from langgraph.prebuilt import tools_condition, ToolNode
+from langgraph.graph import START, StateGraph, MessagesState
+import os
+from dotenv import load_dotenv
 # Load environment variables
 load_dotenv()
 
+prompt_path = os.path.join(os.path.dirname(__file__), "../prompts")
+
+
 # Load system prompt
-with open("system_prompt.txt", "r", encoding="utf-8") as f:
+with open(os.path.join(prompt_path, "system_prompt.txt"), "r", encoding="utf-8") as f:
     system_prompt = f.read()
 sys_msg = SystemMessage(content=system_prompt)
 

app.py

@@ -6,6 +6,7 @@ import pandas as pd
 from langchain_core.messages import HumanMessage
 from agents.agent import build_graph
 
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"