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

package/src/api/agent-tools.ts

@@ -45,7 +45,9 @@ export type FunctionParameter = {
     /** A list of possible values for the parameter (optional) */
     "enum"?: string[] | null;
     /** A dictionary of properties for the parameter (optional) */
-    properties?: object | null;
+    properties?: {
+        [key: string]: any;
+    } | null;
     /** A list of required parameters (optional) */
     required?: string[] | null;
     /** Allow arbitrary data of specific typed */
@@ -73,6 +75,115 @@ export type ExecuteAgentToolRequest = {
 export type AgentToolExecutionResponse = {
     result: string;
 };
+export type AgentVisibilityLevelEnum = "built_in";
+export type VisibilityLevelEnum = "account" | "personal" | "shared" | "workspace" | "favorite";
+export type ListAgentResponse = {
+    id: string;
+    name: string;
+    slug: string;
+    created_by: string | null;
+    visibility_level: string;
+    avatar: string | null;
+    conversation_starter: string[] | null;
+};
+export type AgentType = "CONVERSATIONAL" | "SINGLE_ANSWER";
+export type SimilarityFunctionEnum = "cosine" | "euclidean" | "dot_product";
+export type KnowledgeSourcesConfigRequest = {
+    similarity_function?: SimilarityFunctionEnum | null;
+    post_processing?: boolean | null;
+    max_number_of_kos?: number | null;
+    relevancy_threshold?: number | null;
+    knowledge_sources?: string[] | null;
+    sealed?: boolean | null;
+};
+export type NewAgentRequest = {
+    /** LLM model name */
+    model_name?: string | null;
+    /** LLM model id */
+    model_id?: string | null;
+    /** Agent name */
+    name: string;
+    /** Agent unique slug */
+    slug: string;
+    /** Agent description */
+    description?: string | null;
+    /** Agent avatar image */
+    avatar?: string | null;
+    /** Agent suggested prompt */
+    suggested_prompts?: string[];
+    /** System prompt */
+    system_prompt?: string | null;
+    /** Agent type */
+    "type": AgentType;
+    knowledge_sources_config?: KnowledgeSourcesConfigRequest | null;
+    builtin_tools_ids?: string[];
+    detail_mode?: boolean;
+    structured_output?: string | null;
+    llm_settings?: {
+        [key: string]: any;
+    } | null;
+};
+export type NewAgentResponse = {
+    id: string;
+};
+export type SearchAgentsRequest = {
+    /** Agent ids to filter for */
+    ids: string[];
+};
+export type KnowledgeSourceConfigModel = {
+    similarity_function: SimilarityFunctionEnum;
+    max_number_of_kos: number;
+    relevancy_threshold: number;
+    post_processing: boolean;
+    knowledge_sources: string[];
+    sealed: boolean;
+    created_by?: string | null;
+    created_at?: string | null;
+    updated_by?: string | null;
+    updated_at?: string | null;
+};
+export type AgentToolModel = {
+    agent_id: string;
+    builtin_tool_id: string;
+    builtin_toolkit_id?: string | null;
+    created_by?: string | null;
+    created_at?: string | null;
+    updated_by?: string | null;
+    updated_at?: string | null;
+};
+export type LlmSettingsModel = {
+    property_key?: string | null;
+    property_value?: string | null;
+    property_type?: string | null;
+    agent_id: string;
+    created_by?: string | null;
+    created_at?: string | null;
+    updated_by?: string | null;
+    updated_at?: string | null;
+};
+export type AgentModel = {
+    id: string;
+    name: string;
+    slug: string;
+    description?: string | null;
+    system_prompt: string;
+    visibility_level: string;
+    avatar?: string | null;
+    "type": string;
+    conversation_starter?: string[] | null;
+    use_only: boolean;
+    detail_mode: boolean;
+    structured_output?: string | null;
+    model_id?: string | null;
+    model_name?: string | null;
+    knowledge_source_config?: KnowledgeSourceConfigModel | null;
+    tools?: AgentToolModel[];
+    settings?: LlmSettingsModel[];
+    created_by?: string | null;
+    created_at?: string | null;
+    updated_by?: string | null;
+    updated_at?: string | null;
+};
 /**
  * Health Check
  */
@@ -87,9 +198,9 @@ export function healthCheckHealthzGet(opts?: Oazapfts.RequestOpts) {
     }));
 }
 /**
- * Health Check
+ * Readyz
  */
-export function healthCheckReadyzGet(opts?: Oazapfts.RequestOpts) {
+export function readyzReadyzGet(opts?: Oazapfts.RequestOpts) {
     return oazapfts.ok(oazapfts.fetchJson<{
         status: 200;
         data: any;
@@ -207,7 +318,9 @@ export function listToolsByAgentForSchemaV1AgentsAgentIdToolsSchemaSchemaGet({ a
 }, opts?: Oazapfts.RequestOpts) {
     return oazapfts.ok(oazapfts.fetchJson<{
         status: 200;
-        data: (OpenAiTool | object)[];
+        data: (OpenAiTool | {
+            [key: string]: any;
+        })[];
     } | {
         status: 404;
     } | {
@@ -253,3 +366,192 @@ export function executeToolV1AgentsToolsAgentToolIdExecutePost({ agentToolId, is
         })
     })));
 }
+/**
+ * List Agents
+ */
+export function listAgentsV1AgentsGet({ name, slug, visibility, size, page, xAccountId, authorization }: {
+    name?: string | null;
+    slug?: string | null;
+    visibility?: AgentVisibilityLevelEnum | VisibilityLevelEnum;
+    size?: number;
+    page?: number;
+    xAccountId?: string | null;
+    authorization: string;
+}, opts?: Oazapfts.RequestOpts) {
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 200;
+        data: ListAgentResponse[];
+    } | {
+        status: 404;
+    } | {
+        status: 422;
+        data: HttpValidationError;
+    }>(`/v1/agents${QS.query(QS.explode({
+        name,
+        slug,
+        visibility,
+        size,
+        page
+    }))}`, {
+        ...opts,
+        headers: oazapfts.mergeHeaders(opts?.headers, {
+            "x-account-id": xAccountId,
+            authorization
+        })
+    }));
+}
+/**
+ * Create Agent
+ */
+export function createAgentV1AgentsPost({ isPublic, xAccountId, authorization, newAgentRequest }: {
+    isPublic?: boolean;
+    xAccountId?: string | null;
+    authorization: string;
+    newAgentRequest: NewAgentRequest;
+}, opts?: Oazapfts.RequestOpts) {
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 201;
+        data: NewAgentResponse;
+    } | {
+        status: 404;
+    } | {
+        status: 422;
+        data: HttpValidationError;
+    }>(`/v1/agents${QS.query(QS.explode({
+        is_public: isPublic
+    }))}`, oazapfts.json({
+        ...opts,
+        method: "POST",
+        body: newAgentRequest,
+        headers: oazapfts.mergeHeaders(opts?.headers, {
+            "x-account-id": xAccountId,
+            authorization
+        })
+    })));
+}
+/**
+ * Search Agents
+ */
+export function searchAgentsV1AgentsSearchPost({ xAccountId, authorization, searchAgentsRequest }: {
+    xAccountId?: string | null;
+    authorization: string;
+    searchAgentsRequest: SearchAgentsRequest;
+}, opts?: Oazapfts.RequestOpts) {
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 200;
+        data: ListAgentResponse[];
+    } | {
+        status: 404;
+    } | {
+        status: 422;
+        data: HttpValidationError;
+    }>("/v1/agents/search", oazapfts.json({
+        ...opts,
+        method: "POST",
+        body: searchAgentsRequest,
+        headers: oazapfts.mergeHeaders(opts?.headers, {
+            "x-account-id": xAccountId,
+            authorization
+        })
+    })));
+}
+/**
+ * Get Agent
+ */
+export function getAgentV1AgentsAgentIdGet({ agentId, xAccountId, authorization }: {
+    agentId: string;
+    xAccountId?: string | null;
+    authorization: string;
+}, opts?: Oazapfts.RequestOpts) {
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 200;
+        data: AgentModel;
+    } | {
+        status: 404;
+    } | {
+        status: 422;
+        data: HttpValidationError;
+    }>(`/v1/agents/${encodeURIComponent(agentId)}`, {
+        ...opts,
+        headers: oazapfts.mergeHeaders(opts?.headers, {
+            "x-account-id": xAccountId,
+            authorization
+        })
+    }));
+}
+/**
+ * Update Agent
+ */
+export function updateAgentV1AgentsAgentIdPut({ agentId }: {
+    agentId: string;
+}, opts?: Oazapfts.RequestOpts) {
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 200;
+        data: any;
+    } | {
+        status: 404;
+    } | {
+        status: 422;
+        data: HttpValidationError;
+    }>(`/v1/agents/${encodeURIComponent(agentId)}`, {
+        ...opts,
+        method: "PUT"
+    }));
+}
+/**
+ * Delete Agent
+ */
+export function deleteAgentV1AgentsAgentIdDelete({ agentId }: {
+    agentId: string;
+}, opts?: Oazapfts.RequestOpts) {
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 200;
+        data: any;
+    } | {
+        status: 404;
+    } | {
+        status: 422;
+        data: HttpValidationError;
+    }>(`/v1/agents/${encodeURIComponent(agentId)}`, {
+        ...opts,
+        method: "DELETE"
+    }));
+}
+/**
+ * Add Favorite
+ */
+export function addFavoriteV1AgentsAgentIdFavoritePost({ agentId }: {
+    agentId: string;
+}, opts?: Oazapfts.RequestOpts) {
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 200;
+        data: any;
+    } | {
+        status: 404;
+    } | {
+        status: 422;
+        data: HttpValidationError;
+    }>(`/v1/agents/${encodeURIComponent(agentId)}/favorite`, {
+        ...opts,
+        method: "POST"
+    }));
+}
+/**
+ * Publish Agent
+ */
+export function publishAgentV1AgentsAgentIdPublishPost({ agentId }: {
+    agentId: string;
+}, opts?: Oazapfts.RequestOpts) {
+    return oazapfts.ok(oazapfts.fetchJson<{
+        status: 200;
+        data: any;
+    } | {
+        status: 404;
+    } | {
+        status: 422;
+        data: HttpValidationError;
+    }>(`/v1/agents/${encodeURIComponent(agentId)}/publish`, {
+        ...opts,
+        method: "POST"
+    }));
+}

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/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'
@@ -219,6 +219,7 @@ export interface WorkspaceAiMembersPermissions {
   totalPages: number,
 }
 
+
 export interface WorkspaceAiGroupsPermissions extends GroupsFromResourceResponse {
   actions: Action[],
 }
@@ -357,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