src/embeddings.py

import os
from langchain_openai import OpenAIEmbeddings
from dotenv import load_dotenv

load_dotenv()

def get_embedding_model():
    """
    Returns the Pro-Level Embedding Model.
    Using: text-embedding-3-large (3072 dimensions)
    """
    api_key = os.getenv("OPENAI_API_KEY")
    if not api_key:
        raise ValueError("❌ OPENAI_API_KEY not found in .env file!")

    model = OpenAIEmbeddings(
        model="text-embedding-3-large",
        dimensions=3072,  # Must match Qdrant config
        openai_api_key=api_key
    )
    return model