liuyanyi commited on
Commit
ff55a8e
·
verified ·
1 Parent(s): 2823c68

support gemini3flash

Browse files
Files changed (1) hide show
  1. Paper2Poster/utils/wei_utils.py +12 -0
Paper2Poster/utils/wei_utils.py CHANGED
@@ -186,6 +186,18 @@ def get_agent_config(model_type):
186
  'model_platform': ModelPlatformType.OPENROUTER,
187
  'model_config': OpenRouterConfig().as_dict(),
188
  }
 
 
 
 
 
 
 
 
 
 
 
 
189
  else:
190
  agent_config = {
191
  'model_type': model_type,
 
186
  'model_platform': ModelPlatformType.OPENROUTER,
187
  'model_config': OpenRouterConfig().as_dict(),
188
  }
189
+ elif model_type == 'g3f':
190
+ agent_config = {
191
+ 'model_type': "google/gemini-3-flash-preview",
192
+ 'model_platform': ModelPlatformType.OPENROUTER,
193
+ 'model_config': OpenRouterConfig().as_dict(),
194
+ }
195
+ elif model_type == 'o4o':
196
+ agent_config = {
197
+ 'model_type': "openai/gpt-4o",
198
+ 'model_platform': ModelPlatformType.OPENROUTER,
199
+ 'model_config': OpenRouterConfig().as_dict(),
200
+ }
201
  else:
202
  agent_config = {
203
  'model_type': model_type,