Jaward commited on
Commit
b6c4f64
·
verified ·
1 Parent(s): 13cc517

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -162,8 +162,8 @@ def get_model_client(service, api_key):
162
  return OpenAIChatCompletionClient(model="gpt-4o-2024-08-06", api_key=api_key)
163
  elif service == "Anthropic-claude-3-sonnet-20240229":
164
  return AnthropicChatCompletionClient(model="claude-3-sonnet-20240229", api_key=api_key)
165
- elif service == "Google-gemini-1.5-flash":
166
- return OpenAIChatCompletionClient(model="gemini-1.5-flash", api_key=api_key)
167
  elif service == "Ollama-llama3.2":
168
  return OllamaChatCompletionClient(model="llama3.2")
169
  elif service == "Azure AI Foundry":
@@ -1254,11 +1254,11 @@ with gr.Blocks(
1254
  "Azure AI Foundry",
1255
  "OpenAI-gpt-4o-2024-08-06",
1256
  "Anthropic-claude-3-sonnet-20240229",
1257
- "Google-gemini-1.5-flash",
1258
  "Ollama-llama3.2",
1259
  ],
1260
  label="Model",
1261
- value="Google-gemini-1.5-flash"
1262
  )
1263
  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)")
1264
  serpapi_key = gr.Textbox(label="SerpApi Key (For Research Agent)", type="password", placeholder="Enter your SerpApi key (optional)")
 
162
  return OpenAIChatCompletionClient(model="gpt-4o-2024-08-06", api_key=api_key)
163
  elif service == "Anthropic-claude-3-sonnet-20240229":
164
  return AnthropicChatCompletionClient(model="claude-3-sonnet-20240229", api_key=api_key)
165
+ elif service == "Google-gemini-2.0-flash":
166
+ return OpenAIChatCompletionClient(model="gemini-2.0-flash", api_key=api_key)
167
  elif service == "Ollama-llama3.2":
168
  return OllamaChatCompletionClient(model="llama3.2")
169
  elif service == "Azure AI Foundry":
 
1254
  "Azure AI Foundry",
1255
  "OpenAI-gpt-4o-2024-08-06",
1256
  "Anthropic-claude-3-sonnet-20240229",
1257
+ "Google-gemini-2.0-flash",
1258
  "Ollama-llama3.2",
1259
  ],
1260
  label="Model",
1261
+ value="Google-gemini-2.0-flash"
1262
  )
1263
  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)")
1264
  serpapi_key = gr.Textbox(label="SerpApi Key (For Research Agent)", type="password", placeholder="Enter your SerpApi key (optional)")