Update agent.py

agent.py

@@ -1,7 +1,6 @@
 import os
 from dotenv import load_dotenv
 from tavily import TavilyClient
-from openai import OpenAI
 from cerebras.cloud.sdk import Cerebras
 
 load_dotenv()
@@ -15,7 +14,6 @@ class BasicAgent:
         assert self.hf_token and self.tavily_key
 
         self.tavily = TavilyClient(api_key=self.tavily_key)
-        # self.llm_client = OpenAI(base_url="https://router.huggingface.co/v1", api_key=self.hf_token)
         self.llm_client = Cerebras(api_key=os.environ.get("CEREBRAS_API_KEY"))
         self.model = os.getenv("LLM_MODEL", SAMPLE_MODEL)