@stack-spot/portal-network 0.218.2 → 0.219.0

package/src/api/ai.ts

@@ -384,6 +384,7 @@ export type QuickCommandsStepPromptRequest = {
     allow_use_current_workspace?: boolean;
     knowledge_source_slugs?: string[] | null;
     agent_id?: string | null;
+    agent_version_number?: number | null;
     use_uploaded_files?: boolean;
     agent_built_in?: boolean | null;
     next_step_slug?: string | null;
@@ -871,6 +872,7 @@ export type ChatRequest = {
     } | null;
     user_prompt: string;
     project_id?: string | null;
+    agent_version_number?: number | null;
 };
 export type SourceProjectFile3 = {
     "type": "project_file";
@@ -4061,14 +4063,13 @@ export function devAssistantV3V3ChatPost({ authorization, xAccountId, xMemberId,
 /**
  * List All
  */
-export function listAllV3QuickCommandsGet({ name, slug, size, page, visibilityList, order, isRemote, types, authorization, xAccountId, xMemberId, xUsername }: {
+export function listAllV3QuickCommandsGet({ name, slug, size, page, visibilityList, order, types, authorization, xAccountId, xMemberId, xUsername }: {
     name?: string | null;
     slug?: string | null;
     size?: number;
     page?: number;
     visibilityList?: VisibilityLevelEnum[] | null;
     order?: OrderEnum | null;
-    isRemote?: boolean;
     types?: QuickCommandTypeRequest[] | null;
     authorization: string;
     xAccountId?: string | null;
@@ -4090,7 +4091,6 @@ export function listAllV3QuickCommandsGet({ name, slug, size, page, visibilityLi
         page,
         visibility_list: visibilityList,
         order,
-        is_remote: isRemote,
         types
     }))}`, {
         ...opts,

package/src/api/genAiInference.ts

@@ -69,7 +69,7 @@ export type ValidationError = {
 export type HttpValidationError = {
     detail?: ValidationError[];
 };
-export type ResourceName = "agents" | "deep_ks" | "autocomplete";
+export type ResourceName = "agents" | "deep_ks" | "autocomplete" | "summary";
 export type LlmModelResourceRequest = {
     name: ResourceName;
     is_default?: boolean;
@@ -244,6 +244,8 @@ export type ChatCompletion = {
     } | null;
     /** ID of an alternative model for chat completion. It should be present on available_model_ids agent attr. */
     selected_model_id?: string | null;
+    /** The specific Agent version number to be use in chat. */
+    agent_version_number?: number | null;
 };
 export type ChatRequest = {
     streaming: boolean;
@@ -260,6 +262,8 @@ export type ChatRequest = {
     knowledge_sources?: string[];
     deep_search_ks?: boolean;
     show_chat_processing_state?: boolean;
+    /** The specific Agent version number to be use in chat. */
+    agent_version_number?: number | null;
 };
 export type PromptTokensCurrentUsageResponse = {
     used: number;

package/src/api/workspace-ai.ts

@@ -84,6 +84,13 @@ export type ListWksContentsResponse = {
         [key: string]: any;
     }[] | null;
 };
+export type ContentVersion = {
+    core_id: string;
+    version_number?: number | null;
+};
+export type AddWorkspaceContentRequestV2 = {
+    content_identifier: ContentVersion[];
+};
 export type ActionShared = "VIEW" | "EDIT" | "GRANT_ACESS";
 export type PermissionForSharedRequest = {
     userId: string;
@@ -100,6 +107,12 @@ export type ResourceMembersResponse = {
     name: string;
     actions: string[];
 };
+export type ResourceMembersPaginatedResponse = {
+    items: ResourceMembersResponse[];
+    page: number;
+    size: number;
+    total_pages: number;
+};
 /**
  * Create Workspace
  */
@@ -534,6 +547,36 @@ export function listAllContentsV1WorkspacesWorkspaceIdContentsGet({ workspaceId,
         })
     }));
 }
+/**
+ * Add Content
+ */
+export function addContentV2WorkspacesWorkspaceIdContentTypePost({ workspaceId, contentType, authorization, xAccountId, addWorkspaceContentRequestV2 }: {
+    workspaceId: string;
+    contentType: ContentType;
+    authorization: string;
+    xAccountId?: string | null;
+    addWorkspaceContentRequestV2: AddWorkspaceContentRequestV2;
+}, opts?: Oazapfts.RequestOpts) {
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 200;
+        data: {
+            [key: string]: any;
+        };
+    } | {
+        status: 404;
+    } | {
+        status: 422;
+        data: HttpValidationError;
+    }>(`/v2/workspaces/${encodeURIComponent(workspaceId)}/${encodeURIComponent(contentType)}`, oazapfts.json({
+        ...opts,
+        method: "POST",
+        body: addWorkspaceContentRequestV2,
+        headers: oazapfts.mergeHeaders(opts?.headers, {
+            authorization,
+            "x-account-id": xAccountId
+        })
+    })));
+}
 /**
  * Use Shareable Link
  */
@@ -653,6 +696,46 @@ export function listMembersV1ResourceTypeResourceTypeResourcesResourceIdentifier
         })
     }));
 }
+/**
+ * List Members
+ */
+export function listMembersV2ResourceTypeResourceTypeResourcesResourceIdentifierMembersGet({ resourceType, resourceIdentifier, search, filterBy, filterValue, sortBy, page, size, order, authorization, xAccountId }: {
+    resourceType: ContentType;
+    resourceIdentifier: string;
+    search?: string | null;
+    filterBy?: string | null;
+    filterValue?: string | null;
+    sortBy?: SortBy | null;
+    page?: number;
+    size?: number;
+    order?: OrderEnum;
+    authorization: string;
+    xAccountId?: string | null;
+}, opts?: Oazapfts.RequestOpts) {
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 200;
+        data: ResourceMembersPaginatedResponse;
+    } | {
+        status: 404;
+    } | {
+        status: 422;
+        data: HttpValidationError;
+    }>(`/v2/resource-type/${encodeURIComponent(resourceType)}/resources/${encodeURIComponent(resourceIdentifier)}/members${QS.query(QS.explode({
+        search,
+        filter_by: filterBy,
+        filter_value: filterValue,
+        sort_by: sortBy,
+        page,
+        size,
+        order
+    }))}`, {
+        ...opts,
+        headers: oazapfts.mergeHeaders(opts?.headers, {
+            authorization,
+            "x-account-id": xAccountId
+        })
+    }));
+}
 /**
  * Healthz
  */

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, listToolkitsV2ToolkitsGet, 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, listAgentsV4AgentsGet, listMcpToolsV1McpToolsToolkitIdGet, listMultiAgentsV1AgentsMultiAgentsGet, listToolkitsV1ToolkitsGet, listToolkitsV2ToolkitsGet, listVersionsOfAgentCoreV1AgentsAgentCoreIdVersionsGet, patchVersionRecommendedV1AgentsAgentCoreIdVersionsVersionNumberPatch, publishAgentV1AgentsAgentIdPublishPost, searchAgentsByIdsV1AgentsSearchPost, searchAgentsV2AgentsSearchPost, 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 listAgentsV4 = removeAuthorizationParam(listAgentsV4AgentsGet)
 const listToolkitsAuthorization = removeAuthorizationParam(listToolkitsV2ToolkitsGet)
 
 class AgentToolsClient extends ReactQueryNetworkClient {
@@ -30,6 +31,7 @@ class AgentToolsClient extends ReactQueryNetworkClient {
   tools = this.query(removeAuthorizationParam(getPublicToolKitsV1BuiltinToolkitGet))
 
   /**
+   * @deprecated 
    * Create agent
    */
   createAgent = this.mutation(removeAuthorizationParam(createAgentV1AgentsPost))
@@ -40,6 +42,7 @@ class AgentToolsClient extends ReactQueryNetworkClient {
   deleteAgent = this.mutation(removeAuthorizationParam(deleteAgentV1AgentsAgentIdDelete))
 
   /**
+   * @deprecated 
    * Updates an agent
    */
   updateAgent = this.mutation(removeAuthorizationParam(updateAgentV1AgentsAgentIdPatch))
@@ -78,14 +81,16 @@ class AgentToolsClient extends ReactQueryNetworkClient {
   availableAgentsForMultiAgentAddition = this.query(removeAuthorizationParam(listMultiAgentsV1AgentsMultiAgentsGet))
 
   /** 
+   * @deprecated 
    * Gets agent by id
    */
   agent = this.query(removeAuthorizationParam(getAgentV1AgentsAgentIdGet))
 
   /** 
+   * @deprecated 
    * Gets agents by ids
    */
-  agentsByIds = this.query(removeAuthorizationParam(searchAgentsV1AgentsSearchPost))
+  agentsByIds = this.query(removeAuthorizationParam(searchAgentsByIdsV1AgentsSearchPost))
 
   /** 
    * Gets the default agent slug
@@ -238,6 +243,54 @@ 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))
+  /** 
+   * Deletes an agent version
+   */
+  deleteVersion = this.mutation(removeAuthorizationParam(deleteVersionV1AgentsAgentCoreIdVersionsVersionNumberDelete))
+  /** 
+   * Gets agent versions
+   */
+  listAgentVersions = this.query(removeAuthorizationParam(listVersionsOfAgentCoreV1AgentsAgentCoreIdVersionsGet))
+  /** 
+   * Get agents by list of ids using api v2
+   */
+  agentsByIdsV2 = this.query(removeAuthorizationParam(searchAgentsV2AgentsSearchPost))
+  /** 
+   * Deletes an agent version
+   */
+  forkAgentVersion = this.mutation(removeAuthorizationParam(forkAgentVersionV1AgentsAgentCoreIdVersionsVersionNumberForkPost))
+  /** 
+   * Get agents v4
+   */
+  agentsV4 = this.infiniteQuery(listAgentsV4, {
+    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
+    },
+  })
 }
 
 export const agentToolsClient = new AgentToolsClient()

package/src/client/discover.ts

@@ -1,7 +1,7 @@
 import { HttpError } from '@oazapfts/runtime'
 import { findLast, last } from 'lodash'
 import { getApiAddresses } from '../api-addresses'
-import { ChatResponse3, ConversationResponse } from '../api/ai'
+import { ConversationResponse } from '../api/ai'
 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'
@@ -11,10 +11,10 @@ import { StreamedJson } from '../utils/StreamedJson'
 import { formatJson } from '../utils/string'
 import { agentToolsClient } from './agent-tools'
 import { aiClient } from './ai'
-import { AgentInfo, ChatAgentTool, ChatStep, StepChatStep } from './types'
+import { AgentInfo, ChatAgentTool, ChatResponse, ChatStep, StepChatStep } from './types'
 
-export interface FixedDiscoveryChatResponse extends ChatResponse3 {
-  agent_info: AgentInfo,
+export interface FixedDiscoveryChatResponse extends ChatResponse {
+  agent_info: AgentInfo[],
   tools?: string[],
 }
 
@@ -131,7 +131,7 @@ class DiscoverClient extends ReactQueryNetworkClient {
 
 
   sendChatMessage(request: AiChatRequest & { agentId: string }, minChangeIntervalMS?: number)
-  : StreamedJson<DiscoveryChatResponseWithSteps> {
+    : StreamedJson<DiscoveryChatResponseWithSteps> {
     const abortController = new AbortController()
     const headers = {
       'Content-Type': 'application/json',
@@ -147,171 +147,175 @@ class DiscoverClient extends ReactQueryNetworkClient {
        * StreamedJson object whenever an event is triggered.
        */
     async function transform(event: Partial<FixedDiscoveryChatResponse>, data: Partial<DiscoveryChatResponseWithSteps>) {
-      const info = event.agent_info
+      const agentInfoList = event.agent_info
 
-      if (!info) return
+      if (!agentInfoList?.length) return
 
       const tools = await DiscoverClient.toolsOfAgent(request.agentId)
       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' })
-      }
+      for (const info of agentInfoList) {
+        if (!info) continue
 
-      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 === '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 ?? '',
+          })
 
-      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
+          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 === '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,
+        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' })
-      }
+          }))
+          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 (!info.data) return
+        if (info.type === 'step' && info.action === 'start') {
+          const step = data.steps.find(s => s.id === info.id)
+          if (step) step.status = 'running'
+        }
 
-        //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 }
+        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: 'running',
+            status: 'awaiting_approval',
             duration: info.duration || 0,
             input: info.data?.input,
             user_question: info.data?.user_question,
             attempts: [{
-              tools: [{ ...tool, executionId: info.id, input }],
+              tools: [{
+                executionId: info.id,
+                id: info.data?.tool_id ?? '',
+                name: tool?.name ?? '',
+                goal: tool?.goal,
+                ...tool,
+              }],
             }],
           })
-        } 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) {
+          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 (!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 }
-            currentStep.attempts[info.data.attempt - 1].tools?.push({
-              ...tool,
-              executionId: info.id,
-              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 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: [] }
+            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 - 1]?.tools?.find(t => t.executionId === info.id)
-        if (tool) {
-          tool.output = formatJson(info.data.output)
-          tool.duration = info.duration
+        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
+          }
         }
-      }
 
-      if (info.type === 'final_answer' && info.action === 'start') {
-        const answerStep = last(data.steps)
-        if (answerStep) answerStep.status = 'running'
-      }
+        if (info.type === 'final_answer' && info.action === 'start') {
+          const answerStep = last(data.steps)
+          if (answerStep) answerStep.status = 'running'
+        }
 
-      if (info.type === 'chat' && info.action === 'end') {
-        const answerStep = last(data.steps)
-        if (answerStep) {
-          answerStep.status = 'success'
-          answerStep.duration = info.duration
+        if (info.type === 'chat' && info.action === 'end') {
+          const answerStep = last(data.steps)
+          if (answerStep) {
+            answerStep.status = 'success'
+            answerStep.duration = info.duration
+          }
         }
       }
     }
@@ -319,9 +323,13 @@ class DiscoverClient extends ReactQueryNetworkClient {
     return new StreamedJson({
       eventsPromise: events,
       abortController,
-      minChangeIntervalMS,
       ignoreKeys: ['agent_info'],
       transform,
+      minChangeIntervalMS,
+      minChangeIntervalMSFromEvent: (event) => {
+        if (event?.agent_info?.length) return 0
+        return minChangeIntervalMS
+      },
     })
   }
 }

package/src/client/types.ts

@@ -186,6 +186,7 @@ export interface FixedConversationHistoryResponse extends ConversationHistoryRes
   custom_agent?: {
     id: string,
     built_in: boolean,
+    version_number?: number,
   },
 }