services/embed_model/embed_model.py

import os
import dotenv
from config.env_constant import EnvFilepath
dotenv.load_dotenv(EnvFilepath.ENVPATH)
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()