Update agent.py

agent.py

@@ -1,6 +1,8 @@
 from typing import Any, List, Optional
 
-from smolagents import CodeAgent, InferenceClientModel, HfApiModel, TransformersModel
+from smolagents import CodeAgent, InferenceClientModel, HfApiModel, TransformersModel, LiteLLMModel
+
+
 
 import os
 hf_token = os.getenv("HF_TOKEN")
@@ -31,9 +33,12 @@ class Agent:
     ):
         logger.info("Initializing Agent")
  
-        model_id = "meta-llama/Llama-3.3-70B-Instruct"
+        #model_id = "meta-llama/Llama-3.3-70B-Instruct"
         #self.model = HfApiModel(model_id=model_id, token=hf_token)
-        self.model = TransformersModel(model_id="HuggingFaceTB/SmolLM-135M-Instruct")
+        self.model = LiteLLMModel(
+            model_id="groq/deepseek-r1-distill-llama-70b",
+            temperature=0.2,
+            api_key=os.environ["GROQ_API_KEY"])
         
         self.tools = tools
         self.imports = [