use crate::models::common::{JsonValue, Status, StepType};
use chrono::{DateTime, Utc};
use serde::{Deserialize, Serialize};
use uuid::Uuid;

#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, sqlx::FromRow)]
pub struct AgentExecutionStep {
    pub id: Uuid,
    pub execution_id: Uuid,
    pub step_index: i32,
    pub step_type: StepType,
    pub name: String,
    pub status: Status,
    pub model_id: Option<Uuid>,
    pub tool_name: Option<String>,
    pub input: Option<JsonValue>,
    pub output: Option<JsonValue>,
    pub error_message: Option<String>,
    pub token_input_count: Option<i32>,
    pub token_output_count: Option<i32>,
    pub started_at: Option<DateTime<Utc>>,
    pub finished_at: Option<DateTime<Utc>>,
    pub created_at: DateTime<Utc>,
}