oc-assistant/rag/embeddings.py
2024-10-29 21:57:59 -05:00

11 lines
237 B
Python

from dotenv import load_dotenv
from langchain_openai import OpenAIEmbeddings
def load_embeddins():
load_dotenv()
# model = "text-embedding-ada-002"
model = "text-embedding-3-small"
return OpenAIEmbeddings(model=model)