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

package/src/client/code-shift.ts

@@ -48,10 +48,42 @@ import {
   updateRepositoryServiceV1ReposRepositoryIdPut,
   validateScmUrlServiceV1ReposValidateScmUrlPost,
   listUserServiceV1UsersGet,
+  analyticsProgramGroupsDetailsV1AnalyticsProgramGroupsDetailsGet,
+  analyticsProgramGroupsDetailsDownloadV1AnalyticsProgramGroupsDetailsDownloadGet,
+  getModuleInputsV1ModulesModuleIdInputsGet,
+  analyticsRepositoryDetailedReportDownloadV1AnalyticsRepositoriesDetailsDownloadGet,
+  analyticsRepositoryDetailedReportV1AnalyticsRepositoriesDetailsGet,
+  listRepositoryDownloadServiceV1ReposDownloadGet,
+  listProgramGroupDownloadServiceV1ProgramGroupsDownloadGet,
+  getModuleV1ModulesModuleIdGet,
+  analyticsProgramGroupsTargetDetailsV1AnalyticsProgramGroupsTargetDetailsGet,
+  analyticsProgramGroupsTargetDetailsDownloadV1AnalyticsProgramGroupsTargetDetailsDownloadGet,
+  searchReposScmServiceV2ReposSearchScmPost,
+  importReposWithTagsScmServiceV2ReposSearchScmSearchIdPost,
+  searchReposScmV2V2ReposSearchScmSearchIdGet,
+  analyticsRepositoryTargetDetailsV1AnalyticsRepositoriesTargetDetailsGet,
+  analyticsRepositoryTargetDetailsDownloadV1AnalyticsRepositoriesTargetDetailsDownloadGet,
+  putCustomerRatingReportV1ReportsReportIdCustomerRatingPut,
+  updateModuleServiceV1ModulesModuleIdPut,
+  downloadSearchReposScmV2V2ReposSearchScmSearchIdDownloadGet,
+  moduleFavoriteServiceAddV1ModulesModuleIdFavoritesPost,
+  moduleFavoriteServiceDeleteV1ModulesModuleIdFavoritesDelete,
+  patHealthCheckV1ScmPatHealthCheckGet,
+  getModuleDocsV1ModulesModuleIdDocsGet,
+  analyticsModuleExecutionTimesReportV1AnalyticsModulesExecutionTimesGet,
+  analyticsModuleExecutionTimesDownloadV1AnalyticsModulesExecutionTimesDownloadGet,
+  createReposBatchServiceV2ReposBatchPost,
+  getImportResultV2ReposBatchImportIdGet,
+  downloadImportResultV2ReposBatchImportIdDownloadGet,
+  deleteModuleServiceV1ModulesModuleIdDelete,
+  createRepositoryServiceV2ReposPost,
+  listRepositoryServiceV2ReposGet,
+  listModuleExecutionBatchActivitiesV1ActivitiesModuleExecutionBatchGet,
+  getModuleExecutionBatchActivitiesByIdV1ActivitiesModuleExecutionBatchModuleExecutionBatchIdGet,
 } from '../api/codeShift'
 import { DefaultAPIError } from '../error/DefaultAPIError'
 import { codeShiftDictionary } from '../error/dictionary/code-shift'
-import { StackspotAPIError } from '../error/StackspotAPIError'
+import { StackspotAPIError } from '../error/StackspotAPIError'                      
 import { ReactQueryNetworkClient } from '../network/ReactQueryNetworkClient'
 import { removeAuthorizationParam } from '../utils/remove-authorization-param'
 import { getApiAddresses } from '../api-addresses'
@@ -69,14 +101,34 @@ 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.
    */
   createRepositoriesBatch = this.mutation(removeAuthorizationParam(createReposBatchServiceV1ReposBatchPost))
+  /**
+   * Imports repositories in batch.
+   */
+  createRepositoriesBatchV2 = this.mutation(removeAuthorizationParam(createReposBatchServiceV2ReposBatchPost))
+  /**
+   * Gets the status of a batch import.
+   */
+  getImportBatchResultsV2 = this.query(removeAuthorizationParam(getImportResultV2ReposBatchImportIdGet))
+  /**
+   * Downloads the import result for a batch.
+   */
+  downloadImportBatchResultsV2 = this.query(removeAuthorizationParam(downloadImportResultV2ReposBatchImportIdDownloadGet))
   /**
    * Gets list of repositories.
    */
   repositories = this.query(removeAuthorizationParam(listRepositoryServiceV1ReposGet))
+  /**
+   * Gets list of repositories (v2).
+   */
+  repositoriesV2 = this.query(removeAuthorizationParam(listRepositoryServiceV2ReposGet))  
   /**
    * Gets a repository
    */
@@ -89,6 +141,10 @@ class CodeShift extends ReactQueryNetworkClient {
    * Updates a repository
    */
   updateRepository = this.mutation(removeAuthorizationParam(updateRepositoryServiceV1ReposRepositoryIdPut))
+  /**
+   * List Repository Download Service
+   */
+  repositoryDownload = this.query(removeAuthorizationParam(listRepositoryDownloadServiceV1ReposDownloadGet))
   /**
    * Get a report for a pull request by id.
    */
@@ -97,14 +153,42 @@ class CodeShift extends ReactQueryNetworkClient {
    * Gets modules.
    */
   modules = this.query(removeAuthorizationParam(listModulesServiceV1ModulesGet))
+  /**
+   * Gets module by id.
+   */
+  module = this.query(removeAuthorizationParam(getModuleV1ModulesModuleIdGet))
+  /**
+   * Deletes a module by id.
+   */
+  deleteModule = this.mutation(removeAuthorizationParam(deleteModuleServiceV1ModulesModuleIdDelete))
+  /**
+  * Add a module in favorites.
+  */
+  addFavoriteModule = this.mutation(removeAuthorizationParam(moduleFavoriteServiceAddV1ModulesModuleIdFavoritesPost))
+  /**
+  * Remove a module from favorites.
+  */
+  removeFavoriteModule = this.mutation(removeAuthorizationParam(moduleFavoriteServiceDeleteV1ModulesModuleIdFavoritesDelete))
+  /**
+   * Gets module inputs.
+   */
+  modulesInputs = this.query(removeAuthorizationParam(getModuleInputsV1ModulesModuleIdInputsGet))
   /**
    * Creates a module.
    */
   createModule = this.mutation(removeAuthorizationParam(createModuleServiceV1ModulesPost))
+  /**
+   * Updates a module.
+   */
+  updateModule = this.mutation(removeAuthorizationParam(updateModuleServiceV1ModulesModuleIdPut))
   /**
    * Generates a report.
    */
   generateReport = this.mutation(removeAuthorizationParam(dispatchModuleServiceV1ModulesDispatchesPost))
+  /**
+   * Gets module docs.
+   */
+  moduleDocs = this.query(removeAuthorizationParam(getModuleDocsV1ModulesModuleIdDocsGet))
   /**
    * Gets repository reports.
    */
@@ -117,6 +201,12 @@ class CodeShift extends ReactQueryNetworkClient {
    * Downloads a report as a csv file.
    */
   downloadReport = this.query(removeAuthorizationParam(downloadReportV1ReportsReportIdDownloadGet))
+  /**
+   * Put Customer Rating Report
+   */
+  updateReportRating = this.mutation(
+    removeAuthorizationParam(putCustomerRatingReportV1ReportsReportIdCustomerRatingPut),
+  )
   /**
    * Gets code shift settings
    */
@@ -134,9 +224,28 @@ class CodeShift extends ReactQueryNetworkClient {
    */
   searchRepositoryStatus = this.query(removeAuthorizationParam(getStatusSearchReposScmServiceV1ReposSearchScmSearchRepoIdStatusGet))
   /**
+   * @deprecated
    * Downloads file with found repositories
    */
   downloadSearchRepository = this.mutation(removeAuthorizationParam(downloadSearchReposScmServiceV1ReposSearchScmSearchRepoIdDownloadGet))
+  /**
+   * Searches for repositories (v2)
+   */
+  searchRepositoryV2 = this.mutation(removeAuthorizationParam(searchReposScmServiceV2ReposSearchScmPost))
+  /**
+   * Imports repositories (v2)
+   */
+  importRepositories = this.mutation(removeAuthorizationParam(importReposWithTagsScmServiceV2ReposSearchScmSearchIdPost))
+  /**
+   * Gets repositories search by id (v2)
+   */
+  getRepositoriesBySearchId = this.query(removeAuthorizationParam(searchReposScmV2V2ReposSearchScmSearchIdGet))
+  /**
+   * Downloads file with found repositories (v2)
+   */
+  downloadSearchRepositoryV2 = this.mutation(
+    removeAuthorizationParam(downloadSearchReposScmV2V2ReposSearchScmSearchIdDownloadGet),
+  )
   /**
    * Validate if the user has permission. 
    * We do not use opa in this api, so this is the fn needed to check permissions.
@@ -192,6 +301,10 @@ class CodeShift extends ReactQueryNetworkClient {
    * Deletes a program group.
    */
   deleteProgramGroup = this.mutation(removeAuthorizationParam(deleteProgramGroupServiceV1ProgramGroupsProgramGroupIdDelete))
+  /**
+   * List Program Group Download Service
+   */
+  programGroupDownload = this.query(removeAuthorizationParam(listProgramGroupDownloadServiceV1ProgramGroupsDownloadGet))
   /**
    * Gets list of tags.
    */
@@ -200,11 +313,14 @@ class CodeShift extends ReactQueryNetworkClient {
    * Validates a SCM URL.
    */
   validateSCMUrl = this.mutation(removeAuthorizationParam(validateScmUrlServiceV1ReposValidateScmUrlPost))
+  /**
+   * PAT Health Check
+   */
+  checkScmPatHealth = this.query(removeAuthorizationParam(patHealthCheckV1ScmPatHealthCheckGet))
   /**
    * Get Branches for a Repository
    */
   getBranches = this.query(removeAuthorizationParam(listBranchesServiceV1ReposBranchesGet))
-
   /**
    * General Report Success And Errors
    */
@@ -234,9 +350,11 @@ class CodeShift extends ReactQueryNetworkClient {
     removeAuthorizationParam(analyticsProgramGroupsLastReportStatusV1AnalyticsProgramGroupsLastReportStatusGet),
   )
   /**
-   * Analytics Program Groups Usage
+   * Analytics Program Groups Last Report Status Download
    */
-  analyticsProgramGroupsUsage = this.query(removeAuthorizationParam(analyticsProgramGroupsUsageV1AnalyticsProgramGroupsUsageGet))
+  analyticsProgramGroupsLastReportStatusDownload = this.query(
+    removeAuthorizationParam(analyticsProgramGroupsLastReportStatusDownloadV1AnalyticsProgramGroupsLastReportStatusDownloadGet),
+  )
   /**
    * Analytics User Usage
    */
@@ -252,11 +370,9 @@ class CodeShift extends ReactQueryNetworkClient {
     removeAuthorizationParam(analyticsRepositoryUsageDownloadV1AnalyticsRepositoriesUsageDownloadGet),
   )
   /**
-   * Analytics Program Groups Last Report Status Download
+   * Analytics Program Groups Usage
    */
-  analyticsProgramGroupsLastReportStatusDownload = this.query(
-    removeAuthorizationParam(analyticsProgramGroupsLastReportStatusDownloadV1AnalyticsProgramGroupsLastReportStatusDownloadGet),
-  )
+  analyticsProgramGroupsUsage = this.query(removeAuthorizationParam(analyticsProgramGroupsUsageV1AnalyticsProgramGroupsUsageGet))
   /**
    * Analytics Program Groups Usage Download
    */
@@ -269,6 +385,78 @@ class CodeShift extends ReactQueryNetworkClient {
   getUsersService = this.query(
     removeAuthorizationParam(listUserServiceV1UsersGet),
   )
+  /**
+   * Analytics Program Groups Details
+   */
+  analyticsProgramGroupsDetails = this.query(
+    removeAuthorizationParam(analyticsProgramGroupsDetailsV1AnalyticsProgramGroupsDetailsGet),
+  )
+  /**
+   * Analytics Program Groups Details Download
+   */
+  analyticsProgramGroupsDetailsDownload = this.query(
+    removeAuthorizationParam(analyticsProgramGroupsDetailsDownloadV1AnalyticsProgramGroupsDetailsDownloadGet),
+  )
+  /**
+   * Analytics Program Groups Details
+   */
+  analyticsRepositoryDetails = this.query(
+    removeAuthorizationParam(analyticsRepositoryDetailedReportV1AnalyticsRepositoriesDetailsGet),
+  )
+  /**
+   * Analytics Program Groups Details Download
+  */
+  analyticsRepositoryDetailsDownload = this.query(
+    removeAuthorizationParam(analyticsRepositoryDetailedReportDownloadV1AnalyticsRepositoriesDetailsDownloadGet),
+  )
+  /**
+   * Analytics Program Groups Target Details
+   */
+  analyticsProgramGroupsTargetDetails = this.query(
+    removeAuthorizationParam(analyticsProgramGroupsTargetDetailsV1AnalyticsProgramGroupsTargetDetailsGet),
+  )
+  /**
+   * Analytics Program Groups Target Details Download
+   */
+  analyticsProgramGroupsTargetDetailsDownload = this.query(
+    removeAuthorizationParam(analyticsProgramGroupsTargetDetailsDownloadV1AnalyticsProgramGroupsTargetDetailsDownloadGet),
+  )
+  /** 
+   * Analytics Repository Target Details
+   */
+  analyticsRepositoryTargetDetails = this.query(
+    removeAuthorizationParam(analyticsRepositoryTargetDetailsV1AnalyticsRepositoriesTargetDetailsGet),
+  )
+  /**
+   * Analytics Repository Target Details Download
+   */
+  analyticsRepositoryTargetDetailsDownload = this.query(
+    removeAuthorizationParam(analyticsRepositoryTargetDetailsDownloadV1AnalyticsRepositoriesTargetDetailsDownloadGet),
+  )
+  /**
+   * Analytics Modules Execution Times
+   */
+  analyticsModulesExecutionTimes = this.query(
+    removeAuthorizationParam(analyticsModuleExecutionTimesReportV1AnalyticsModulesExecutionTimesGet),
+  )
+  /**
+   * Analytics Modules Execution Times Download
+   */
+  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

@@ -1,4 +1,5 @@
 import { HttpError } from '@oazapfts/runtime'
+import { getApiAddresses } from '../api-addresses'
 import {
   addFavoriteV1KnowledgeSourcesSlugFavoritePost,
   createKsV1KnowledgeSourcesPost,
@@ -10,10 +11,13 @@ import {
   findKsV1KnowledgeSourcesSlugGet,
   forkKsV1KnowledgeSourcesSlugForkPost,
   getFileUploadStatusV1FileUploadFileUploadIdGet,
+  getIndexedFilesV1KnowledgeSourcesKsSlugFilesGet,
   getUploadFormV1FileUploadFormPost,
   getUploadFormV2V2FileUploadFormPost,
   listKosV1KnowledgeSourcesKsSlugObjectsGet,
+  listKosV2KnowledgeSourcesKsSlugObjectsGet,
   listKsV1KnowledgeSourcesGet,
+  listKsV2KnowledgeSourcesGet,
   publishKsV1KnowledgeSourcesSlugPublishPost,
   resetKosV1KnowledgeSourcesKsSlugObjectsDelete,
   saveChunkedFilesDeprecatedV1FileUploadFileUploadIdPost,
@@ -31,7 +35,10 @@ import { FileUploadError } from '../error/FileUploadError'
 import { StackspotAPIError } from '../error/StackspotAPIError'
 import { ReactQueryNetworkClient } from '../network/ReactQueryNetworkClient'
 import { removeAuthorizationParam } from '../utils/remove-authorization-param'
-import { getApiAddresses } from '../api-addresses'
+
+const listKSV2WithoutAuthorization = removeAuthorizationParam(listKsV2KnowledgeSourcesGet)
+const listKOV2WithoutAuthorization = removeAuthorizationParam(listKosV2KnowledgeSourcesKsSlugObjectsGet)
+
 
 class DataIntegrationClient extends ReactQueryNetworkClient {
   constructor() {
@@ -89,6 +96,17 @@ class DataIntegrationClient extends ReactQueryNetworkClient {
    * List of knowledge sources
    */
   knowledgeSources = this.query(removeAuthorizationParam(listKsV1KnowledgeSourcesGet))
+  /**
+   * List of knowledge sources with pagination
+   */
+  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.
    */
@@ -101,6 +119,10 @@ class DataIntegrationClient extends ReactQueryNetworkClient {
    * List knowledge objects from a knowledge source
    */
   listKnowledgeObjects = this.query(removeAuthorizationParam(listKosV1KnowledgeSourcesKsSlugObjectsGet))
+  /**
+   * List knowledge objects from a knowledge source with pagination
+   */
+  listKnowledgeObjectsV2 = this.infiniteQuery(listKOV2WithoutAuthorization, { accumulator: 'items' })
   /**
    * Get a knowledge object by id
    */
@@ -180,6 +202,11 @@ class DataIntegrationClient extends ReactQueryNetworkClient {
    * Delete Knowledge Object from Knowledge Source
    */
   deleteKnowledgeObjectFromKnowledgeSource = this.mutation(removeAuthorizationParam(deleteKoByIdV1KnowledgeSourcesKsSlugObjectsKoIdDelete))
+  /**
+   * Get Indexed Files
+   */
+  listIndexedProjectFiles = this.query(removeAuthorizationParam(getIndexedFilesV1KnowledgeSourcesKsSlugFilesGet))
+
 }
 
 export const dataIntegrationClient = new DataIntegrationClient()

package/src/client/discover.ts

@@ -0,0 +1,319 @@
+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, AiChatRequest, refreshInsights } from '../api/discover'
+import { DefaultAPIError } from '../error/DefaultAPIError'
+import { StackspotAPIError } from '../error/StackspotAPIError'
+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'
+
+export interface ChatConversionDetails extends ConversationResponse {
+  opportunityName?: string,
+  hypothesisCount?: number,
+}
+
+class DiscoverClient extends ReactQueryNetworkClient {
+  constructor() {
+    super(getApiAddresses().discover.url, defaults)
+  }
+
+  protected buildStackSpotError(error: HttpError): StackspotAPIError {
+    return new DefaultAPIError(error.data, error.status, baseDictionary, error.headers)
+  }
+
+  opportunities = this.query(getAll)
+
+  opportunity = this.query(getById)
+
+  hypotheses = this.query(getAll1)
+
+  hypothesisById = this.query(getById1)
+
+  documents = this.query(getAllByHypothesis)
+
+  document = this.query(getById2)
+
+  artifacts = this.query(getAll2)
+
+  createOpportunity = this.mutation(create)
+
+  createHypothesis = this.mutation(create1)
+
+  createDocument = this.mutation(create2)
+
+  deleteOpportunity = this.mutation(deleteById)
+
+  deleteHypothesis = this.mutation(deleteById1)
+
+  deleteDocument = this.mutation(deleteById2)
+
+  /**
+   * Fetch insights
+   */
+  insights = this.query(fetchInsights)
+  /**
+   * Refresh insights
+   */
+  refreshInsights = this.mutation(refreshInsights)
+  /**
+   * Get Insight by Id
+   */
+  insightById = this.query(getInsightById)
+
+  chats = this.query({
+    name: 'chats',
+    request: async (_signal, variables: { filter?: string, page?: number, size?: number }) => {
+      const opportunities = await this.opportunities.query({})
+
+      const opportunitiesByChatId: Record<string, any> = opportunities.content.reduce(
+        (acc, opp: GetOpportunityResponse) => {
+          if (opp.chatId) {
+            acc[opp.chatId] = opp
+          }
+          return acc
+        },
+        {} as Record<string, any>,
+      )
+
+      const chatsHistory = await aiClient.chats.query(
+        { ...variables, page: variables.page, size: variables.size ?? 40 },
+      )
+
+      const filteredItems = variables.filter
+        ? chatsHistory.filter((chat) => chat.title.toLowerCase().includes(variables.filter!.toLowerCase()))
+        : chatsHistory
+
+      const enrichedChats = filteredItems?.map(chat => {
+        const relatedOpportunity = opportunitiesByChatId[chat.id]
+
+        return {
+          ...chat,
+          opportunityName: relatedOpportunity?.title ?? null,
+          hypothesisCount: relatedOpportunity?.hypotheses?.length ?? 0,
+        }
+      })
+
+      return enrichedChats as ChatConversionDetails[]
+    },
+  })
+
+  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: AiChatRequest & { agentId: string }, minChangeIntervalMS?: number): StreamedJson<ChatResponseWithSteps> {
+    const abortController = new AbortController()
+    const headers = {
+      'Content-Type': 'application/json',
+      'Accept': 'text/event-stream',
+    }
+    const events = this.stream(
+      this.resolveURL('/v2/ai/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 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' })
+      }
+
+      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' })
+      }
+
+      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 }
+          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 - 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 === 'chat' && info.action === 'end') {
+        const answerStep = last(data.steps)
+        if (answerStep) {
+          answerStep.status = 'success'
+          answerStep.duration = info.duration
+        }
+      }
+    }
+
+    return new StreamedJson({
+      eventsPromise: events,
+      abortController,
+      minChangeIntervalMS,
+      ignoreKeys: ['agent_info'],
+      transform,
+    })
+  }
+}
+
+export const discoverClient = new DiscoverClient()