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