VishnuRamDebyez commited on
Commit
57eaaa5
·
verified ·
1 Parent(s): 256426b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -27,12 +27,19 @@ text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=20
27
  splits = text_splitter.split_documents(doc)
28
 
29
  # Create a vector store and retriever
30
- vectorstore = Chroma.from_documents(documents=splits, embedding=gemini_embeddings)
 
 
 
 
 
31
  retriever = vectorstore.as_retriever()
32
 
33
 
34
 
35
 
 
 
36
  from langchain.prompts.chat import ChatPromptTemplate, MessagesPlaceholder
37
 
38
  # Define system prompt, ensuring that `context` is specified as an input
 
27
  splits = text_splitter.split_documents(doc)
28
 
29
  # Create a vector store and retriever
30
+
31
+ from langchain_chroma import Chroma
32
+ from langchain_chroma import chromadb
33
+
34
+ # Use Chroma in-memory mode for temporary storage
35
+ vectorstore = Chroma.from_documents(documents=splits, embedding=gemini_embeddings, persist_directory=None)
36
  retriever = vectorstore.as_retriever()
37
 
38
 
39
 
40
 
41
+
42
+
43
  from langchain.prompts.chat import ChatPromptTemplate, MessagesPlaceholder
44
 
45
  # Define system prompt, ensuring that `context` is specified as an input