coo7 commited on
Commit
f6c3932
·
verified ·
1 Parent(s): 00493d3

Update internal/types/monica.go

Browse files
Files changed (1) hide show
  1. internal/types/monica.go +23 -8
internal/types/monica.go CHANGED
@@ -209,14 +209,19 @@ func GetSupportedModels() OpenAIModelList {
209
  models := []OpenAIModel{
210
  {ID: "gpt-4o-mini", Object: "model", OwnedBy: "monica"},
211
  {ID: "gpt-4o", Object: "model", OwnedBy: "monica"},
 
 
212
  {ID: "claude-3-5-sonnet", Object: "model", OwnedBy: "monica"},
213
  {ID: "claude-3-5-haiku", Object: "model", OwnedBy: "monica"},
214
- {ID: "gemini-2.0", Object: "model", OwnedBy: "monica"},
215
- {ID: "gemini-1.5", Object: "model", OwnedBy: "monica"},
216
- {ID: "o1-mini", Object: "model", OwnedBy: "monica"},
 
217
  {ID: "o1-preview", Object: "model", OwnedBy: "monica"},
218
  {ID: "deepseek-reasoner", Object: "model", OwnedBy: "monica"},
219
  {ID: "deepseek-chat", Object: "model", OwnedBy: "monica"},
 
 
220
  }
221
 
222
  return OpenAIModelList{
@@ -339,23 +344,33 @@ func modelToBot(model string) string {
339
  return "gpt_4_o_mini_chat"
340
  case strings.HasPrefix(model, "gpt-4o"):
341
  return "gpt_4_o_chat"
 
 
 
 
342
  case strings.HasPrefix(model, "claude-3-5-sonnet"):
343
  return "claude_3.5_sonnet"
344
- case strings.Contains(model, "haiku"):
345
  return "claude_3.5_haiku"
346
- case strings.HasPrefix(model, "gemini-2"):
 
 
347
  return "gemini_2_0"
348
  case strings.HasPrefix(model, "gemini-1"):
349
  return "gemini_1_5"
350
- case strings.HasPrefix(model, "o1-mini"):
351
- return "openai_o_1_mini"
352
  case strings.HasPrefix(model, "o1-preview"):
353
  return "openai_o_1"
 
 
354
  case model == "deepseek-reasoner":
355
  return "deepseek_reasoner"
356
  case model == "deepseek-chat":
357
  return "deepseek_chat"
 
 
 
 
358
  default:
359
- return "claude_3.5_sonnet"
360
  }
361
  }
 
209
  models := []OpenAIModel{
210
  {ID: "gpt-4o-mini", Object: "model", OwnedBy: "monica"},
211
  {ID: "gpt-4o", Object: "model", OwnedBy: "monica"},
212
+ {ID: "claude-3-7-sonnet", Object: "model", OwnedBy: "monica"},
213
+ {ID: "claude-3-7-sonnet-thinking", Object: "model", OwnedBy: "monica"},
214
  {ID: "claude-3-5-sonnet", Object: "model", OwnedBy: "monica"},
215
  {ID: "claude-3-5-haiku", Object: "model", OwnedBy: "monica"},
216
+ {ID: "gemini-2.0-pro", Object: "model", OwnedBy: "monica"},
217
+ {ID: "gemini-2.0-flash", Object: "model", OwnedBy: "monica"},
218
+ {ID: "gemini-1.5-pro", Object: "model", OwnedBy: "monica"},
219
+ {ID: "o3-mini", Object: "model", OwnedBy: "monica"},
220
  {ID: "o1-preview", Object: "model", OwnedBy: "monica"},
221
  {ID: "deepseek-reasoner", Object: "model", OwnedBy: "monica"},
222
  {ID: "deepseek-chat", Object: "model", OwnedBy: "monica"},
223
+ {ID: "deepclaude", Object: "model", OwnedBy: "monica"},
224
+ {ID: "sonar", Object: "model", OwnedBy: "monica"},
225
  }
226
 
227
  return OpenAIModelList{
 
344
  return "gpt_4_o_mini_chat"
345
  case strings.HasPrefix(model, "gpt-4o"):
346
  return "gpt_4_o_chat"
347
+ case strings.HasPrefix(model, "claude-3-7-sonnet-thinking"):
348
+ return "claude_3_7_sonnet_think"
349
+ case strings.HasPrefix(model, "claude-3-7-sonnet"):
350
+ return "claude_3_7_sonnet"
351
  case strings.HasPrefix(model, "claude-3-5-sonnet"):
352
  return "claude_3.5_sonnet"
353
+ case strings.HasPrefix(model, "claude-3-5-haiku"):
354
  return "claude_3.5_haiku"
355
+ case strings.HasPrefix(model, "gemini-2.0-pro"):
356
+ return "gemini_2_0_pro"
357
+ case strings.HasPrefix(model, "gemini-2.0-flash"):
358
  return "gemini_2_0"
359
  case strings.HasPrefix(model, "gemini-1"):
360
  return "gemini_1_5"
 
 
361
  case strings.HasPrefix(model, "o1-preview"):
362
  return "openai_o_1"
363
+ case strings.HasPrefix(model, "o3-mini"):
364
+ return "openai_o_3_mini"
365
  case model == "deepseek-reasoner":
366
  return "deepseek_reasoner"
367
  case model == "deepseek-chat":
368
  return "deepseek_chat"
369
+ case model == "deepclaude":
370
+ return "deepclaude"
371
+ case model == "sonar":
372
+ return "sonar"
373
  default:
374
+ return model
375
  }
376
  }