from langchain_openai import ChatOpenAI
from langchain_openai import OpenAIEmbeddings

import os
from dotenv import load_dotenv
load_dotenv()

# os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
os.environ["OPENAI_API_KEY"] = "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"
model_name = os.getenv("MODEL_NAME")
mini_model_name = os.getenv("MINI_MODEL_NAME")
embedding_model = os.getenv("EMBEDDING_MODEL")

llm = ChatOpenAI(model=model_name)
llm2 = ChatOpenAI(model=mini_model_name)

embeddings = OpenAIEmbeddings(model=embedding_model)