mii/openai-api-rs
1
0
Fork 0
mirror of https://github.com/mii443/openai-api-rs.git synced 2025-08-22 15:15:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: update examples

Browse Source
This commit is contained in:
Anush008
2023-08-25 17:29:36 +05:30
parent ffe2bd49be
commit f453dc9323
5 changed files with 15 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
examples/function_call_role.rs
View File

@ -4,7 +4,7 @@ use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use std::{env, vec};
async fn get_coin_price(coin: &str) -> f64 {
fn get_coin_price(coin: &str) -> f64 {
let coin = coin.to_lowercase();
match coin.as_str() {
"btc" | "bitcoin" => 10000.0,
@ -13,8 +13,7 @@ async fn get_coin_price(coin: &str) -> f64 {
}
}
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
fn main() -> Result<(), Box<dyn std::error::Error>> {
let client = Client::new(env::var("OPENAI_API_KEY").unwrap().to_string());
let mut properties = HashMap::new();
@ -60,7 +59,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
user: None,
};
let result = client.chat_completion(req).await?;
let result = client.chat_completion(req)?;
match result.choices[0].finish_reason {
chat_completion::FinishReason::stop => {
@ -93,7 +92,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
chat_completion::ChatCompletionMessage {
role: chat_completion::MessageRole::function,
content: {
let price = get_coin_price(&coin).await;
let price = get_coin_price(&coin);
format!("{{\"price\": {}}}", price)
},
name: Some(String::from("get_coin_price")),
@ -113,7 +112,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
logit_bias: None,
user: None,
};
let result = client.chat_completion(req).await?;
let result = client.chat_completion(req)?;
println!("{:?}", result.choices[0].message.content);
}
chat_completion::FinishReason::content_filter => {