system

agent.py

@@ -9,19 +9,25 @@ from tools import (
 
 # ──────────────────────────────── Config ────────────────────────────────
 SYSTEM_PROMPT = """
-You are a smart AI assistant using a tool-augmented reasoning strategy. Follow this loop:
-Thought: ...
-Action: ...
-Action Input: ...
+You are a helpful AI assistant. You will answer questions by thinking step-by-step, taking actions using tools when necessary, and finishing with a final answer.
 
-You will get an Observation, and then continue reasoning.
+When you want to use a tool, respond *exactly* in the following format:
 
-Only end the loop with:
-FINAL ANSWER: [your short final answer here].
+Thought: [your reasoning]
+Action: [tool_name]
+Action Input: [input]
 
-DO NOT answer directly without using this loop at least once, unless the answer is trivially obvious.
+When you receive an observation, continue reasoning. When you are confident in your answer, write:
+
+Thought: [final reasoning]
+FINAL ANSWER: [your answer]
+
+IMPORTANT:
+- If using tools that require a `task_id`, only use the value provided.
+- Do not make up tool names.
+- Do not loop unnecessarily.
+- Provide FINAL ANSWER as soon as you are confident.
 
-When using task-based tools (audio_transcriber_tool, excel_tool, analyze_code_tool, image_tool), ONLY use the 'task_id' value.
 """
 
 TOOLS = [
@@ -34,7 +40,7 @@ TOOLS = [
 class SimpleLangGraphAgent:
     def __init__(self, model_name="gpt-4o-mini"):
         self.agent = create_react_agent(
-            model=ChatOpenAI(model_name=model_name, temperature=0),
+            model=ChatOpenAI(model_name=model_name, temperature=0.3),
             tools=TOOLS,
         )
 
@@ -49,6 +55,7 @@ class SimpleLangGraphAgent:
 
         for msg in final_state["messages"][::-1]:
             if "FINAL ANSWER:" in msg.content.upper():
+                print(f"\n\n msg.content = {msg.content} \n\n")
                 return msg.content.split("FINAL ANSWER:")[-1].strip()
 
         return "No FINAL ANSWER found."

app.py

@@ -14,38 +14,7 @@ from state import AgentState
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
-SYSTEM_PROMPT = """
-You are a helpful AI assistant using a ReAct pattern. You will answer questions by thinking step-by-step, taking actions using tools when necessary, and finishing with a final answer.
-
-When you want to use a tool, respond *exactly* in the following format:
-
-Thought: [your reasoning]
-Action: [tool_name]
-Action Input: [input]
-
-When you receive an observation, continue reasoning. When you are confident in your answer, write:
-
-Thought: [final reasoning]
-FINAL ANSWER: [your answer]
-
-Use only the available tools:
-- wikipedia_search_tool
-- arxiv_search_tool
-- audio_transcriber_tool
-- excel_tool
-- analyze_code_tool
-- image_tool
-- add_tool
-- subtract_tool
-- multiply_tool
-- divide_tool
-
-IMPORTANT:
-- If using tools that require a `task_id`, only use the value provided.
-- Do not make up tool names.
-- Do not loop unnecessarily.
-- Provide FINAL ANSWER as soon as you are confident.
-"""
+
 
 
 class BasicAgent:

tools.py

@@ -34,7 +34,7 @@ def _download_file_for_task(task_id: str, ext: str) -> str:
     try:
         resp = requests.get(url, timeout=10)
         if resp.status_code == 200 and resp.content:
-            print(f"Downloaded file from {url} to {local_path}")
+            print(f"\n Downloaded file from {url} to {local_path} \n")
             with open(local_path, "wb") as f:
                 f.write(resp.content)
             return local_path
@@ -179,6 +179,7 @@ def audio_transcriber_tool(task_id: str) -> str:
             break
 
     if not local_audio or not os.path.exists(local_audio):
+        print("Error: No audio file found (download failed).")
         return "Error: No audio file found (download failed)."
         
 
@@ -194,7 +195,7 @@ def audio_transcriber_tool(task_id: str) -> str:
                 model="whisper-1", 
                 file=audio_file,
             )
-            print("reached response")
+            # print("reached response")
         text = response.text.strip()
     except Exception as e:
         text = f"Error during transcription: {e}"
@@ -213,12 +214,13 @@ def wikipedia_search_tool(wiki_query: str) -> str:
     Purpose: When the user asks for historical, biographical, scientific, or factual information, use this tool.
 
     Input: A string describing a topic to search on Wikipedia.
-
+    Tips: If you dont get enough information, try again with a different querry
     Example usage:
     - "Who was Marie Curie?"
     - "Explain quantum entanglement"
     - "Tell me about the French Revolution"
     """
+    print("Reached Wikipedia tool, with query = ", wiki_query)
     try:
         docs = WikipediaLoader(query=wiki_query, load_max_docs=3).load()  # Reduced from 5 to 3
         
@@ -253,6 +255,7 @@ def wikipedia_search_tool(wiki_query: str) -> str:
                 break
         
         if not result.strip():
+            print("No wiki result found")
             return "No Wikipedia results found for the given query. [END_OF_SEARCH]"
         
         # Add clear end marker
@@ -279,12 +282,14 @@ def arxiv_search_tool(query: str) -> str:
     - "What are recent studies on climate change?"
     - "Search for papers on quantum computing"
     """
+    print("Reached ArXiv tool, with query = ", query)
     try:
         # Search arXiv for the top result
         search = arxiv.Search(query=query, max_results=1, sort_by=arxiv.SortCriterion.Relevance)
         result = next(search.results(), None)
 
         if not result:
+            print("No arXiv result found")
             return "No results found. [END_OF_SEARCH]"
 
         # Download PDF
@@ -330,21 +335,19 @@ def analyze_code_tool(task_id: str) -> str:
     - "Analyze the code file for bugs"
     - "Explain the functions in this script"
     """
-    print("reached analyze_code_tool")
+    print("Reached analyze_code_tool")
     code_txt = ""
     if not task_id:
         code_txt = "No code provided."
     else:
         path = _download_file_for_task(task_id, "py")
         if not path:
+            print("Error: .py file not found for this task.")
             return "Error: .py file not found for this task."
         code_txt = Path(path).read_text(encoding="utf-8", errors="ignore")
-    # else:
-    #     return "Error: neither snippet nor file provided."
 
-    # Truncate for safety
-    lines = code_txt.splitlines()[:400]
-    code_sample = "\n".join(lines)[:10_000]
+    lines = code_txt.splitlines()
+    code_sample = "\n".join(lines)
 
     prompt = [
         SystemMessage(content="You are a senior Python code reviewer."),
@@ -376,6 +379,7 @@ def add_tool(a: float, b: float) -> str:
     - "Add 3.14 and 2.86"
     - "Calculate the sum of 100 and 250"
     """
+    print("Reached add_tool")
     result = a + b
     return f"Addition result: {a} + {b} = {result}"
 
@@ -393,6 +397,7 @@ def subtract_tool(a: float, b: float) -> str:
     - "Subtract 15.5 from 40.2"
     - "Calculate 1000 minus 347"
     """
+    print("Reached subtract_tool")
     result = a - b
     return f"Subtraction result: {a} - {b} = {result}"
 
@@ -410,6 +415,7 @@ def multiply_tool(a: float, b: float) -> str:
     - "Multiply 12.5 by 4"
     - "Calculate the product of 15 and 20"
     """
+    print("Reached multiply_tool")
     result = a * b
     return f"Multiplication result: {a} × {b} = {result}"
 
@@ -427,6 +433,7 @@ def divide_tool(a: float, b: float) -> str:
     - "Divide 75 by 3"
     - "Calculate 144 divided by 12"
     """
+    print("Reached divide_tool")
     if b == 0:
         return "Division error: Cannot divide by zero"
     result = a / b