_utils/langchain_utils/embeddings.py

import os

from pydantic import Secret
from setup.easy_imports import OpenAIEmbeddings
from setup.tokens import openai_api_key


class EmbeddingClass:
    def open_ai(self):
        return OpenAIEmbeddings(api_key=Secret(openai_api_key))  # type: ignore


embedding = EmbeddingClass()