use chrono::{DateTime, Utc};
use serde::{Deserialize, Serialize};
use std::path::PathBuf;
pub type SeedId = uuid::Uuid;
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Seed {
pub id: SeedId,
pub goal: String,
pub constraints: Vec<String>,
pub acceptance_criteria: Vec<String>,
pub ontology: Vec<Entity>,
pub created_at: DateTime<Utc>,
#[serde(default)]
pub generation: u32,
#[serde(skip_serializing_if = "Option::is_none")]
pub parent_seed_id: Option<SeedId>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub cspace_hint: Option<String>,
#[serde(default, skip_serializing_if = "String::is_empty")]
pub original_request: String,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub output_schema: Option<serde_json::Value>,
#[serde(skip_serializing_if = "Option::is_none")]
pub project_id: Option<uuid::Uuid>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub workspace_context: Option<String>,
#[serde(default, skip_serializing_if = "Vec::is_empty")]
pub mount_paths: Vec<PathBuf>,
}
impl Seed {
pub fn new(goal: impl Into<String>) -> Self {
let goal = goal.into();
Self {
id: uuid::Uuid::new_v4(),
goal: goal.clone(),
original_request: goal,
constraints: Vec::new(),
acceptance_criteria: Vec::new(),
ontology: Vec::new(),
created_at: Utc::now(),
generation: 0,
parent_seed_id: None,
cspace_hint: None,
output_schema: None,
project_id: None,
workspace_context: None,
mount_paths: Vec::new(),
}
}
pub fn from_message(message: impl Into<String>) -> Self {
let msg = message.into();
Self {
id: uuid::Uuid::new_v4(),
goal: msg.clone(),
original_request: msg,
constraints: Vec::new(),
acceptance_criteria: Vec::new(),
ontology: Vec::new(),
created_at: Utc::now(),
generation: 0,
parent_seed_id: None,
cspace_hint: None,
output_schema: None,
project_id: None,
workspace_context: None,
mount_paths: Vec::new(),
}
}
pub fn evolved_from(parent: &Seed) -> Seed {
Self {
id: uuid::Uuid::new_v4(),
goal: parent.goal.clone(),
original_request: parent.original_request.clone(),
constraints: parent.constraints.clone(),
acceptance_criteria: parent.acceptance_criteria.clone(),
ontology: parent.ontology.clone(),
created_at: Utc::now(),
generation: parent.generation + 1,
parent_seed_id: Some(parent.id),
cspace_hint: parent.cspace_hint.clone(),
output_schema: parent.output_schema.clone(),
project_id: parent.project_id,
workspace_context: parent.workspace_context.clone(),
mount_paths: parent.mount_paths.clone(),
}
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Entity {
pub name: String,
pub entity_type: String,
pub description: String,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AmbiguityScore {
pub goal_clarity: f64,
pub constraint_clarity: f64,
pub success_criteria: f64,
}
impl AmbiguityScore {
pub fn new(goal_clarity: f64, constraint_clarity: f64, success_criteria: f64) -> Self {
Self {
goal_clarity: goal_clarity.clamp(0.0, 1.0),
constraint_clarity: constraint_clarity.clamp(0.0, 1.0),
success_criteria: success_criteria.clamp(0.0, 1.0),
}
}
pub fn ambiguity(&self) -> f64 {
1.0 - (self.goal_clarity * 0.4
+ self.constraint_clarity * 0.3
+ self.success_criteria * 0.3)
}
pub fn is_ready(&self) -> bool {
self.ambiguity() <= 0.2
}
}
impl Default for AmbiguityScore {
fn default() -> Self {
Self {
goal_clarity: 0.0,
constraint_clarity: 0.0,
success_criteria: 0.0,
}
}
}