Makhfi_AI / workflow /vectorstore.py
Aasher's picture
Refactor workflow to integrate ChatGoogleGenerativeAI and update vectorstore embeddings
38ccc4e
raw
history blame contribute delete
585 Bytes
from pinecone import Pinecone
from langchain_pinecone import PineconeVectorStore
from langchain_google_genai import GoogleGenerativeAIEmbeddings
from core.config import get_settings
settings = get_settings()
index_name = settings.INDEX_NAME
namespace = settings.NAMESPACE
embeddings = GoogleGenerativeAIEmbeddings(
model=f"models/{settings.EMBEDDINGS_MODEL}", google_api_key=settings.GOOGLE_API_KEY
)
pc = Pinecone(api_key=settings.PINECONE_API_KEY)
index = pc.Index(index_name)
vector_store = PineconeVectorStore(
index=index, namespace=namespace, embedding=embeddings
)