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

Add new_with_endpoint

Browse Source
This commit is contained in:
Dongri Jin
2023-07-29 07:32:41 +09:00
parent 39c8e2b572
commit c3a08c36a3
1 changed files with 25 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
src/v1/api.rs
View File

@ -26,15 +26,23 @@ use crate::v1::moderation::{CreateModerationRequest, CreateModerationResponse};
use reqwest::Response; use reqwest::Response;
const APU_URL_V1: &str = "https://api.openai.com/v1"; const API_URL_V1: &str = "https://api.openai.com/v1";
pub struct Client { pub struct Client {
pub api_endpoint: String,
pub api_key: String, pub api_key: String,
} }
impl Client { impl Client {
pub fn new(api_key: String) -> Self { pub fn new(api_key: String) -> Self {
Self { api_key } Self::new_with_endpoint(API_URL_V1.to_owned(), api_key)
}
pub fn new_with_endpoint(api_endpoint: String, api_key: String) -> Self {
Self {
api_endpoint,
api_key,
}
} }
pub async fn post<T: serde::ser::Serialize>( pub async fn post<T: serde::ser::Serialize>(
@ -43,7 +51,11 @@ impl Client {
params: &T, params: &T,
) -> Result<Response, APIError> { ) -> Result<Response, APIError> {
let client = reqwest::Client::new(); let client = reqwest::Client::new();
let url = format!("{APU_URL_V1}{path}"); let url = format!(
"{api_endpoint}{path}",
api_endpoint = self.api_endpoint,
path = path
);
let res = client let res = client
.post(&url) .post(&url)
.header(reqwest::header::CONTENT_TYPE, "application/json") .header(reqwest::header::CONTENT_TYPE, "application/json")
@ -67,7 +79,11 @@ impl Client {
pub async fn get(&self, path: &str) -> Result<Response, APIError> { pub async fn get(&self, path: &str) -> Result<Response, APIError> {
let client = reqwest::Client::new(); let client = reqwest::Client::new();
let url = format!("{APU_URL_V1}{path}"); let url = format!(
"{api_endpoint}{path}",
api_endpoint = self.api_endpoint,
path = path
);
let res = client let res = client
.get(&url) .get(&url)
.header(reqwest::header::CONTENT_TYPE, "application/json") .header(reqwest::header::CONTENT_TYPE, "application/json")
@ -90,7 +106,11 @@ impl Client {
pub async fn delete(&self, path: &str) -> Result<Response, APIError> { pub async fn delete(&self, path: &str) -> Result<Response, APIError> {
let client = reqwest::Client::new(); let client = reqwest::Client::new();
let url = format!("{APU_URL_V1}{path}"); let url = format!(
"{api_endpoint}{path}",
api_endpoint = self.api_endpoint,
path = path
);
let res = client let res = client
.delete(&url) .delete(&url)
.header(reqwest::header::CONTENT_TYPE, "application/json") .header(reqwest::header::CONTENT_TYPE, "application/json")