@stack-spot/portal-network 1.0.0-betaadp.3 → 1.0.0-dev.1768422812092

package/src/client/account.ts

@@ -11,6 +11,7 @@ import {
   bindRoles,
   bindToGroups,
   bindToRoles,
+  cancelSecretDelete,
   create,
   create1,
   createAccountRole, createExtension, createExtensionLink,
@@ -41,13 +42,13 @@ import {
   enableSecret,
   enterpriseContact,
   findAssociations,
-  findSecrets1,
+  findSecrets,
   getAccess,
   getAccount1,
   getAccountMembers,
-  getAccountMembers2,
+  getAccountMembers1,
   getAccountRateLimit,
-  getAccounts1,
+  getAccounts,
   getAccountSso,
   getActiveExtensionVersion,
   getAllAccountSso,
@@ -74,10 +75,11 @@ import {
   getMembers1,
   getPartnerAccount, getPartnersSharingAllowed,
   getPersonalAccountDetails,
+  getPersonalAccountExpirationData,
   getPersonalClientCredentials,
-  getResources1,
+  getResources,
   getResources2,
-  getResourcesAndActionsWithStatus1,
+  getResourcesAndActionsWithStatus,
   getResourceTypes1,
   getRoleGroups,
   getRoleGroups1,
@@ -87,7 +89,8 @@ import {
   getRoles1,
   getRoles2,
   getRoles3,
-  getRoles4, getRoles5,
+  getRoles4,
+  getRoles5,
   getScmProvider,
   getServiceCredential,
   getServiceCredentialByIdRateLimit,
@@ -99,12 +102,13 @@ import {
   isCreatedScmCredentials1,
   listAccountMemberFavorites,
   listExtensionLinks,
-  listExtensions,
   listExtensionVersions,
+  listManageableExtensions,
   listMemberFavoritesByResource,
-  listScmCredentials,
+  listScmCredentials1,
   listScmCredentials2, partialUpdateSso, personalAccessTokenAuthorization,
   personalContact,
+  reactivateTrialAccount,
   removeRoleFromMember,
   removeTrialAccount,
   resetOtp,
@@ -141,8 +145,12 @@ import { DefaultAPIError } from '../error/DefaultAPIError'
 import { accountDictionary } from '../error/dictionary/account'
 import { StackspotAPIError } from '../error/StackspotAPIError'
 import { ReactQueryNetworkClient } from '../network/ReactQueryNetworkClient'
+import { removeAuthorizationParam } from '../utils/remove-authorization-param'
 import { CreateSCMRequest, SCMStatus, UpdateSCMRequest } from './types'
 
+const listAllResourcesOfGroup = removeAuthorizationParam(getGroupResources)
+const getsListAllResourcesOfGroup = removeAuthorizationParam(getResources)
+
 class AccountClient extends ReactQueryNetworkClient {
   constructor() {
     super(getApiAddresses().account.url, defaults)
@@ -222,15 +230,15 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Gets all members (paginated).
    */
-  allMembers = this.infiniteQuery(getAccountMembers2)
+  allMembers = this.infiniteQuery(getAccountMembers)
   /**
    * Gets member Groups (paginated).
    */
-  memberGroups = this.infiniteQuery(getMemberGroups1)
+  memberGroups = this.infiniteQuery(getMemberGroups)
   /**
    * Gets member roles (paginated).
    */
-  memberRoles = this.infiniteQuery(getRoles4)
+  memberRoles = this.infiniteQuery(getRoles1)
   /**
    * Gets member resources.
    */
@@ -282,7 +290,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Get all groups (paginated).
    */
-  allGroups = this.infiniteQuery(getGroups1)
+  allGroups = this.infiniteQuery(getGroups)
   /**
    * Creates a group.
    */
@@ -298,7 +306,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 +318,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Gets all roles of a group.
    */
-  groupRoles = this.infiniteQuery(getRoles5)
+  groupRoles = this.infiniteQuery(getRoles2)
   /**
    * Adds several roles to a group.
    */
@@ -320,9 +328,9 @@ class AccountClient extends ReactQueryNetworkClient {
    */
   removeRoleFromGroup = this.mutation(deleteRole)
   /**
-   * Gets all resources of a group (paginated).
+   * Get all resources by groups with pagination.
    */
-  groupResources = this.infiniteQuery(getGroupResources1)
+  allResourcesByGroupWithPagination = this.infiniteQuery(listAllResourcesOfGroup, { accumulator: 'items' })
   /**
    * Adds several resources to a group.
    */
@@ -334,14 +342,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 +357,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 +377,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 +385,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 +397,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Gets all resources (paginated)
    */
-  allResources = this.infiniteQuery(getResources1)
+  allResources = this.infiniteQuery(getsListAllResourcesOfGroup, { accumulator: 'items' })
   /**
    * Gets all resource types (paginated)
    */
@@ -409,7 +417,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Gets all SCM credentials (account level).
    */
-  allSCMCredentials = this.query(listScmCredentials2)
+  allSCMCredentials = this.query(listScmCredentials1)
   /**
    * Gets the status for the SCM credential.
    * 
@@ -485,7 +493,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Gets all SCM credentials for the user currently logged in.
    */
-  allUserSCMCredentials = this.query(listScmCredentials)
+  allUserSCMCredentials = this.query(listScmCredentials2)
   /**
    * Gets the SCM provider.
    */
@@ -623,43 +631,43 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Get all account members with pagination
    */
-  allMembersWithPagination = this.query(getAccountMembers)
+  allMembersWithPagination = this.query(getAccountMembers1)
   /**
    * Get Members Groups By Profile
    */
-  allMembersGroupsByProfileWithPagination = this.query(getMemberGroups)
+  allMembersGroupsByProfileWithPagination = this.query(getMemberGroups1)
   /**
    * Get Members Roles By Profile
    */
-  allMembersRolesByProfileWithPagination = this.query(getRoles1)
+  allMembersRolesByProfileWithPagination = this.query(getRoles4)
   /**
    * Get all account groups with pagination
    */
-  allGroupsWithPagination = this.query(getGroups)
+  allGroupsWithPagination = this.query(getGroups1)
   /**
    * Get all members by group with pagination
    */
-  allMembersByGroupWithPagination = this.query(getMembers)
+  allMembersByGroupWithPagination = this.query(getMembers1)
   /**
    * Get all roles by groups with pagination
    */
-  allRolesByGroupWithPagination = this.query(getRoles2)
+  allRolesByGroupWithPagination = this.query(getRoles5)
   /**
-   * Get all resources by groups with pagination
+   * Gets all resources of a group (paginated)
    */
-  allResourcesByGroupWithPagination = this.query(getGroupResources)
+  groupResources = this.query(getGroupResources1)
   /**
    * Get all roles account with pagination
    */
-  allRolesWithPagination = this.query(getRoles)
+  allRolesWithPagination = this.query(getRoles3)
   /**
    * Get all members by role with pagination
    */
-  allMembersByRoleWithPagination = this.query(getRoleMembers)
+  allMembersByRoleWithPagination = this.query(getRoleMembers1)
   /**
    * Get all groups by roles with pagination
    */
-  allGroupsByRoleWithPagination = this.query(getRoleGroups)
+  allGroupsByRoleWithPagination = this.query(getRoleGroups1)
   /**
    * Get Service Credentials
    */
@@ -683,7 +691,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Lists all extensions.
    */
-  extensions = this.query(listExtensions)
+  extensions = this.query(listManageableExtensions)
 
   /**
    * Lists all versions of an extension.
@@ -738,7 +746,18 @@ 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 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 }
+    },
+  })
   /**
    * Get secret by id
    */
@@ -748,10 +767,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)
-    }
+    },
   })
 
   /**
@@ -774,6 +793,10 @@ class AccountClient extends ReactQueryNetworkClient {
    * Disable secret
    */
   disableSecret = this.mutation(disableSecret)
+  /**
+   * Cancel Secret delete
+   */
+  cancelSecretDelete = this.mutation(cancelSecretDelete)
 
   // TRIAL
   /**
@@ -872,7 +895,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Lists enterprise accounts
    */
-  getEnterpriseAccounts = this.query(getAccounts1)
+  getEnterpriseAccounts = this.query(getAccounts)
   /**
    * Get an enterprise account
    */
@@ -897,6 +920,14 @@ class AccountClient extends ReactQueryNetworkClient {
    * Find Secret Associations
    */
   findSecretAssociations = this.infiniteQuery(findAssociations, { accumulator: 'items' })
+  /**
+   * Get expiration data
+   */
+  getExpirationData = this.query(getPersonalAccountExpirationData)
+  /**
+   * Reactive trial account
+   */
+  reactivateTrialAccount = this.mutation(reactivateTrialAccount)
 }
 
 export const accountClient = new AccountClient()

package/src/client/agent-tools.ts

@@ -1,6 +1,6 @@
 import { HttpError } from '@oazapfts/runtime'
 import { getApiAddresses } from '../api-addresses'
-import { addFavoriteV1AgentsAgentIdFavoritePost, AgentVisibilityLevelEnum, createAgentV1AgentsPost, createToolkitToolsV1ToolkitsToolkitIdToolsPost, createToolkitV1ToolkitsPost, defaults, deleteAgentV1AgentsAgentIdDelete, deleteFavoriteV1AgentsAgentIdFavoriteDelete, deleteToolkitToolsV1ToolkitsToolkitIdToolsDelete, deleteToolkitV1ToolkitsToolkitIdDelete, editToolkitToolV1ToolkitsToolkitIdToolsToolIdPut, forkAgentV1AgentsAgentIdForkPost, forkToolkitV1ToolkitsToolkitIdForkPost, getAgentV1AgentsAgentIdGet, getPublicToolKitsV1BuiltinToolkitGet, getToolkitToolV1ToolkitsToolkitIdToolsToolIdGet, getToolkitV1ToolkitsToolkitIdGet, listAgentsUsingToolsV1ToolkitsToolkitIdToolsAgentsPost, listAgentsV1AgentsGet, listMultiAgentsV1AgentsMultiAgentsGet, listToolkitsV1ToolkitsGet, publishAgentV1AgentsAgentIdPublishPost, searchAgentsV1AgentsSearchPost, shareV1AgentsAgentIdSharePost, updateAgentV1AgentsAgentIdPatch, updateToolkitV1ToolkitsToolkitIdPatch, VisibilityLevelEnum } from '../api/agent-tools'
+import { addFavoriteV1AgentsAgentIdFavoritePost, AgentVisibilityLevelEnum, createAgentV1AgentsPost, createToolkitToolsV1ToolkitsToolkitIdToolsPost, createToolkitV1ToolkitsPost, defaults, deleteAgentV1AgentsAgentIdDelete, deleteFavoriteV1AgentsAgentIdFavoriteDelete, deleteToolkitToolsV1ToolkitsToolkitIdToolsDelete, deleteToolkitV1ToolkitsToolkitIdDelete, editToolkitToolV1ToolkitsToolkitIdToolsToolIdPut, forkAgentV1AgentsAgentIdForkPost, forkToolkitV1ToolkitsToolkitIdForkPost, getAgentV1AgentsAgentIdGet, getPublicToolKitsV1BuiltinToolkitGet, getToolkitToolV1ToolkitsToolkitIdToolsToolIdGet, getToolkitV1ToolkitsToolkitIdGet, listAgentsUsingToolsV1ToolkitsToolkitIdToolsAgentsPost, listAgentsV1AgentsGet, listAgentsV3AgentsGet, listMcpToolsV1McpToolsToolkitIdGet, listMultiAgentsV1AgentsMultiAgentsGet, listToolkitsV1ToolkitsGet, listToolkitsV2ToolkitsGet, publishAgentV1AgentsAgentIdPublishPost, searchAgentsV1AgentsSearchPost, shareV1AgentsAgentIdSharePost, updateAgentV1AgentsAgentIdPatch, updateToolkitV1ToolkitsToolkitIdPatch, VisibilityLevelEnum } from '../api/agent-tools'
 import { DefaultAPIError } from '../error/DefaultAPIError'
 import { agentToolsDictionary } from '../error/dictionary/agent-tools'
 import { StackspotAPIError } from '../error/StackspotAPIError'
@@ -12,6 +12,8 @@ import { AgentResponseWithBuiltIn, AgentToolsOpenAPIPreview, AgentVisibilityLeve
 import { workspaceAiClient } from './workspace-ai'
 
 const AGENT_DEFAULT_SLUG = 'stk_flex'
+const listAgentsV3AgentsWithoutAuthorization = removeAuthorizationParam(listAgentsV3AgentsGet)
+const listToolkitsAuthorization = removeAuthorizationParam(listToolkitsV2ToolkitsGet)
 
 class AgentToolsClient extends ReactQueryNetworkClient {
   constructor() {
@@ -22,6 +24,9 @@ class AgentToolsClient extends ReactQueryNetworkClient {
     return new DefaultAPIError(error.data, error.status, agentToolsDictionary, error.headers)
   }
 
+  /**
+   * Lists all public toolkits.
+   */
   tools = this.query(removeAuthorizationParam(getPublicToolKitsV1BuiltinToolkitGet))
 
   /**
@@ -54,6 +59,19 @@ class AgentToolsClient extends ReactQueryNetworkClient {
    */
   agents = this.infiniteQuery(removeAuthorizationParam(listAgentsV1AgentsGet))
 
+  /** 
+   * List agents with support for multiple filters
+   */
+  agentsMultipleFilters = this.infiniteQuery(listAgentsV3AgentsWithoutAuthorization, {
+    pageParamName: 'filters.page',
+    accumulator: 'items', 
+    getNextPageParam: ({ variables, lastPage, lastPageParam }) => {
+      const size = variables.filters.size ?? 1
+      const parsedLastPageParam = (lastPageParam as number) ?? variables.filters.page ?? 1
+      return lastPage.items && lastPage.items.length < size ? undefined : parsedLastPageParam + 1
+    },
+  })
+
   /** 
    * List agents available to be added to other agents
    */
@@ -80,11 +98,11 @@ class AgentToolsClient extends ReactQueryNetworkClient {
   agentDefault = this.query({
     name: 'agentDefault',
     request: async (signal) => {
-      const agentDefault = await listAgentsV1AgentsGet(
-        { visibility: 'built_in', slug: this.agentDefaultSlug, authorization: '' }, { signal },
+      const agentDefault = await listAgentsV3AgentsGet(
+        { filters: { visibility_list: ['built_in'], slug: this.agentDefaultSlug }, authorization: '' }, { signal },
       )
 
-      const agentId = agentDefault?.find((agent) => agent.slug ===  this.agentDefaultSlug)?.id
+      const agentId = agentDefault?.items?.find((agent) => agent.slug ===  this.agentDefaultSlug)?.id
       const agent = agentId ? await this.agent.query({ agentId }) : undefined
       return agent
     },
@@ -95,40 +113,61 @@ class AgentToolsClient extends ReactQueryNetworkClient {
    */
   allAgents = this.query({
     name: 'allAgents',
-    request: async (signal, variables: { visibilities: (AgentVisibilityLevel | 'all')[] }) => {
+    request: async (signal, variables: { visibilities: (AgentVisibilityLevel | 'all')[], filter?: string }) => {
       const allVisibilities = ['account', 'built_in', 'recently_used', 'favorite', 'personal', 'shared', 'workspace'] as const
-      const visibilities = variables.visibilities.includes('all')
+      const visibilities = !variables.visibilities || variables.visibilities?.includes('all')
         ? allVisibilities
         : variables.visibilities as Array<AgentVisibilityLevelEnum | VisibilityLevelEnum>
-
       const shouldFetchWorkspaceAgents = visibilities.includes('workspace')
-
+      
       const workspaceAgentsPromise = shouldFetchWorkspaceAgents
         ? workspaceAiClient.workspacesContentsByType.query({ contentType: 'agent' })
         : Promise.resolve([])
 
-      const [workspaceAgents, ...agentsByVisibility] = await Promise.all([
+      const listAgentsVisibilities = visibilities.filter((visibility) => visibility!=='workspace')
+      const listAgentsPromise = listAgentsVisibilities.length 
+        ? listAgentsV3AgentsGet({ filters: { 
+          visibility_list: listAgentsVisibilities as (AgentVisibilityLevelEnum | VisibilityLevelEnum)[],
+          name: variables.filter,
+        }, 
+        authorization: '' }, { signal }) 
+        : Promise.resolve({ items: [] })
+
+      const [workspaceAgents, agentsByVisibility] = await Promise.all([
         workspaceAgentsPromise,
-        ...visibilities.map((visibility) => listAgentsV1AgentsGet({ visibility, authorization: '' }, { signal })),
+        listAgentsPromise,
       ])
 
       const workspaceAgentsWithSpaceName = workspaceAgents.flatMap(({ agents, space_name }) =>
         agents?.map((agent) => ({ ...agent, spaceName: space_name, builtIn: false }))) as AgentResponseWithBuiltIn[]
-
-      const allAgents: AgentResponseWithBuiltIn[] = workspaceAgentsWithSpaceName ?? []
-
-      agentsByVisibility.forEach(agents => allAgents.push(...agents.map(agent => ({
+      
+      const filteredWorkspaceAgents = variables.filter ? workspaceAgentsWithSpaceName.filter((agent) => 
+        agent.name.toLowerCase().includes(variables.filter!)) :
+        workspaceAgentsWithSpaceName
+      
+      const allAgents: AgentResponseWithBuiltIn[] = filteredWorkspaceAgents ?? []
+      
+      agentsByVisibility.items?.forEach(agent => allAgents.push(({
         ...agent,
         builtIn: agent?.visibility_level === 'built_in',
-      }))))
+      })))
 
       return allAgents
     },
   })
 
-  /* Get list of Toolkits
-  */
+  /**
+   * Lists all custom toolkits.
+   */
   toolkits = this.query(removeAuthorizationParam(listToolkitsV1ToolkitsGet))
+  /**
+  * Get list of Toolkits V2
+  */
+  toolkitsV2 = this.infiniteQuery(listToolkitsAuthorization, { accumulator: 'items' })
+  /**
+   * Lists all tools of an MCP toolkit.
+   */
+  mcpTools = this.query(removeAuthorizationParam(listMcpToolsV1McpToolsToolkitIdGet))
   /**
    * Get a toolkit by Id
    */
@@ -161,6 +200,7 @@ class AgentToolsClient extends ReactQueryNetworkClient {
    * Delete tool from Toolkit
    */
   deleteToolFromToolkit = this.mutation(removeAuthorizationParam(deleteToolkitToolsV1ToolkitsToolkitIdToolsDelete))
+  
   /**
    * Previews an uploaded open api file as a stream where each array item is an OpenAPI operation.
    */