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

refactoring function

Browse Source
This commit is contained in:
Dongri Jin
2024-10-16 08:46:01 +09:00
parent 34aae9174a
commit 5867baa3ef
6 changed files with 106 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
examples/function_call.rs
View File

@ -1,6 +1,7 @@
use openai_api_rs::v1::api::OpenAIClient;
use openai_api_rs::v1::chat_completion::{self, ChatCompletionRequest};
use openai_api_rs::v1::common::GPT4_O;
use openai_api_rs::v1::types;
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use std::{env, vec};
@ -21,8 +22,8 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
let mut properties = HashMap::new();
properties.insert(
"coin".to_string(),
Box::new(chat_completion::JSONSchemaDefine {
schema_type: Some(chat_completion::JSONSchemaType::String),
Box::new(types::JSONSchemaDefine {
schema_type: Some(types::JSONSchemaType::String),
description: Some("The cryptocurrency to get the price of".to_string()),
..Default::default()
}),
@ -40,11 +41,11 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
)
.tools(vec![chat_completion::Tool {
r#type: chat_completion::ToolType::Function,
function: chat_completion::Function {
function: types::Function {
name: String::from("get_coin_price"),
description: Some(String::from("Get the price of a cryptocurrency")),
parameters: chat_completion::FunctionParameters {
schema_type: chat_completion::JSONSchemaType::Object,
parameters: types::FunctionParameters {
schema_type: types::JSONSchemaType::Object,
properties: Some(properties),
required: Some(vec![String::from("coin")]),
},

11
examples/function_call_role.rs
View File

@ -1,6 +1,7 @@
use openai_api_rs::v1::api::OpenAIClient;
use openai_api_rs::v1::chat_completion::{self, ChatCompletionRequest};
use openai_api_rs::v1::common::GPT4_O;
use openai_api_rs::v1::types;
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use std::{env, vec};
@ -21,8 +22,8 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
let mut properties = HashMap::new();
properties.insert(
"coin".to_string(),
Box::new(chat_completion::JSONSchemaDefine {
schema_type: Some(chat_completion::JSONSchemaType::String),
Box::new(types::JSONSchemaDefine {
schema_type: Some(types::JSONSchemaType::String),
description: Some("The cryptocurrency to get the price of".to_string()),
..Default::default()
}),
@ -40,11 +41,11 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
)
.tools(vec![chat_completion::Tool {
r#type: chat_completion::ToolType::Function,
function: chat_completion::Function {
function: types::Function {
name: String::from("get_coin_price"),
description: Some(String::from("Get the price of a cryptocurrency")),
parameters: chat_completion::FunctionParameters {
schema_type: chat_completion::JSONSchemaType::Object,
parameters: types::FunctionParameters {
schema_type: types::JSONSchemaType::Object,
properties: Some(properties),
required: Some(vec![String::from("coin")]),
},