Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -159,8 +159,8 @@ def get_model_client(service, api_key):
|
|
159 |
return OpenAIChatCompletionClient(model="gpt-4o-2024-08-06", api_key=api_key)
|
160 |
elif service == "Anthropic-claude-3-sonnet-20240229":
|
161 |
return AnthropicChatCompletionClient(model="claude-3-sonnet-20240229", api_key=api_key)
|
162 |
-
elif service == "Google-gemini-
|
163 |
-
return OpenAIChatCompletionClient(model="gemini-
|
164 |
elif service == "Ollama-llama3.2":
|
165 |
return OllamaChatCompletionClient(model="llama3.2")
|
166 |
elif service == "Azure AI Foundry":
|
@@ -1233,11 +1233,11 @@ with gr.Blocks(
|
|
1233 |
"Azure AI Foundry",
|
1234 |
"OpenAI-gpt-4o-2024-08-06",
|
1235 |
"Anthropic-claude-3-sonnet-20240229",
|
1236 |
-
"Google-gemini-
|
1237 |
"Ollama-llama3.2",
|
1238 |
],
|
1239 |
label="Model",
|
1240 |
-
value="Google-gemini-
|
1241 |
)
|
1242 |
api_key = gr.Textbox(label="Model Provider API Key", type="password", placeholder="Not required for Ollama or Azure AI Foundry (use GITHUB_TOKEN env var)")
|
1243 |
serpapi_key = gr.Textbox(label="SerpApi Key (For Research Agent)", type="password", placeholder="Enter your SerpApi key (optional)")
|
|
|
159 |
return OpenAIChatCompletionClient(model="gpt-4o-2024-08-06", api_key=api_key)
|
160 |
elif service == "Anthropic-claude-3-sonnet-20240229":
|
161 |
return AnthropicChatCompletionClient(model="claude-3-sonnet-20240229", api_key=api_key)
|
162 |
+
elif service == "Google-gemini-2.0-flash":
|
163 |
+
return OpenAIChatCompletionClient(model="gemini-2.0-flash", api_key=api_key)
|
164 |
elif service == "Ollama-llama3.2":
|
165 |
return OllamaChatCompletionClient(model="llama3.2")
|
166 |
elif service == "Azure AI Foundry":
|
|
|
1233 |
"Azure AI Foundry",
|
1234 |
"OpenAI-gpt-4o-2024-08-06",
|
1235 |
"Anthropic-claude-3-sonnet-20240229",
|
1236 |
+
"Google-gemini-2.0-flash",
|
1237 |
"Ollama-llama3.2",
|
1238 |
],
|
1239 |
label="Model",
|
1240 |
+
value="Google-gemini-2.0-flash"
|
1241 |
)
|
1242 |
api_key = gr.Textbox(label="Model Provider API Key", type="password", placeholder="Not required for Ollama or Azure AI Foundry (use GITHUB_TOKEN env var)")
|
1243 |
serpapi_key = gr.Textbox(label="SerpApi Key (For Research Agent)", type="password", placeholder="Enter your SerpApi key (optional)")
|