@stack-spot/portal-network 0.148.0 → 0.149.0-beta.2

package/src/client/agent-tools.ts

@@ -1,10 +1,13 @@
 import { HttpError } from '@oazapfts/runtime'
-import { createToolkitToolsV1ToolkitsToolkitIdToolsPost, createToolkitV1ToolkitsPost, defaults, deleteToolkitToolsV1ToolkitsToolkitIdToolsDelete, deleteToolkitV1ToolkitsToolkitIdDelete, editToolkitToolV1ToolkitsToolkitIdToolsToolIdPut, getPublicToolKitsV1BuiltinToolkitGet, getToolkitToolV1ToolkitsToolkitIdToolsToolIdGet, getToolkitV1ToolkitsToolkitIdGet, HttpValidationError, listToolkitsV1ToolkitsGet, updateToolkitV1ToolkitsToolkitIdPatch } from '../api/agent-tools'
+import { addFavoriteV1AgentsAgentIdFavoritePost, AgentVisibilityLevelEnum, createAgentV1AgentsPost, createToolkitToolsV1ToolkitsToolkitIdToolsPost, createToolkitV1ToolkitsPost, defaults, deleteAgentV1AgentsAgentIdDelete, deleteToolkitToolsV1ToolkitsToolkitIdToolsDelete, deleteToolkitV1ToolkitsToolkitIdDelete, editToolkitToolV1ToolkitsToolkitIdToolsToolIdPut, getAgentV1AgentsAgentIdGet, getPublicToolKitsV1BuiltinToolkitGet, getToolkitToolV1ToolkitsToolkitIdToolsToolIdGet, getToolkitV1ToolkitsToolkitIdGet, HttpValidationError, listAgentsV1AgentsGet, listToolkitsV1ToolkitsGet, publishAgentV1AgentsAgentIdPublishPost, searchAgentsV1AgentsSearchPost, updateAgentV1AgentsAgentIdPut, updateToolkitV1ToolkitsToolkitIdPatch, 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 { AgentToolsOpenAPIPreview } from './types'
+import { AgentResponseWithBuiltIn, AgentToolsOpenAPIPreview, AgentVisibilityLevel } from './types'
+import { workspaceAiClient } from './workspace-ai'
+
+const AGENT_DEFAULT_SLUG = 'stk_code_buddy'
 
 class AgentToolsClient extends ReactQueryNetworkClient {
   constructor() {
@@ -21,11 +24,107 @@ class AgentToolsClient extends ReactQueryNetworkClient {
     })
   }
 
-  tools = this.query(getPublicToolKitsV1BuiltinToolkitGet)
+  tools = this.query(removeAuthorizationParam(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)
 
   /**
-   * Get list of Toolkits
+  * 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
+    },
+  })
+
+  /* Get list of Toolkits
+  */
   toolkits = this.query(removeAuthorizationParam(listToolkitsV1ToolkitsGet))
   /**
    * Get a toolkit by Id

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

@@ -33,6 +33,7 @@ import {
   quickCommandsRunV2V2QuickCommandsSlugStepsStepSlugRunPost,
   resetKnowledgeObjectsV1KnowledgeSourcesSlugObjectsDelete,
   runFetchStepV1QuickCommandsSlugStepsStepSlugFetchRunPost,
+  searchKnowledgeSourcesV1KnowledgeSourcesSearchPost,
   updateQuickCommandV1QuickCommandsSlugPatch,
   updateTitleV1ConversationsConversationIdPatch,
   vectorizeCustomKnowledgeSourceV1KnowledgeSourcesSlugCustomPost,
@@ -134,6 +135,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/types.ts

@@ -1,7 +1,6 @@
 import { RequestOpts } from '@oazapfts/runtime'
 import { AccountScmInfoSaveRequest, AccountScmInfoUpdateRequest, AccountScmStatusResponse, GroupsFromResourceResponse, MembersFromResourceResponse } from '../api/account'
-import { AgentResponse, VisibilityLevel } from '../api/agent'
-import { HttpMethod } from '../api/agent-tools'
+import { AgentVisibilityLevelEnum, HttpMethod, 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'
@@ -220,6 +219,7 @@ export interface WorkspaceAiMembersPermissions {
   totalPages: number,
 }
 
+
 export interface WorkspaceAiGroupsPermissions extends GroupsFromResourceResponse {
   actions: Action[],
 }
@@ -358,12 +358,14 @@ 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
 
 export interface AgentToolsOpenAPIPreview {
   name: string,