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)")
|