Initial public release of Warp.

Repo-Sync-Origin: warpdotdev/warp-internal@12af1d983b

crates/graphql/src/api/mutations/update_agent_task.rs

@@ -0,0 +1,60 @@
+use crate::{
+    ai::{AgentTaskState, PlatformErrorCode},
+    error::UserFacingError,
+    request_context::RequestContext,
+    response_context::ResponseContext,
+    schema,
+};
+
+#[derive(cynic::QueryVariables, Debug)]
+pub struct UpdateAgentTaskVariables {
+    pub input: UpdateAgentTaskInput,
+    pub request_context: RequestContext,
+}
+
+#[derive(cynic::InputObject, Debug)]
+pub struct UpdateAgentTaskInput {
+    pub task_id: cynic::Id,
+
+    // Important: for our server-side changeset logic to work, any fields which we aren't trying to
+    // update must be omitted, NOT set to null.
+    #[cynic(skip_serializing_if = "Option::is_none")]
+    pub task_state: Option<AgentTaskState>,
+    #[cynic(skip_serializing_if = "Option::is_none")]
+    pub session_id: Option<cynic::Id>,
+    #[cynic(skip_serializing_if = "Option::is_none")]
+    pub conversation_id: Option<cynic::Id>,
+    #[cynic(skip_serializing_if = "Option::is_none")]
+    pub status_message: Option<AgentTaskStatusMessageInput>,
+}
+
+#[derive(cynic::InputObject, Debug)]
+pub struct AgentTaskStatusMessageInput {
+    #[cynic(skip_serializing_if = "Option::is_none")]
+    pub error_code: Option<PlatformErrorCode>,
+    pub message: String,
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+#[cynic(graphql_type = "RootMutation", variables = "UpdateAgentTaskVariables")]
+pub struct UpdateAgentTask {
+    #[arguments(input: $input, requestContext: $request_context)]
+    pub update_agent_task: UpdateAgentTaskResult,
+}
+
+crate::client::define_operation! {
+    update_agent_task(UpdateAgentTaskVariables) -> UpdateAgentTask;
+}
+
+#[derive(cynic::InlineFragments, Debug)]
+pub enum UpdateAgentTaskResult {
+    UpdateAgentTaskOutput(UpdateAgentTaskOutput),
+    UserFacingError(UserFacingError),
+    #[cynic(fallback)]
+    Unknown,
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+pub struct UpdateAgentTaskOutput {
+    pub response_context: ResponseContext,
+}