Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
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-
|
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-
|
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 |
|