Skip to content

Commit 66e2d21

Browse files
committed
fix example
1 parent 2a732da commit 66e2d21

File tree

1 file changed

+26
-15
lines changed

1 file changed

+26
-15
lines changed

examples/reasoning/src/main.rs

Lines changed: 26 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -8,38 +8,49 @@ use serde_json::json;
88

99
const OPENROUTER_REASONING_KEY: &str = "reasoning";
1010
const OPENROUTER_BASEURL: &str = "https://openrouter.ai/api/v1";
11+
const OPENROUTER_MODEL_NAME: &str = "deepseek/deepseek-r1";
1112
const DEEPSEEK_REASONING_KEY: &str = "reasoning_content";
1213
const DEEPSEEK_BASEURL: &str = "https://api.deepseek.com";
14+
const DEEPSEEK_MODEL_NAME: &str = "deepseek-reasoner";
1315

1416

1517
#[tokio::main(flavor = "current_thread")]
1618
async fn main() {
1719
let test_key = std::env::var("TEST_API_KEY").unwrap();
1820
let use_deepseek = std::env::var("USE_DEEPSEEK").is_ok();
19-
let (reasoning_key, base_url) = if use_deepseek {
20-
(DEEPSEEK_REASONING_KEY, DEEPSEEK_BASEURL)
21+
let (reasoning_key, base_url, model_name) = if use_deepseek {
22+
(DEEPSEEK_REASONING_KEY, DEEPSEEK_BASEURL, DEEPSEEK_MODEL_NAME)
2123
} else {
22-
(OPENROUTER_REASONING_KEY, OPENROUTER_BASEURL)
24+
(OPENROUTER_REASONING_KEY, OPENROUTER_BASEURL, OPENROUTER_MODEL_NAME)
2325
};
2426
let client = Client::with_config(
2527
OpenAIConfig::new()
2628
.with_api_base(base_url)
2729
.with_api_key(test_key),
2830
);
29-
let request = CreateChatCompletionRequestArgs::default()
30-
.messages(vec![ChatCompletionRequestUserMessageArgs::default()
31-
.content("Hello! Do you know the Rust programming language?")
31+
let messages = vec![ChatCompletionRequestUserMessageArgs::default()
32+
.content("Hello! Do you know the Rust programming language?")
33+
.build()
34+
.unwrap()
35+
.into()];
36+
let request = if use_deepseek {
37+
CreateChatCompletionRequestArgs::default()
38+
.messages(messages)
39+
.model(model_name)
3240
.build()
3341
.unwrap()
34-
.into()])
35-
.model("deepseek/deepseek-r1")
36-
// The extra params that OpenRouter requires to get reasoning content
37-
// See https://openrouter.ai/docs/api-reference/parameters#include-reasoning
38-
.extra_params(json!({
39-
"include_reasoning" : true
40-
}))
41-
.build()
42-
.unwrap();
42+
} else {
43+
CreateChatCompletionRequestArgs::default()
44+
.messages(messages)
45+
.model(model_name)
46+
// The extra params that OpenRouter requires to get reasoning content
47+
// See https://openrouter.ai/docs/api-reference/parameters#include-reasoning
48+
.extra_params(json!({
49+
"include_reasoning" : true
50+
}))
51+
.build()
52+
.unwrap()
53+
};
4354

4455
let mut result = client.chat().create_stream(request).await.unwrap();
4556
let mut reasoning = String::new();

0 commit comments

Comments
 (0)