Spaces:
Sleeping
Sleeping
Adding chroma db pull
Browse files
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(
|