Abbasid commited on
Commit
83ebef7
·
verified ·
1 Parent(s): 7502786

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -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 = ChatGroq(model_name="meta-llama/llama-4-scout-17b-16e-instruct", temperature=0)
69
  self.retriever = self._get_retriever()
70
 
71
  def _get_retriever(self):
@@ -91,7 +91,7 @@ class MyAgent:
91
  def _build_llm(self, provider: str):
92
  """Helper to build the main text-based LLM based on the chosen provider."""
93
  if provider == "google": return ChatGoogleGenerativeAI(model="gemini-1.5-pro-latest", temperature=0)
94
- elif provider == "groq": return ChatGroq(model_name="meta-llama/llama-4-scout-17b-16e-instruct", temperature=0)
95
  elif provider == "huggingface": return ChatHuggingFace(llm=HuggingFaceEndpoint(repo_id="Qwen/Qwen2.5-Coder-32B-Instruct", temperature=0))
96
  else: raise ValueError("Provider must be 'google', 'groq', or 'huggingface'")
97
 
 
65
  print(f"Initializing agent with provider: {provider}")
66
 
67
  self.llm = self._build_llm(provider)
68
+ self.vision_llm = ChatGroq(model_name="meta-llama/llama-4-maverick-17b-128e-instruct", temperature=0)
69
  self.retriever = self._get_retriever()
70
 
71
  def _get_retriever(self):
 
91
  def _build_llm(self, provider: str):
92
  """Helper to build the main text-based LLM based on the chosen provider."""
93
  if provider == "google": return ChatGoogleGenerativeAI(model="gemini-1.5-pro-latest", temperature=0)
94
+ elif provider == "groq": return ChatGroq(model_name="meta-llama/llama-4-maverick-17b-128e-instruct", temperature=0)
95
  elif provider == "huggingface": return ChatHuggingFace(llm=HuggingFaceEndpoint(repo_id="Qwen/Qwen2.5-Coder-32B-Instruct", temperature=0))
96
  else: raise ValueError("Provider must be 'google', 'groq', or 'huggingface'")
97