import os
from smolagents import (OpenAIServerModel)
from dotenv import load_dotenv
load_dotenv(override=True)

OPENAI_MODEL = os.getenv("OPENAI_MODEL")
OPENAI_BASE = os.getenv("OPENAI_API_BASE")
OPENAI_KEY = os.getenv("OPENAI_API_KEY")


def create_react_model():
    return OpenAIServerModel(model_id=OPENAI_MODEL, api_base=OPENAI_BASE, api_key=OPENAI_KEY)

def create_vision_model():
    return OpenAIServerModel(model_id=OPENAI_MODEL, api_base=OPENAI_BASE, api_key=OPENAI_KEY)

react_model = create_react_model()