async_openai/
container_files.rs

1use bytes::Bytes;
2
3use crate::{
4    config::Config,
5    error::OpenAIError,
6    types::containers::{
7        ContainerFileListResource, ContainerFileResource, CreateContainerFileRequest,
8        DeleteContainerFileResponse,
9    },
10    Client, RequestOptions,
11};
12
13/// Create and manage container files for use with the Code Interpreter tool.
14pub struct ContainerFiles<'c, C: Config> {
15    client: &'c Client<C>,
16    container_id: String,
17    pub(crate) request_options: RequestOptions,
18}
19
20impl<'c, C: Config> ContainerFiles<'c, C> {
21    pub fn new(client: &'c Client<C>, container_id: &str) -> Self {
22        Self {
23            client,
24            container_id: container_id.to_string(),
25            request_options: RequestOptions::new(),
26        }
27    }
28
29    /// Create a container file by uploading a raw file or by referencing an existing file ID.
30    #[crate::byot(
31        T0 = Clone,
32        R = serde::de::DeserializeOwned,
33        where_clause = "reqwest::multipart::Form: crate::traits::AsyncTryFrom<T0, Error = OpenAIError>",
34    )]
35    pub async fn create(
36        &self,
37        request: CreateContainerFileRequest,
38    ) -> Result<ContainerFileResource, OpenAIError> {
39        self.client
40            .post_form(
41                &format!("/containers/{}/files", self.container_id),
42                request,
43                &self.request_options,
44            )
45            .await
46    }
47
48    /// List container files.
49    #[crate::byot(R = serde::de::DeserializeOwned)]
50    pub async fn list(&self) -> Result<ContainerFileListResource, OpenAIError> {
51        self.client
52            .get(
53                &format!("/containers/{}/files", self.container_id),
54                &self.request_options,
55            )
56            .await
57    }
58
59    /// Retrieve a container file.
60    #[crate::byot(T0 = std::fmt::Display, R = serde::de::DeserializeOwned)]
61    pub async fn retrieve(&self, file_id: &str) -> Result<ContainerFileResource, OpenAIError> {
62        self.client
63            .get(
64                format!("/containers/{}/files/{file_id}", self.container_id).as_str(),
65                &self.request_options,
66            )
67            .await
68    }
69
70    /// Delete a container file.
71    #[crate::byot(T0 = std::fmt::Display, R = serde::de::DeserializeOwned)]
72    pub async fn delete(&self, file_id: &str) -> Result<DeleteContainerFileResponse, OpenAIError> {
73        self.client
74            .delete(
75                format!("/containers/{}/files/{file_id}", self.container_id).as_str(),
76                &self.request_options,
77            )
78            .await
79    }
80
81    /// Returns the content of a container file.
82    pub async fn content(&self, file_id: &str) -> Result<Bytes, OpenAIError> {
83        let (bytes, _headers) = self
84            .client
85            .get_raw(
86                format!("/containers/{}/files/{file_id}/content", self.container_id).as_str(),
87                &self.request_options,
88            )
89            .await?;
90        Ok(bytes)
91    }
92}
async_openai/container_files.rs

async_openai/
container_files.rs
