Abbasid commited on
Commit
c133122
·
verified ·
1 Parent(s): a640e6e

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -56,7 +56,7 @@ class MyAgent:
56
  This class-based approach ensures clean management of dependencies.
57
  """
58
 
59
- def __init__(self, provider: str = "google"):
60
  """
61
  Initializes the agent, setting up LLMs and the FAISS retriever.
62
  Args:
@@ -65,7 +65,7 @@ class MyAgent:
65
  print(f"Initializing agent with provider: {provider}")
66
 
67
  self.llm = self._build_llm(provider)
68
- self.vision_llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash", temperature=0)
69
  self.retriever = self._get_retriever()
70
 
71
  def _get_retriever(self):
 
56
  This class-based approach ensures clean management of dependencies.
57
  """
58
 
59
+ def __init__(self, provider: str = "groq"):
60
  """
61
  Initializes the agent, setting up LLMs and the FAISS retriever.
62
  Args:
 
65
  print(f"Initializing agent with provider: {provider}")
66
 
67
  self.llm = self._build_llm(provider)
68
+ self.vision_llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", temperature=0)
69
  self.retriever = self._get_retriever()
70
 
71
  def _get_retriever(self):