Update app.py

app.py

@@ -21,9 +21,9 @@ class SearchAgentUI:
 
    
     @st.cache_resource
-    def get_agent(_self): 
+    def get_agent(_self,api_keys): 
         index_path = INDEX_PATH
-        return agent.Agent(index_path)    
+        return agent.Agent(index_path,api_keys)    
 
     def download_index_from_gdrive(self) -> bool:
         try:
@@ -257,7 +257,7 @@ class SearchAgentUI:
             return
         
         import agent
-        agent = self.get_agent()
+        agent = self.get_agent(api_keys)
 
         # Provider selection in sidebar
         with st.sidebar: