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((r.items, r.pagination.pages, r.pagination.current))
24        })
25        .await
26    }
27
28    /// List the objects that use a prompt by id.
29    pub async fn uses(&self, id: &str) -> Result<models::ObjectListResponse, ManagerError> {
30        let cfg = self.cfg.get().await?;
31        let cfg = cfg.as_ref();
32        with_retry(&self.retry, true, || manager_api::get_prompt_uses(cfg, id))
33            .await
34            .map_err(map_manager_err)
35    }
36
37    /// Get a prompt by id.
38    pub async fn get(&self, id: &str) -> Result<models::PromptItemResponse, ManagerError> {
39        let cfg = self.cfg.get().await?;
40        let cfg = cfg.as_ref();
41        with_retry(&self.retry, true, || api::get_prompt(cfg, id))
42            .await
43            .map_err(map_manager_err)
44    }
45
46    /// Upload an audio prompt from a file, optionally overriding its filename and label.
47    pub async fn upload(
48        &self,
49        file: PathBuf,
50        filename: Option<&str>,
51        label: Option<&str>,
52    ) -> Result<models::PromptItemResponse, ManagerError> {
53        let cfg = self.cfg.get().await?;
54        let cfg = cfg.as_ref();
55        with_retry(&self.retry, false, || {
56            manager_api::upload_prompt(cfg, file.clone(), filename, label)
57        })
58        .await
59        .map_err(map_manager_err)
60    }
61
62    /// Update a prompt's metadata.
63    pub async fn update(
64        &self,
65        id: &str,
66        body: models::RestUpdatePrompt,
67    ) -> Result<models::PromptItemResponse, ManagerError> {
68        let cfg = self.cfg.get().await?;
69        let cfg = cfg.as_ref();
70        with_retry(&self.retry, false, || {
71            api::update_prompt(cfg, id, body.clone())
72        })
73        .await
74        .map_err(map_manager_err)
75    }
76
77    /// Delete a prompt.
78    pub async fn delete(&self, id: &str) -> Result<(), ManagerError> {
79        let cfg = self.cfg.get().await?;
80        let cfg = cfg.as_ref();
81        with_retry(&self.retry, false, || api::delete_prompt(cfg, id))
82            .await
83            .map_err(map_manager_err)?;
84        Ok(())
85    }
86}
babelforce_manager_sdk/resources/prompts.rs

babelforce_manager_sdk/resources/
prompts.rs
