Upload 3 files

agent folder added.

agent/__init__.py

@@ -0,0 +1,3 @@
+"""
+Agent module 
+"""

agent/agent.py

@@ -0,0 +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

agent/tools/search_tool.py

@@ -0,0 +1,25 @@
+from langchain_community.tools import DuckDuckGoSearchResults, DuckDuckGoSearchRun, BingSearchResults, BingSearchRun
+from langchain.tools import tool
+
+@tool(description="Use this tool to get search results for a query. Input should be a string representing the search query. Output will be a string containing the search results.")
+def search_tool(query: str) -> str:
+    print(f"Invoking search tool with query: {query}")
+    try:
+        search_tool = DuckDuckGoSearchResults(engine="duckduckgo")
+        search_results = search_tool.run(query)
+        print(f"Search tool returned results: {search_results[:100]}...")  # Print first 100 chars of results
+        return search_results
+    except Exception as e:
+        error_message = f"Error invoking search tool: {e}"
+        return error_message
+
+# search_tool = Tool(
+#     name = "get_search_results",
+#     func = get_search_results,
+#     description = "Use this tool to get search results for a query. Input should be a string representing the search query. Output will be a string containing the search results."
+# )
+
+
+if __name__ == "__main__":
+    results = search_tool("Who's the current President of France?")
+    print(results)