Jaward commited on
Commit
26c3269
·
verified ·
1 Parent(s): c4def78

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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-1.5-flash":
163
- return OpenAIChatCompletionClient(model="gemini-1.5-flash", api_key=api_key)
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-1.5-flash",
1237
  "Ollama-llama3.2",
1238
  ],
1239
  label="Model",
1240
- value="Google-gemini-1.5-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)")
 
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)")