Genarrative/apps/ai-game-creator-shell/src-tauri/src/agent/runtime_models.rs

use super::*;

#[derive(Clone, Debug, Default, Deserialize, Eq, PartialEq, Serialize)]
#[serde(rename_all = "camelCase")]
pub(crate) struct AgentRuntimeToolAction {
    #[serde(default)]
    pub(crate) tool: String,
    #[serde(default)]
    pub(crate) reason: Option<String>,
    #[serde(default)]
    pub(crate) input: serde_json::Value,
}

#[derive(Clone, Debug, Deserialize, Eq, PartialEq, Serialize)]
#[serde(rename_all = "camelCase")]
pub(crate) struct AgentRuntimeToolObservation {
    pub(crate) tool: String,
    pub(crate) status: String,
    pub(crate) summary: String,
    pub(crate) detail: Option<String>,
}

#[derive(Clone, Debug, Deserialize, Eq, PartialEq, Serialize)]
#[serde(deny_unknown_fields, rename_all = "camelCase")]
pub(crate) struct AgentRuntimeProjectRevision {
    pub(crate) schema_version: String,
    pub(crate) project_id: String,
    pub(crate) revision: u64,
    pub(crate) updated_at: u64,
}

#[derive(Clone, Debug, Eq, PartialEq)]
pub(crate) struct AgentRuntimeProviderRequestSnapshot {
    pub(in crate::agent) project_id: String,
    pub(in crate::agent) agent_id: String,
    pub(in crate::agent) task_id: String,
    pub(in crate::agent) session_id: String,
    pub(in crate::agent) run_id: String,
    pub(in crate::agent) source: String,
    pub(in crate::agent) goal_id: Option<String>,
    pub(in crate::agent) goal_revision: u64,
    pub(in crate::agent) goal_snapshot_fingerprint: String,
    pub(in crate::agent) applied_steer_cursor: u64,
    pub(in crate::agent) request_kind: String,
    pub(in crate::agent) request_slot: String,
    pub(in crate::agent) web_search_enabled: bool,
    pub(in crate::agent) allow_idle_context_compaction: bool,
}

impl AgentRuntimeProviderRequestSnapshot {
    pub(in crate::agent) fn with_request_slot(&self, request_slot: impl Into<String>) -> Self {
        let mut snapshot = self.clone();
        snapshot.request_slot = request_slot.into();
        snapshot
    }

    pub(in crate::agent) fn with_web_search_enabled(&self, web_search_enabled: bool) -> Self {
        let mut snapshot = self.clone();
        snapshot.web_search_enabled = web_search_enabled;
        snapshot
    }

    pub(super) fn with_idle_context_compaction(&self, allow: bool) -> Self {
        let mut snapshot = self.clone();
        snapshot.allow_idle_context_compaction = allow;
        snapshot
    }
}

#[derive(Clone, Debug, Deserialize, Eq, PartialEq, Serialize)]
#[serde(deny_unknown_fields, rename_all = "camelCase")]
pub(crate) struct AgentRuntimeVerificationGate {
    pub(crate) schema_version: String,
    pub(crate) project_id: String,
    pub(crate) agent_id: String,
    pub(crate) run_id: String,
    pub(crate) requires_verification: bool,
    pub(crate) mutation_revision: Option<u64>,
    pub(crate) verified_revision: Option<u64>,
    pub(crate) last_mutation_tool: Option<String>,
    pub(crate) last_verification_tool: Option<String>,
    pub(crate) last_verification_status: Option<String>,
    /// `last_verification_*` is deliberately the most recent verification so
    /// preview failures can drive the repair loop. Keep the static smoke
    /// result separately: a successful `preview.validate` must not erase the
    /// proof that the exact same revision passed `game.static_smoke`.
    #[serde(default)]
    pub(crate) static_smoke_verified_revision: Option<u64>,
    /// Bind the durable static-smoke credential to the exact entry bytes.
    /// A project revision alone is insufficient because files can be changed
    /// outside the Runtime mutation path without advancing that revision.
    #[serde(default)]
    pub(crate) static_smoke_verified_game_index_sha256: Option<String>,
    #[serde(default)]
    pub(crate) failed_playtest_revision: Option<u64>,
    pub(crate) updated_at: u64,
}

#[derive(Clone, Debug, Deserialize, Eq, PartialEq, Serialize)]
#[serde(rename_all = "camelCase")]
pub(crate) struct AgentRuntimePendingToolAction {
    pub(crate) schema_version: String,
    pub(crate) fingerprint_version: String,
    pub(crate) agent_id: String,
    pub(crate) task_id: String,
    pub(crate) session_id: String,
    pub(crate) run_id: String,
    pub(crate) source: String,
    #[serde(default = "default_agent_runtime_run_profile")]
    pub(crate) run_profile: String,
    #[serde(default)]
    pub(crate) run_profile_binding_fingerprint: String,
    pub(crate) task: String,
    #[serde(default)]
    pub(crate) goal_id: Option<String>,
    #[serde(default)]
    pub(crate) goal_revision: u64,
    #[serde(default)]
    pub(crate) goal_snapshot_fingerprint: String,
    pub(crate) loop_iteration: u32,
    pub(crate) action_index: u32,
    pub(crate) occurrence_nonce: u64,
    pub(crate) thinking_summary: String,
    pub(crate) plan: Vec<String>,
    pub(crate) fallback_response: String,
    pub(crate) observations: Vec<AgentRuntimeToolObservation>,
    pub(crate) project_revision_before: AgentRuntimeProjectRevision,
    pub(crate) verification_gate_before: AgentRuntimeVerificationGate,
    pub(crate) planned_repository_context_fingerprint: String,
    #[serde(default)]
    pub(crate) planned_steer_cursor: u64,
    pub(crate) action: AgentRuntimeToolAction,
    pub(crate) action_id: String,
    pub(crate) action_fingerprint: String,
    pub(crate) input_summary: Option<String>,
    #[serde(default)]
    pub(crate) execution_mode: String,
    pub(crate) status: String,
    pub(crate) observation: Option<AgentRuntimeToolObservation>,
    pub(crate) created_at: u64,
    pub(crate) updated_at: u64,
}