Spaces:
Running
Running
add open router
Browse files- app.py +5 -3
- app_openrouter.py +22 -0
- pyproject.toml +1 -1
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -30,6 +30,7 @@ from app_qwen_coder import demo as demo_qwen_coder
|
|
| 30 |
from app_nvidia_coder import demo as demo_nvidia_coder
|
| 31 |
from app_openai import demo as demo_openai
|
| 32 |
from app_sambanova_coder import demo as demo_sambanova_coder
|
|
|
|
| 33 |
from utils import get_app
|
| 34 |
import gradio as gr
|
| 35 |
|
|
@@ -60,6 +61,7 @@ gr.load(
|
|
| 60 |
|
| 61 |
# Create mapping of providers to their demos
|
| 62 |
PROVIDERS = {
|
|
|
|
| 63 |
"OpenAI Coder": demo_openai_coder,
|
| 64 |
"Sambanova Coder": demo_sambanova_coder,
|
| 65 |
"Gemini Coder": demo_gemini_coder,
|
|
@@ -101,13 +103,13 @@ with demo:
|
|
| 101 |
|
| 102 |
provider_dropdown = gr.Dropdown(
|
| 103 |
choices=list(PROVIDERS.keys()),
|
| 104 |
-
value="
|
| 105 |
label="Select code snippet"
|
| 106 |
)
|
| 107 |
code_display = gr.Code(
|
| 108 |
label="Provider Code Snippet",
|
| 109 |
language="python",
|
| 110 |
-
value=PROVIDER_SNIPPETS["
|
| 111 |
)
|
| 112 |
|
| 113 |
def update_code(provider):
|
|
@@ -122,7 +124,7 @@ with demo:
|
|
| 122 |
|
| 123 |
selected_demo = get_app(
|
| 124 |
models=list(PROVIDERS.keys()),
|
| 125 |
-
default_model="
|
| 126 |
src=PROVIDERS,
|
| 127 |
dropdown_label="Select Provider",
|
| 128 |
)
|
|
|
|
| 30 |
from app_nvidia_coder import demo as demo_nvidia_coder
|
| 31 |
from app_openai import demo as demo_openai
|
| 32 |
from app_sambanova_coder import demo as demo_sambanova_coder
|
| 33 |
+
from app_openrouter import demo as demo_openrouter
|
| 34 |
from utils import get_app
|
| 35 |
import gradio as gr
|
| 36 |
|
|
|
|
| 61 |
|
| 62 |
# Create mapping of providers to their demos
|
| 63 |
PROVIDERS = {
|
| 64 |
+
"OpenRouter Coder": demo_openrouter,
|
| 65 |
"OpenAI Coder": demo_openai_coder,
|
| 66 |
"Sambanova Coder": demo_sambanova_coder,
|
| 67 |
"Gemini Coder": demo_gemini_coder,
|
|
|
|
| 103 |
|
| 104 |
provider_dropdown = gr.Dropdown(
|
| 105 |
choices=list(PROVIDERS.keys()),
|
| 106 |
+
value="OpenRouter Coder",
|
| 107 |
label="Select code snippet"
|
| 108 |
)
|
| 109 |
code_display = gr.Code(
|
| 110 |
label="Provider Code Snippet",
|
| 111 |
language="python",
|
| 112 |
+
value=PROVIDER_SNIPPETS["OpenRouter Coder"]
|
| 113 |
)
|
| 114 |
|
| 115 |
def update_code(provider):
|
|
|
|
| 124 |
|
| 125 |
selected_demo = get_app(
|
| 126 |
models=list(PROVIDERS.keys()),
|
| 127 |
+
default_model="OpenRouter Coder",
|
| 128 |
src=PROVIDERS,
|
| 129 |
dropdown_label="Select Provider",
|
| 130 |
)
|
app_openrouter.py
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import ai_gradio
|
| 2 |
+
|
| 3 |
+
from utils_ai_gradio import get_app
|
| 4 |
+
|
| 5 |
+
# Get the OpenAI models but keep their full names for loading
|
| 6 |
+
OPENROUTER_MODELS_FULL = [k for k in ai_gradio.registry.keys() if k.startswith("openrouter:")]
|
| 7 |
+
|
| 8 |
+
# Create display names without the prefix
|
| 9 |
+
OPENROUTER_MODELS_DISPLAY = [k.replace("openrouter:", "") for k in OPENROUTER_MODELS_FULL]
|
| 10 |
+
|
| 11 |
+
# Create and launch the interface using get_app utility
|
| 12 |
+
demo = get_app(
|
| 13 |
+
models=OPENROUTER_MODELS_FULL, # Use the full names with prefix
|
| 14 |
+
default_model=OPENROUTER_MODELS_FULL[1],
|
| 15 |
+
dropdown_label="Select OpenRouter Model",
|
| 16 |
+
choices=OPENROUTER_MODELS_DISPLAY, # Display names without prefix
|
| 17 |
+
fill_height=True,
|
| 18 |
+
coder=True,
|
| 19 |
+
)
|
| 20 |
+
|
| 21 |
+
if __name__ == "__main__":
|
| 22 |
+
demo.launch()
|
pyproject.toml
CHANGED
|
@@ -38,7 +38,7 @@ dependencies = [
|
|
| 38 |
"langchain>=0.3.14",
|
| 39 |
"chromadb>=0.5.23",
|
| 40 |
"openai>=1.55.0",
|
| 41 |
-
"ai-gradio[crewai,deepseek,gemini,groq,hyperbolic,openai,smolagents,transformers, langchain, mistral,minimax,nvidia, qwen]>=0.2.
|
| 42 |
]
|
| 43 |
|
| 44 |
[tool.uv.sources]
|
|
|
|
| 38 |
"langchain>=0.3.14",
|
| 39 |
"chromadb>=0.5.23",
|
| 40 |
"openai>=1.55.0",
|
| 41 |
+
"ai-gradio[crewai,deepseek,gemini,groq,hyperbolic,openai,smolagents,transformers, langchain, mistral,minimax,nvidia, qwen]>=0.2.44",
|
| 42 |
]
|
| 43 |
|
| 44 |
[tool.uv.sources]
|
requirements.txt
CHANGED
|
@@ -2,7 +2,7 @@
|
|
| 2 |
# uv pip compile pyproject.toml -o requirements.txt
|
| 3 |
accelerate==1.2.1
|
| 4 |
# via ai-gradio
|
| 5 |
-
ai-gradio==0.2.
|
| 6 |
# via anychat (pyproject.toml)
|
| 7 |
aiofiles==23.2.1
|
| 8 |
# via gradio
|
|
|
|
| 2 |
# uv pip compile pyproject.toml -o requirements.txt
|
| 3 |
accelerate==1.2.1
|
| 4 |
# via ai-gradio
|
| 5 |
+
ai-gradio==0.2.44
|
| 6 |
# via anychat (pyproject.toml)
|
| 7 |
aiofiles==23.2.1
|
| 8 |
# via gradio
|