first pass of merging in warp (doesn't build)

crates/graphql/src/api/ai.rs

@@ -2,6 +2,7 @@ use crate::object::ObjectMetadata;
 use crate::object_permissions::ObjectPermissions;
 use crate::scalars::Time;
 use crate::schema;
+use crate::user::PublicUserProfile;
 
 #[derive(cynic::Enum, Clone, Copy, Debug)]
 pub enum RequestLimitRefreshDuration {
@@ -123,6 +124,7 @@ pub enum AgentHarness {
     Oz,
     ClaudeCode,
     Gemini,
+    Codex,
     #[cynic(fallback)]
     Other(String),
 }
@@ -148,6 +150,7 @@ pub struct AIConversation {
     pub working_directory: Option<String>,
     pub usage: ConversationUsage,
     pub metadata: ObjectMetadata,
+    pub creator: Option<PublicUserProfile>,
     pub permissions: ObjectPermissions,
     pub ambient_agent_task_id: Option<cynic::Id>,
     pub artifacts: Option<Vec<AIConversationArtifact>>,
@@ -164,6 +167,48 @@ pub struct ConversationUsage {
 #[derive(cynic::QueryFragment, Debug, Clone)]
 pub struct ConversationUsageMetadata {
     pub context_window_usage: f64,
+    pub context_window_segments: Vec<ContextWindowSegment>,
     pub credits_spent: f64,
+    pub platform_credits_spent: f64,
     pub summarized: bool,
 }
+
+#[derive(cynic::Enum, Clone, Copy, Debug, PartialEq, Eq)]
+pub enum ContextWindowSegmentType {
+    Unknown,
+    SystemPrompt,
+    ToolDefinitions,
+    ConversationHistory,
+    LatestInput,
+    Images,
+    Other,
+}
+
+#[derive(cynic::QueryFragment, Debug, Clone)]
+pub struct ContextWindowSegment {
+    pub segment_type: ContextWindowSegmentType,
+    pub token_count: i32,
+}
+
+impl From<ContextWindowSegmentType> for persistence::model::ContextWindowSegmentType {
+    fn from(value: ContextWindowSegmentType) -> Self {
+        match value {
+            ContextWindowSegmentType::Unknown => Self::Unknown,
+            ContextWindowSegmentType::SystemPrompt => Self::SystemPrompt,
+            ContextWindowSegmentType::ToolDefinitions => Self::ToolDefinitions,
+            ContextWindowSegmentType::ConversationHistory => Self::ConversationHistory,
+            ContextWindowSegmentType::LatestInput => Self::LatestInput,
+            ContextWindowSegmentType::Images => Self::Images,
+            ContextWindowSegmentType::Other => Self::Other,
+        }
+    }
+}
+
+impl From<&ContextWindowSegment> for persistence::model::ContextWindowSegment {
+    fn from(gql: &ContextWindowSegment) -> Self {
+        Self {
+            segment_type: gql.segment_type.into(),
+            token_count: u32::try_from(gql.token_count).unwrap_or_default(),
+        }
+    }
+}

crates/graphql/src/api/billing.rs

@@ -1,4 +1,6 @@
-use crate::{scalars::Time, schema, workspace::UgcCollectionEnablementSetting};
+use crate::scalars::Time;
+use crate::schema;
+use crate::workspace::UgcCollectionEnablementSetting;
 
 #[derive(cynic::QueryFragment, Debug, Clone)]
 pub struct BillingMetadata {
@@ -109,6 +111,7 @@ pub struct Tier {
     pub enterprise_credits_auto_reload_policy: Option<EnterpriseCreditsAutoReloadPolicy>,
     pub multi_admin_policy: Option<MultiAdminPolicy>,
     pub ambient_agents_policy: Option<AmbientAgentsPolicy>,
+    pub usage_visibility_policy: Option<UsageVisibilityPolicy>,
 }
 
 #[derive(cynic::QueryFragment, Debug, Clone)]
@@ -147,6 +150,7 @@ pub struct WarpAiPolicy {
     pub is_code_suggestions_toggleable: bool,
     pub is_prompt_suggestions_toggleable: bool,
     pub is_next_command_enabled: bool,
+    pub is_git_operations_ai_enabled: bool,
     pub is_voice_enabled: bool,
 }
 
@@ -294,3 +298,90 @@ pub enum StripeSubscriptionPlan {
     #[cynic(fallback)]
     Other(String),
 }
+
+#[derive(cynic::QueryFragment, Debug, Clone)]
+pub struct UsageVisibilityPolicy {
+    pub admin_granularity: UsageVisibilityGranularity,
+    pub max_prior_cycles: i32,
+}
+
+#[derive(cynic::Enum, Clone, Debug, PartialEq, Eq)]
+pub enum UsageVisibilityGranularity {
+    OwnOnly,
+    TeamAggregate,
+    PerUserTotals,
+    FullBreakdown,
+    #[cynic(fallback)]
+    Other(String),
+}
+
+#[derive(cynic::QueryFragment, Debug, Clone)]
+pub struct BillingCycleUsageHistory {
+    pub current_period_start: Time,
+    pub current_period_end: Time,
+    pub summaries: Vec<BillingCycleUsageSummary>,
+}
+
+#[derive(cynic::QueryFragment, Debug, Clone)]
+pub struct BillingCycleUsageSummary {
+    pub period_start: Time,
+    pub period_end: Time,
+    pub entries: Vec<UsageEntry>,
+}
+
+#[derive(cynic::QueryFragment, Debug, Clone)]
+pub struct UsageEntry {
+    pub subject_type: AiCreditsUsageAndCostSubjectType,
+    pub subject_uid: Option<String>,
+    pub subject_display_name: Option<String>,
+    pub cost_type: AiCreditsUsageAndCostType,
+    pub usage_bucket: AiCreditsUsageBucket,
+    pub usage_source: AiCreditsUsageSource,
+    pub credits_used: i32,
+    pub cost_cents: i32,
+}
+
+#[derive(cynic::Enum, Clone, Debug, PartialEq, Eq)]
+#[cynic(graphql_type = "AICreditsUsageAndCostSubjectType")]
+pub enum AiCreditsUsageAndCostSubjectType {
+    Team,
+    User,
+    ServiceAccount,
+    #[cynic(fallback)]
+    Other(String),
+}
+
+#[derive(cynic::Enum, Clone, Debug, PartialEq, Eq)]
+#[cynic(graphql_type = "AICreditsUsageAndCostType")]
+pub enum AiCreditsUsageAndCostType {
+    BaseLimit,
+    BonusGrant,
+    Payg,
+    AmbientBonusGrant,
+    Aggregate,
+    #[cynic(fallback)]
+    Other(String),
+}
+
+#[derive(cynic::Enum, Clone, Debug, PartialEq, Eq)]
+#[cynic(graphql_type = "AICreditsUsageBucket")]
+pub enum AiCreditsUsageBucket {
+    Ai,
+    Compute,
+    Platform,
+    SuggestedCodeDiffs,
+    Voice,
+    Aggregate,
+    #[cynic(fallback)]
+    Other(String),
+}
+
+#[derive(cynic::Enum, Clone, Debug, PartialEq, Eq)]
+#[cynic(graphql_type = "AICreditsUsageSource")]
+pub enum AiCreditsUsageSource {
+    Local,
+    Cloud,
+    Aggregate,
+    #[cynic(fallback)]
+    Other(String),
+}

crates/graphql/src/api/error.rs

@@ -1,5 +1,6 @@
 use super::object::ObjectType;
-use crate::{response_context::ResponseContext, schema};
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryFragment, Debug)]
 pub struct UserFacingError {

crates/graphql/src/api/experiment.rs

@@ -67,6 +67,8 @@ pub enum Experiment {
     FreeUserInitialCreditsThreeHundred,
     OzMultiHarnessControl,
     OzMultiHarnessExperiment,
+    SshRemoteServerControl,
+    SshRemoteServerExperiment,
     #[cynic(fallback)]
     Other(String),
 }

crates/graphql/src/api/folder.rs

@@ -1,8 +1,6 @@
-use crate::{
-    object::{CloudObject, ObjectMetadata},
-    object_permissions::ObjectPermissions,
-    schema,
-};
+use crate::object::{CloudObject, ObjectMetadata};
+use crate::object_permissions::ObjectPermissions;
+use crate::schema;
 
 #[derive(cynic::QueryFragment, Debug, Clone)]
 pub struct Folder {

crates/graphql/src/api/full_source_code_embedding.rs

@@ -13,6 +13,8 @@ pub enum EmbeddingConfig {
     Voyage35512,
     #[cynic(rename = "VOYAGE_3_5_LITE_512")]
     Voyage35Lite512,
+    #[cynic(rename = "VOYAGE_4_512")]
+    Voyage4512,
 }
 
 #[derive(cynic::Scalar, Debug, Clone)]

crates/graphql/src/api/generic_string_object.rs

@@ -1,7 +1,5 @@
-use super::{
-    object::{CloudObjectEventEntrypoint, ObjectMetadata},
-    object_permissions::ObjectPermissions,
-};
+use super::object::{CloudObjectEventEntrypoint, ObjectMetadata};
+use super::object_permissions::ObjectPermissions;
 use crate::schema;
 
 #[derive(cynic::QueryFragment, Debug, Clone)]
@@ -32,6 +30,12 @@ pub enum GenericStringObjectFormat {
     JsonCloudEnvironment,
     #[cynic(rename = "JsonScheduledAmbientAgent")]
     JsonScheduledAmbientAgent,
+    /// Fallback for GSO formats this client build does not recognize (for example
+    /// server-only formats such as `JsonRunner`). Without this, decoding a Drive
+    /// sync response that contains an unknown format fails for the entire response.
+    /// This variant only arises when deserializing; we never serialize it.
+    #[cynic(fallback)]
+    Unknown,
 }
 
 #[derive(cynic::InputObject, Debug)]
@@ -58,6 +62,7 @@ impl std::fmt::Display for GenericStringObjectFormat {
             GenericStringObjectFormat::JsonTemplatableMCPServer => "JsonTemplatableMCPServer",
             GenericStringObjectFormat::JsonCloudEnvironment => "JsonCloudEnvironment",
             GenericStringObjectFormat::JsonScheduledAmbientAgent => "JsonScheduledAmbientAgent",
+            GenericStringObjectFormat::Unknown => "Unknown",
         };
         write!(f, "{s}")
     }

crates/graphql/src/api/mod.rs

@@ -19,8 +19,8 @@ pub mod user;
 pub mod workflow;
 pub mod workspace;
 
+use cynic::impl_scalar;
 // Re-export the schema that's generated by the schema crate.
 pub use galaxy_graphql_schema::schema;
-
-use cynic::impl_scalar;
 impl_scalar!(crate::scalars::Time, schema::Time);
+impl_scalar!(crate::scalars::Uint32, schema::Uint);

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation addInviteLinkDomainRestriction($requestContext: RequestContext!, $input: AddInviteLinkDomainRestrictionInput!) {

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

@@ -1,11 +1,9 @@
-use crate::{
-    error::UserFacingError,
-    object_permissions::{AccessLevel, ObjectPermissions},
-    request_context::RequestContext,
-    response_context::ResponseContext,
-    schema,
-    user::PublicUserProfile,
-};
+use crate::error::UserFacingError;
+use crate::object_permissions::{AccessLevel, ObjectPermissions};
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
+use crate::user::PublicUserProfile;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct AddObjectGuestsVariables {

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

@@ -1,9 +1,10 @@
 use super::create_generic_string_object::CreateGenericStringObjectOutput;
-use crate::{
-    error::UserFacingError, generic_string_object::GenericStringObjectInput,
-    object_permissions::Owner, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::generic_string_object::GenericStringObjectInput;
+use crate::object_permissions::Owner;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct BulkCreateObjectsVariables {

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

@@ -1,8 +1,8 @@
 use super::create_file_artifact_upload_target::FileArtifact;
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct ConfirmFileArtifactUploadVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct CreateAgentTaskVariables {

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

@@ -1,8 +1,8 @@
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
 use crate::scalars::Time;
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct CreateAnonymousUserVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct CreateFileArtifactUploadTargetVariables {
@@ -53,6 +53,7 @@ pub struct FileArtifactUploadTarget {
     pub url: String,
     pub method: String,
     pub headers: Vec<FileArtifactUploadHeader>,
+    pub fields: Vec<FileArtifactUploadField>,
 }
 
 #[derive(cynic::QueryFragment, Debug)]
@@ -62,6 +63,51 @@ pub struct FileArtifactUploadHeader {
     pub value: String,
 }
 
+/// A single multipart form field for a presigned POST upload target.
+#[derive(cynic::QueryFragment, Debug)]
+#[cynic(graphql_type = "UploadField")]
+pub struct FileArtifactUploadField {
+    pub name: String,
+    pub value: FileArtifactUploadFieldValue,
+}
+
+/// The value of a multipart form field on a presigned POST upload target.
+#[derive(cynic::InlineFragments, Debug)]
+#[cynic(graphql_type = "UploadFieldValue")]
+pub enum FileArtifactUploadFieldValue {
+    StaticUploadFieldValue(StaticUploadFieldValue),
+    ContentCRC32CFieldValue(ContentCRC32CFieldValue),
+    ContentDataFieldValue(ContentDataFieldValue),
+    #[cynic(fallback)]
+    Unknown,
+}
+
+/// Literal string value known at URL-generation time.
+#[derive(cynic::QueryFragment, Debug)]
+pub struct StaticUploadFieldValue {
+    pub value: String,
+}
+
+/// Signals the client to compute CRC32C of the upload, base64-encode the 4-byte
+/// big-endian result, and send it as the field's value. The GraphQL type
+/// carries no payload beyond a placeholder `_: Boolean`; the client only cares
+/// about the variant tag, but cynic's `QueryFragment` derive requires at least
+/// one field to select.
+#[derive(cynic::QueryFragment, Debug, Default)]
+pub struct ContentCRC32CFieldValue {
+    #[cynic(rename = "_")]
+    _placeholder: Option<bool>,
+}
+
+/// Signals the client to use the raw upload bytes as the field's value. Must
+/// be the final entry in `fields`. See [`ContentCRC32CFieldValue`] for why we
+/// select a placeholder `_: Boolean`.
+#[derive(cynic::QueryFragment, Debug, Default)]
+pub struct ContentDataFieldValue {
+    #[cynic(rename = "_")]
+    _placeholder: Option<bool>,
+}
+
 #[derive(cynic::InputObject, Debug)]
 #[cynic(graphql_type = "CreateFileArtifactUploadTargetInput")]
 pub struct CreateFileArtifactUploadTargetInput {

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

@@ -1,7 +1,9 @@
-use crate::{
-    error::UserFacingError, folder::Folder, object_permissions::Owner,
-    request_context::RequestContext, response_context::ResponseContext, schema,
-};
+use crate::error::UserFacingError;
+use crate::folder::Folder;
+use crate::object_permissions::Owner;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct CreateFolderVariables {

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

@@ -1,12 +1,10 @@
+use crate::error::UserFacingError;
+use crate::generic_string_object::{GenericStringObject, GenericStringObjectInput};
+use crate::object_permissions::Owner;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
 use crate::scalars::Time;
-use crate::{
-    error::UserFacingError,
-    generic_string_object::{GenericStringObject, GenericStringObjectInput},
-    object_permissions::Owner,
-    request_context::RequestContext,
-    response_context::ResponseContext,
-    schema,
-};
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct CreateGenericStringObjectVariables {

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

@@ -0,0 +1,59 @@
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::scalars::Time;
+use crate::schema;
+
+#[derive(cynic::QueryVariables, Debug)]
+pub struct CreateManagedMcpClientConfigVariables {
+    pub input: CreateManagedMcpClientConfigInput,
+    pub request_context: RequestContext,
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+#[cynic(
+    graphql_type = "RootMutation",
+    variables = "CreateManagedMcpClientConfigVariables"
+)]
+pub struct CreateManagedMcpClientConfig {
+    #[arguments(input: $input, requestContext: $request_context)]
+    pub create_managed_mcp_client_config: CreateManagedMcpClientConfigResult,
+}
+
+crate::client::define_operation! {
+    create_managed_mcp_client_config(CreateManagedMcpClientConfigVariables) -> CreateManagedMcpClientConfig;
+}
+
+#[derive(cynic::InputObject, Debug)]
+pub struct CreateManagedMcpClientConfigInput {
+    pub uid: cynic::Id,
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+pub struct CreateManagedMcpClientConfigOutput {
+    pub transport_kind: ManagedMcpTransportKind,
+    pub mcp_config_json: String,
+    pub proxy_url: Option<String>,
+    pub proxy_token: Option<String>,
+    pub authorization_header_name: Option<String>,
+    pub authorization_header_value: Option<String>,
+    pub expires_at: Option<Time>,
+    pub response_context: ResponseContext,
+}
+
+#[derive(cynic::InlineFragments, Debug)]
+#[allow(clippy::large_enum_variant)]
+pub enum CreateManagedMcpClientConfigResult {
+    CreateManagedMcpClientConfigOutput(CreateManagedMcpClientConfigOutput),
+    UserFacingError(UserFacingError),
+    #[cynic(fallback)]
+    Unknown,
+}
+
+#[derive(cynic::Enum, Clone, Copy, Debug, Eq, PartialEq)]
+pub enum ManagedMcpTransportKind {
+    #[cynic(rename = "URL")]
+    Url,
+    #[cynic(rename = "COMMAND")]
+    Command,
+}

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

@@ -1,11 +1,9 @@
-use crate::{
-    error::UserFacingError,
-    managed_secrets::{ManagedSecret, ManagedSecretType},
-    object_permissions::Owner,
-    request_context::RequestContext,
-    response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::managed_secrets::{ManagedSecret, ManagedSecretType};
+use crate::object_permissions::Owner;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct CreateManagedSecretVariables {

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

@@ -1,9 +1,11 @@
+use crate::error::UserFacingError;
+use crate::notebook::Notebook;
+use crate::object::CloudObjectEventEntrypoint;
+use crate::object_permissions::Owner;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
 use crate::scalars::Time;
-use crate::{
-    error::UserFacingError, notebook::Notebook, object::CloudObjectEventEntrypoint,
-    object_permissions::Owner, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct CreateNotebookVariables {

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

@@ -1,4 +1,6 @@
-use crate::{error::UserFacingError, request_context::RequestContext, schema};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct CreateSimpleIntegrationVariables {

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

@@ -1,7 +1,9 @@
-use crate::{
-    error::UserFacingError, object::CloudObjectEventEntrypoint, request_context::RequestContext,
-    response_context::ResponseContext, schema, workspace::Workspace,
-};
+use crate::error::UserFacingError;
+use crate::object::CloudObjectEventEntrypoint;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
+use crate::workspace::Workspace;
 
 /*
 mutation CreateTeam($input: CreateTeamInput!, $request_context: RequestContext!) {
@@ -35,6 +37,7 @@ mutation CreateTeam($input: CreateTeamInput!, $request_context: RequestContext!)
               isCodeSuggestionsToggleable
               isPromptSuggestionsToggleable
               isNextCommandEnabled
+              isGitOperationsAiEnabled
               isVoiceEnabled
             }
             teamSizePolicy {

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

@@ -1,8 +1,11 @@
+use crate::error::UserFacingError;
+use crate::object::CloudObjectEventEntrypoint;
+use crate::object_permissions::Owner;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
 use crate::scalars::Time;
-use crate::{
-    error::UserFacingError, object::CloudObjectEventEntrypoint, object_permissions::Owner,
-    request_context::RequestContext, response_context::ResponseContext, schema, workflow::Workflow,
-};
+use crate::schema;
+use crate::workflow::Workflow;
 
 /*
 mutation CreateWorkflow($input: CreateWorkflowInput!, $requestContext: RequestContext!) {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct DeleteAIConversationVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation DeleteInviteLinkDomainRestriction($input:DeleteInviteLinkDomainRestrictionInput!, $request_context:RequestContext!) {

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

@@ -1,7 +1,8 @@
-use crate::{
-    error::UserFacingError, object_permissions::Owner, request_context::RequestContext,
-    response_context::ResponseContext, schema,
-};
+use crate::error::UserFacingError;
+use crate::object_permissions::Owner;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct DeleteManagedSecretVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct DeleteObjectVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation DeleteTeamInvite(

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

@@ -1,7 +1,8 @@
-use crate::{
-    error::UserFacingError, object_permissions::Owner, request_context::RequestContext,
-    response_context::ResponseContext, schema,
-};
+use crate::error::UserFacingError;
+use crate::object_permissions::Owner;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation EmptyTrash($input: EmptyTrashInput!, $requestContext: RequestContext!) {

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

@@ -1,4 +1,6 @@
-use crate::{error::UserFacingError, request_context::RequestContext, schema};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct ExpireApiKeyVariables {

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

@@ -1,10 +1,9 @@
-use crate::scalars::Time;
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
-
+use crate::error::UserFacingError;
 use crate::queries::api_keys::ApiKeyProperties;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::scalars::Time;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct GenerateApiKeyVariables {
@@ -16,6 +15,7 @@ pub struct GenerateApiKeyVariables {
 pub struct GenerateApiKeyInput {
     pub name: String,
     pub team_id: Option<cynic::Id>,
+    pub agent_uid: Option<cynic::Id>,
     pub expires_at: Option<Time>,
 }
 

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

@@ -1,10 +1,10 @@
-use crate::{
-    error::UserFacingError,
-    full_source_code_embedding::{ContentHash, EmbeddingConfig, Fragment, NodeHash, RepoMetadata},
-    request_context::RequestContext,
-    response_context::ResponseContext,
-    schema,
+use crate::error::UserFacingError;
+use crate::full_source_code_embedding::{
+    ContentHash, EmbeddingConfig, Fragment, NodeHash, RepoMetadata,
 };
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct GenerateCodeEmbeddingsVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation GenerateCommands($input: GenerateCommandsInput!, $requestContext: RequestContext!) {

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

@@ -1,7 +1,8 @@
-use crate::{
-    ai::RequestLimitInfo, error::UserFacingError, request_context::RequestContext,
-    response_context::ResponseContext, schema,
-};
+use crate::ai::RequestLimitInfo;
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation GenerateDialogue($input: GenerateDialogueInput!, $requestContext: RequestContext!) {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation GenerateMetadataForCommand($input: GenerateMetadataForCommandInput!, $requestContext: RequestContext!) {

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

@@ -1,8 +1,6 @@
-use crate::{
-    notebook::{UpdateNotebookEditAccessInput, UpdateNotebookEditAccessResult},
-    request_context::RequestContext,
-    schema,
-};
+use crate::notebook::{UpdateNotebookEditAccessInput, UpdateNotebookEditAccessResult};
+use crate::request_context::RequestContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct GiveUpNotebookEditAccessVariables {

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

@@ -1,8 +1,6 @@
-use crate::{
-    notebook::{UpdateNotebookEditAccessInput, UpdateNotebookEditAccessResult},
-    request_context::RequestContext,
-    schema,
-};
+use crate::notebook::{UpdateNotebookEditAccessInput, UpdateNotebookEditAccessResult};
+use crate::request_context::RequestContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct GrabNotebookEditAccessVariables {

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

@@ -1,7 +1,8 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    scalars::Time, schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::scalars::Time;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct IssueTaskIdentityTokenVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation JoinTeamWithTeamDiscovery($input: JoinTeamWithTeamDiscoveryInput!, $requestContext: RequestContext!) {

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

@@ -1,4 +1,6 @@
-use crate::{error::UserFacingError, request_context::RequestContext, schema};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct LeaveObjectVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct MintCustomTokenVariables {

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

@@ -7,6 +7,7 @@ pub mod create_anonymous_user;
 pub mod create_file_artifact_upload_target;
 pub mod create_folder;
 pub mod create_generic_string_object;
+pub mod create_managed_mcp_client_config;
 pub mod create_managed_secret;
 pub mod create_notebook;
 pub mod create_simple_integration;

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

@@ -1,11 +1,9 @@
-use crate::{
-    error::UserFacingError,
-    object::{ObjectMetadata, ObjectType},
-    object_permissions::Owner,
-    request_context::RequestContext,
-    response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::object::{ObjectMetadata, ObjectType};
+use crate::object_permissions::Owner;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct MoveObjectVariables {

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

@@ -1,9 +1,7 @@
-use crate::{
-    error::UserFacingError,
-    full_source_code_embedding::{EmbeddingConfig, NodeHash, RepoMetadata},
-    request_context::RequestContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::full_source_code_embedding::{EmbeddingConfig, NodeHash, RepoMetadata};
+use crate::request_context::RequestContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct PopulateMerkleTreeCacheVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::InputObject, Debug)]
 pub struct PurchaseAddonCreditsInput {

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

@@ -1,11 +1,9 @@
-use crate::{
-    error::UserFacingError,
-    object_actions::{ActionType, ObjectActionHistory},
-    request_context::RequestContext,
-    response_context::ResponseContext,
-    scalars::Time,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::object_actions::{ActionType, ObjectActionHistory};
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::scalars::Time;
+use crate::schema;
 
 /*
 mutation RecordObjectAction($input: RecordObjectActionInput!, $requestContext: RequestContext!) {

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

@@ -1,7 +1,8 @@
-use crate::{
-    error::UserFacingError, object_permissions::ObjectPermissions, request_context::RequestContext,
-    response_context::ResponseContext, schema,
-};
+use crate::error::UserFacingError;
+use crate::object_permissions::ObjectPermissions;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct RemoveObjectGuestVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct RemoveObjectLinkPermissionsVariables {

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

@@ -1,7 +1,8 @@
-use crate::{
-    error::UserFacingError, object::CloudObjectEventEntrypoint, request_context::RequestContext,
-    response_context::ResponseContext, schema,
-};
+use crate::error::UserFacingError;
+use crate::object::CloudObjectEventEntrypoint;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation RemoveUserFromTeam($input: RemoveUserFromTeamInput!, $requestContext: RequestContext!) {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation RenameTeam($input: RenameTeamInput!, $requestContext: RequestContext!) {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation ProvideNegativeFeedbackResponseForAiConversation

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation ResetInviteLinks($input: ResetInviteLinksInput!, $requestContext: RequestContext!) {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation SendReferralInviteEmails($input: SendReferralInviteEmailsInput!, $requestContext: RequestContext!) {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation SendTeamInviteEmail($input: SendTeamInviteEmailInput!, $requestContext: RequestContext!) {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation SetIsInviteLinkEnabled($input: SetIsInviteLinkEnabledInput!, $requestContext: RequestContext!) {

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

@@ -1,7 +1,8 @@
-use crate::{
-    error::UserFacingError, object_permissions::AccessLevel, request_context::RequestContext,
-    response_context::ResponseContext, schema,
-};
+use crate::error::UserFacingError;
+use crate::object_permissions::AccessLevel;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct SetObjectLinkPermissionsVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation SetTeamDiscoverability($input: SetTeamDiscoverabilityInput!, $requestContext: RequestContext!) {

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

@@ -1,7 +1,8 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema, workspace::MembershipRole,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
+use crate::workspace::MembershipRole;
 
 /*
 mutation SetTeamMemberRole($input: SetTeamMemberRoleInput!, $requestContext: RequestContext!) {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation SetUserIsOnboarded($requestContext: RequestContext!) {

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

@@ -1,8 +1,8 @@
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
 use crate::scalars::Time;
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct ShareBlockVariables<'a> {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation stripeBillingPortal($input: StripeBillingPortalInput!, $requestContext: RequestContext!) {

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

@@ -1,7 +1,9 @@
-use crate::{
-    error::UserFacingError, object::ObjectMetadata, object_permissions::Owner,
-    request_context::RequestContext, response_context::ResponseContext, schema,
-};
+use crate::error::UserFacingError;
+use crate::object::ObjectMetadata;
+use crate::object_permissions::Owner;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct TransferGenericStringObjectOwnerVariables {

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

@@ -1,7 +1,9 @@
-use crate::{
-    error::UserFacingError, object::ObjectMetadata, object_permissions::Owner,
-    request_context::RequestContext, response_context::ResponseContext, schema,
-};
+use crate::error::UserFacingError;
+use crate::object::ObjectMetadata;
+use crate::object_permissions::Owner;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct TransferNotebookOwnerVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation TransferTeamOwnership($input: TransferTeamOwnershipInput!, $requestContext: RequestContext!) {

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

@@ -1,7 +1,9 @@
-use crate::{
-    error::UserFacingError, object::ObjectMetadata, object_permissions::Owner,
-    request_context::RequestContext, response_context::ResponseContext, schema,
-};
+use crate::error::UserFacingError;
+use crate::object::ObjectMetadata;
+use crate::object_permissions::Owner;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct TransferWorkflowOwnerVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct TrashObjectVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct UnshareBlockVariables {

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

@@ -1,7 +1,8 @@
-use crate::{
-    error::UserFacingError, object::ObjectMetadata, request_context::RequestContext,
-    response_context::ResponseContext, schema,
-};
+use crate::error::UserFacingError;
+use crate::object::ObjectMetadata;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct UntrashObjectVariables {

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

@@ -1,10 +1,8 @@
-use crate::{
-    ai::{AgentTaskState, PlatformErrorCode},
-    error::UserFacingError,
-    request_context::RequestContext,
-    response_context::ResponseContext,
-    schema,
-};
+use crate::ai::{AgentTaskState, PlatformErrorCode};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct UpdateAgentTaskVariables {

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

@@ -1,7 +1,8 @@
-use crate::{
-    error::UserFacingError, object::ObjectUpdateSuccess, request_context::RequestContext,
-    response_context::ResponseContext, schema,
-};
+use crate::error::UserFacingError;
+use crate::object::ObjectUpdateSuccess;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation UpdateFolder($input: UpdateFolderInput!, $requestContext: RequestContext!) {

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

@@ -1,9 +1,10 @@
+use crate::error::UserFacingError;
+use crate::generic_string_object::GenericStringObject;
+use crate::object::ObjectUpdateSuccess;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
 use crate::scalars::Time;
-use crate::{
-    error::UserFacingError, generic_string_object::GenericStringObject,
-    object::ObjectUpdateSuccess, request_context::RequestContext,
-    response_context::ResponseContext, schema,
-};
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct UpdateGenericStringObjectVariables {

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

@@ -1,7 +1,9 @@
-use crate::{
-    error::UserFacingError, managed_secrets::ManagedSecret, object_permissions::Owner,
-    request_context::RequestContext, response_context::ResponseContext, schema,
-};
+use crate::error::UserFacingError;
+use crate::managed_secrets::ManagedSecret;
+use crate::object_permissions::Owner;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct UpdateManagedSecretVariables {

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

@@ -1,10 +1,8 @@
-use crate::{
-    error::UserFacingError,
-    full_source_code_embedding::{EmbeddingConfig, NodeHash},
-    request_context::RequestContext,
-    response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::full_source_code_embedding::{EmbeddingConfig, NodeHash};
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct UpdateMerkleTreeVariables {

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

@@ -1,8 +1,10 @@
+use crate::error::UserFacingError;
+use crate::notebook::Notebook;
+use crate::object::ObjectUpdateSuccess;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
 use crate::scalars::Time;
-use crate::{
-    error::UserFacingError, notebook::Notebook, object::ObjectUpdateSuccess,
-    request_context::RequestContext, response_context::ResponseContext, schema,
-};
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct UpdateNotebookVariables {

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

@@ -1,10 +1,8 @@
-use crate::{
-    error::UserFacingError,
-    object_permissions::{AccessLevel, ObjectPermissions},
-    request_context::RequestContext,
-    response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::object_permissions::{AccessLevel, ObjectPermissions};
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct UpdateObjectGuestsVariables {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation updateOnboardingSurveyStatus($input: UpdateOnboardingSurveyStatusInput!, $clientContext: ClientContext!, $osContext: OSContext!) {

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

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 /*
 mutation UpdateUserSettings($input: UpdateUserSettingsInput!, $requestContext: RequestContext!) {

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

@@ -1,8 +1,10 @@
+use crate::error::UserFacingError;
+use crate::object::ObjectUpdateSuccess;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
 use crate::scalars::Time;
-use crate::{
-    error::UserFacingError, object::ObjectUpdateSuccess, request_context::RequestContext,
-    response_context::ResponseContext, schema, workflow::Workflow,
-};
+use crate::schema;
+use crate::workflow::Workflow;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct UpdateWorkflowVariables {

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

@@ -1,7 +1,8 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    schema, workspace::WorkspaceSettings,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
+use crate::workspace::WorkspaceSettings;
 
 // Note that `isInviteLinkEnabled` and `IsDiscoverable` aren't fetchable from this mutation;
 // they don't come from the same code path as fetching org settings. We could change this

crates/graphql/src/api/notebook.rs

@@ -1,7 +1,7 @@
-use super::{
-    error::UserFacingError, object::ObjectMetadata, object_permissions::ObjectPermissions,
-    response_context::ResponseContext,
-};
+use super::error::UserFacingError;
+use super::object::ObjectMetadata;
+use super::object_permissions::ObjectPermissions;
+use super::response_context::ResponseContext;
 use crate::schema;
 
 #[derive(cynic::QueryFragment, Debug, Clone)]

crates/graphql/src/api/object.rs

@@ -1,12 +1,10 @@
-use crate::{
-    ai::AIConversation,
-    folder::{Folder, FolderWithDescendants},
-    generic_string_object::GenericStringObject,
-    notebook::Notebook,
-    scalars::Time,
-    schema,
-    workflow::Workflow,
-};
+use crate::ai::AIConversation;
+use crate::folder::{Folder, FolderWithDescendants};
+use crate::generic_string_object::GenericStringObject;
+use crate::notebook::Notebook;
+use crate::scalars::Time;
+use crate::schema;
+use crate::workflow::Workflow;
 
 #[derive(cynic::QueryFragment, Debug, Clone)]
 pub struct ObjectMetadata {
@@ -43,6 +41,7 @@ pub struct ObjectUpdateSuccess {
     pub revision_ts: Time,
 }
 
+#[allow(clippy::large_enum_variant)]
 #[derive(cynic::InlineFragments, Debug, Clone)]
 pub enum CloudObject {
     AIConversation(AIConversation),
@@ -54,6 +53,7 @@ pub enum CloudObject {
     Unknown,
 }
 
+#[allow(clippy::large_enum_variant)]
 #[derive(cynic::InlineFragments, Debug, Clone)]
 pub enum CloudObjectWithDescendants {
     AIConversation(AIConversation),

crates/graphql/src/api/object_permissions.rs

@@ -1,3 +1,5 @@
+use std::fmt;
+
 use super::object::{Container, Space};
 use crate::scalars::Time;
 use crate::schema;
@@ -32,6 +34,16 @@ pub enum OwnerType {
     User,
 }
 
+impl fmt::Display for OwnerType {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        let owner_type = match self {
+            OwnerType::Team => "Team",
+            OwnerType::User => "Personal",
+        };
+        write!(f, "{owner_type}")
+    }
+}
+
 #[derive(cynic::QueryFragment, Debug, Clone)]
 pub struct UserGuest {
     pub firebase_uid: cynic::Id,

crates/graphql/src/api/queries/api_keys.rs

@@ -1,7 +1,9 @@
-use crate::{
-    api::object_permissions::OwnerType, error::UserFacingError, request_context::RequestContext,
-    response_context::ResponseContext, scalars::Time, schema,
-};
+use crate::api::object_permissions::OwnerType;
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::scalars::Time;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct ApiKeysVariables {
@@ -22,10 +24,17 @@ pub struct ApiKeyProperties {
     pub name: String,
     pub key_suffix: String,
     pub owner_type: OwnerType,
+    pub agent_info: Option<ApiKeyAgentInfo>,
     pub expires_at: Option<Time>,
     pub last_used_at: Option<Time>,
     pub created_at: Time,
 }
+#[derive(cynic::QueryFragment, Debug)]
+#[cynic(graphql_type = "APIKeyAgentInfo")]
+pub struct ApiKeyAgentInfo {
+    pub name: String,
+    pub uid: cynic::Id,
+}
 
 #[derive(cynic::InlineFragments, Debug)]
 #[cynic(graphql_type = "APIKeyPropertiesResult")]

crates/graphql/src/api/queries/codebase_context_config.rs

@@ -1,7 +1,7 @@
-use crate::{
-    error::UserFacingError, full_source_code_embedding::EmbeddingConfig,
-    request_context::RequestContext, schema,
-};
+use crate::error::UserFacingError;
+use crate::full_source_code_embedding::EmbeddingConfig;
+use crate::request_context::RequestContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct CodebaseContextConfigVariables {

crates/graphql/src/api/queries/free_available_models.rs

@@ -1,7 +1,7 @@
-use crate::{
-    api::queries::get_feature_model_choices::FeatureModelChoice, request_context::RequestContext,
-    response_context::ResponseContext, schema,
-};
+use crate::api::queries::get_feature_model_choices::FeatureModelChoice;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct FreeAvailableModelsVariables {

crates/graphql/src/api/queries/get_ai_conversation_format.rs

@@ -1,9 +1,9 @@
-use crate::{
-    ai::AIConversationFormat, error::UserFacingError, request_context::RequestContext,
-    response_context::ResponseContext, schema,
-};
-
 use super::list_ai_conversations::ListAIConversationsInput;
+use crate::ai::AIConversationFormat;
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct GetAIConversationFormatVariables {

crates/graphql/src/api/queries/get_ai_overages_for_workspace.rs

@@ -1,4 +1,6 @@
-use crate::{billing::BillingMetadata, request_context::RequestContext, schema};
+use crate::billing::BillingMetadata;
+use crate::request_context::RequestContext;
+use crate::schema;
 
 /*
 query GetAiOveragesForWorkspace($requestContext: RequestContext!) {

crates/graphql/src/api/queries/get_available_harnesses.rs

@@ -0,0 +1,58 @@
+use crate::ai::AgentHarness;
+use crate::request_context::RequestContext;
+use crate::schema;
+
+#[derive(cynic::QueryVariables, Debug)]
+pub struct GetAvailableHarnessesVariables {
+    pub request_context: RequestContext,
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+#[cynic(
+    graphql_type = "RootQuery",
+    variables = "GetAvailableHarnessesVariables"
+)]
+pub struct GetAvailableHarnesses {
+    #[arguments(requestContext: $request_context)]
+    pub user: UserResult,
+}
+crate::client::define_operation! {
+    get_available_harnesses(GetAvailableHarnessesVariables) -> GetAvailableHarnesses;
+}
+
+#[derive(cynic::InlineFragments, Debug)]
+pub enum UserResult {
+    UserOutput(UserOutput),
+    #[cynic(fallback)]
+    Unknown,
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+pub struct UserOutput {
+    pub user: User,
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+pub struct User {
+    pub available_harnesses: AvailableHarnesses,
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+pub struct AvailableHarnesses {
+    pub harnesses: Vec<HarnessInfo>,
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+pub struct HarnessInfo {
+    pub harness: AgentHarness,
+    pub display_name: String,
+    pub enabled: bool,
+    pub available_models: Vec<HarnessModel>,
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+pub struct HarnessModel {
+    pub id: cynic::Id,
+    pub display_name: String,
+    pub reasoning_level: Option<String>,
+}

crates/graphql/src/api/queries/get_cloud_object.rs

@@ -1,8 +1,9 @@
-use crate::{
-    error::UserFacingError, object::CloudObjectWithDescendants,
-    object_actions::ObjectActionHistory, request_context::RequestContext,
-    response_context::ResponseContext, schema,
-};
+use crate::error::UserFacingError;
+use crate::object::CloudObjectWithDescendants;
+use crate::object_actions::ObjectActionHistory;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct GetCloudObjectVariables {

crates/graphql/src/api/queries/get_conversation_usage.rs

@@ -1,6 +1,8 @@
 use std::collections::HashMap;
 
-use crate::{request_context::RequestContext, scalars::Time, schema};
+use crate::request_context::RequestContext;
+use crate::scalars::Time;
+use crate::schema;
 
 /*
 query GetConversationUsage(
@@ -23,6 +25,7 @@ query GetConversationUsage(
           usageMetadata {
             contextWindowUsage
             creditsSpent
+            platformCreditsSpent
             summarized
             tokenUsage { modelId totalTokens }
             warpTokenUsage { modelId totalTokens tokenUsageByCategory { category tokens } }
@@ -111,7 +114,9 @@ pub struct ConversationUsage {
 #[derive(cynic::QueryFragment, Debug, Clone)]
 pub struct ConversationUsageMetadata {
     pub context_window_usage: f64,
+    pub context_window_segments: Vec<ContextWindowSegment>,
     pub credits_spent: f64,
+    pub platform_credits_spent: f64,
     pub summarized: bool,
     pub token_usage: Vec<ModelTokenUsage>,
     pub warp_token_usage: Vec<TokenUsage>,
@@ -119,6 +124,23 @@ pub struct ConversationUsageMetadata {
     pub tool_usage_metadata: ToolUsageMetadata,
 }
 
+#[derive(cynic::Enum, Clone, Copy, Debug, PartialEq, Eq)]
+pub enum ContextWindowSegmentType {
+    Unknown,
+    SystemPrompt,
+    ToolDefinitions,
+    ConversationHistory,
+    LatestInput,
+    Images,
+    Other,
+}
+
+#[derive(cynic::QueryFragment, Debug, Clone)]
+pub struct ContextWindowSegment {
+    pub segment_type: ContextWindowSegmentType,
+    pub token_count: i32,
+}
+
 fn convert_token_usage(
     warp_token_usage: &[TokenUsage],
     byok_token_usage: &[TokenUsage],
@@ -168,13 +190,34 @@ impl From<&ConversationUsageMetadata> for persistence::model::ConversationUsageM
             was_summarized: gql.summarized,
             context_window_usage: gql.context_window_usage as f32,
             credits_spent: gql.credits_spent as f32,
+            platform_credits_spent: gql.platform_credits_spent as f32,
             credits_spent_for_last_block: None,
             token_usage: convert_token_usage(&gql.warp_token_usage, &gql.byok_token_usage),
             tool_usage_metadata: (&gql.tool_usage_metadata).into(),
-            total_cache_read_tokens: 0,
-            total_cache_write_tokens: 0,
-            total_cache_miss_tokens: 0,
-            total_cost_cents: 0.0,
+            context_window_segments: gql.context_window_segments.iter().map(Into::into).collect(),
+        }
+    }
+}
+
+impl From<ContextWindowSegmentType> for persistence::model::ContextWindowSegmentType {
+    fn from(value: ContextWindowSegmentType) -> Self {
+        match value {
+            ContextWindowSegmentType::Unknown => Self::Unknown,
+            ContextWindowSegmentType::SystemPrompt => Self::SystemPrompt,
+            ContextWindowSegmentType::ToolDefinitions => Self::ToolDefinitions,
+            ContextWindowSegmentType::ConversationHistory => Self::ConversationHistory,
+            ContextWindowSegmentType::LatestInput => Self::LatestInput,
+            ContextWindowSegmentType::Images => Self::Images,
+            ContextWindowSegmentType::Other => Self::Other,
+        }
+    }
+}
+
+impl From<&ContextWindowSegment> for persistence::model::ContextWindowSegment {
+    fn from(gql: &ContextWindowSegment) -> Self {
+        Self {
+            segment_type: gql.segment_type.into(),
+            token_count: u32::try_from(gql.token_count).unwrap_or_default(),
         }
     }
 }

crates/graphql/src/api/queries/get_discoverable_teams.rs

@@ -1,6 +1,7 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, schema, user::DiscoverableTeamData,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::schema;
+use crate::user::DiscoverableTeamData;
 
 /*
 query GetDiscoverableTeams($requestContext: RequestContext!) {

crates/graphql/src/api/queries/get_feature_model_choices.rs

@@ -1,4 +1,5 @@
-use crate::{request_context::RequestContext, schema};
+use crate::request_context::RequestContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct GetFeatureModelChoicesVariables {
@@ -69,7 +70,9 @@ pub enum DisableReason {
 #[derive(cynic::Enum, Clone, Debug)]
 pub enum LlmModelHost {
     AwsBedrock,
+    CustomEndpoint,
     DirectApi,
+    GeminiEnterprise,
     #[cynic(fallback)]
     Other(String),
 }
@@ -80,6 +83,14 @@ pub struct RoutingHostConfig {
     pub model_routing_host: LlmModelHost,
 }
 
+#[derive(cynic::QueryFragment, Debug)]
+pub struct LlmContextWindow {
+    pub is_configurable: bool,
+    pub min: crate::scalars::Uint32,
+    pub max: crate::scalars::Uint32,
+    pub default: crate::scalars::Uint32,
+}
+
 #[derive(cynic::QueryFragment, Debug)]
 pub struct LlmInfo {
     pub display_name: String,
@@ -94,6 +105,7 @@ pub struct LlmInfo {
     pub provider: LlmProvider,
     pub host_configs: Vec<RoutingHostConfig>,
     pub pricing: LlmPricing,
+    pub context_window: LlmContextWindow,
 }
 
 #[derive(cynic::QueryFragment, Debug)]

crates/graphql/src/api/queries/get_integrations_using_environment.rs

@@ -1,4 +1,6 @@
-use crate::{error::UserFacingError, request_context::RequestContext, schema};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct GetIntegrationsUsingEnvironmentVariables {

crates/graphql/src/api/queries/get_oauth_connect_tx_status.rs

@@ -1,4 +1,6 @@
-use crate::{error::UserFacingError, request_context::RequestContext, schema};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::schema;
 
 #[derive(cynic::InputObject, Debug)]
 pub struct GetOAuthConnectTxStatusInput {

crates/graphql/src/api/queries/get_referral_info.rs

@@ -1,4 +1,5 @@
-use crate::{request_context::RequestContext, schema};
+use crate::request_context::RequestContext;
+use crate::schema;
 
 /*
 query GetReferralInfo($requestContext: RequestContext!) {

crates/graphql/src/api/queries/get_relevant_fragments.rs

@@ -1,9 +1,7 @@
-use crate::{
-    error::UserFacingError,
-    full_source_code_embedding::{ContentHash, EmbeddingConfig, NodeHash, RepoMetadata},
-    request_context::RequestContext,
-    schema,
-};
+use crate::error::UserFacingError;
+use crate::full_source_code_embedding::{ContentHash, EmbeddingConfig, NodeHash, RepoMetadata};
+use crate::request_context::RequestContext;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct GetRelevantFragmentsVariables {

crates/graphql/src/api/queries/get_request_limit_info.rs

@@ -1,10 +1,8 @@
-use crate::{
-    ai::RequestLimitInfo,
-    billing::{BonusGrant, BonusGrantsInfo},
-    error::UserFacingError,
-    request_context::RequestContext,
-    schema,
-};
+use crate::ai::RequestLimitInfo;
+use crate::billing::{BonusGrant, BonusGrantsInfo};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::schema;
 
 /*
 query GetRequestLimitInfo($requestContext: RequestContext!) {

crates/graphql/src/api/queries/get_scheduled_agent_history.rs

@@ -1,7 +1,8 @@
-use crate::{
-    error::UserFacingError, request_context::RequestContext, response_context::ResponseContext,
-    scalars::Time, schema,
-};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::scalars::Time;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct GetScheduledAgentHistoryVariables {

crates/graphql/src/api/queries/get_simple_integrations.rs

@@ -1,4 +1,7 @@
-use crate::{error::UserFacingError, request_context::RequestContext, scalars::Time, schema};
+use crate::error::UserFacingError;
+use crate::request_context::RequestContext;
+use crate::scalars::Time;
+use crate::schema;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct SimpleIntegrationsVariables {

crates/graphql/src/api/queries/get_updated_cloud_objects.rs

@@ -1,10 +1,15 @@
-use crate::{
-    error::UserFacingError, folder::Folder, generic_string_object::GenericStringObject,
-    mcp_gallery_template::MCPGalleryTemplate, notebook::Notebook,
-    object_actions::ObjectActionHistory, request_context::RequestContext,
-    response_context::ResponseContext, scalars::Time, schema, user::PublicUserProfile,
-    workflow::Workflow,
-};
+use crate::error::UserFacingError;
+use crate::folder::Folder;
+use crate::generic_string_object::GenericStringObject;
+use crate::mcp_gallery_template::MCPGalleryTemplate;
+use crate::notebook::Notebook;
+use crate::object_actions::ObjectActionHistory;
+use crate::request_context::RequestContext;
+use crate::response_context::ResponseContext;
+use crate::scalars::Time;
+use crate::schema;
+use crate::user::PublicUserProfile;
+use crate::workflow::Workflow;
 
 #[derive(cynic::QueryVariables, Debug)]
 pub struct GetUpdatedCloudObjectsVariables {
@@ -64,7 +69,7 @@ pub struct UpdatedCloudObjectsInput {
     pub workflows: Option<Vec<UpdatedObjectInput>>,
 }
 
-#[derive(cynic::InputObject, Debug)]
+#[derive(cynic::InputObject, Debug, Clone)]
 pub struct UpdatedObjectInput {
     pub actions_ts: Option<Time>,
     pub metadata_ts: Option<Time>,