remove unused chroma
Browse files
climateqa/engine/vectorstore.py
CHANGED
|
@@ -4,7 +4,6 @@
|
|
| 4 |
import os
|
| 5 |
from pinecone import Pinecone
|
| 6 |
from langchain_community.vectorstores import Pinecone as PineconeVectorstore
|
| 7 |
-
from langchain_chroma import Chroma
|
| 8 |
|
| 9 |
# LOAD ENVIRONMENT VARIABLES
|
| 10 |
try:
|
|
@@ -14,9 +13,6 @@ except:
|
|
| 14 |
pass
|
| 15 |
|
| 16 |
|
| 17 |
-
def get_chroma_vectorstore(embedding_function, persist_directory="/home/dora/climate-question-answering/data/vectorstore"):
|
| 18 |
-
vectorstore = Chroma(persist_directory=persist_directory, embedding_function=embedding_function)
|
| 19 |
-
return vectorstore
|
| 20 |
|
| 21 |
|
| 22 |
def get_pinecone_vectorstore(embeddings,text_key = "content", index_name = os.getenv("PINECONE_API_INDEX")):
|
|
|
|
| 4 |
import os
|
| 5 |
from pinecone import Pinecone
|
| 6 |
from langchain_community.vectorstores import Pinecone as PineconeVectorstore
|
|
|
|
| 7 |
|
| 8 |
# LOAD ENVIRONMENT VARIABLES
|
| 9 |
try:
|
|
|
|
| 13 |
pass
|
| 14 |
|
| 15 |
|
|
|
|
|
|
|
|
|
|
| 16 |
|
| 17 |
|
| 18 |
def get_pinecone_vectorstore(embeddings,text_key = "content", index_name = os.getenv("PINECONE_API_INDEX")):
|