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

Merge pull request #17 from dongri/fix-required

Browse Source 
Fix required
This commit is contained in:
Dongri Jin
2023-07-25 23:22:10 +09:00
committed by GitHub
parent 8280733f05 5ee3321ed0
commit 786335b8ab
8 changed files with 31 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/chat_completion.rs
View File

@ -9,7 +9,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
model: chat_completion::GPT4.to_string(),
messages: vec![chat_completion::ChatCompletionMessage {
role: chat_completion::MessageRole::user,
content: Some(String::from("What is Bitcoin?")),
content: String::from("What is Bitcoin?"),
name: None,
function_call: None,
}],

2
examples/completion.rs
View File

@ -7,7 +7,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
let client = Client::new(env::var("OPENAI_API_KEY").unwrap().to_string());
let req = CompletionRequest {
model: completion::GPT3_TEXT_DAVINCI_003.to_string(),
prompt: Some(String::from("What is Bitcoin?")),
prompt: String::from("What is Bitcoin?"),
suffix: None,
max_tokens: Some(3000),
temperature: Some(0.9),

6
examples/function_call.rs
View File

@ -34,18 +34,18 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
model: chat_completion::GPT3_5_TURBO_0613.to_string(),
messages: vec![chat_completion::ChatCompletionMessage {
role: chat_completion::MessageRole::user,
content: Some(String::from("What is the price of Ethereum?")),
content: String::from("What is the price of Ethereum?"),
name: None,
function_call: None,
}],
functions: Some(vec![chat_completion::Function {
name: String::from("get_coin_price"),
description: Some(String::from("Get the price of a cryptocurrency")),
parameters: Some(chat_completion::FunctionParameters {
parameters: chat_completion::FunctionParameters {
schema_type: chat_completion::JSONSchemaType::Object,
properties: Some(properties),
required: Some(vec![String::from("coin")]),
}),
},
}]),
function_call: Some("auto".to_string()),
temperature: None,

12
examples/function_call_role.rs
View File

@ -34,18 +34,18 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
model: chat_completion::GPT3_5_TURBO_0613.to_string(),
messages: vec![chat_completion::ChatCompletionMessage {
role: chat_completion::MessageRole::user,
content: Some(String::from("What is the price of Ethereum?")),
content: String::from("What is the price of Ethereum?"),
name: None,
function_call: None,
}],
functions: Some(vec![chat_completion::Function {
name: String::from("get_coin_price"),
description: Some(String::from("Get the price of a cryptocurrency")),
parameters: Some(chat_completion::FunctionParameters {
parameters: chat_completion::FunctionParameters {
schema_type: chat_completion::JSONSchemaType::Object,
properties: Some(properties),
required: Some(vec![String::from("coin")]),
}),
},
}]),
function_call: None,
temperature: None,
@ -86,16 +86,16 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
messages: vec![
chat_completion::ChatCompletionMessage {
role: chat_completion::MessageRole::user,
content: Some(String::from("What is the price of Ethereum?")),
content: String::from("What is the price of Ethereum?"),
name: None,
function_call: None,
},
chat_completion::ChatCompletionMessage {
role: chat_completion::MessageRole::function,
content: Some({
content: {
let price = get_coin_price(&coin).await;
format!("{{\"price\": {}}}", price)
}),
},
name: Some(String::from("get_coin_price")),
function_call: None,
},

4
src/v1/audio.rs
View File

@ -4,8 +4,8 @@ pub const WHISPER_1: &str = "whisper-1";
#[derive(Debug, Serialize)]
pub struct AudioTranscriptionRequest {
pub model: String,
pub file: String,
pub model: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub prompt: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
@ -23,8 +23,8 @@ pub struct AudioTranscriptionResponse {
#[derive(Debug, Serialize)]
pub struct AudioTranslationRequest {
pub model: String,
pub file: String,
pub model: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub prompt: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]

21
src/v1/chat_completion.rs
View File

@ -43,7 +43,7 @@ pub struct ChatCompletionRequest {
pub user: Option<String>,
}
#[derive(Debug, Serialize, Deserialize)]
#[derive(Debug, Serialize, Deserialize, Clone)]
#[allow(non_camel_case_types)]
pub enum MessageRole {
user,
@ -52,8 +52,18 @@ pub enum MessageRole {
function,
}
#[derive(Debug, Serialize, Deserialize)]
#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct ChatCompletionMessage {
pub role: MessageRole,
pub content: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub name: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub function_call: Option<FunctionCall>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct ChatCompletionMessageForResponse {
pub role: MessageRole,
#[serde(skip_serializing_if = "Option::is_none")]
pub content: Option<String>,
@ -66,7 +76,7 @@ pub struct ChatCompletionMessage {
#[derive(Debug, Deserialize)]
pub struct ChatCompletionChoice {
pub index: i64,
pub message: ChatCompletionMessage,
pub message: ChatCompletionMessageForResponse,
pub finish_reason: FinishReason,
}
@ -85,8 +95,7 @@ pub struct Function {
pub name: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub description: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub parameters: Option<FunctionParameters>,
pub parameters: FunctionParameters,
}
#[derive(Debug, Serialize, Deserialize)]
@ -136,7 +145,7 @@ pub enum FinishReason {
null,
}
#[derive(Debug, Serialize, Deserialize)]
#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct FunctionCall {
#[serde(skip_serializing_if = "Option::is_none")]
pub name: Option<String>,

3
src/v1/completion.rs
View File

@ -20,8 +20,7 @@ pub const GPT3_BABBAGE: &str = "babbage";
#[derive(Debug, Serialize)]
pub struct CompletionRequest {
pub model: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub prompt: Option<String>,
pub prompt: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub suffix: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]

2
src/v1/moderation.rs
View File

@ -3,6 +3,8 @@ use serde::{Deserialize, Serialize};
#[derive(Debug, Serialize)]
pub struct CreateModerationRequest {
pub input: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub model: Option<String>,
}
#[derive(Debug, Deserialize)]