Update agent.py

agent.py

@@ -0,0 +1,30 @@
+from smolagents import DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, LiteLLMModel
+from langchain_community.document_loaders import TextLoader, PyPDFLoader, Docx2txtLoader, CSVLoader
+from langchain_text_splitters import RecursiveCharacterTextSplitter
+from langchain_community.retrievers import BM25Retriever
+from agent_tools import *
+
+
+# Initialize tools
+search_tool = DuckDuckGoSearchTool(max_results = 5)
+wiki_tool = WikipediaSearchTool()
+rag_tool = RagTool()
+calculator_tools = [addition_tool, subtraction_tool, multiplication_tool, division_tool, exponent_tool, modulus_tool]
+
+# Agent Definitions
+calculator_agent = CodeAgent(
+    model = model,
+    tools = calculator_tools,
+    name = "arithmetic_calculator_agent",
+    description = "Performs basic arithmetic calculations between two numbers.",
+    max_steps = 10,
+    verbosity_level=3
+)
+
+main_agent = CodeAgent(
+    tools=[search_tool, wiki_tool, rag_tool],
+    model=model,
+    managed_agents = [calculator_agent],
+    max_steps=10,
+    verbosity_level=3,
+)