chore: update ai model name (#6316)

This commit is contained in:
Nathan.fooo 2024-09-15 23:36:29 +08:00 committed by GitHub
parent 877236baf9
commit 7271e0a108
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 6 deletions

View File

@ -59,7 +59,7 @@ List<AIModelPB> _availableModels = [
AIModelPB.DefaultModel,
AIModelPB.Claude3Opus,
AIModelPB.Claude3Sonnet,
AIModelPB.GPT35,
AIModelPB.GPT4oMini,
AIModelPB.GPT4o,
];
@ -71,8 +71,8 @@ String _titleForAIModel(AIModelPB model) {
return "Claude 3 Opus";
case AIModelPB.Claude3Sonnet:
return "Claude 3 Sonnet";
case AIModelPB.GPT35:
return "GPT-3.5";
case AIModelPB.GPT4oMini:
return "GPT-4o-mini";
case AIModelPB.GPT4o:
return "GPT-4o";
default:

View File

@ -424,7 +424,7 @@ impl From<UpdateUserWorkspaceSettingPB> for AFWorkspaceSettingsChange {
pub enum AIModelPB {
#[default]
DefaultModel = 0,
GPT35 = 1,
GPT4oMini = 1,
GPT4o = 2,
Claude3Sonnet = 3,
Claude3Opus = 4,
@ -434,7 +434,7 @@ impl AIModelPB {
pub fn to_str(&self) -> &str {
match self {
AIModelPB::DefaultModel => "default-model",
AIModelPB::GPT35 => "gpt-3.5-turbo",
AIModelPB::GPT4oMini => "gpt-4o-mini",
AIModelPB::GPT4o => "gpt-4o",
AIModelPB::Claude3Sonnet => "claude-3-sonnet",
AIModelPB::Claude3Opus => "claude-3-opus",
@ -447,7 +447,7 @@ impl FromStr for AIModelPB {
fn from_str(s: &str) -> Result<Self, Self::Err> {
match s {
"gpt-3.5-turbo" => Ok(AIModelPB::GPT35),
"gpt-3.5-turbo" => Ok(AIModelPB::GPT4oMini),
"gpt-4o" => Ok(AIModelPB::GPT4o),
"claude-3-sonnet" => Ok(AIModelPB::Claude3Sonnet),
"claude-3-opus" => Ok(AIModelPB::Claude3Opus),