Spaces:
Running
Running
File size: 1,493 Bytes
48ec86e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
"""
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 ")
|