Spaces:
Build error
Build error
File size: 1,575 Bytes
b9b64be 24dcddf 34cd91e 24dcddf 34cd91e 24dcddf | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | # from langchain_mongodb import MongoDBAtlasVectorSearch
# from pymongo import MongoClient
from .llm import embeddings
import os
from langchain_pinecone import PineconeVectorStore
# client = MongoClient(os.getenv("MONGO_CONNECTION_STR"))
# DB_NAME = os.getenv("DB_NAME")
# COLLECTION_NAME = os.getenv("COLLECTION_NAME")
# ATLAS_VECTOR_CHATBOT_INDEX_NAME = os.getenv("ATLAS_VECTOR_CHATBOT_INDEX_NAME")
# ATLAS_VECTOR_TUTOR_INDEX_NAME = os.getenv("ATLAS_VECTOR_TUTOR_INDEX_NAME")
# MONGODB_COLLECTION_CHATBOT = client[DB_NAME][ATLAS_VECTOR_CHATBOT_INDEX_NAME]
# MONGODB_COLLECTION_TUTOR = client[DB_NAME][ATLAS_VECTOR_TUTOR_INDEX_NAME]
# vector_store_chatbot = MongoDBAtlasVectorSearch(
# collection=MONGODB_COLLECTION_CHATBOT,
# embedding=embeddings,
# index_name=ATLAS_VECTOR_CHATBOT_INDEX_NAME,
# relevance_score_fn="cosine",
# )
# vector_store_tutor = MongoDBAtlasVectorSearch(
# collection=MONGODB_COLLECTION_TUTOR,
# embedding=embeddings,
# index_name=ATLAS_VECTOR_TUTOR_INDEX_NAME,
# relevance_score_fn="cosine",
# )
API_PINCONE_KEY = os.getenv("PINECONE_API_KEY")
index_tutor = "tutor-vector-store"
index_chatbot = "chatbot-vector-store"
vector_store_tutor = PineconeVectorStore(
index_name=index_tutor, embedding=embeddings, pinecone_api_key=API_PINCONE_KEY
)
vector_store_chatbot = PineconeVectorStore(
index_name=index_chatbot, embedding=embeddings, pinecone_api_key=API_PINCONE_KEY
)
vector_store_fresher = PineconeVectorStore(
index_name="fresher-handbook", embedding=embeddings, pinecone_api_key=API_PINCONE_KEY
) |