use serde::{Deserialize, Serialize};
use crate::card::prompt::{ParameterName, validate};
use crate::error::WyrdError;
#[derive(Debug, Clone, PartialEq, Serialize, schemars::JsonSchema)]
#[cfg_attr(feature = "server", derive(utoipa::ToSchema))]
pub struct PromptSpec {
#[serde(flatten)]
#[cfg_attr(feature = "server", schema(value_type = serde_json::Value))]
pub prompt: skald_spec::Prompt,
}
impl PromptSpec {
pub fn new(prompt: skald_spec::Prompt) -> Result<Self, WyrdError> {
let prompt = prompt.authored()?.compile()?;
let spec = Self { prompt };
validate(&spec)?;
Ok(spec)
}
pub fn parameters(&self) -> Vec<ParameterName> {
self.prompt
.variables
.iter()
.filter_map(|name| ParameterName::new(name.clone()).ok())
.collect()
}
pub fn content_hash(&self) -> String {
crate::card::prompt::content_hash(self)
}
pub fn is_fully_bound(&self) -> bool {
self.prompt.variables.is_empty()
}
}
impl<'de> Deserialize<'de> for PromptSpec {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>,
{
let spec = Self {
prompt: skald_spec::Prompt::deserialize(deserializer)?,
};
validate(&spec).map_err(serde::de::Error::custom)?;
Ok(spec)
}
}