@stack-spot/portal-network 0.146.2-beta.0 → 0.147.0

package/src/client/agent-tools.ts

@@ -1,13 +1,10 @@
 import { HttpError } from '@oazapfts/runtime'
-import { addFavoriteV1AgentsAgentIdFavoritePost, AgentVisibilityLevelEnum, createAgentV1AgentsPost, defaults, deleteAgentV1AgentsAgentIdDelete, getAgentV1AgentsAgentIdGet, getPublicToolKitsV1BuiltinToolkitGet, HttpValidationError, listAgentsV1AgentsGet, publishAgentV1AgentsAgentIdPublishPost, searchAgentsV1AgentsSearchPost, updateAgentV1AgentsAgentIdPut, VisibilityLevelEnum } from '../api/agent-tools'
+import { createToolkitToolsV1ToolkitsToolkitIdToolsPost, createToolkitV1ToolkitsPost, defaults, deleteToolkitToolsV1ToolkitsToolkitIdToolsDelete, deleteToolkitV1ToolkitsToolkitIdDelete, editToolkitToolV1ToolkitsToolkitIdToolsToolIdPut, getPublicToolKitsV1BuiltinToolkitGet, getToolkitToolV1ToolkitsToolkitIdToolsToolIdGet, getToolkitV1ToolkitsToolkitIdGet, HttpValidationError, listToolkitsV1ToolkitsGet, updateToolkitV1ToolkitsToolkitIdPatch } from '../api/agent-tools'
 import apis from '../apis.json'
 import { StackspotAPIError } from '../error/StackspotAPIError'
 import { ReactQueryNetworkClient } from '../network/ReactQueryNetworkClient'
 import { removeAuthorizationParam } from '../utils/remove-authorization-param'
-import { AgentResponseWithBuiltIn, AgentVisibilityLevel } from './types'
-import { workspaceAiClient } from './workspace-ai'
-
-const AGENT_DEFAULT_SLUG = 'stk_code_buddy'
+import { AgentToolsOpenAPIPreview } from './types'
 
 class AgentToolsClient extends ReactQueryNetworkClient {
   constructor() {
@@ -19,106 +16,64 @@ class AgentToolsClient extends ReactQueryNetworkClient {
       status: error.status,
       headers: error.headers,
       stack: error.stack,
-      message: (error.data as HttpValidationError | undefined)?.detail?.map(d => d.msg)?.join('\n'),
+      // @ts-ignore API documentation is wrong
+      message: (error.data as HttpValidationError | undefined)?.details?.map(d => d.msg)?.join('\n'),
     })
   }
 
   tools = this.query(getPublicToolKitsV1BuiltinToolkitGet)
 
   /**
-   * Create agent
+   * Get list of Toolkits
    */
-  createAgent = this.mutation(removeAuthorizationParam(createAgentV1AgentsPost))
-
+  toolkits = this.query(removeAuthorizationParam(listToolkitsV1ToolkitsGet))
   /**
-   * Delete agent
+   * Get a toolkit by Id
    */
-  deleteAgent = this.mutation(deleteAgentV1AgentsAgentIdDelete)
-
+  toolkitById = this.query(removeAuthorizationParam(getToolkitV1ToolkitsToolkitIdGet))
   /**
-   * Updates an agent
+   * Create a toolkit
    */
-  updateAgent = this.mutation(updateAgentV1AgentsAgentIdPut)
-
+  createToolkit = this.mutation(removeAuthorizationParam(createToolkitV1ToolkitsPost))
   /**
-  * Favorite an agent
-  */
-  favoriteAgent = this.mutation(addFavoriteV1AgentsAgentIdFavoritePost)
-
+   * Update a toolkit
+   */
+  updateToolkit = this.mutation(removeAuthorizationParam(updateToolkitV1ToolkitsToolkitIdPatch))
   /**
-  * Publish an agent
-  */
-  publishAgent = this.mutation(publishAgentV1AgentsAgentIdPublishPost)
-
-  /** 
-   * List agents
+   * Delete a toolkit
    */
-  agents = this.infiniteQuery(removeAuthorizationParam(listAgentsV1AgentsGet))
-
-  /** 
-   * Gets agent by id
+  deleteToolkit = this.mutation(removeAuthorizationParam(deleteToolkitV1ToolkitsToolkitIdDelete))
+  /**
+   * Get Tool from Toolkit
    */
-  agent = this.query(removeAuthorizationParam(getAgentV1AgentsAgentIdGet))
-
-  /** 
-   * Gets agents by ids
+  toolFromToolkit = this.query(removeAuthorizationParam(getToolkitToolV1ToolkitsToolkitIdToolsToolIdGet))
+  /**
+   * Add tool on Toolkit
    */
-  agentsByIds = this.query(removeAuthorizationParam(searchAgentsV1AgentsSearchPost))
-
-  /** 
-   * Gets the default agent slug
+  addToolOnToolkit = this.mutation(removeAuthorizationParam(createToolkitToolsV1ToolkitsToolkitIdToolsPost))
+  /**
+   * Edit tool on Toolkit
    */
-  agentDefaultSlug = AGENT_DEFAULT_SLUG
-
-  /** 
-   * Gets the default agent
+  editToolOnToolkit = this.mutation(removeAuthorizationParam(editToolkitToolV1ToolkitsToolkitIdToolsToolIdPut))
+  /**
+   * Delete tool from Toolkit
    */
-  agentDefault = this.query({
-    name: 'agentDefault',
-    request: async (signal) => {
-      const agentDefault = await listAgentsV1AgentsGet(
-        { visibility: 'built_in', slug: this.agentDefaultSlug, authorization: '' }, { signal },
-      )
-
-      const agentId = agentDefault.at(0)?.id
-      const agent = agentId ? await this.agent.query({ agentId }) : undefined
-      return agent
-    },
-  })
-
-  /** 
-   * List agents filtered by visibility.
+  deleteToolFromToolkit = this.mutation(removeAuthorizationParam(deleteToolkitToolsV1ToolkitsToolkitIdToolsDelete))
+  /**
+   * Previews an uploaded open api file.
    */
-  allAgents = this.query({
-    name: 'allAgents',
-    request: async (signal, variables: { visibilities: (AgentVisibilityLevel | 'all')[] }) => {
-      const allVisibilities = ['account', 'built_in', 'favorite', 'personal', 'shared', 'workspace'] as const
-      const 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([
-        workspaceAgentsPromise,
-        ...visibilities.map((visibility) => listAgentsV1AgentsGet({ visibility, authorization: '' }, { signal })),
-      ])
-
-      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 => ({
-        ...agent,
-        builtIn: agent?.visibility_level === 'built_in',
-      }))))
-
-      return allAgents
+  previewUploadedAPI = this.query({
+    name: 'previewUploadedAPI',
+    request: async (signal, params: { fileUploadId: string }) => {
+      const url = `/v1/toolkits/tools/preview/${encodeURIComponent(params.fileUploadId)}`
+      const events = await this.stream(url, { signal })
+      const result: AgentToolsOpenAPIPreview[] = []
+      for await (const event of events) {
+        if (event.data) {
+          result.push(JSON.parse(event.data))
+        }
+      }
+      return result
     },
   })
 }

package/src/client/agent.ts

@@ -1,8 +1,12 @@
 import { HttpError } from '@oazapfts/runtime'
-import { defaults, deleteV1AgentByAgentIdFavorite, getV1AgentByAgentId, getV1Agents, getV1PublicAgentByAgentId, getV1PublicAgents, postV1AgentByAgentIdFavorite, postV1AgentsTrial, putV1AgentByAgentId } from '../api/agent'
+import { defaults, deleteV1AgentByAgentIdFavorite, getV1AgentByAgentId, getV1Agents, getV1PublicAgentByAgentId, getV1PublicAgents, postV1AgentByAgentIdFavorite, postV1AgentsTrial, putV1AgentByAgentId, VisibilityLevel } from '../api/agent'
 import apis from '../apis.json'
 import { StackspotAPIError } from '../error/StackspotAPIError'
 import { ReactQueryNetworkClient } from '../network/ReactQueryNetworkClient'
+import { AgentResponseWithBuiltIn, AgentVisibilityLevel } from './types'
+import { workspaceAiClient } from './workspace-ai'
+
+export const isAgentDefault = (agentSlug?: string) => agentSlug === 'stk_code_buddy'
 
 interface AgentError {
   code?: string,
@@ -28,6 +32,56 @@ class AgentClient extends ReactQueryNetworkClient {
     })
   }
 
+  /** 
+   * List agents filtered by visibility.
+   */
+  allAgents = this.query({
+    name: 'allAgents',
+    request: async (signal, variables: { visibilities: AgentVisibilityLevel[] }) => {
+      const visibilities: VisibilityLevel[] = variables.visibilities.includes('ALL')
+        ? ['PERSONAL', 'SHARED', 'WORKSPACE', 'ACCOUNT', 'FAVORITE']
+        : variables.visibilities.filter((visibility) => visibility !== 'BUILT-IN' && visibility !== 'ALL') as VisibilityLevel[]
+
+      const shouldIncludeBuiltInAgent = variables.visibilities.includes('ALL') || variables.visibilities.includes('BUILT-IN')
+      const shouldFetchBuiltInAgent = shouldIncludeBuiltInAgent || variables.visibilities.includes('FAVORITE')
+      const shouldFetchWorkspaceAgents = variables.visibilities.includes('ALL') || variables.visibilities.includes('WORKSPACE')
+
+      const publicAgentsPromise = shouldFetchBuiltInAgent ? getV1PublicAgents({}, { signal }) : Promise.resolve([])
+      const workspaceAgentsPromise = shouldFetchWorkspaceAgents
+        ? workspaceAiClient.workspacesContentsByType.query({ contentType: 'agent' })
+        : Promise.resolve([])
+
+      const [publicAgents, workspaceAgents, ...agentsByVisibility] = await Promise.all([
+        publicAgentsPromise,
+        workspaceAgentsPromise,
+        ...visibilities.map((visibility) => getV1Agents({ visibility }, { signal })),
+      ])
+
+      const workspaceAgentsWithSpaceName = workspaceAgents?.flatMap(({ agents, space_name }) =>
+        agents?.map((agent) => ({
+          ...agent,
+          spaceName: space_name,
+          builtIn: publicAgents?.some(publicAgent => publicAgent.id === agent.id),
+        }))) as AgentResponseWithBuiltIn[]
+
+      const allAgents: AgentResponseWithBuiltIn[] = workspaceAgentsWithSpaceName || []
+
+      if (shouldIncludeBuiltInAgent) {
+        const builtInsAgents = publicAgents?.map((agent) => ({ ...agent, builtIn: true, visibility_level: 'BUILT-IN' }))
+        allAgents.push(...builtInsAgents)
+      }
+
+      agentsByVisibility.forEach(agents => {
+        allAgents.push(...agents.map(agent => ({
+          ...agent,
+          builtIn: publicAgents?.some(publicAgent => publicAgent.id === agent.id),
+        })))
+      })
+
+      return allAgents
+    },
+  })
+
   /** 
    * Gets an agent by id
    */
@@ -38,6 +92,18 @@ class AgentClient extends ReactQueryNetworkClient {
       : getV1AgentByAgentId({ ...variables }, { signal }),
   })
 
+  /** 
+   * Gets the default agent
+   */
+  agentDefault = this.query({
+    name: 'agentDefault',
+    request: async (signal) => {
+      const publicAgents = await getV1PublicAgents({}, { signal })
+      const agentDefault = publicAgents.find((agent) => isAgentDefault(agent.slug))
+      return agentDefault ? { ...agentDefault, builtIn: true } : undefined
+    },
+  })
+
   /** 
    * List commons agents
    */

package/src/client/ai.ts

@@ -33,7 +33,6 @@ import {
   quickCommandsRunV2V2QuickCommandsSlugStepsStepSlugRunPost,
   resetKnowledgeObjectsV1KnowledgeSourcesSlugObjectsDelete,
   runFetchStepV1QuickCommandsSlugStepsStepSlugFetchRunPost,
-  searchKnowledgeSourcesV1KnowledgeSourcesSearchPost,
   updateQuickCommandV1QuickCommandsSlugPatch,
   updateTitleV1ConversationsConversationIdPatch,
   vectorizeCustomKnowledgeSourceV1KnowledgeSourcesSlugCustomPost,
@@ -135,10 +134,6 @@ class AIClient extends ReactQueryNetworkClient {
    * Gets a knowledge source document by the slug of the parent knowledge source and the document id.
    */
   knowledgeSourceDocument = this.query(removeAuthorizationParam(findKnowledgeObjectByCustomIdV1KnowledgeSourcesSlugObjectsCustomIdGet))
-  /**
-   * Lists knowledge sources matching the provided IDs.
-   */
-  searchKnowledgeSources = this.query(removeAuthorizationParam(searchKnowledgeSourcesV1KnowledgeSourcesSearchPost))
   /**
    * Gets the chat history. This is a paginated resource.
    */

package/src/client/data-integration.ts

@@ -9,6 +9,7 @@ import {
   listKsV1KnowledgeSourcesGet,
   saveChunkedFilesDeprecatedV1FileUploadFileUploadIdPost,
   saveChunkedFilesV1FileUploadFileUploadIdKnowledgeObjectsPost,
+  splitUploadedFilePreviewV1FileUploadFileUploadIdSplitPreviewSplitStrategyGet,
   splitUploadedFileV1FileUploadFileUploadIdSplitPost,
 } from '../api/dataIntegration'
 import apis from '../apis.json'
@@ -65,7 +66,12 @@ class DataIntegrationClient extends ReactQueryNetworkClient {
         'Accept': 'text/event-stream',
       },
       signal: options?.signal,
-    })}
+    })
+  }
+  /**
+   * Generate a preview with splitted objects
+   */
+  previewFromSplit = this.query(removeAuthorizationParam(splitUploadedFilePreviewV1FileUploadFileUploadIdSplitPreviewSplitStrategyGet))
   /**
    * List of knowledge sources
    */
@@ -81,24 +87,26 @@ class DataIntegrationClient extends ReactQueryNetworkClient {
 
   uploadFiles = this.mutation({
     name: 'uploadFiles',
-    request: (abortSignal, params: { files: File[], type: FileUploadType }) => Promise.all(params.files.map(async (f) => {
-      try {
-        const { id, url, form } = await getUploadFormV2V2FileUploadFormPost({
-          authorization: '',
-          newFileUploadRequest: { file_name: f.name, target_type: params.type, target_id: '' },
-        }, { signal: abortSignal })
-        const formData = new FormData()
-        Object.entries(form).forEach(([key, value]) => {
-          formData.append(key, value)
-        })
-        formData.append('file', f)
-        const response = await fetch(url, { method: 'post', body: formData, signal: abortSignal })
-        if (!response.ok) throw new StackspotAPIError({ status: response.status, headers: response.headers, message: 'Failed to upload' })
-        return id
-      } catch (e: any) {
-        throw new FileUploadError(f.name, e)
-      } 
-    })),
+    request: (abortSignal, params: { files: File[], type: FileUploadType, targetId?: string }) => (
+      Promise.all(params.files.map(async (f) => {
+        try {
+          const { id, url, form } = await getUploadFormV2V2FileUploadFormPost({
+            authorization: '',
+            newFileUploadRequest: { file_name: f.name, target_type: params.type, target_id: params.targetId ?? '' },
+          }, { signal: abortSignal })
+          const formData = new FormData()
+          Object.entries(form).forEach(([key, value]) => {
+            formData.append(key, value)
+          })
+          formData.append('file', f)
+          const response = await fetch(url, { method: 'post', body: formData, signal: abortSignal })
+          if (!response.ok) throw new StackspotAPIError({ status: response.status, headers: response.headers, message: 'Failed to upload' })
+          return id
+        } catch (e: any) {
+          throw new FileUploadError(f.name, e)
+        } 
+      }))
+    ),
   })
 }
 

package/src/client/types.ts

@@ -1,6 +1,7 @@
 import { RequestOpts } from '@oazapfts/runtime'
 import { AccountScmInfoSaveRequest, AccountScmInfoUpdateRequest, AccountScmStatusResponse, GroupsFromResourceResponse, MembersFromResourceResponse } from '../api/account'
-import { AgentVisibilityLevelEnum, ListAgentResponse, VisibilityLevelEnum } from '../api/agent-tools'
+import { AgentResponse, VisibilityLevel } from '../api/agent'
+import { HttpMethod } from '../api/agent-tools'
 import { ChatRequest, ChatResponse3, ContentDependencyResponse, ConversationHistoryResponse, ConversationResponse, DependencyResponse } from '../api/ai'
 import { ConnectAccountRequestV2, ManagedAccountProvisionRequest } from '../api/cloudAccount'
 import { AllocationCostRequest, AllocationCostResponse, ChargePeriod, getAllocationCostFilters, ManagedService, ServiceResource } from '../api/cloudServices'
@@ -219,7 +220,6 @@ export interface WorkspaceAiMembersPermissions {
   totalPages: number,
 }
 
-
 export interface WorkspaceAiGroupsPermissions extends GroupsFromResourceResponse {
   actions: Action[],
 }
@@ -358,11 +358,18 @@ export type FixVariables<
 
 export type ReplaceResult<T extends (...args: any[]) => Promise<any>, Fix> = (...args: Parameters<T>) => Promise<Fix>
 
-export interface AgentResponseWithBuiltIn extends Omit<ListAgentResponse, 'conversation_starter' | 'avatar'>   {
+export interface AgentResponseWithBuiltIn extends AgentResponse {
   builtIn?: boolean,
   spaceName?: string,
-  conversation_starter?: string[] | null,
-  avatar?: string | null | undefined,
-} 
+}
+
+export type AgentVisibilityLevel = VisibilityLevel | 'BUILT-IN' | 'ALL'
 
-export type AgentVisibilityLevel = AgentVisibilityLevelEnum | VisibilityLevelEnum
+export interface AgentToolsOpenAPIPreview {
+  name: string,
+  description: string,
+  method: HttpMethod,
+  url: string,
+  parameters?: Record<string, any>,
+  request_body?: Record<string, any>,
+}

package/src/utils/StreamedJson.tsx

@@ -92,8 +92,8 @@ export class StreamedJson<T> {
       else if (typeof source[k] === 'string') target[k] += source[k]
       else if (Array.isArray(source[k])) target[k].push(...source[k])
       else if (typeof source[k] === 'number') parseFloat(target[k] + source[k])
-      else if (typeof source[k] === 'object') this.merge(source[k], target[k])
-      else target[k] = source[k]
+      else if (source[k] && typeof source[k] === 'object') this.merge(source[k], target[k])
+      else if (source[k]) target[k] = source[k]
     })
   }