services/embed_model/embed_model.py

import os
import dotenv
dotenv.load_dotenv()
from datetime import datetime
from langchain_openai import AzureOpenAIEmbeddings


def serve_embed_model():
    try:
        embed_model = AzureOpenAIEmbeddings(
            model=os.environ.get('azureai--embedmodel--name'),
            azure_endpoint=os.environ.get('azureai--embedmodel--endpoint'),
            api_key=os.environ.get('azureai--embedmodel--api-key'),
            api_version=os.environ.get('azureai--embedmodel--api--version')
        )
        # asyncio.run(embed_model.aembed_documents("Hello my name is EMA"))
        print(f"✅ [INFO] Initialized embedding model succeeded")
        return embed_model
    except Exception as E:
        print(f"❌ [ERROR] Initialized embedding model failed")


embed_model = serve_embed_model()