babelforce_manager_sdk/resources/
prompts.rs

1use crate::error::{map_manager_err, ManagerError};
2use crate::gen::manager::apis::configuration::Configuration;
3use crate::gen::manager::apis::{manager_api, prompt_api as api};
4use crate::gen::manager::models;
5use crate::http::collect_all;
6use crate::retry::{with_retry, RetryPolicy};
7use crate::token::SharedCfg;
8use std::path::PathBuf;
9
10/// Audio prompts — `/api/v2/prompts`.
11pub struct PromptsResource {
12    pub(crate) cfg: SharedCfg<Configuration>,
13    pub(crate) retry: RetryPolicy,
14}
15
16impl PromptsResource {
17    /// List all prompts (auto-paginated).
18    pub async fn list_all(&self) -> Result<Vec<models::Prompt>, ManagerError> {
19        let cfg = self.cfg.get().await?;
20        let cfg = cfg.as_ref();
21        collect_all(&self.retry, map_manager_err, |page| async move {
22            let r = api::list_prompts(cfg, Some(page), None).await?;
23            Ok((
24                r.items,
25                r.pagination.pages.unwrap_or(1),
26                r.pagination.current.unwrap_or(1),
27            ))
28        })
29        .await
30    }
31
32    /// List the objects that use a prompt by id.
33    pub async fn uses(&self, id: &str) -> Result<models::ObjectListResponse, ManagerError> {
34        let cfg = self.cfg.get().await?;
35        let cfg = cfg.as_ref();
36        with_retry(&self.retry, true, || manager_api::get_prompt_uses(cfg, id))
37            .await
38            .map_err(map_manager_err)
39    }
40
41    /// Get a prompt by id.
42    pub async fn get(&self, id: &str) -> Result<models::PromptItemResponse, ManagerError> {
43        let cfg = self.cfg.get().await?;
44        let cfg = cfg.as_ref();
45        with_retry(&self.retry, true, || api::get_prompt(cfg, id))
46            .await
47            .map_err(map_manager_err)
48    }
49
50    /// Upload an audio prompt from a file, optionally overriding its filename and label.
51    pub async fn upload(
52        &self,
53        file: PathBuf,
54        filename: Option<&str>,
55        label: Option<&str>,
56    ) -> Result<models::PromptItemResponse, ManagerError> {
57        let cfg = self.cfg.get().await?;
58        let cfg = cfg.as_ref();
59        with_retry(&self.retry, false, || {
60            manager_api::upload_prompt(cfg, file.clone(), filename, label)
61        })
62        .await
63        .map_err(map_manager_err)
64    }
65
66    /// Update a prompt's metadata.
67    pub async fn update(
68        &self,
69        id: &str,
70        body: models::RestUpdatePrompt,
71    ) -> Result<models::PromptItemResponse, ManagerError> {
72        let cfg = self.cfg.get().await?;
73        let cfg = cfg.as_ref();
74        with_retry(&self.retry, false, || {
75            api::update_prompt(cfg, id, body.clone())
76        })
77        .await
78        .map_err(map_manager_err)
79    }
80
81    /// Delete a prompt.
82    pub async fn delete(&self, id: &str) -> Result<(), ManagerError> {
83        let cfg = self.cfg.get().await?;
84        let cfg = cfg.as_ref();
85        with_retry(&self.retry, false, || api::delete_prompt(cfg, id))
86            .await
87            .map_err(map_manager_err)?;
88        Ok(())
89    }
90}
babelforce_manager_sdk/resources/prompts.rs

babelforce_manager_sdk/resources/
prompts.rs
