@stack-spot/portal-network 0.203.0 → 0.204.0-beta.1

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, listAgentsV3AgentsGet, listMcpToolsV1McpToolsToolkitIdGet, listMultiAgentsV1AgentsMultiAgentsGet, listToolkitsV1ToolkitsGet, publishAgentV1AgentsAgentIdPublishPost, searchAgentsV1AgentsSearchPost, shareV1AgentsAgentIdSharePost, updateAgentV1AgentsAgentIdPatch, updateToolkitV1ToolkitsToolkitIdPatch, VisibilityLevelEnum } from '../api/agent-tools'
+import { addFavoriteV1AgentsAgentIdFavoritePost, AgentVisibilityLevelEnum, createAgentV1AgentsPost, createAgentV2AgentsPost, createToolkitToolsV1ToolkitsToolkitIdToolsPost, createToolkitV1ToolkitsPost, createVersionV1AgentsAgentCoreIdVersionsPost, defaults, deleteAgentV1AgentsAgentIdDelete, deleteFavoriteV1AgentsAgentIdFavoriteDelete, deleteToolkitToolsV1ToolkitsToolkitIdToolsDelete, deleteToolkitV1ToolkitsToolkitIdDelete, deleteVersionV1AgentsAgentCoreIdVersionsVersionNumberDelete, editToolkitToolV1ToolkitsToolkitIdToolsToolIdPut, findByAgentCoreIdV2AgentsAgentCoreIdGet, forkAgentV1AgentsAgentIdForkPost, forkAgentVersionV1AgentsAgentCoreIdVersionsVersionNumberForkPost, forkToolkitV1ToolkitsToolkitIdForkPost, getAgentV1AgentsAgentIdGet, getPublicToolKitsV1BuiltinToolkitGet, getToolkitToolV1ToolkitsToolkitIdToolsToolIdGet, getToolkitV1ToolkitsToolkitIdGet, listAgentsUsingToolsV1ToolkitsToolkitIdToolsAgentsPost, listAgentsV1AgentsGet, listAgentsV3AgentsGet, listMcpToolsV1McpToolsToolkitIdGet, listMultiAgentsV1AgentsMultiAgentsGet, listToolkitsV1ToolkitsGet, listToolkitsV2ToolkitsGet, listVersionsOfAgentCoreV1AgentsAgentCoreIdVersionsGet, patchVersionRecommendedV1AgentsAgentCoreIdVersionsVersionNumberPatch, publishAgentV1AgentsAgentIdPublishPost, searchAgentsByIdsV1AgentsSearchPost, shareV1AgentsAgentIdSharePost, updateAgentV1AgentsAgentIdPatch, updateAgentV2AgentsAgentIdPatch, updateToolkitV1ToolkitsToolkitIdPatch, VisibilityLevelEnum } from '../api/agent-tools'
 import { DefaultAPIError } from '../error/DefaultAPIError'
 import { agentToolsDictionary } from '../error/dictionary/agent-tools'
 import { StackspotAPIError } from '../error/StackspotAPIError'
@@ -13,6 +13,7 @@ import { workspaceAiClient } from './workspace-ai'
 
 const AGENT_DEFAULT_SLUG = 'stk_flex'
 const listAgentsV3AgentsWithoutAuthorization = removeAuthorizationParam(listAgentsV3AgentsGet)
+const listToolkitsAuthorization = removeAuthorizationParam(listToolkitsV2ToolkitsGet)
 
 class AgentToolsClient extends ReactQueryNetworkClient {
   constructor() {
@@ -29,6 +30,7 @@ class AgentToolsClient extends ReactQueryNetworkClient {
   tools = this.query(removeAuthorizationParam(getPublicToolKitsV1BuiltinToolkitGet))
 
   /**
+   * @deprecated 
    * Create agent
    */
   createAgent = this.mutation(removeAuthorizationParam(createAgentV1AgentsPost))
@@ -39,6 +41,7 @@ class AgentToolsClient extends ReactQueryNetworkClient {
   deleteAgent = this.mutation(removeAuthorizationParam(deleteAgentV1AgentsAgentIdDelete))
 
   /**
+   * @deprecated 
    * Updates an agent
    */
   updateAgent = this.mutation(removeAuthorizationParam(updateAgentV1AgentsAgentIdPatch))
@@ -77,6 +80,7 @@ class AgentToolsClient extends ReactQueryNetworkClient {
   availableAgentsForMultiAgentAddition = this.query(removeAuthorizationParam(listMultiAgentsV1AgentsMultiAgentsGet))
 
   /** 
+   * @deprecated 
    * Gets agent by id
    */
   agent = this.query(removeAuthorizationParam(getAgentV1AgentsAgentIdGet))
@@ -84,7 +88,7 @@ class AgentToolsClient extends ReactQueryNetworkClient {
   /** 
    * Gets agents by ids
    */
-  agentsByIds = this.query(removeAuthorizationParam(searchAgentsV1AgentsSearchPost))
+  agentsByIds = this.query(removeAuthorizationParam(searchAgentsByIdsV1AgentsSearchPost))
 
   /** 
    * Gets the default agent slug
@@ -97,11 +101,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
     },
@@ -112,32 +116,44 @@ 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
     },
@@ -147,6 +163,10 @@ class AgentToolsClient extends ReactQueryNetworkClient {
    * 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.
    */
@@ -183,6 +203,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.
    */
@@ -220,6 +241,38 @@ class AgentToolsClient extends ReactQueryNetworkClient {
   * remove an agent from favorites 
   */
   removeFavorite = this.mutation(removeAuthorizationParam(deleteFavoriteV1AgentsAgentIdFavoriteDelete))
+  /**
+  * Creates an agent version
+  */
+  createAgentVersion = this.mutation(removeAuthorizationParam(createVersionV1AgentsAgentCoreIdVersionsPost))
+  /**
+  * V2 of agent creation
+  */
+  createAgentV2 = this.mutation(removeAuthorizationParam(createAgentV2AgentsPost))
+  /**
+  * V2 of agent update
+  */
+  updateAgentV2 = this.mutation(removeAuthorizationParam(updateAgentV2AgentsAgentIdPatch))
+  /**
+  * Update agent recommended version
+  */
+  updateRecommendedVersion = this.mutation(removeAuthorizationParam(patchVersionRecommendedV1AgentsAgentCoreIdVersionsVersionNumberPatch))
+  /** 
+   * V2 of get agent by id
+   */
+  agentV2 = this.query(removeAuthorizationParam(findByAgentCoreIdV2AgentsAgentCoreIdGet))
+  /** 
+   * Gets agent versions
+   */
+  listAgentVersions = this.query(removeAuthorizationParam(listVersionsOfAgentCoreV1AgentsAgentCoreIdVersionsGet))
+  /** 
+   * Deletes an agent version
+   */
+  deleteVersion = this.mutation(removeAuthorizationParam(deleteVersionV1AgentsAgentCoreIdVersionsVersionNumberDelete))
+  /** 
+   * Deletes an agent version
+   */
+  forkAgentVersion = this.mutation(removeAuthorizationParam(forkAgentVersionV1AgentsAgentCoreIdVersionsVersionNumberForkPost))
 }
 
 export const agentToolsClient = new AgentToolsClient()

package/src/client/ai.ts

@@ -1,5 +1,5 @@
 import { HttpError } from '@oazapfts/runtime'
-import { findLast, isArray, last } from 'lodash'
+import { isArray } from 'lodash'
 import { getApiAddresses } from '../api-addresses'
 import {
   addFavoriteV1AiStacksStackIdFavoritePost,
@@ -7,13 +7,17 @@ import {
   calculateNextStepV1QuickCommandsSlugStepsStepSlugCalculateNextStepPost,
   callbackV1QuickCommandsCallbackExecutionIdGet,
   conversationHistoryV1ConversationsConversationIdGet,
+  createAnswerForReviewV1ResourcesResourceTypeSlugResourceSlugReviewsReviewIdAnswersPost,
   createExecutionV1QuickCommandsCreateExecutionSlugPost,
+  createResourceReviewV1ResourcesResourceTypeSlugResourceSlugReviewsPost,
   currentV1TokensUsageCurrentGet,
   defaults,
   deleteConversationV1ConversationsConversationIdDelete,
   deleteFavoriteV1AiStacksStackIdFavoriteDelete,
   deleteFavoriteV1QuickCommandsSlugFavoriteDelete,
   deleteKnowledgeObjectByCustomIdV1KnowledgeSourcesSlugObjectsCustomIdDelete,
+  deleteResourceReviewV1ResourcesResourceTypeSlugResourceSlugReviewsReviewIdDelete,
+  deleteReviewCommentV1ResourcesResourceTypeSlugResourceSlugReviewsReviewIdAnswersAnswerIdDelete,
   downloadConversationV1ConversationsConversationIdDownloadGet,
   findKnowledgeObjectByCustomIdV1KnowledgeSourcesSlugObjectsCustomIdGet,
   findKnowledgeSourceV1KnowledgeSourcesSlugGet,
@@ -22,12 +26,14 @@ import {
   getContentDependenciesV1ContentContentTypeContentIdDependenciesGet,
   getFlagsV1FlagsGet,
   getQuickCommandV1QuickCommandsSlugGet,
+  getReviewsByResourceV1ResourcesResourceTypeSlugResourceSlugReviewsGet,
   getScriptExecutionStatusV1QuickCommandsScriptExecutionsScriptExecutionIdGet,
   getUploadFormV1FileUploadFormPost,
   HttpValidationError,
   listAiStacksV1AiStacksGet,
   listAllV1QuickCommandsAllGet,
   listAllV2QuickCommandsAllGet,
+  listAllV3QuickCommandsGet,
   listAssociationV1WorkspaceWorkspaceIdGet,
   listConversationsV1ConversationsGet,
   listKnowledgeSourcesV1KnowledgeSourcesGet,
@@ -40,28 +46,29 @@ import {
   runFetchStepV1QuickCommandsSlugStepsStepSlugFetchRunPost,
   searchKnowledgeSourcesV1KnowledgeSourcesSearchPost,
   startScriptStepV1QuickCommandsSlugStepsStepSlugStartScriptPost,
+  tokensByUserV1AnalyticsTokensByUserGet,
+  tokensDailyUsageV1AnalyticsTokensDailyUsageGet,
   totalV1TokensUsageTotalGet,
   updateQuickCommandV1QuickCommandsSlugPatch,
+  updateResourceReviewV1ResourcesResourceTypeSlugResourceSlugReviewsReviewIdPatch,
+  updateReviewCommentV1ResourcesResourceTypeSlugResourceSlugReviewsReviewIdAnswersAnswerIdPatch,
   updateTitleV1ConversationsConversationIdPatch,
   vectorizeCustomKnowledgeSourceV1KnowledgeSourcesSlugCustomPost,
 } from '../api/ai'
+
+
 import { StackspotAPIError } from '../error/StackspotAPIError'
 import { ReactQueryNetworkClient } from '../network/ReactQueryNetworkClient'
 import { removeAuthorizationParam } from '../utils/remove-authorization-param'
 import { StreamedJson } from '../utils/StreamedJson'
-import { formatJson } from '../utils/string'
-import { agentToolsClient } from './agent-tools'
 import {
-  ChatAgentTool,
-  ChatResponseWithSteps,
-  FixedChatRequest,
-  FixedChatResponse,
   FixedConversationResponse,
   FixedDependencyResponse,
   ReplaceResult,
-  StepChatStep,
 } from './types'
 
+const listQAV3WithoutAuthorization = removeAuthorizationParam(listAllV3QuickCommandsGet)
+
 class AIClient extends ReactQueryNetworkClient {
   constructor() {
     super(getApiAddresses().ai.url, defaults)
@@ -127,6 +134,10 @@ class AIClient extends ReactQueryNetworkClient {
    * Lists all the quick commands according to filters passed as parameter (including QC with secrets).
    */
   allQuickCommands = this.query(removeAuthorizationParam(listAllV2QuickCommandsAllGet))
+  /**
+   * Lists all the quick commands according to filters passed as parameter (including QC with secrets) with pagination.
+   */
+  allQuickCommandsV3 = this.infiniteQuery(listQAV3WithoutAuthorization, { accumulator: 'items' })
   /**
    * Gets a quick command by its slug.
    */
@@ -261,6 +272,14 @@ class AIClient extends ReactQueryNetworkClient {
    */
   removeFavoriteQuickCommand = this.mutation(removeAuthorizationParam(deleteFavoriteV1QuickCommandsSlugFavoriteDelete))
   /**
+   * Get Tokens Daily usage
+   */
+  analyticsTokensDailyUsage = this.query(removeAuthorizationParam(tokensDailyUsageV1AnalyticsTokensDailyUsageGet))
+  /**
+   * Get Tokens By User
+   */
+  analyticsTokensByUser = this.query(removeAuthorizationParam(tokensByUserV1AnalyticsTokensByUserGet))
+  /*
    * Lists all the feature flags for this user.
    */
   flags = this.query(getFlagsV1FlagsGet)
@@ -273,217 +292,45 @@ class AIClient extends ReactQueryNetworkClient {
    */
   getStatusScriptStep = this.query(removeAuthorizationParam(getScriptExecutionStatusV1QuickCommandsScriptExecutionsScriptExecutionIdGet))
 
-  private static async toolsOfAgent(agentId?: string) {
-    try {
-      const agent = agentId ? await agentToolsClient.agent.query({ agentId }) : undefined
-      if (!agent) return []
-      const tools: (Omit<ChatAgentTool, 'duration' | 'prompt' | 'output'>)[] = []
-      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 []
-    }
-  }
-
-  sendChatMessage(request: FixedChatRequest, minChangeIntervalMS?: number): StreamedJson<ChatResponseWithSteps> {
-    const abortController = new AbortController()
-    const headers = {
-      'Content-Type': 'application/json',
-      'Accept': 'text/event-stream',
-    }
-    const events = this.stream(
-      this.resolveURL('v3/chat'),
-      { method: 'post', body: JSON.stringify(request), headers, signal: abortController.signal },
-    )
-
-    /**
-     * This function treats events in the streaming that deals with the execution of tools. Since these events are not concatenated like
-     * normal streamings of data, we need this separate function to deal with them. It transforms the internal data model of the
-     * StreamedJson object whenever an event is triggered.
-     */
-    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] : []
-
-      if (info.type === 'planning' && info.action === 'end') {
-        data.steps.push({
-          id: 'planning',
-          type: 'planning',
-          status: 'success',
-          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',
-          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 === 'step' && info.action === 'start') {
-        const step = data.steps.find(s => s.id === info.id)
-        if (step) step.status = 'running'
-      }
-
-      if (info.type === 'step' && info.action === 'end') {
-        const step = data.steps.find(s => s.id === info.id) as StepChatStep
-        if (step) {
-          step.status = 'success'
-          step.duration = info.duration
-          const lastToolId = last(step.attempts[0].tools)?.id
-          const lastAttemptOfLastTool = findLast(step.attempts.map(a => a.tools).flat(), t => t?.id === lastToolId)
-          step.output = lastAttemptOfLastTool?.output
-        }
-      }
-
-      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' })
-      }
+  /**
+   * Get reviews by resource
+   */
+  getReviewsByResource = this.query(removeAuthorizationParam(getReviewsByResourceV1ResourcesResourceTypeSlugResourceSlugReviewsGet))
 
-      if (info.type === 'tool' && info.action === 'start') {
-        const currentStep = data.steps.find(s => s.status === 'running') as StepChatStep
-        if (!info.data) return
+  /**
+   * Create a resource review
+   */
+  createResourceReview = this.mutation(removeAuthorizationParam(createResourceReviewV1ResourcesResourceTypeSlugResourceSlugReviewsPost))
 
-        //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,
-              })
-            }
-          }
-        }
-      }
+  /**
+   * Create an answer for a review
+   */
+  createAnswerForReview = this.mutation(
+    removeAuthorizationParam(createAnswerForReviewV1ResourcesResourceTypeSlugResourceSlugReviewsReviewIdAnswersPost))
 
-      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 - 1]?.tools?.find(t => t.executionId === info.id)
-        if (tool) {
-          tool.output = formatJson(info.data.output)
-          tool.duration = info.duration
-        }
-      }
+  /**
+   * Update a resource review
+   */
+  updateResourceReview = this.mutation(
+    removeAuthorizationParam(updateResourceReviewV1ResourcesResourceTypeSlugResourceSlugReviewsReviewIdPatch))
 
-      if (info.type === 'final_answer' && info.action === 'start') {
-        const answerStep = last(data.steps)
-        if (answerStep) answerStep.status = 'running'
-      }
+  /**
+   * Delete a resource review
+   */
+  deleteResourceReview = this.mutation(
+    removeAuthorizationParam(deleteResourceReviewV1ResourcesResourceTypeSlugResourceSlugReviewsReviewIdDelete))
 
-      if (info.type === 'chat' && info.action === 'end') {
-        const answerStep = last(data.steps)
-        if (answerStep) {
-          answerStep.status = 'success'
-          answerStep.duration = info.duration
-        }
-      }
-    }
+  /**
+   * Update a review comment (answer)
+   */
+  updateReviewComment = this.mutation(
+    removeAuthorizationParam(updateReviewCommentV1ResourcesResourceTypeSlugResourceSlugReviewsReviewIdAnswersAnswerIdPatch))
 
-    return new StreamedJson({
-      eventsPromise: events,
-      abortController,
-      minChangeIntervalMS,
-      ignoreKeys: ['agent_info'],
-      transform,
-    })
-  }
+  /**
+   * Delete a review comment (answer)
+   */
+  deleteReviewComment = this.mutation(
+    removeAuthorizationParam(deleteReviewCommentV1ResourcesResourceTypeSlugResourceSlugReviewsReviewIdAnswersAnswerIdDelete))
 
   contentDependencies = this.query(removeAuthorizationParam(
     getContentDependenciesV1ContentContentTypeContentIdDependenciesGet as ReplaceResult<

package/src/client/code-shift.ts

@@ -75,6 +75,11 @@ import {
   createReposBatchServiceV2ReposBatchPost,
   getImportResultV2ReposBatchImportIdGet,
   downloadImportResultV2ReposBatchImportIdDownloadGet,
+  deleteModuleServiceV1ModulesModuleIdDelete,
+  createRepositoryServiceV2ReposPost,
+  listRepositoryServiceV2ReposGet,
+  listModuleExecutionBatchActivitiesV1ActivitiesModuleExecutionBatchGet,
+  getModuleExecutionBatchActivitiesByIdV1ActivitiesModuleExecutionBatchModuleExecutionBatchIdGet,
 } from '../api/codeShift'
 import { DefaultAPIError } from '../error/DefaultAPIError'
 import { codeShiftDictionary } from '../error/dictionary/code-shift'
@@ -96,6 +101,10 @@ class CodeShift extends ReactQueryNetworkClient {
    * Creates a repository.
    */
   createRepository = this.mutation(removeAuthorizationParam(createRepositoryServiceV1ReposPost))
+  /**
+  * Creates a repository (v2).
+  */
+  createRepositoryV2 = this.mutation(removeAuthorizationParam(createRepositoryServiceV2ReposPost))
   /**
    * Creates repositories in batch.
    */
@@ -116,6 +125,10 @@ class CodeShift extends ReactQueryNetworkClient {
    * Gets list of repositories.
    */
   repositories = this.query(removeAuthorizationParam(listRepositoryServiceV1ReposGet))
+  /**
+   * Gets list of repositories (v2).
+   */
+  repositoriesV2 = this.query(removeAuthorizationParam(listRepositoryServiceV2ReposGet))  
   /**
    * Gets a repository
    */
@@ -144,6 +157,10 @@ class CodeShift extends ReactQueryNetworkClient {
    * Gets module by id.
    */
   module = this.query(removeAuthorizationParam(getModuleV1ModulesModuleIdGet))
+  /**
+   * Deletes a module by id.
+   */
+  deleteModule = this.mutation(removeAuthorizationParam(deleteModuleServiceV1ModulesModuleIdDelete))
   /**
   * Add a module in favorites.
   */
@@ -428,6 +445,18 @@ class CodeShift extends ReactQueryNetworkClient {
   analyticsModulesExecutionTimesDownload = this.query(
     removeAuthorizationParam(analyticsModuleExecutionTimesDownloadV1AnalyticsModulesExecutionTimesDownloadGet),
   )
+  /**
+   * List Module Execution Batch Activities
+   */
+  activitiesModuleExecutionBatch = this.query(
+    removeAuthorizationParam(listModuleExecutionBatchActivitiesV1ActivitiesModuleExecutionBatchGet),
+  )
+  /**
+   * Get Module Execution Batch Activity by ID
+   */
+  activityModuleExecutionBatch = this.query(
+    removeAuthorizationParam(getModuleExecutionBatchActivitiesByIdV1ActivitiesModuleExecutionBatchModuleExecutionBatchIdGet),
+  )
 }
 
 export const codeShiftClient = new CodeShift()

package/src/client/data-integration.ts

@@ -99,7 +99,14 @@ class DataIntegrationClient extends ReactQueryNetworkClient {
   /**
    * List of knowledge sources with pagination
    */
-  knowledgeSourcesV2 = this.infiniteQuery(listKSV2WithoutAuthorization, { accumulator: 'items' })
+  knowledgeSourcesV2 = this.infiniteQuery(listKSV2WithoutAuthorization, {
+    accumulator: 'items',
+    getNextPageParam: ({ variables, lastPage, lastPageParam }) => {
+      const size = variables.size ?? 1
+      const parsedLastPageParam = (lastPageParam as number) ?? variables.page ?? 1
+      return lastPage.items && lastPage.items.length < size ? undefined : parsedLastPageParam + 1
+    },
+  })
   /**
    * Adds the resource of type Knowledge Source to the list of favorites.
    */
@@ -199,7 +206,7 @@ class DataIntegrationClient extends ReactQueryNetworkClient {
    * Get Indexed Files
    */
   listIndexedProjectFiles = this.query(removeAuthorizationParam(getIndexedFilesV1KnowledgeSourcesKsSlugFilesGet))
-  
+
 }
 
 export const dataIntegrationClient = new DataIntegrationClient()

package/src/client/discover.ts

@@ -2,16 +2,16 @@ import { HttpError } from '@oazapfts/runtime'
 import { findLast, last } from 'lodash'
 import { getApiAddresses } from '../api-addresses'
 import { ConversationResponse } from '../api/ai'
-import { create, create1, create2, defaults, deleteById, deleteById1, deleteById2, fetchInsights, getAll, getAll1, getAll2, getAllByHypothesis, getById, getById1, getById2, getInsightById, GetOpportunityResponse, MessageRequest, refreshInsights } from '../api/discover'
+import { AiChatRequest, create, create1, create2, defaults, deleteById, deleteById1, deleteById2, fetchInsights, getAll, getAll1, getAll2, getAllByHypothesis, getById, getById1, getById2, getInsightById, GetOpportunityResponse, refreshInsights } from '../api/discover'
 import { DefaultAPIError } from '../error/DefaultAPIError'
+import { baseDictionary } from '../error/dictionary/base'
 import { StackspotAPIError } from '../error/StackspotAPIError'
+import { ReactQueryNetworkClient } from '../network/ReactQueryNetworkClient'
 import { StreamedJson } from '../utils/StreamedJson'
-import { baseDictionary } from '../error/dictionary/base'
 import { formatJson } from '../utils/string'
-import { ReactQueryNetworkClient } from '../network/ReactQueryNetworkClient'
-import { aiClient } from './ai'
-import { ChatAgentTool, ChatResponseWithSteps, FixedChatResponse, StepChatStep } from './types'
 import { agentToolsClient } from './agent-tools'
+import { aiClient } from './ai'
+import { AgentInfo, ChatAgentTool, ChatResponseWithSteps, FixedChatResponse, StepChatStep } from './types'
 
 export interface ChatConversionDetails extends ConversationResponse {
   opportunityName?: string,
@@ -121,7 +121,7 @@ class DiscoverClient extends ReactQueryNetworkClient {
   }
 
 
-  sendChatMessage(request: MessageRequest & { agentId: string }, minChangeIntervalMS?: number): StreamedJson<ChatResponseWithSteps> {
+  sendChatMessage(request: AiChatRequest & { agentId: string }, minChangeIntervalMS?: number): StreamedJson<ChatResponseWithSteps> {
     const abortController = new AbortController()
     const headers = {
       'Content-Type': 'application/json',
@@ -137,7 +137,8 @@ class DiscoverClient extends ReactQueryNetworkClient {
        * StreamedJson object whenever an event is triggered.
        */
     async function transform(event: Partial<FixedChatResponse>, data: Partial<ChatResponseWithSteps>) {
-      const info = event.agent_info
+      // todo fix
+      const info = event.agent_info as unknown as AgentInfo
 
       if (!info) return