Final_Assignment_Template

Sleeping

Abbasid commited on Jul 22

Commit

bd6822e

verified ·

1 Parent(s): d6fbb7e

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -65,7 +65,7 @@ class MyAgent:
         print(f"Initializing agent with provider: {provider}")
         self.llm = self._build_llm(provider)
-        self.vision_llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", temperature=0)
         self.retriever = self._get_retriever()
     def _get_retriever(self):

         print(f"Initializing agent with provider: {provider}")
         self.llm = self._build_llm(provider)
+        self.vision_llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash", temperature=0)
         self.retriever = self._get_retriever()
     def _get_retriever(self):