tasks.py

"""
Define invoke tasks.

invoke list
invoke build

invoke --help build

"""
from invoke import task

NAME_DEF = "final assignment tasks"

@task(
    default=True,
)
def gemini(c):
    """Run python litellm_model.py gemini params."""
    c.run("python litellm_model.py gemini-2.5-flash-preview-04-17  https://generativelanguage.googleapis.com/v1beta %GEMINI_API_KEY% gemini")

@task
def gemini1(c):
    """Run python litellm_model.py gemini-as-openai params."""
    c.run("python litellm_model.py gemini-2.5-flash-preview-04-17  https://generativelanguage.googleapis.com/v1beta %GEMINI_API_KEY% openai")

@task
def gemini2(c):
    """Run python litellm_model.py gemini-via-proxy params."""
    # c.run("python litellm_model.py gemini-2.5-flash-preview-04-17  https://generativelanguage.googleapis.com/v1beta %GEMINI_API_KEY%")
    c.run("python litellm_model.py gemini-2.5-flash-preview-04-17  https://api-proxy.me/gemini/v1beta %GEMINI_API_KEY% openai")

@task
def xai(c):
    """Run python litellm_model.py grok params."""
    c.run("python litellm_model.py grok-3-beta https://api.x.ai/v1 %XAI_API_KEY% xai")

@task
def grok1(c):
    """Run python litellm_model.py grok params."""
    c.run("python litellm_model.py grok-3-beta https://api.x.ai/v1 %XAI_API_KEY% openai")

@task
def siliconflow(c):
    """Run python litellm_model.py (siliconflow setup)."""
    c.run("python litellm_model.py ")


@task
def agent(c):
    """Run python basic_agent.py."""
    c.run("python basic_agent.py ")