Skip to content

Commit d3d99b8

Browse files
committed
Support override default model name
1 parent b43830f commit d3d99b8

File tree

2 files changed

+11
-5
lines changed

2 files changed

+11
-5
lines changed

aiscript-vm/src/ai/mod.rs

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -77,14 +77,18 @@ impl AiConfig {
7777
match model {
7878
m if m.starts_with("gpt") => {
7979
if let Some(openai) = self.openai.as_ref() {
80-
Ok(openai.clone())
80+
let mut config = openai.clone();
81+
config.model = Some(m);
82+
Ok(config)
8183
} else {
8284
Ok(ModelConfig::default())
8385
}
8486
}
8587
m if m.starts_with("claude") => {
86-
if let Some(anthropic) = self.openai.as_ref() {
87-
Ok(anthropic.clone())
88+
if let Some(anthropic) = self.anthropic.as_ref() {
89+
let mut config = anthropic.clone();
90+
config.model = Some(m);
91+
Ok(config)
8892
} else {
8993
Ok(ModelConfig {
9094
api_key: env::var("CLAUDE_API_KEY")
@@ -96,7 +100,9 @@ impl AiConfig {
96100
}
97101
m if m.starts_with("deepseek") => {
98102
if let Some(deepseek) = self.deepseek.as_ref() {
99-
Ok(deepseek.clone())
103+
let mut config = deepseek.clone();
104+
config.model = Some(m);
105+
Ok(config)
100106
} else {
101107
Ok(ModelConfig {
102108
api_key: env::var("DEEPSEEK_API_KEY")

examples/claude.ai

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
let a = prompt {
22
input: "What is rust?",
3-
model: "claude-3-5-sonnet-latest"
3+
model: "claude-3-7-sonnet-latest"
44
};
55
print(a);

0 commit comments

Comments
 (0)