File size: 616 Bytes
de84956 |
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 |
# 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"))
|