# embeddings.py – OpenAI Version (text-embedding-3-small)

import os
from langchain_openai import OpenAIEmbeddings

EMBED_MODEL = "text-embedding-3-small"

def get_embeddings():
    api_key = os.environ.get("OPENAI_API_KEY")
    if not api_key:
        raise RuntimeError(
            "OPENAI_API_KEY fehlt. Bitte als Secret im HuggingFace Space setzen."
        )

    print(f">>> Lade OpenAI Embedding Model: {EMBED_MODEL}")
    emb = OpenAIEmbeddings(
        model=EMBED_MODEL,
        api_key=api_key,
    )
    return emb

if __name__ == "__main__":
    e = get_embeddings()
    print(e.embed_query("Test"))