Merge pull request #192 from dongri/add-responses

Add response api

Author: dongri

README.md

@@ -166,6 +166,7 @@ Check out the [full API documentation](https://platform.openai.com/docs/api-refe
 - [x] [Assistants](https://platform.openai.com/docs/assistants/overview)
 - [x] [Batch](https://platform.openai.com/docs/api-reference/batch)
 - [x] [Realtime](https://platform.openai.com/docs/api-reference/realtime)
+- [x] [Responses](https://platform.openai.com/docs/api-reference/responses)
 
 ## License
 

examples/responses.rs

@@ -0,0 +1,21 @@
+use openai_api_rs::v1::api::OpenAIClient;
+use openai_api_rs::v1::common::GPT4_1_MINI;
+use openai_api_rs::v1::responses::CreateResponseRequest;
+use serde_json::json;
+use std::env;
+
+#[tokio::main]
+async fn main() -> Result<(), Box<dyn std::error::Error>> {
+    let api_key = env::var("OPENAI_API_KEY").unwrap();
+    let mut client = OpenAIClient::builder().with_api_key(api_key).build()?;
+
+    let mut req = CreateResponseRequest::new();
+    req.model = Some(GPT4_1_MINI.to_string());
+    req.input = Some(json!("Write a haiku about Rust."));
+    req.extra.insert("temperature".to_string(), json!(0.7));
+
+    let resp = client.create_response(req).await?;
+    println!("response id: {} status: {:?}", resp.id, resp.status);
+    println!("response output: {:?}", resp.output);
+    Ok(())
+}

src/v1/api.rs

@@ -35,6 +35,9 @@ use crate::v1::message::{
 };
 use crate::v1::model::{ModelResponse, ModelsResponse};
 use crate::v1::moderation::{CreateModerationRequest, CreateModerationResponse};
+use crate::v1::responses::{
+    CountTokensRequest, CountTokensResponse, CreateResponseRequest, ListResponses, ResponseObject,
+};
 use crate::v1::run::{
     CreateRunRequest, CreateThreadAndRunRequest, ListRun, ListRunStep, ModifyRunRequest, RunObject,
     RunStepObject,
@@ -819,6 +822,70 @@ impl OpenAIClient {
         self.get(&url).await
     }
 
+    // Responses API
+    pub async fn create_response(
+        &mut self,
+        req: CreateResponseRequest,
+    ) -> Result<ResponseObject, APIError> {
+        self.post("responses", &req).await
+    }
+
+    pub async fn retrieve_response(
+        &mut self,
+        response_id: String,
+    ) -> Result<ResponseObject, APIError> {
+        self.get(&format!("responses/{response_id}")).await
+    }
+
+    pub async fn delete_response(
+        &mut self,
+        response_id: String,
+    ) -> Result<common::DeletionStatus, APIError> {
+        self.delete(&format!("responses/{response_id}")).await
+    }
+
+    pub async fn cancel_response(
+        &mut self,
+        response_id: String,
+    ) -> Result<ResponseObject, APIError> {
+        self.post(
+            &format!("responses/{response_id}/cancel"),
+            &common::EmptyRequestBody {},
+        )
+        .await
+    }
+
+    pub async fn list_response_input_items(
+        &mut self,
+        response_id: String,
+        after: Option<String>,
+        limit: Option<i64>,
+        order: Option<String>,
+    ) -> Result<ListResponses, APIError> {
+        let mut url = format!("responses/{}/input_items", response_id);
+        let mut params = vec![];
+        if let Some(after) = after {
+            params.push(format!("after={}", after));
+        }
+        if let Some(limit) = limit {
+            params.push(format!("limit={}", limit));
+        }
+        if let Some(order) = order {
+            params.push(format!("order={}", order));
+        }
+        if !params.is_empty() {
+            url = format!("{}?{}", url, params.join("&"));
+        }
+        self.get(&url).await
+    }
+
+    pub async fn count_response_input_tokens(
+        &mut self,
+        req: CountTokensRequest,
+    ) -> Result<CountTokensResponse, APIError> {
+        self.post("responses/input_tokens", &req).await
+    }
+
     pub async fn list_models(&mut self) -> Result<ModelsResponse, APIError> {
         self.get("models").await
     }

src/v1/mod.rs

@@ -13,6 +13,7 @@ pub mod fine_tuning;
 pub mod image;
 pub mod model;
 pub mod moderation;
+pub mod responses;
 
 // beta
 pub mod assistant;