Update agent.py

agent.py

@@ -33,14 +33,15 @@ def build_gemini_llm():
     return ChatGoogleGenerativeAI(model = "gemini-3.7-flash", temperature = 0, max_output_tokens = 1025, include_thoughts=True)
 
 
-# @tool
-# def search_web(query:str) -> str:
-#         """
-#         Searches the web for the answer to a given question or topic.
-#         Args:
-#             query (str): the question or topic to search for.
-#         """
-#         return DuckDuckGoSearchRun()
+@tool
+def search_web(query:str) -> str:
+    """
+    Searches the web for the answer to a given question or topic.
+    Args:
+        query (str): the question or topic to search for.
+    """
+    print(f"Query to ddsg: {query}")
+    return DuckDuckGoSearchRun()
 
 @tool
 def extract_text_from_image(img_path: str) -> str:
@@ -172,6 +173,8 @@ def wikipedia_search(query: str):
 
         if not search_results:
             return f"No Wikipedia results found for {query}. Consider another query or try a web search."
+        print("Result from wikipedia:")
+        print(search_results)
         return "\n\n---\n\n".join(
             f"Title: {doc.metadata.get('title', 'Unknown')}\n"
             f"Content: {doc.page_content}"
@@ -181,10 +184,8 @@ def wikipedia_search(query: str):
         print(f"Wikipedia search failed for {query}: {e}")
         return f"Wikipedia search failed for {query} due to {e}."
 
-search_web = DuckDuckGoSearchRun()
 model = build_gemini_llm()
 tools = [
-    extract_text_from_image,
     search_web,
     wikipedia_search
 ]
@@ -193,9 +194,8 @@ model_with_tools = model.bind_tools(tools)
 def assistant(state: AgentState):
     response = model_with_tools.invoke(state["messages"])
     print("--- Thinking Section ---")
-    print(response.content[0].get("thinking"))
+    print(response.content[0].get("thinking").strip())
     print("--- End of Thinking ---")
-    print(response)
     return {
         "messages": [response],
         "file_path": state["file_path"],