fix(core): optimize upgrade to team page (#9099)

packages/frontend/graphql/src/graphql/cancel-subscription.gql

@@ -1,5 +1,8 @@
-mutation cancelSubscription($plan: SubscriptionPlan = Pro) {
-  cancelSubscription(plan: $plan) {
+mutation cancelSubscription(
+  $plan: SubscriptionPlan = Pro
+  $workspaceId: String
+) {
+  cancelSubscription(plan: $plan, workspaceId: $workspaceId) {
     id
     status
     nextBillAt

packages/frontend/graphql/src/graphql/index.ts

@@ -98,8 +98,8 @@ export const cancelSubscriptionMutation = {
   definitionName: 'cancelSubscription',
   containsFile: false,
   query: `
-mutation cancelSubscription($plan: SubscriptionPlan = Pro) {
-  cancelSubscription(plan: $plan) {
+mutation cancelSubscription($plan: SubscriptionPlan = Pro, $workspaceId: String) {
+  cancelSubscription(plan: $plan, workspaceId: $workspaceId) {
     id
     status
     nextBillAt
@@ -188,7 +188,7 @@ export const createCopilotMessageMutation = {
   id: 'createCopilotMessageMutation' as const,
   operationName: 'createCopilotMessage',
   definitionName: 'createCopilotMessage',
-  containsFile: true,
+  containsFile: false,
   query: `
 mutation createCopilotMessage($options: CreateChatMessageInput!) {
   createCopilotMessage(options: $options)
@@ -960,8 +960,8 @@ export const resumeSubscriptionMutation = {
   definitionName: 'resumeSubscription',
   containsFile: false,
   query: `
-mutation resumeSubscription($plan: SubscriptionPlan = Pro) {
-  resumeSubscription(plan: $plan) {
+mutation resumeSubscription($plan: SubscriptionPlan = Pro, $workspaceId: String) {
+  resumeSubscription(plan: $plan, workspaceId: $workspaceId) {
     id
     status
     nextBillAt
@@ -1184,8 +1184,12 @@ export const updateSubscriptionMutation = {
   definitionName: 'updateSubscriptionRecurring',
   containsFile: false,
   query: `
-mutation updateSubscription($plan: SubscriptionPlan = Pro, $recurring: SubscriptionRecurring!) {
-  updateSubscriptionRecurring(plan: $plan, recurring: $recurring) {
+mutation updateSubscription($plan: SubscriptionPlan = Pro, $recurring: SubscriptionRecurring!, $workspaceId: String) {
+  updateSubscriptionRecurring(
+    plan: $plan
+    recurring: $recurring
+    workspaceId: $workspaceId
+  ) {
     id
     plan
     recurring

packages/frontend/graphql/src/graphql/resume-subscription.gql

@@ -1,5 +1,8 @@
-mutation resumeSubscription($plan: SubscriptionPlan = Pro) {
-  resumeSubscription(plan: $plan) {
+mutation resumeSubscription(
+  $plan: SubscriptionPlan = Pro
+  $workspaceId: String
+) {
+  resumeSubscription(plan: $plan, workspaceId: $workspaceId) {
     id
     status
     nextBillAt

packages/frontend/graphql/src/graphql/update-subscription-billing.gql

@@ -1,8 +1,13 @@
 mutation updateSubscription(
   $plan: SubscriptionPlan = Pro
   $recurring: SubscriptionRecurring!
+  $workspaceId: String
 ) {
-  updateSubscriptionRecurring(plan: $plan, recurring: $recurring) {
+  updateSubscriptionRecurring(
+    plan: $plan
+    recurring: $recurring
+    workspaceId: $workspaceId
+  ) {
     id
     plan
     recurring

packages/frontend/graphql/src/schema.ts

@@ -1476,6 +1476,7 @@ export type SetBlobMutation = { __typename?: 'Mutation'; setBlob: string };
 
 export type CancelSubscriptionMutationVariables = Exact<{
   plan?: InputMaybe<SubscriptionPlan>;
+  workspaceId?: InputMaybe<Scalars['String']['input']>;
 }>;
 
 export type CancelSubscriptionMutation = {
@@ -2256,6 +2257,7 @@ export type RemoveAvatarMutation = {
 
 export type ResumeSubscriptionMutationVariables = Exact<{
   plan?: InputMaybe<SubscriptionPlan>;
+  workspaceId?: InputMaybe<Scalars['String']['input']>;
 }>;
 
 export type ResumeSubscriptionMutation = {
@@ -2467,6 +2469,7 @@ export type UpdateServerRuntimeConfigsMutation = {
 export type UpdateSubscriptionMutationVariables = Exact<{
   plan?: InputMaybe<SubscriptionPlan>;
   recurring: SubscriptionRecurring;
+  workspaceId?: InputMaybe<Scalars['String']['input']>;
 }>;
 
 export type UpdateSubscriptionMutation = {