update

app.py

@@ -17,6 +17,21 @@ GOOGLE_API_KEY=os.environ['GOOGLE_API_KEY']
 
 loader = WebBaseLoader("https://www.google.com/finance/?hl=en")
 docs = loader.load()
+
+gemini_embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001")
+
+# Save to disk
+vectorstore = Chroma.from_documents(
+                     documents=docs,                 # Data
+                     embedding=gemini_embeddings,    # Embedding model
+                     persist_directory="./chroma_db" # Directory to save data
+                     )
+
+vectorstore_disk = Chroma(
+                        persist_directory="./chroma_db",       # Directory of db
+                        embedding_function=gemini_embeddings   # Embedding model
+                   )
+retriever = vectorstore_disk.as_retriever(search_kwargs={"k": 1})
 # If there is no environment variable set for the API key, you can pass the API
 # key to the parameter `google_api_key` of the `ChatGoogleGenerativeAI` function:
 # `google_api_key="key"`.