@stack-spot/portal-network 0.187.0 → 0.188.0-beta.1

package/src/client/account.ts

@@ -11,6 +11,7 @@ import {
   bindRoles,
   bindToGroups,
   bindToRoles,
+  cancelSecretDelete,
   create,
   create1,
   createAccountRole, createExtension, createExtensionLink,
@@ -41,7 +42,7 @@ import {
   enableSecret,
   enterpriseContact,
   findAssociations,
-  findSecrets1,
+  findSecrets,
   getAccess,
   getAccount1,
   getAccountMembers,
@@ -63,31 +64,23 @@ import {
   getFeatures3,
   getGroupById,
   getGroupResources,
-  getGroupResources1,
   getGroups,
-  getGroups1,
   getMemberById,
   getMemberGroups,
-  getMemberGroups1,
   getMemberPreferences,
   getMembers,
-  getMembers1,
   getPartnerAccount, getPartnersSharingAllowed,
   getPersonalAccountDetails,
   getPersonalClientCredentials,
-  getResources1,
+  getResources,
   getResources2,
-  getResourcesAndActionsWithStatus1,
+  getResourcesAndActionsWithStatus,
   getResourceTypes1,
   getRoleGroups,
-  getRoleGroups1,
   getRoleMembers,
-  getRoleMembers1,
   getRoles,
   getRoles1,
   getRoles2,
-  getRoles3,
-  getRoles4, getRoles5,
   getScmProvider,
   getServiceCredential,
   getServiceCredentialByIdRateLimit,
@@ -226,11 +219,11 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Gets member Groups (paginated).
    */
-  memberGroups = this.infiniteQuery(getMemberGroups1)
+  memberGroups = this.infiniteQuery(getMemberGroups)
   /**
    * Gets member roles (paginated).
    */
-  memberRoles = this.infiniteQuery(getRoles4)
+  memberRoles = this.infiniteQuery(getRoles)
   /**
    * Gets member resources.
    */
@@ -282,7 +275,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Get all groups (paginated).
    */
-  allGroups = this.infiniteQuery(getGroups1)
+  allGroups = this.infiniteQuery(getGroups)
   /**
    * Creates a group.
    */
@@ -298,7 +291,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Gets all members in a group (paginated).
    */
-  groupMembers = this.infiniteQuery(getMembers1)
+  groupMembers = this.infiniteQuery(getMembers)
   /**
    * Adds several members to a group
    */
@@ -310,7 +303,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Gets all roles of a group.
    */
-  groupRoles = this.infiniteQuery(getRoles5)
+  groupRoles = this.infiniteQuery(getRoles)
   /**
    * Adds several roles to a group.
    */
@@ -322,7 +315,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Gets all resources of a group (paginated).
    */
-  groupResources = this.infiniteQuery(getGroupResources1)
+  groupResources = this.infiniteQuery(getGroupResources)
   /**
    * Adds several resources to a group.
    */
@@ -334,14 +327,14 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Gets all roles in the account (paginated)
    */
-  allRoles = this.infiniteQuery(getRoles3)
+  allRoles = this.infiniteQuery(getRoles)
   /**
    * Get a role by id
    */
   role = this.query({
     name: 'account.role',
     request: async (signal, { id }: { id: string }) => {
-      const roles = await getRoles3({ filterBy: 'id', filterValue: id }, { signal })
+      const roles = await getRoles({ filterBy: 'id', filterValue: id }, { signal })
       if (!roles.length) throw new StackspotAPIError({ status: 404 })
       return roles[0]
     },
@@ -349,7 +342,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Gets all members with the provided role (paginated)
    */
-  roleMembers = this.infiniteQuery(getRoleMembers1)
+  roleMembers = this.infiniteQuery(getRoleMembers)
   /**
    * Creates a role
    */
@@ -369,7 +362,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Gets all groups with the provided role (paginated)
    */
-  roleGroups = this.infiniteQuery(getRoleGroups1)
+  roleGroups = this.infiniteQuery(getRoleGroups)
   /**
    * Adds a role to several groups
    */
@@ -377,7 +370,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Get the actions a role is allowed to perform
    */
-  rolePermissions = this.infiniteQuery(getResourcesAndActionsWithStatus1)
+  rolePermissions = this.infiniteQuery(getResourcesAndActionsWithStatus)
   /**
    * Updates a role with new list of permissions
    */
@@ -389,7 +382,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Gets all resources (paginated)
    */
-  allResources = this.infiniteQuery(getResources1)
+  allResources = this.infiniteQuery(getResources)
   /**
    * Gets all resource types (paginated)
    */
@@ -738,15 +731,15 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Get secrets
    */
-  getSecrets = this.query(findSecrets1)
+  getSecrets = this.query(findSecrets)
   /**
  * Get workspace and spot secrets
  */
   getWorkspaceAndSpotSecrets = this.query({
     name: 'account.getWorkspaceAndSpotSecrets',
     request: async (signal) => {
-      const workspaceSecrets = await findSecrets1({ filterByScope: 'SCOPED', filterByScopedBy: 'WORKSPACE', size: 1500 }, { signal })
-      const spotSecrets = await findSecrets1({ filterByScope: 'SCOPED', filterByScopedBy: 'SPOT', size: 1500 }, { signal })
+      const workspaceSecrets = await findSecrets({ filterByScope: 'SCOPED', filterByScopedBy: 'WORKSPACE', size: 1500 }, { signal })
+      const spotSecrets = await findSecrets({ filterByScope: 'SCOPED', filterByScopedBy: 'SPOT', size: 1500 }, { signal })
       return { workspaces: workspaceSecrets?.items, spots: spotSecrets?.items }
     },
   })
@@ -759,10 +752,10 @@ class AccountClient extends ReactQueryNetworkClient {
    */
   getSecretsByIds = this.query({
     name: 'getSecretsByIds',
-    request: (signal, variables: { ids: string[], }) => {
+    request: (signal, variables: { ids: string[] }) => {
       const requests = variables?.ids?.map((secretId) => getById({ secretId }, { signal }))
       return Promise.all(requests)
-    }
+    },
   })
 
   /**
@@ -785,6 +778,10 @@ class AccountClient extends ReactQueryNetworkClient {
    * Disable secret
    */
   disableSecret = this.mutation(disableSecret)
+  /**
+   * Cancel Secret delete
+   */
+  cancelSecretDelete = this.mutation(cancelSecretDelete)
 
   // TRIAL
   /**

package/src/client/ai.ts

@@ -266,6 +266,9 @@ class AIClient extends ReactQueryNetworkClient {
       agent.toolkits?.builtin_toolkits?.forEach(kit => kit.tools?.forEach(({ id, name, description }) => {
         if (id) tools.push({ image: kit.image_url, id, name: name || id, description })
       }))
+      agent.toolkits?.custom_toolkits?.forEach(kit => kit.tools?.forEach(({ id, name, description }) => {
+        if (id) tools.push({ image: kit.avatar ?? undefined, id, name: name || id, description })
+      }))
       return tools
     } catch {
       return []
@@ -303,6 +306,33 @@ class AIClient extends ReactQueryNetworkClient {
           steps: info.data?.steps?.map(s => s.goal) ?? [],
           goal: info.data?.plan_goal ?? '',
         })
+        
+        info.data?.steps.forEach(s => data.steps?.push({
+          id: s.id,
+          type: 'step',
+          status: 'pending',
+          input: s.goal,
+          attempts: [{
+            tools: s.tools?.map(t => ({
+              ...(tools.find(({ id }) => id === t.tool_id) ?? { id: t.tool_id, name: t.tool_id }),
+              executionId: t.tool_execution_id,
+              goal: t.goal,
+            })),
+          }],
+        }))
+        data.steps.push({ id: 'answer', type: 'answer', status: 'pending' })
+      }
+
+      if (info.type === 'planning' && info.action === 'awaiting_approval') {
+        data.steps.push({
+          id: 'planning',
+          type: 'planning',
+          status: 'awaiting_approval',
+          user_question: info.data?.user_question,
+          duration: info.duration || 0,
+          steps: info.data?.steps?.map(s => s.goal) ?? [],
+          goal: info.data?.plan_goal ?? '',
+        })
         info.data?.steps.forEach(s => data.steps?.push({
           id: s.id,
           type: 'step',
@@ -312,6 +342,7 @@ class AIClient extends ReactQueryNetworkClient {
             tools: s.tools?.map(t => ({
               ...(tools.find(({ id }) => id === t.tool_id) ?? { id: t.tool_id, name: t.tool_id }),
               executionId: t.tool_execution_id,
+              goal: t.goal,
             })),
           }],
         }))
@@ -334,27 +365,78 @@ class AIClient extends ReactQueryNetworkClient {
         }
       }
 
+      if (info.type === 'tool' && info.action === 'awaiting_approval') {
+        const tool = tools.find(({ id }) => id === info.data?.tool_id) 
+        data.steps.push({
+          id: info.id,
+          type: 'tool',
+          status: 'awaiting_approval',
+          duration: info.duration || 0,
+          input: info.data?.input,
+          user_question: info.data?.user_question,
+          attempts: [{
+            tools: [{
+              executionId: info.id,
+              id: info.data?.tool_id ?? '',
+              name: tool?.name ?? '',
+              goal: tool?.goal,
+              ...tool,
+            }],
+          }],
+        })
+        data.steps.push({ id: 'answer', type: 'answer', status: 'pending' })
+      }
+
       if (info.type === 'tool' && info.action === 'start') {
         const currentStep = data.steps.find(s => s.status === 'running') as StepChatStep
-        if (!currentStep || !info.data || !currentStep.attempts[0].tools) return
-        const toolInFirstAttempt = currentStep.attempts[0].tools.find(t => t.executionId === info.id)
-        if (!toolInFirstAttempt) return
-        const input = formatJson(info.data.input)
-        if (info.data.attempt === 0) {
-          toolInFirstAttempt.input = input
-        } else {
-          currentStep.attempts[info.data.attempt] ??= { tools: [] }
-          currentStep.attempts[info.data.attempt].tools?.push({
-            ...toolInFirstAttempt,
-            input,
+        if (!info.data) return
+        
+        //There might be a tool with status awaiting_approval, so we want to inform tool has already started
+        if (!currentStep || !currentStep.attempts[0].tools) {
+          const input = formatJson(info.data.input)
+          const tool = tools.find(({ id }) => id === info.data?.tool_id) ?? { id: info.data?.tool_id, name: info.data?.tool_id }
+          data.steps.push({
+            id: info.id,
+            type: 'tool',
+            status: 'running',
+            duration: info.duration || 0,
+            input: info.data?.input,
+            user_question: info.data?.user_question,
+            attempts: [{
+              tools:[{ ...tool, executionId: info.id, input }],
+            }],
           })
+        } else {
+          const toolInFirstAttempt = currentStep.attempts[0].tools?.find(t => t.executionId === info.id)
+          //One step might have multiple tools. When in an approval mode, we might not have all the tools in the array yet.
+          //So we make sure to add any tools that are not in there.
+          if (!toolInFirstAttempt) {
+            const input = formatJson(info.data.input)
+            const tool = tools?.find(({ id }) => id === info.data?.tool_id) ?? { id: info.data?.tool_id, name: info.data?.tool_id }
+            currentStep.attempts[info.data.attempt-1].tools?.push({
+              ...tool,
+              input,
+            })
+          } else {
+            const input = formatJson(info.data.input)
+            if (info.data.attempt === 1) {
+              toolInFirstAttempt.input = input
+            } else {
+              const tool = tools.find(({ id }) => id === info.data?.tool_id) ?? { id: info.data?.tool_id, name: info.data?.tool_id }
+              currentStep.attempts[info.data.attempt-1] ??= { tools: [] }
+              currentStep.attempts[info.data.attempt-1].tools?.push({
+                ...tool,
+                input,
+              })
+            }
+          }
         }
       }
 
       if (info.type === 'tool' && info.action === 'end') {
         const currentStep = data.steps.find(s => s.status === 'running') as StepChatStep
         if (!currentStep || !info.data) return
-        const tool = currentStep.attempts[info.data.attempt]?.tools?.find(t => t.executionId === info.id)
+        const tool = currentStep.attempts[info.data.attempt-1]?.tools?.find(t => t.executionId === info.id)
         if (tool) {
           tool.output = formatJson(info.data.output)
           tool.duration = info.duration
@@ -393,3 +475,4 @@ class AIClient extends ReactQueryNetworkClient {
 }
 
 export const aiClient = new AIClient()
+

package/src/client/code-shift.ts

@@ -58,11 +58,14 @@ import {
   getModuleV1ModulesModuleIdGet,
   analyticsProgramGroupsTargetDetailsV1AnalyticsProgramGroupsTargetDetailsGet,
   analyticsProgramGroupsTargetDetailsDownloadV1AnalyticsProgramGroupsTargetDetailsDownloadGet,
+  putCustomerRatingReportV1ReportsReportIdCustomerRatingPut,
   searchReposScmServiceV2ReposSearchScmPost,
   importReposWithTagsScmServiceV2ReposSearchScmSearchIdPost,
   searchReposScmV2V2ReposSearchScmSearchIdGet,
   analyticsRepositoryTargetDetailsV1AnalyticsRepositoriesTargetDetailsGet,
   analyticsRepositoryTargetDetailsDownloadV1AnalyticsRepositoriesTargetDetailsDownloadGet,
+  updateModuleServiceV1ModulesModuleIdPut,
+  downloadSearchReposScmV2V2ReposSearchScmSearchIdDownloadGet,
 } from '../api/codeShift'
 import { DefaultAPIError } from '../error/DefaultAPIError'
 import { codeShiftDictionary } from '../error/dictionary/code-shift'
@@ -128,6 +131,10 @@ class CodeShift extends ReactQueryNetworkClient {
    * Creates a module.
    */
   createModule = this.mutation(removeAuthorizationParam(createModuleServiceV1ModulesPost))
+  /**
+   * Updates a module.
+   */
+  updateModule = this.mutation(removeAuthorizationParam(updateModuleServiceV1ModulesModuleIdPut))
   /**
    * Generates a report.
    */
@@ -144,6 +151,12 @@ class CodeShift extends ReactQueryNetworkClient {
    * Downloads a report as a csv file.
    */
   downloadReport = this.query(removeAuthorizationParam(downloadReportV1ReportsReportIdDownloadGet))
+  /**
+   * Put Customer Rating Report
+   */
+  updateReportRating = this.mutation(
+    removeAuthorizationParam(putCustomerRatingReportV1ReportsReportIdCustomerRatingPut),
+  )
   /**
    * Gets code shift settings
    */
@@ -161,6 +174,7 @@ class CodeShift extends ReactQueryNetworkClient {
    */
   searchRepositoryStatus = this.query(removeAuthorizationParam(getStatusSearchReposScmServiceV1ReposSearchScmSearchRepoIdStatusGet))
   /**
+   * @deprecated
    * Downloads file with found repositories
    */
   downloadSearchRepository = this.mutation(removeAuthorizationParam(downloadSearchReposScmServiceV1ReposSearchScmSearchRepoIdDownloadGet))
@@ -176,6 +190,12 @@ class CodeShift extends ReactQueryNetworkClient {
    * Gets repositories search by id (v2)
    */
   getRepositoriesBySearchId = this.query(removeAuthorizationParam(searchReposScmV2V2ReposSearchScmSearchIdGet))
+  /**
+   * Downloads file with found repositories (v2)
+   */
+  downloadSearchRepositoryV2 = this.mutation(
+    removeAuthorizationParam(downloadSearchReposScmV2V2ReposSearchScmSearchIdDownloadGet),
+  )
   /**
    * Validate if the user has permission. 
    * We do not use opa in this api, so this is the fn needed to check permissions.

package/src/client/types.ts

@@ -245,6 +245,7 @@ export interface ChatAgentTool {
   image?: string,
   input?: string,
   output?: string,
+  goal?: string,
 }
 
 export interface ChatStepAttempt {
@@ -256,8 +257,8 @@ export interface ChatStepAttempt {
 
 export interface BaseChatStep {
   id: string,
-  type: 'planning' | 'step' | 'answer',
-  status: 'pending' | 'running' | 'success' | 'error',
+  type: 'planning' | 'step' | 'answer' | 'tool',
+  status: 'pending' | 'running' | 'success' | 'error' | 'awaiting_approval',
   /**
    * Duration in seconds.
    */
@@ -266,9 +267,24 @@ export interface BaseChatStep {
 
 export interface PlanningChatStep extends BaseChatStep {
   type: 'planning',
-  status: 'success',
+  status: 'success' | 'awaiting_approval',
   steps: string[],
   goal: string,
+  user_question?: string,
+}
+
+export interface ToolChatStep extends BaseChatStep {
+  type: 'tool',
+  status: 'running' | 'success' | 'error' | 'awaiting_approval',
+  /**
+   * Each step might attempt to run for multiple times, with different inputs and outputs. If first attempt succeeds, this array will have
+   * only one element.
+   * 
+   * This array never has less than one element, despite the step's status.
+   */
+  attempts: ChatStepAttempt[],
+  input?: Record<string, any>,
+  user_question?: string,
 }
 
 export interface StepChatStep extends BaseChatStep {
@@ -288,17 +304,19 @@ export interface AnswerChatStep extends BaseChatStep {
   type: 'answer',
 }
 
-export type ChatStep = PlanningChatStep | StepChatStep | AnswerChatStep
+export type ChatStep = PlanningChatStep | StepChatStep | AnswerChatStep | ToolChatStep
 
 export interface BaseAgentInfo {
   type: 'chat' | 'planning' | 'step' | 'tool' | 'final_answer',
-  action: 'start' | 'end',
+  action: 'start' | 'end' | 'awaiting_approval',
   duration?: number,
+  id: string,
 }
 
 export interface AgentTool {
   tool_id: string,
   tool_execution_id: string,
+  goal: string,
 }
 
 export interface PlanningAgentInfo extends BaseAgentInfo {
@@ -311,6 +329,7 @@ export interface PlanningAgentInfo extends BaseAgentInfo {
       goal: string,
       tools?: AgentTool[],
     }[],
+    user_question?: string,
   },
 }
 
@@ -326,6 +345,8 @@ export interface ToolAgentInfo extends BaseAgentInfo {
     input?: any,
     attempt: number,
     output?: string,
+    user_question?: string,
+    tool_id: string,
   },
 }