Update agent/agent.py

agent/agent.py

@@ -1,29 +1,29 @@
-import os
-from typing import TypedDict, List, Dict, Any, Optional
-from lanchain_google_genai import ChatGoogleGenerativeAI
-from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage, AIMessage
-from langgraph.grap.message import add_messages
-from langgraph.graph import START, StateGraph
-from langgraph.prebuilt import ToolNode, tools_condition
-
-from tools.search_tool import search_tool
-
-
-# --- zBottaAgent Definition ---
-
-class zBottaAgent:
-    def __init__(self):
-        print("zBottaAgent initialized.")
-        chat = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0.7)
-        tools = [search_tool]
-        self.chat_with_tools = chat.bind_tools(tools)
-
-    def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        messages = [
-            SystemMessage(content="You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer. Your answer should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."),
-            HumanMessage(content=question)
-        ]
-        response = self.chat_with_tools.invoke(messages)
-        print(f"Agent returning response: {response}")
-        return response
+import os
+from typing import TypedDict, List, Dict, Any, Optional
+from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage, AIMessage
+from langgraph.grap.message import add_messages
+from langgraph.graph import START, StateGraph
+from langgraph.prebuilt import ToolNode, tools_condition
+
+from tools.search_tool import search_tool
+
+
+# --- zBottaAgent Definition ---
+
+class zBottaAgent:
+    def __init__(self):
+        print("zBottaAgent initialized.")
+        chat = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0.7)
+        tools = [search_tool]
+        self.chat_with_tools = chat.bind_tools(tools)
+
+    def __call__(self, question: str) -> str:
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        messages = [
+            SystemMessage(content="You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer. Your answer should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."),
+            HumanMessage(content=question)
+        ]
+        response = self.chat_with_tools.invoke(messages)
+        print(f"Agent returning response: {response}")
+        return response