@stack-spot/portal-network 0.138.1 → 0.139.1-beta.0

package/src/client/agent-tools.ts

@@ -1,8 +1,13 @@
 import { HttpError } from '@oazapfts/runtime'
-import { defaults, getPublicToolKitsV1BuiltinToolkitGet, HttpValidationError } from '../api/agent-tools'
+import { addFavoriteV1AgentsAgentIdFavoritePost, AgentVisibilityLevelEnum, createAgentV1AgentsPost, defaults, deleteAgentV1AgentsAgentIdDelete, getAgentV1AgentsAgentIdGet, getPublicToolKitsV1BuiltinToolkitGet, HttpValidationError, listAgentsV1AgentsGet, publishAgentV1AgentsAgentIdPublishPost, searchAgentsV1AgentsSearchPost, updateAgentV1AgentsAgentIdPut, VisibilityLevelEnum } 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'
 
 class AgentToolsClient extends ReactQueryNetworkClient {
   constructor() {
@@ -19,6 +24,103 @@ class AgentToolsClient extends ReactQueryNetworkClient {
   }
 
   tools = this.query(getPublicToolKitsV1BuiltinToolkitGet)
+
+  /**
+   * Create agent
+   */
+  createAgent = this.mutation(removeAuthorizationParam(createAgentV1AgentsPost))
+
+  /**
+   * Delete agent
+   */
+  deleteAgent = this.mutation(deleteAgentV1AgentsAgentIdDelete)
+
+  /**
+   * Updates an agent
+   */
+  updateAgent = this.mutation(updateAgentV1AgentsAgentIdPut)
+
+  /**
+  * Favorite an agent
+  */
+  favoriteAgent = this.mutation(addFavoriteV1AgentsAgentIdFavoritePost)
+
+  /**
+  * Publish an agent
+  */
+  publishAgent = this.mutation(publishAgentV1AgentsAgentIdPublishPost)
+
+  /** 
+   * List agents
+   */
+  agents = this.infiniteQuery(removeAuthorizationParam(listAgentsV1AgentsGet))
+
+  /** 
+   * Gets agent by id
+   */
+  agent = this.query(removeAuthorizationParam(getAgentV1AgentsAgentIdGet))
+
+  /** 
+   * Gets agents by ids
+   */
+  agentsByIds = this.query(removeAuthorizationParam(searchAgentsV1AgentsSearchPost))
+
+  /** 
+   * Gets the default agent slug
+   */
+  agentDefaultSlug = AGENT_DEFAULT_SLUG
+
+  /** 
+   * Gets the default agent
+   */
+  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.
+   */
+  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
+    },
+  })
 }
 
 export const agentToolsClient = new AgentToolsClient()

package/src/client/agent.ts

@@ -1,12 +1,8 @@
 import { HttpError } from '@oazapfts/runtime'
-import { defaults, deleteV1AgentByAgentIdFavorite, getV1AgentByAgentId, getV1Agents, getV1PublicAgentByAgentId, getV1PublicAgents, postV1AgentByAgentIdFavorite, postV1AgentsTrial, putV1AgentByAgentId, VisibilityLevel } from '../api/agent'
+import { defaults, deleteV1AgentByAgentIdFavorite, getV1AgentByAgentId, getV1Agents, getV1PublicAgentByAgentId, getV1PublicAgents, postV1AgentByAgentIdFavorite, postV1AgentsTrial, putV1AgentByAgentId } 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,
@@ -32,56 +28,6 @@ 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
    */
@@ -92,18 +38,6 @@ 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

@@ -31,6 +31,7 @@ import {
   QuickCommandsExecutionRequest,
   quickCommandsRunV2V2QuickCommandsSlugStepsStepSlugRunPost,
   resetKnowledgeObjectsV1KnowledgeSourcesSlugObjectsDelete,
+  searchKnowledgeSourcesV1KnowledgeSourcesSearchPost,
   updateQuickCommandV1QuickCommandsSlugPatch,
   updateTitleV1ConversationsConversationIdPatch,
   vectorizeCustomKnowledgeSourceV1KnowledgeSourcesSlugCustomPost,
@@ -127,6 +128,10 @@ 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

@@ -1,21 +1,22 @@
 import { HttpError } from '@oazapfts/runtime'
-import
-{
+import {
   addFavoriteV1KnowledgeSourcesSlugFavoritePost,
   defaults,
   deleteFavoriteV1KnowledgeSourcesSlugFavoriteDelete,
+  FileUploadType,
   getFileUploadStatusV1FileUploadFileUploadIdGet,
   getUploadFormV2V2FileUploadFormPost,
   listKsV1KnowledgeSourcesGet,
-  saveChunkedFilesV1FileUploadFileUploadIdPost,
+  saveChunkedFilesV1FileUploadFileUploadIdKnowledgeObjectsPost,
   splitUploadedFileV1FileUploadFileUploadIdSplitPost,
 } from '../api/dataIntegration'
 import apis from '../apis.json'
 import { DefaultAPIError } from '../error/DefaultAPIError'
+import { baseDictionary } from '../error/dictionary/base'
+import { FileUploadError } from '../error/FileUploadError'
 import { StackspotAPIError } from '../error/StackspotAPIError'
 import { ReactQueryNetworkClient } from '../network/ReactQueryNetworkClient'
 import { removeAuthorizationParam } from '../utils/remove-authorization-param'
-import { baseDictionary } from '../error/dictionary/base'
 
 class DataIntegrationClient extends ReactQueryNetworkClient {
   constructor() {
@@ -36,7 +37,7 @@ class DataIntegrationClient extends ReactQueryNetworkClient {
   /**
    * Save the splitted knowledge objects
    */
-  saveKnowledgeObjects = this.mutation(removeAuthorizationParam(saveChunkedFilesV1FileUploadFileUploadIdPost))
+  saveKnowledgeObjects = this.mutation(removeAuthorizationParam(saveChunkedFilesV1FileUploadFileUploadIdKnowledgeObjectsPost))
   /**
    * Splits the file with the passed settings
    */
@@ -70,7 +71,29 @@ class DataIntegrationClient extends ReactQueryNetworkClient {
   /**
    * Removes the resource of type Knowledge Source from the list of favorites.
    */
-  removeFavoriteKnowledgeSource = this.mutation(removeAuthorizationParam(deleteFavoriteV1KnowledgeSourcesSlugFavoriteDelete)) 
+  removeFavoriteKnowledgeSource = this.mutation(removeAuthorizationParam(deleteFavoriteV1KnowledgeSourcesSlugFavoriteDelete))
+
+  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)
+      } 
+    })),
+  })
 }
 
 export const dataIntegrationClient = new DataIntegrationClient()

package/src/client/types.ts

@@ -1,6 +1,6 @@
 import { RequestOpts } from '@oazapfts/runtime'
 import { AccountScmInfoSaveRequest, AccountScmInfoUpdateRequest, AccountScmStatusResponse, GroupsFromResourceResponse, MembersFromResourceResponse } from '../api/account'
-import { AgentResponse, VisibilityLevel } from '../api/agent'
+import { AgentVisibilityLevelEnum, ListAgentResponse, VisibilityLevelEnum } 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'
@@ -184,6 +184,7 @@ export interface FixedChatRequest extends ChatRequest {
     platform_version?: string,
     stackspot_ai_version?: string,
     os?: string,
+    upload_ids?: string[],
   },
 }
 
@@ -218,6 +219,7 @@ export interface WorkspaceAiMembersPermissions {
   totalPages: number,
 }
 
+
 export interface WorkspaceAiGroupsPermissions extends GroupsFromResourceResponse {
   actions: Action[],
 }
@@ -356,9 +358,11 @@ export type FixVariables<
 
 export type ReplaceResult<T extends (...args: any[]) => Promise<any>, Fix> = (...args: Parameters<T>) => Promise<Fix>
 
-export interface AgentResponseWithBuiltIn extends AgentResponse {
+export interface AgentResponseWithBuiltIn extends Omit<ListAgentResponse, 'conversation_starter' | 'avatar'>   {
   builtIn?: boolean,
   spaceName?: string,
-}
+  conversation_starter?: string[] | null,
+  avatar?: string | null | undefined,
+} 
 
-export type AgentVisibilityLevel = VisibilityLevel | 'BUILT-IN' | 'ALL'
+export type AgentVisibilityLevel = AgentVisibilityLevelEnum | VisibilityLevelEnum

package/src/error/FileUploadError.ts

@@ -0,0 +1,18 @@
+import { HttpValidationError } from '../api/ai'
+import { StackspotAPIError } from './StackspotAPIError'
+
+export class FileUploadError extends StackspotAPIError {
+  readonly fileName: string
+
+  constructor(fileName: string, error: any) {
+    super({
+      status: error.status,
+      headers: error.headers,
+      stack: error.stack,
+      message: error instanceof StackspotAPIError
+        ? error.message
+        : (error.data as HttpValidationError | undefined)?.detail?.map(d => d.msg)?.join('\n'),
+    })
+    this.fileName = fileName
+  }
+}

package/src/index.ts

@@ -13,13 +13,13 @@ export { cloudServicesClient } from './client/cloud-services'
 export { codeShiftClient } from './client/code-shift'
 export { contentClient } from './client/content'
 export { dataIntegrationClient } from './client/data-integration'
+export { discoveryClient } from './client/discovery'
 export { eventBusClient } from './client/event-bus'
 export { genAiInferenceClient } from './client/gen-ai-inference'
 export { insightsClient } from './client/insights'
 export { notificationClient } from './client/notification'
 export { runtimeManagerClient } from './client/runtime-manager'
 export { secretsClient } from './client/secrets'
-export { discoveryClient } from './client/discovery'
 export * from './client/types'
 export { workflowClient } from './client/workflow'
 export { workspaceClient } from './client/workspace'
@@ -27,6 +27,7 @@ export { workspaceAiClient } from './client/workspace-ai'
 export { workspaceManagerClient } from './client/workspace-manager'
 export { workspaceSearchClient } from './client/workspace-search'
 export { DefaultAPIError } from './error/DefaultAPIError'
+export { FileUploadError } from './error/FileUploadError'
 export { StackspotAPIError } from './error/StackspotAPIError'
 export { StreamCanceledError } from './error/StreamCanceledError'
 export { StreamError } from './error/StreamError'