src/final_assignment_template/models.py

from smolagents import LiteLLMModel
import os


planner_model = LiteLLMModel(
    # model_id="openrouter/openai/o4-mini-high",
    model_id="openrouter/deepseek/deepseek-r1:free",
    api_base="https://openrouter.ai/api/v1",
    api_key=os.getenv("OPENROUTER_API_KEY")
)


openrouter_qwenCoder_model = LiteLLMModel(
    model_id="openrouter/qwen/qwen-2.5-coder-32b-instruct:free",
    api_base="https://openrouter.ai/api/v1",
    api_key=os.getenv("OPENROUTER_API_KEY")
)

# nvidia/llama-3.3-nemotron-super-49b-v1:free
# microsoft/mai-ds-r1:free
modelLiteLLm = LiteLLMModel(
    model_id="openrouter/microsoft/mai-ds-r1:free",
     api_base="https://openrouter.ai/api/v1",
    api_key=os.getenv("OPENROUTER_API_KEY")
)


videoLiteLLm = LiteLLMModel(
    model_id="openrouter/google/gemini-2.0-flash-exp:free",
     api_base="https://openrouter.ai/api/v1",
    api_key=os.getenv("OPENROUTER_API_KEY")
)


imageLiteLLm = LiteLLMModel(
    model_id="openrouter/meta-llama/llama-4-maverick:free",
     api_base="https://openrouter.ai/api/v1",
    api_key=os.getenv("OPENROUTER_API_KEY")
)


summarizeModle = LiteLLMModel(
    model_id="openrouter/meta-llama/llama-4-maverick:free",
     api_base="https://openrouter.ai/api/v1",
    api_key=os.getenv("OPENROUTER_API_KEY")
)