File size: 582 Bytes
5f0df75 ddcd2fe 5f0df75 d24af38 ddcd2fe 5f0df75 |
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 |
from conversation import Conversation_RAG
from vector_index import *
from pinecone import Pinecone
pinecone_api_key = os.environ.get("PINECONE_API_KEY")
class ModelSetup:
def __init__(self, model_name):
self.model_name = model_name
def setup(self):
conv_rag = Conversation_RAG(self.model_name)
print(f"pc api key: {pinecone_api_key}")
pc = Pinecone(
api_key=pinecone_api_key,
)
self.vectordb = conv_rag.get_vectordb(pc)
self.pipeline = conv_rag.create_model()
return "Model Setup Complete" |