use crate::runtime::{self, Error};
#[derive(Clone, Debug, Default)]
pub struct AiGetAgentDefaultOptions {
pub language: Option<String>,
pub model: Option<String>,
}
pub struct AiManager {
session: std::sync::Arc<runtime::Client>,
}
impl AiManager {
pub(crate) fn new(session: std::sync::Arc<runtime::Client>) -> Self {
Self { session }
}
pub async fn ask(
&self,
body: crate::models::schemas::AiAsk,
) -> Result<crate::models::schemas::AiResponseFull, Error> {
let mut url = self.session.base_url("api");
url.push_str("/ai");
url.push_str("/ask");
let mut req = self.session.new_request("POST", &url);
let payload = serde_json::to_vec(&body)?;
req = runtime::with_json_body(req, &payload);
let resp = self.session.fetch(req).await?;
let data = runtime::response_bytes(&resp)?;
Ok(serde_json::from_slice(&data)?)
}
pub async fn create_text_gen(
&self,
body: crate::models::schemas::AiTextGen,
) -> Result<crate::models::schemas::AiResponse, Error> {
let mut url = self.session.base_url("api");
url.push_str("/ai");
url.push_str("/text_gen");
let mut req = self.session.new_request("POST", &url);
let payload = serde_json::to_vec(&body)?;
req = runtime::with_json_body(req, &payload);
let resp = self.session.fetch(req).await?;
let data = runtime::response_bytes(&resp)?;
Ok(serde_json::from_slice(&data)?)
}
pub async fn get_agent_default(
&self,
mode: crate::models::schemas::GetAgentDefaultMode,
opts: Option<AiGetAgentDefaultOptions>,
) -> Result<crate::models::schemas::AiAgent, Error> {
let mut url = self.session.base_url("api");
url.push_str("/ai_agent_default");
let mut req = self.session.new_request("GET", &url);
req = runtime::with_query(req, "mode", &mode.0);
let opts = opts.unwrap_or_default();
if let Some(value) = opts.language {
req = runtime::with_query(req, "language", &value);
}
if let Some(value) = opts.model {
req = runtime::with_query(req, "model", &value);
}
let resp = self.session.fetch(req).await?;
let data = runtime::response_bytes(&resp)?;
Ok(serde_json::from_slice(&data)?)
}
pub async fn extract(
&self,
body: crate::models::schemas::AiExtract,
) -> Result<crate::models::schemas::AiResponse, Error> {
let mut url = self.session.base_url("api");
url.push_str("/ai");
url.push_str("/extract");
let mut req = self.session.new_request("POST", &url);
let payload = serde_json::to_vec(&body)?;
req = runtime::with_json_body(req, &payload);
let resp = self.session.fetch(req).await?;
let data = runtime::response_bytes(&resp)?;
Ok(serde_json::from_slice(&data)?)
}
pub async fn create_extract_structured(
&self,
body: crate::models::schemas::AiExtractStructured,
) -> Result<crate::models::schemas::AiExtractStructuredResponse, Error> {
let mut url = self.session.base_url("api");
url.push_str("/ai");
url.push_str("/extract_structured");
let mut req = self.session.new_request("POST", &url);
let payload = serde_json::to_vec(&body)?;
req = runtime::with_json_body(req, &payload);
let resp = self.session.fetch(req).await?;
let data = runtime::response_bytes(&resp)?;
Ok(serde_json::from_slice(&data)?)
}
}