Update agent.py

agent.py

@@ -53,6 +53,7 @@ from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
 from langchain.agents import initialize_agent, Tool, AgentType
 from transformers import pipeline
+from smolagents import LiteLLMModel
 
 load_dotenv()
 
@@ -461,16 +462,12 @@ tools = [wiki_tool, calc_tool, file_tool, web_tool, arvix_tool, youtube_tool, vi
 # Get the Hugging Face API token from the environment variable
 hf_token = os.getenv("HF_TOKEN")
 
-# Load the Qwen2.5-Coder-32B-Instruct model using HuggingFaceHub
-from langchain_huggingface import HuggingFaceEndpoint
 
-# Initialize the HuggingFaceEndpoint with the desired model and parameters
-llm = HuggingFaceEndpoint(
-    repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-    #task="conversational", 
-    huggingfacehub_api_token=hf_token,
-    temperature=0.7,               
-    max_new_tokens=1024           
+# Initialize the desired model and parameters
+llm = LiteLLMModel(
+    model_id="gemini/gemini-2.0-flash", 
+    api_key=os.getenv("GEMINI_API_KEY"),
+    max_tokens=8192
 )