1

server-rs/crates/spacetime-client/src/custom_world.rs

@@ -460,6 +460,41 @@ impl SpacetimeClient {
         .await
     }
 
+    pub async fn upsert_custom_world_agent_operation_progress(
+        &self,
+        input: CustomWorldAgentOperationProgressRecordInput,
+    ) -> Result<CustomWorldAgentOperationRecord, SpacetimeClientError> {
+        let procedure_input = CustomWorldAgentOperationProgressInput {
+            session_id: input.session_id,
+            owner_user_id: input.owner_user_id,
+            operation_id: input.operation_id,
+            operation_type: parse_rpg_agent_operation_type_record(input.operation_type.as_str())?,
+            operation_status: parse_rpg_agent_operation_status_record(
+                input.operation_status.as_str(),
+            )?,
+            phase_label: input.phase_label,
+            phase_detail: input.phase_detail,
+            operation_progress: input.operation_progress,
+            error_message: input.error_message,
+            updated_at_micros: input.updated_at_micros,
+        };
+
+        self.call_after_connect(move |connection, sender| {
+            connection
+                .procedures()
+                .upsert_custom_world_agent_operation_progress_then(
+                    procedure_input,
+                    move |_, result| {
+                        let mapped = result
+                            .map_err(|error| SpacetimeClientError::Procedure(error.to_string()))
+                            .and_then(map_custom_world_agent_operation_procedure_result);
+                        send_once(&sender, mapped);
+                    },
+                );
+        })
+        .await
+    }
+
     pub async fn get_custom_world_agent_operation(
         &self,
         session_id: String,

server-rs/crates/spacetime-client/src/mapper.rs

@@ -2762,6 +2762,41 @@ pub(crate) fn format_rpg_agent_operation_status(
     }
 }
 
+pub(crate) fn parse_rpg_agent_operation_type_record(
+    value: &str,
+) -> Result<crate::module_bindings::RpgAgentOperationType, SpacetimeClientError> {
+    match value.trim() {
+        "process_message" => Ok(crate::module_bindings::RpgAgentOperationType::ProcessMessage),
+        "draft_foundation" => Ok(crate::module_bindings::RpgAgentOperationType::DraftFoundation),
+        "update_draft_card" => Ok(crate::module_bindings::RpgAgentOperationType::UpdateDraftCard),
+        "sync_result_profile" => {
+            Ok(crate::module_bindings::RpgAgentOperationType::SyncResultProfile)
+        }
+        "generate_characters" => {
+            Ok(crate::module_bindings::RpgAgentOperationType::GenerateCharacters)
+        }
+        "generate_landmarks" => {
+            Ok(crate::module_bindings::RpgAgentOperationType::GenerateLandmarks)
+        }
+        "generate_role_assets" => {
+            Ok(crate::module_bindings::RpgAgentOperationType::GenerateRoleAssets)
+        }
+        "sync_role_assets" => Ok(crate::module_bindings::RpgAgentOperationType::SyncRoleAssets),
+        "generate_scene_assets" => {
+            Ok(crate::module_bindings::RpgAgentOperationType::GenerateSceneAssets)
+        }
+        "sync_scene_assets" => Ok(crate::module_bindings::RpgAgentOperationType::SyncSceneAssets),
+        "expand_long_tail" => Ok(crate::module_bindings::RpgAgentOperationType::ExpandLongTail),
+        "publish_world" => Ok(crate::module_bindings::RpgAgentOperationType::PublishWorld),
+        "revert_checkpoint" => Ok(crate::module_bindings::RpgAgentOperationType::RevertCheckpoint),
+        "delete_characters" => Ok(crate::module_bindings::RpgAgentOperationType::DeleteCharacters),
+        "delete_landmarks" => Ok(crate::module_bindings::RpgAgentOperationType::DeleteLandmarks),
+        other => Err(SpacetimeClientError::Runtime(format!(
+            "未知 rpg agent operation type: {other}"
+        ))),
+    }
+}
+
 pub(crate) fn parse_rpg_agent_operation_status_record(
     value: &str,
 ) -> Result<crate::module_bindings::RpgAgentOperationStatus, SpacetimeClientError> {
@@ -3710,6 +3745,20 @@ pub struct CustomWorldAgentMessageFinalizeRecordInput {
     pub updated_at_micros: i64,
 }
 
+#[derive(Clone, Debug, PartialEq, Eq)]
+pub struct CustomWorldAgentOperationProgressRecordInput {
+    pub session_id: String,
+    pub owner_user_id: String,
+    pub operation_id: String,
+    pub operation_type: String,
+    pub operation_status: String,
+    pub phase_label: String,
+    pub phase_detail: String,
+    pub operation_progress: u32,
+    pub error_message: Option<String>,
+    pub updated_at_micros: i64,
+}
+
 #[derive(Clone, Debug, PartialEq, Eq)]
 pub struct CustomWorldAgentActionExecuteRecordInput {
     pub session_id: String,

server-rs/crates/spacetime-client/src/module_bindings/mod.rs

@@ -115,6 +115,7 @@ pub mod custom_world_agent_message_snapshot_type;
 pub mod custom_world_agent_message_submit_input_type;
 pub mod custom_world_agent_operation_type;
 pub mod custom_world_agent_operation_get_input_type;
+pub mod custom_world_agent_operation_progress_input_type;
 pub mod custom_world_agent_operation_procedure_result_type;
 pub mod custom_world_agent_operation_progress_input_type;
 pub mod custom_world_agent_operation_snapshot_type;
@@ -342,6 +343,7 @@ pub mod start_ai_task_reducer;
 pub mod start_ai_task_stage_reducer;
 pub mod turn_in_quest_reducer;
 pub mod unpublish_custom_world_profile_reducer;
+pub mod upsert_custom_world_agent_operation_progress_procedure;
 pub mod upsert_chapter_progression_reducer;
 pub mod upsert_custom_world_profile_reducer;
 pub mod upsert_npc_state_reducer;
@@ -587,6 +589,7 @@ pub use custom_world_agent_message_snapshot_type::CustomWorldAgentMessageSnapsho
 pub use custom_world_agent_message_submit_input_type::CustomWorldAgentMessageSubmitInput;
 pub use custom_world_agent_operation_type::CustomWorldAgentOperation;
 pub use custom_world_agent_operation_get_input_type::CustomWorldAgentOperationGetInput;
+pub use custom_world_agent_operation_progress_input_type::CustomWorldAgentOperationProgressInput;
 pub use custom_world_agent_operation_procedure_result_type::CustomWorldAgentOperationProcedureResult;
 pub use custom_world_agent_operation_progress_input_type::CustomWorldAgentOperationProgressInput;
 pub use custom_world_agent_operation_snapshot_type::CustomWorldAgentOperationSnapshot;
@@ -856,6 +859,7 @@ pub use start_ai_task_reducer::start_ai_task;
 pub use start_ai_task_stage_reducer::start_ai_task_stage;
 pub use turn_in_quest_reducer::turn_in_quest;
 pub use unpublish_custom_world_profile_reducer::unpublish_custom_world_profile;
+pub use upsert_custom_world_agent_operation_progress_procedure::upsert_custom_world_agent_operation_progress;
 pub use upsert_chapter_progression_reducer::upsert_chapter_progression;
 pub use upsert_custom_world_profile_reducer::upsert_custom_world_profile;
 pub use upsert_npc_state_reducer::upsert_npc_state;