@stack-spot/portal-network 0.189.0 → 0.190.0-beta.1

package/src/api/codeShift.ts

@@ -12,7 +12,7 @@ export const defaults: Oazapfts.Defaults<Oazapfts.CustomHeaders> = {
 };
 const oazapfts = Oazapfts.runtime(defaults);
 export const servers = {};
-export type CodeShiftRole = "list-applications" | "create-application" | "list-repository" | "create-repository" | "delete-repository" | "dispatch-module" | "create-module" | "account-settings" | "list-integration" | "create-integration" | "delete-integration" | "list-program-group" | "create-program-group" | "delete-program-group" | "view-analytics";
+export type CodeShiftRole = "list-applications" | "create-application" | "list-repository" | "create-repository" | "delete-repository" | "dispatch-module" | "create-module" | "account-settings" | "list-integration" | "create-integration" | "delete-integration" | "list-program-group" | "create-program-group" | "delete-program-group" | "view-analytics" | "account-discovery-repos";
 export type RolesResponse = {
     result: boolean;
 };
@@ -836,8 +836,9 @@ export type RepositoryLastReportStatusResponse = {
     totalFilesChanged: number | null;
     totalIssues: number | null;
     pullRequestLink: string | null;
-    merged: boolean | null;
+    merged?: boolean | null;
     customerScore?: number | null;
+    languages?: string | null;
 };
 export type AnalyticsRepositoryLastReportStatusResponse = {
     totalFiles?: number;
@@ -987,7 +988,7 @@ export type RepositoryDetailedReportItemResponse = {
     moduleVersion?: string | null;
     processingTimeSeconds?: number | null;
     pullRequestLink?: string | null;
-    merged?: string | null;
+    merged?: boolean | null;
     reportCreatedAt?: string | null;
     reportCreatedByEmail?: string | null;
     reportCreatedByName?: string | null;
@@ -1000,6 +1001,7 @@ export type RepositoryDetailedReportItemResponse = {
     totalFilesCount?: number;
     status?: string | null;
     customerScore?: number | null;
+    languages?: string | null;
 };
 export type AnalyticsRepositoryDetailedReportResponse = {
     reposCount?: number;

package/src/api/genAiInference.ts

@@ -134,6 +134,9 @@ export type UpdateLlmModelRequest = {
     };
     resources: LlmModelResourceRequest[];
 };
+export type GetModelByIdsRequest = {
+    model_ids: string[];
+};
 export type UpdateLlmModelResourceRequest = {
     name: string;
     is_default?: boolean;
@@ -239,6 +242,8 @@ export type ChatCompletion = {
     function_call?: string | {
         [key: string]: string;
     } | null;
+    /** ID of an alternative model for chat completion. It should be present on available_model_ids agent attr. */
+    selected_model_id?: string | null;
 };
 export type ChatRequest = {
     streaming: boolean;
@@ -246,6 +251,8 @@ export type ChatRequest = {
     stackspot_knowledge?: boolean;
     return_ks_in_response?: boolean;
     upload_ids?: string[];
+    execution_id?: string | null;
+    selected_model?: string | null;
 };
 /**
  * Health Check
@@ -469,6 +476,32 @@ export function deleteV1LlmModelsModelIdDelete({ modelId, xAccountId, authorizat
         })
     }));
 }
+/**
+ * Get Models List
+ */
+export function getModelsListV1LlmModelsListPost({ xAccountId, authorization, getModelByIdsRequest }: {
+    xAccountId?: string | null;
+    authorization: string;
+    getModelByIdsRequest: GetModelByIdsRequest;
+}, opts?: Oazapfts.RequestOpts) {
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 200;
+        data: LlmModelsResponse[];
+    } | {
+        status: 404;
+    } | {
+        status: 422;
+        data: HttpValidationError;
+    }>("/v1/llm/models-list", oazapfts.json({
+        ...opts,
+        method: "POST",
+        body: getModelByIdsRequest,
+        headers: oazapfts.mergeHeaders(opts?.headers, {
+            "x-account-id": xAccountId,
+            authorization
+        })
+    })));
+}
 /**
  * Save Or Update Model Resources
  */
@@ -549,10 +582,15 @@ export function listLlmProvidersV1LlmProvidersGet({ acceptsSelfHosted, xAccountI
 /**
  * Handle completions requests
  */
-export function createCompletionsV1ChatCompletionsPost({ xRequestOrigin, xAccountId, authorization, chatCompletion }: {
-    xRequestOrigin?: string | null;
+export function createCompletionsV1ChatCompletionsPost({ xAccountId, authorization, xRequestOrigin, xConversationId, xMessageId, xQcExecutionId, xQcSlug, xPlatformVersion, chatCompletion }: {
     xAccountId?: string | null;
     authorization: string;
+    xRequestOrigin?: string | null;
+    xConversationId?: string | null;
+    xMessageId?: string | null;
+    xQcExecutionId?: string | null;
+    xQcSlug?: string | null;
+    xPlatformVersion?: string | null;
     chatCompletion: ChatCompletion;
 }, opts?: Oazapfts.RequestOpts) {
     return oazapfts.ok(oazapfts.fetchJson<{
@@ -568,9 +606,14 @@ export function createCompletionsV1ChatCompletionsPost({ xRequestOrigin, xAccoun
         method: "POST",
         body: chatCompletion,
         headers: oazapfts.mergeHeaders(opts?.headers, {
-            "x-request-origin": xRequestOrigin,
             "x-account-id": xAccountId,
-            authorization
+            authorization,
+            "x-request-origin": xRequestOrigin,
+            "x-conversation-id": xConversationId,
+            "x-message-id": xMessageId,
+            "x-qc-execution-id": xQcExecutionId,
+            "x-qc-slug": xQcSlug,
+            "x-platform-version": xPlatformVersion
         })
     })));
 }

package/src/api/notification.ts

@@ -332,3 +332,5 @@ export function delete1({ notificationIntentId }: {
         method: "DELETE"
     }));
 }
+
+

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,
@@ -102,9 +105,10 @@ import {
   listExtensions,
   listExtensionVersions,
   listMemberFavoritesByResource,
-  listScmCredentials,
+  listScmCredentials1,
   listScmCredentials2, partialUpdateSso, personalAccessTokenAuthorization,
   personalContact,
+  reactivateTrialAccount,
   removeRoleFromMember,
   removeTrialAccount,
   resetOtp,
@@ -222,15 +226,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 +286,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Get all groups (paginated).
    */
-  allGroups = this.infiniteQuery(getGroups1)
+  allGroups = this.infiniteQuery(getGroups)
   /**
    * Creates a group.
    */
@@ -298,7 +302,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 +314,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.
    */
@@ -322,7 +326,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 +338,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 +353,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 +373,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 +381,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 +393,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Gets all resources (paginated)
    */
-  allResources = this.infiniteQuery(getResources1)
+  allResources = this.infiniteQuery(getResources)
   /**
    * Gets all resource types (paginated)
    */
@@ -409,7 +413,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 +489,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 +627,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
    */
-  allResourcesByGroupWithPagination = this.query(getGroupResources)
+  allResourcesByGroupWithPagination = 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
    */
@@ -738,15 +742,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 +763,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 +789,10 @@ class AccountClient extends ReactQueryNetworkClient {
    * Disable secret
    */
   disableSecret = this.mutation(disableSecret)
+  /**
+   * Cancel Secret delete
+   */
+  cancelSecretDelete = this.mutation(cancelSecretDelete)
 
   // TRIAL
   /**
@@ -883,7 +891,7 @@ class AccountClient extends ReactQueryNetworkClient {
   /**
    * Lists enterprise accounts
    */
-  getEnterpriseAccounts = this.query(getAccounts1)
+  getEnterpriseAccounts = this.query(getAccounts)
   /**
    * Get an enterprise account
    */
@@ -908,6 +916,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/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 []
@@ -290,7 +293,9 @@ class AIClient extends ReactQueryNetworkClient {
      */
     async function transform(event: Partial<FixedChatResponse>, data: Partial<ChatResponseWithSteps>) {
       const info = event.agent_info
+
       if (!info) return
+
       const tools = await AIClient.toolsOfAgent(request.context?.agent_id)
       data.steps = data.steps ? [...data.steps] : []
 
@@ -303,6 +308,59 @@ 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',
+          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 +370,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 +393,80 @@ 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,
+              executionId: info.id,
+              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,
+                executionId: info.id,
+                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
@@ -366,7 +478,7 @@ class AIClient extends ReactQueryNetworkClient {
         if (answerStep) answerStep.status = 'running'
       }
 
-      if (info.type === 'final_answer' && info.action === 'end') {
+      if (info.type === 'chat' && info.action === 'end') {
         const answerStep = last(data.steps)
         if (answerStep) {
           answerStep.status = 'success'
@@ -393,3 +505,4 @@ class AIClient extends ReactQueryNetworkClient {
 }
 
 export const aiClient = new AIClient()
+