Update agent.py

agent.py

@@ -42,7 +42,6 @@ from io import StringIO
 from transformers import BertTokenizer, BertModel
 import torch
 import torch.nn.functional as F
-from langchain.agents import initialize_agent, AgentType
 from langchain_community.chat_models import ChatOpenAI
 from langchain_community.tools import Tool
 import time
@@ -50,7 +49,6 @@ from huggingface_hub import InferenceClient
 from langchain_community.llms import HuggingFaceHub
 from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
-from langchain.agents import initialize_agent, Tool, AgentType
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 from huggingface_hub import login
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, BitsAndBytesConfig
@@ -59,57 +57,6 @@ from langchain_huggingface import HuggingFaceEndpoint
 
 load_dotenv()
 
-@tool
-def multiply(a: int, b: int) -> int:
-    """Multiply two numbers.
-    Args:
-        a: first int
-        b: second int
-    """
-    return a * b
-
-@tool
-def add(a: int, b: int) -> int:
-    """Add two numbers.
-    
-    Args:
-        a: first int
-        b: second int
-    """
-    return a + b
-
-@tool
-def subtract(a: int, b: int) -> int:
-    """Subtract two numbers.
-    
-    Args:
-        a: first int
-        b: second int
-    """
-    return a - b
-
-@tool
-def divide(a: int, b: int) -> int:
-    """Divide two numbers.
-    
-    Args:
-        a: first int
-        b: second int
-    """
-    if b == 0:
-        raise ValueError("Cannot divide by zero.")
-    return a / b
-
-@tool
-def modulus(a: int, b: int) -> int:
-    """Get the modulus of two numbers.
-    
-    Args:
-        a: first int
-        b: second int
-    """
-    return a % b
-
 
 @tool
 def calculator(inputs: dict):
@@ -132,6 +79,7 @@ def calculator(inputs: dict):
         return a % b
     else:
         return "Unknown operation"
+        
 
 @tool
 def wiki_search(query: str) -> str: