From 91fdaf3c3bfec4760f863820f2163539affa2fe6 Mon Sep 17 00:00:00 2001 From: Dongri Jin Date: Mon, 15 Apr 2024 16:01:53 +0900 Subject: [PATCH] Fix readme --- README.md | 28 ++++++++++++++-------------- 1 file changed, 14 insertions(+), 14 deletions(-) diff --git a/README.md b/README.md index 89286f7..9811dd4 100644 --- a/README.md +++ b/README.md @@ -18,27 +18,18 @@ The library needs to be configured with your account's secret key, which is avai $ export OPENAI_API_KEY=sk-xxxxxxx ``` -### Set OPENAI_API_BASE to environment variable (optional) -```bash -$ export OPENAI_API_BASE=https://api.openai.com/v1 -``` - ### Create client ```rust -use openai_api_rs::v1::api::Client; -use std::env; let client = Client::new(env::var("OPENAI_API_KEY").unwrap().to_string()); ``` ### Create request ```rust -use openai_api_rs::v1::chat_completion::{self, ChatCompletionRequest}; -use openai_api_rs::v1::common::GPT4; let req = ChatCompletionRequest::new( GPT4.to_string(), vec![chat_completion::ChatCompletionMessage { role: chat_completion::MessageRole::user, - content: chat_completion::Content::Text(String::from("Hello OpenAI!")), + content: chat_completion::Content::Text(String::from("What is bitcoin?")), name: None, }], ); @@ -46,8 +37,13 @@ let req = ChatCompletionRequest::new( ### Send request ```rust -let result = client.completion(req)?; -println!("{:?}", result.choices[0].text); +let result = client.chat_completion(req)?; +println!("Content: {:?}", result.choices[0].message.content); +``` + +### Set OPENAI_API_BASE to environment variable (optional) +```bash +$ export OPENAI_API_BASE=https://api.openai.com/v1 ``` ## Example of chat completion @@ -59,16 +55,20 @@ use std::env; fn main() -> Result<(), Box> { let client = Client::new(env::var("OPENAI_API_KEY").unwrap().to_string()); + let req = ChatCompletionRequest::new( GPT4.to_string(), vec![chat_completion::ChatCompletionMessage { role: chat_completion::MessageRole::user, - content: chat_completion::Content::Text(String::from("What is Bitcoin?")), + content: chat_completion::Content::Text(String::from("What is bitcoin?")), name: None, }], ); + let result = client.chat_completion(req)?; - println!("{:?}", result.choices[0].message.content); + println!("Content: {:?}", result.choices[0].message.content); + println!("Response Headers: {:?}", result.headers); + Ok(()) } ```