kdevoe commited on
Commit
fc64087
·
verified ·
1 Parent(s): 6faa281

Adding chroma db pull

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -14,6 +14,11 @@ from langchain.vectorstores import Chroma
14
  openai_api_key = os.getenv("openapi_token")
15
  embedding = OpenAIEmbeddings(openai_api_key=openai_api_key)
16
 
 
 
 
 
 
17
  # Streamed response emulator
18
  def response_generator():
19
  response = random.choice(
 
14
  openai_api_key = os.getenv("openapi_token")
15
  embedding = OpenAIEmbeddings(openai_api_key=openai_api_key)
16
 
17
+ # Setup vector database
18
+ persist_directory = './chroma_db'
19
+ vectordb = Chroma(persist_directory=persist_directory, embedding_function=embedding)
20
+
21
+
22
  # Streamed response emulator
23
  def response_generator():
24
  response = random.choice(