Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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
|