Spaces:
Running
Running
add new sonoma model
Browse files
app.py
CHANGED
@@ -1630,11 +1630,16 @@ AVAILABLE_MODELS = [
|
|
1630 |
"name": "Sonoma Dusk Alpha",
|
1631 |
"id": "openrouter/sonoma-dusk-alpha",
|
1632 |
"description": "OpenRouter Sonoma Dusk Alpha model with vision capabilities"
|
|
|
|
|
|
|
|
|
|
|
1633 |
}
|
1634 |
]
|
1635 |
|
1636 |
# Default model selection
|
1637 |
-
DEFAULT_MODEL_NAME = "
|
1638 |
DEFAULT_MODEL = None
|
1639 |
for _m in AVAILABLE_MODELS:
|
1640 |
if _m.get("name") == DEFAULT_MODEL_NAME:
|
@@ -1766,6 +1771,12 @@ def get_inference_client(model_id, provider="auto"):
|
|
1766 |
api_key=os.getenv("OPENROUTER_API_KEY"),
|
1767 |
base_url="https://openrouter.ai/api/v1",
|
1768 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
1769 |
elif model_id == "step-3":
|
1770 |
# Use StepFun API client for Step-3 model
|
1771 |
return OpenAI(
|
|
|
1630 |
"name": "Sonoma Dusk Alpha",
|
1631 |
"id": "openrouter/sonoma-dusk-alpha",
|
1632 |
"description": "OpenRouter Sonoma Dusk Alpha model with vision capabilities"
|
1633 |
+
},
|
1634 |
+
{
|
1635 |
+
"name": "Sonoma Sky Alpha",
|
1636 |
+
"id": "openrouter/sonoma-sky-alpha",
|
1637 |
+
"description": "OpenRouter Sonoma Sky Alpha model with vision capabilities"
|
1638 |
}
|
1639 |
]
|
1640 |
|
1641 |
# Default model selection
|
1642 |
+
DEFAULT_MODEL_NAME = "Sonoma Sky Alpha"
|
1643 |
DEFAULT_MODEL = None
|
1644 |
for _m in AVAILABLE_MODELS:
|
1645 |
if _m.get("name") == DEFAULT_MODEL_NAME:
|
|
|
1771 |
api_key=os.getenv("OPENROUTER_API_KEY"),
|
1772 |
base_url="https://openrouter.ai/api/v1",
|
1773 |
)
|
1774 |
+
elif model_id == "openrouter/sonoma-sky-alpha":
|
1775 |
+
# Use OpenRouter client for Sonoma Sky Alpha model
|
1776 |
+
return OpenAI(
|
1777 |
+
api_key=os.getenv("OPENROUTER_API_KEY"),
|
1778 |
+
base_url="https://openrouter.ai/api/v1",
|
1779 |
+
)
|
1780 |
elif model_id == "step-3":
|
1781 |
# Use StepFun API client for Step-3 model
|
1782 |
return OpenAI(
|