service_provider_config.py

from llama_index import OpenAIEmbedding
from llama_index.llms import OpenAI

from constants import ChatbotVersion


def get_service_provider_config(model_name: str = ChatbotVersion.CHATGPT_35.value):
    llm = OpenAI(model=model_name)
    embed_model = OpenAIEmbedding()
    return llm, embed_model