@agents-at-scale/ark 0.1.53 → 0.1.55

package/dist/commands/export/index.js

@@ -19,8 +19,8 @@ async function exportResources(options, config) {
     try {
         const allResourceTypes = config.defaultExportTypes || RESOURCE_ORDER;
         const outputPath = options.output || 'ark-export.yaml';
-        let resourceTypes = options.types
-            ? (options.types.split(','))
+        const resourceTypes = options.types
+            ? options.types.split(',')
             : allResourceTypes;
         // ensure that we get resources in the correct order; e.g. agents before teams that use the agents
         resourceTypes.sort((a, b) => {
@@ -37,7 +37,7 @@ async function exportResources(options, config) {
             output.info(`fetching ${resourceType}...`);
             const resources = await listResources(resourceType, {
                 namespace: options.namespace,
-                labels: options.labels
+                labels: options.labels,
             });
             const resourceCount = resources.length;
             if (resources.length > 0) {
@@ -50,7 +50,9 @@ async function exportResources(options, config) {
             output.warning('no resources found to export');
             return;
         }
-        const yamlContent = allResources.map((resource) => yaml.stringify(resource)).join("\n---\n");
+        const yamlContent = allResources
+            .map((resource) => yaml.stringify(resource))
+            .join('\n---\n');
         await fs.writeFile(outputPath, yamlContent, 'utf-8');
         output.success(`exported ${allResourceCount} resources to ${outputPath}`);
     }

package/dist/commands/generate/generators/agent.js

@@ -19,6 +19,8 @@ export function createAgentGenerator() {
     };
 }
 class AgentGenerator {
+    templateDiscovery;
+    templateEngine;
     constructor() {
         this.templateDiscovery = new TemplateDiscovery();
         this.templateEngine = new TemplateEngine();

package/dist/commands/generate/generators/marketplace.js

@@ -17,6 +17,8 @@ export function createMarketplaceGenerator() {
     };
 }
 class MarketplaceGenerator {
+    templateDiscovery;
+    templateEngine;
     constructor() {
         this.templateDiscovery = new TemplateDiscovery();
         this.templateEngine = new TemplateEngine();

package/dist/commands/generate/generators/mcpserver.js

@@ -16,6 +16,8 @@ export function createMcpServerGenerator() {
     };
 }
 class McpServerGenerator {
+    templateDiscovery;
+    templateEngine;
     constructor() {
         this.templateDiscovery = new TemplateDiscovery();
         this.templateEngine = new TemplateEngine();

package/dist/commands/generate/generators/project.js

@@ -21,6 +21,9 @@ export function createProjectGenerator() {
     };
 }
 class ProjectGenerator {
+    templateDiscovery;
+    templateEngine;
+    samplesPath;
     constructor() {
         this.templateDiscovery = new TemplateDiscovery();
         this.templateEngine = new TemplateEngine();
@@ -727,7 +730,9 @@ Generated with ARK CLI generator`;
             },
         ];
         if (config.projectType === 'empty') {
-            steps.push({ desc: 'Add YAML files to agents/, teams/, queries/ directories' }, { desc: 'Use either the default model already in models/ or a configuration template from samples/models/ of ARK repository' }, { desc: 'Edit .env file to set your API keys' }, { desc: 'Deploy your project', cmd: 'devspace dev' });
+            steps.push({ desc: 'Add YAML files to agents/, teams/, queries/ directories' }, {
+                desc: 'Use either the default model already in models/ or a configuration template from samples/models/ of ARK repository',
+            }, { desc: 'Edit .env file to set your API keys' }, { desc: 'Deploy your project', cmd: 'devspace dev' });
         }
         else if (config.selectedModels && config.selectedModels !== 'none') {
             steps.push({ desc: 'Edit .env file to set your API keys' }, { desc: 'Load environment variables', cmd: 'source .env' }, { desc: 'Deploy your project', cmd: 'devspace dev' }, {
@@ -736,7 +741,9 @@ Generated with ARK CLI generator`;
             });
         }
         else {
-            steps.push({ desc: 'Use either the default model already in models/ or a configuration template from samples/models/ of ARK repository' }, { desc: 'Edit .env file to set your API keys' }, { desc: 'Deploy your project', cmd: 'devspace dev' });
+            steps.push({
+                desc: 'Use either the default model already in models/ or a configuration template from samples/models/ of ARK repository',
+            }, { desc: 'Edit .env file to set your API keys' }, { desc: 'Deploy your project', cmd: 'devspace dev' });
         }
         console.log(chalk.magenta.bold('🚀 NEXT STEPS:\n'));
         let stepNumber = 1;

package/dist/commands/generate/generators/query.js

@@ -19,6 +19,8 @@ export function createQueryGenerator() {
     };
 }
 class QueryGenerator {
+    templateDiscovery;
+    templateEngine;
     constructor() {
         this.templateDiscovery = new TemplateDiscovery();
         this.templateEngine = new TemplateEngine();

package/dist/commands/generate/generators/team.js

@@ -18,6 +18,8 @@ export function createTeamGenerator() {
     };
 }
 class TeamGenerator {
+    templateDiscovery;
+    templateEngine;
     constructor() {
         this.templateDiscovery = new TemplateDiscovery();
         this.templateEngine = new TemplateEngine();

package/dist/commands/generate/templateDiscovery.js

@@ -2,6 +2,7 @@ import fs from 'fs';
 import path from 'path';
 import { fileURLToPath } from 'url';
 export class TemplateDiscovery {
+    templatesPath;
     constructor() {
         // Get the path to the templates directory
         // This handles both development and production scenarios

package/dist/commands/generate/templateEngine.js

@@ -4,9 +4,7 @@ import chalk from 'chalk';
 import { SecurityUtils } from '../../lib/security.js';
 import { TemplateError } from '../../lib/errors.js';
 export class TemplateEngine {
-    constructor() {
-        this.variables = {};
-    }
+    variables = {};
     /**
      * Set template variables for substitution
      */

package/dist/commands/import/index.js

@@ -5,7 +5,7 @@ async function importResources(filepath) {
     try {
         output.info(`importing ark resources from ${filepath}...`);
         const args = ['create', '-f', filepath];
-        const result = await execa('kubectl', args, {
+        await execa('kubectl', args, {
             stdio: 'pipe',
         });
         output.success(`imported resources from ${filepath}`);

package/dist/commands/install/index.js

@@ -40,6 +40,7 @@ async function checkAndCleanFailedRelease(releaseName, namespace, verbose = fals
         }
     }
     catch {
+        // Ignore errors - prerequisite may not exist
     }
 }
 async function installService(service, verbose = false) {
@@ -170,7 +171,7 @@ export async function installArk(config, serviceName, options = {}) {
                 checked: Boolean(service.enabled),
             })),
         ];
-        let selectedComponents = [];
+        let selectedComponents;
         try {
             const answers = await inquirer.prompt([
                 {

package/dist/commands/models/kubernetes/manifest-builder.js

@@ -1,5 +1,6 @@
 // Kubernetes model manifest builder
 export class KubernetesModelManifestBuilder {
+    modelName;
     constructor(modelName) {
         this.modelName = modelName;
     }
@@ -23,24 +24,40 @@ export class KubernetesModelManifestBuilder {
     }
     buildProviderConfig(config) {
         if (config.type === 'azure') {
-            return {
-                azure: {
+            const azureConfig = config;
+            const azure = {
+                baseUrl: { value: azureConfig.baseUrl },
+                apiVersion: { value: azureConfig.apiVersion },
+            };
+            const authMethod = azureConfig.authMethod ?? 'apiKey';
+            if (authMethod === 'apiKey') {
+                azure.auth = {
                     apiKey: {
                         valueFrom: {
                             secretKeyRef: {
-                                name: config.secretName,
+                                name: azureConfig.secretName || 'azure-openai-secret',
                                 key: 'api-key',
                             },
                         },
                     },
-                    baseUrl: {
-                        value: config.baseUrl,
-                    },
-                    apiVersion: {
-                        value: config.apiVersion,
+                };
+            }
+            else if (authMethod === 'managedIdentity') {
+                azure.auth = {
+                    managedIdentity: azureConfig.clientId ?
+                        { clientId: { value: azureConfig.clientId } }
+                        : {},
+                };
+            }
+            else if (authMethod === 'workloadIdentity') {
+                azure.auth = {
+                    workloadIdentity: {
+                        clientId: { value: azureConfig.clientId },
+                        tenantId: { value: azureConfig.tenantId },
                     },
-                },
-            };
+                };
+            }
+            return { azure };
         }
         if (config.type === 'bedrock') {
             return this.buildBedrockConfig(config);

package/dist/commands/models/providers/azure.d.ts

@@ -1,12 +1,16 @@
 import { BaseProviderConfig, BaseCollectorOptions, ProviderConfigCollector } from './types.js';
+export type AzureAuthMethod = 'apiKey' | 'managedIdentity' | 'workloadIdentity';
 /**
  * Configuration for Azure OpenAI models.
  */
 export interface AzureConfig extends BaseProviderConfig {
     type: 'azure';
     baseUrl: string;
-    apiKey: string;
     apiVersion: string;
+    authMethod?: AzureAuthMethod;
+    apiKey?: string;
+    clientId?: string;
+    tenantId?: string;
 }
 /**
  * Options specific to Azure collector.
@@ -15,16 +19,15 @@ export interface AzureCollectorOptions extends BaseCollectorOptions {
     baseUrl?: string;
     apiKey?: string;
     apiVersion?: string;
+    authMethod?: AzureAuthMethod;
+    clientId?: string;
+    tenantId?: string;
 }
 /**
  * Configuration collector for Azure OpenAI models.
  *
- * Collects the necessary configuration to connect to Azure OpenAI Service:
- * - baseUrl: The Azure OpenAI endpoint URL (e.g., https://<resource>.openai.azure.com)
- * - apiVersion: The API version to use (defaults to 2024-12-01-preview)
- * - apiKey: The authentication key for the Azure OpenAI resource
- *
- * Values can be provided via command-line options or will be prompted interactively.
+ * Supports API Key, Managed Identity (AKS), and Workload Identity auth.
+ * Values can be provided via command-line options or prompted interactively.
  */
 export declare class AzureConfigCollector implements ProviderConfigCollector {
     collectConfig(options: BaseCollectorOptions): Promise<AzureConfig>;

package/dist/commands/models/providers/azure.js

@@ -2,12 +2,8 @@ import inquirer from 'inquirer';
 /**
  * Configuration collector for Azure OpenAI models.
  *
- * Collects the necessary configuration to connect to Azure OpenAI Service:
- * - baseUrl: The Azure OpenAI endpoint URL (e.g., https://<resource>.openai.azure.com)
- * - apiVersion: The API version to use (defaults to 2024-12-01-preview)
- * - apiKey: The authentication key for the Azure OpenAI resource
- *
- * Values can be provided via command-line options or will be prompted interactively.
+ * Supports API Key, Managed Identity (AKS), and Workload Identity auth.
+ * Values can be provided via command-line options or prompted interactively.
  */
 export class AzureConfigCollector {
     async collectConfig(options) {
@@ -50,33 +46,99 @@ export class AzureConfigCollector {
             ]);
             apiVersion = answer.apiVersion;
         }
-        let apiKey = azureOptions.apiKey;
-        if (!apiKey) {
+        let authMethod = azureOptions.authMethod;
+        if (!authMethod && azureOptions.apiKey) {
+            authMethod = 'apiKey';
+        }
+        if (!authMethod) {
             const answer = await inquirer.prompt([
                 {
-                    type: 'password',
-                    name: 'apiKey',
-                    message: 'API key:',
-                    mask: '*',
-                    validate: (input) => {
-                        if (!input)
-                            return 'API key is required';
-                        return true;
-                    },
+                    type: 'list',
+                    name: 'authMethod',
+                    message: 'Authentication:',
+                    choices: [
+                        { name: 'API Key', value: 'apiKey' },
+                        { name: 'Managed Identity (AKS)', value: 'managedIdentity' },
+                        { name: 'Workload Identity', value: 'workloadIdentity' },
+                    ],
                 },
             ]);
-            apiKey = answer.apiKey;
+            authMethod = answer.authMethod;
+        }
+        let apiKey;
+        let clientId;
+        let tenantId;
+        if (authMethod === 'apiKey') {
+            apiKey = azureOptions.apiKey;
+            if (!apiKey) {
+                const answer = await inquirer.prompt([
+                    {
+                        type: 'password',
+                        name: 'apiKey',
+                        message: 'API key:',
+                        mask: '*',
+                        validate: (input) => {
+                            if (!input)
+                                return 'API key is required';
+                            return true;
+                        },
+                    },
+                ]);
+                apiKey = answer.apiKey;
+            }
+            if (!apiKey) {
+                throw new Error('API key is required');
+            }
         }
-        if (!apiKey) {
-            throw new Error('API key is required');
+        else if (authMethod === 'managedIdentity') {
+            clientId = azureOptions.clientId;
+            if (!clientId) {
+                const answer = await inquirer.prompt([
+                    {
+                        type: 'input',
+                        name: 'clientId',
+                        message: 'Managed Identity Client ID (optional for system-assigned):',
+                    },
+                ]);
+                clientId = answer.clientId || undefined;
+            }
+        }
+        else if (authMethod === 'workloadIdentity') {
+            clientId = azureOptions.clientId;
+            tenantId = azureOptions.tenantId;
+            if (!clientId) {
+                const answer = await inquirer.prompt([
+                    {
+                        type: 'input',
+                        name: 'clientId',
+                        message: 'Workload Identity Client ID:',
+                        validate: (input) => input ? true : 'Client ID is required for Workload Identity',
+                    },
+                ]);
+                clientId = answer.clientId;
+            }
+            if (!tenantId) {
+                const answer = await inquirer.prompt([
+                    {
+                        type: 'input',
+                        name: 'tenantId',
+                        message: 'Azure Tenant ID:',
+                        validate: (input) => input ? true : 'Tenant ID is required for Workload Identity',
+                    },
+                ]);
+                tenantId = answer.tenantId;
+            }
         }
         return {
             type: 'azure',
             modelValue: options.model,
             secretName: '',
             baseUrl,
-            apiKey,
             apiVersion,
+            authMethod,
+            apiKey,
+            clientId,
+            tenantId,
         };
     }
 }

package/dist/commands/uninstall/index.js

@@ -88,7 +88,7 @@ async function uninstallArk(config, serviceName, options = {}) {
     const services = getInstallableServices();
     const serviceEntries = Object.entries(services).reverse();
     for (const [, service] of serviceEntries) {
-        let shouldUninstall = false;
+        let shouldUninstall;
         try {
             // Ask for confirmation
             shouldUninstall =

package/dist/components/ChatUI.js

@@ -410,31 +410,32 @@ const ChatUI = ({ initialTargetId, arkApiClient, arkApiProxy, config, }) => {
             // Create abort controller for this request
             const controller = new AbortController();
             setAbortController(controller);
-            // Convert messages to format expected by OpenAI API - only include user and agent messages
             const apiMessages = messages
                 .filter((msg) => msg.type === 'user' || msg.type === 'agent' || msg.type === 'team')
-                .map((msg) => {
+                .flatMap((msg) => {
                 if (msg.type === 'user') {
-                    return {
-                        role: 'user',
-                        content: msg.content,
-                    };
+                    return [
+                        {
+                            role: 'user',
+                            content: msg.content,
+                        },
+                    ];
                 }
                 else if (msg.type === 'agent') {
-                    return {
-                        role: 'assistant',
-                        content: msg.content,
-                    };
+                    return [
+                        {
+                            role: 'assistant',
+                            content: msg.content,
+                        },
+                    ];
                 }
                 else if (msg.type === 'team') {
-                    // For teams, concatenate all member responses
-                    const content = msg.members.map((m) => m.content).join(' ');
-                    return {
+                    return msg.members.map((m) => ({
                         role: 'assistant',
-                        content: content || '',
-                    };
+                        content: m.content || '',
+                    }));
                 }
-                return { role: 'user', content: '' };
+                return [];
             });
             // Add the new user message
             apiMessages.push({

package/dist/components/statusChecker.js

@@ -24,7 +24,7 @@ export const getKubectlVersion = () => ({
             throw new Error('kubectl version output missing clientVersion field');
         }
         catch (e) {
-            throw new Error(`Failed to parse kubectl version JSON: ${e instanceof Error ? e.message : 'Unknown error'}`);
+            throw new Error(`Failed to parse kubectl version JSON: ${e instanceof Error ? e.message : 'Unknown error'}`, { cause: e });
         }
     },
 });
@@ -72,7 +72,7 @@ export class StatusChecker {
             return config.versionExtract(stdout);
         }
         catch (error) {
-            throw new Error(`Failed to get ${config.command} version: ${error instanceof Error ? error.message : 'Unknown error'}`);
+            throw new Error(`Failed to get ${config.command} version: ${error instanceof Error ? error.message : 'Unknown error'}`, { cause: error });
         }
     }
     /**
@@ -341,7 +341,7 @@ export class StatusChecker {
             clusterAccess = true;
         }
         catch {
-            clusterAccess = false;
+            // clusterAccess remains false
         }
         // Get cluster info if accessible
         let clusterInfo;

package/dist/lib/arkApiClient.js

@@ -1,5 +1,7 @@
 import OpenAI from 'openai';
 export class ArkApiClient {
+    openai;
+    baseUrl;
     constructor(arkApiUrl) {
         this.baseUrl = arkApiUrl;
         this.openai = new OpenAI({
@@ -29,7 +31,7 @@ export class ArkApiClient {
             return targets;
         }
         catch (error) {
-            throw new Error(`Failed to get query targets: ${error instanceof Error ? error.message : 'Unknown error'}`);
+            throw new Error(`Failed to get query targets: ${error instanceof Error ? error.message : 'Unknown error'}`, { cause: error });
         }
     }
     async getAgents() {
@@ -42,7 +44,7 @@ export class ArkApiClient {
             return data.items || [];
         }
         catch (error) {
-            throw new Error(`Failed to get agents: ${error instanceof Error ? error.message : 'Unknown error'}`);
+            throw new Error(`Failed to get agents: ${error instanceof Error ? error.message : 'Unknown error'}`, { cause: error });
         }
     }
     async getModels() {
@@ -55,7 +57,7 @@ export class ArkApiClient {
             return data.items || [];
         }
         catch (error) {
-            throw new Error(`Failed to get models: ${error instanceof Error ? error.message : 'Unknown error'}`);
+            throw new Error(`Failed to get models: ${error instanceof Error ? error.message : 'Unknown error'}`, { cause: error });
         }
     }
     async getTools() {
@@ -68,7 +70,7 @@ export class ArkApiClient {
             return data.items || [];
         }
         catch (error) {
-            throw new Error(`Failed to get tools: ${error instanceof Error ? error.message : 'Unknown error'}`);
+            throw new Error(`Failed to get tools: ${error instanceof Error ? error.message : 'Unknown error'}`, { cause: error });
         }
     }
     async getTeams() {
@@ -81,7 +83,7 @@ export class ArkApiClient {
             return data.items || [];
         }
         catch (error) {
-            throw new Error(`Failed to get teams: ${error instanceof Error ? error.message : 'Unknown error'}`);
+            throw new Error(`Failed to get teams: ${error instanceof Error ? error.message : 'Unknown error'}`, { cause: error });
         }
     }
     async getSessions() {
@@ -94,7 +96,7 @@ export class ArkApiClient {
             return data.items || [];
         }
         catch (error) {
-            throw new Error(`Failed to get sessions: ${error instanceof Error ? error.message : 'Unknown error'}`);
+            throw new Error(`Failed to get sessions: ${error instanceof Error ? error.message : 'Unknown error'}`, { cause: error });
         }
     }
     async deleteSession(sessionId) {
@@ -108,7 +110,7 @@ export class ArkApiClient {
             return await response.json();
         }
         catch (error) {
-            throw new Error(`Failed to delete session: ${error instanceof Error ? error.message : 'Unknown error'}`);
+            throw new Error(`Failed to delete session: ${error instanceof Error ? error.message : 'Unknown error'}`, { cause: error });
         }
     }
     async deleteQueryMessages(sessionId, queryId) {
@@ -122,7 +124,7 @@ export class ArkApiClient {
             return await response.json();
         }
         catch (error) {
-            throw new Error(`Failed to delete query messages: ${error instanceof Error ? error.message : 'Unknown error'}`);
+            throw new Error(`Failed to delete query messages: ${error instanceof Error ? error.message : 'Unknown error'}`, { cause: error });
         }
     }
     async deleteAllSessions() {
@@ -136,7 +138,7 @@ export class ArkApiClient {
             return await response.json();
         }
         catch (error) {
-            throw new Error(`Failed to delete all sessions: ${error instanceof Error ? error.message : 'Unknown error'}`);
+            throw new Error(`Failed to delete all sessions: ${error instanceof Error ? error.message : 'Unknown error'}`, { cause: error });
         }
     }
     async createChatCompletion(params) {

package/dist/lib/arkApiProxy.js

@@ -2,6 +2,7 @@ import { ArkApiClient } from './arkApiClient.js';
 import { ArkServiceProxy } from './arkServiceProxy.js';
 import { arkServices } from '../arkServices.js';
 export class ArkApiProxy {
+    serviceProxy;
     constructor(localPort, reusePortForwards = false) {
         const arkApiService = arkServices['ark-api'];
         this.serviceProxy = new ArkServiceProxy(arkApiService, localPort, reusePortForwards);

package/dist/lib/arkServiceProxy.js

@@ -3,9 +3,13 @@ import find from 'find-process';
 import Debug from 'debug';
 const debug = Debug('ark:service-proxy');
 export class ArkServiceProxy {
+    reusePortForwards;
+    kubectlProcess;
+    localPort;
+    isReady = false;
+    service;
     constructor(service, localPort, reusePortForwards = false) {
         this.reusePortForwards = reusePortForwards;
-        this.isReady = false;
         this.service = service;
         this.localPort =
             localPort || service.k8sPortForwardLocalPort || this.getRandomPort();

package/dist/lib/chatClient.js

@@ -1,5 +1,6 @@
 import { QUERY_ANNOTATIONS } from './constants.js';
 export class ChatClient {
+    arkApiClient;
     constructor(arkApiClient) {
         this.arkApiClient = arkApiClient;
     }
@@ -57,6 +58,14 @@ export class ChatClient {
                 if (onChunk) {
                     onChunk(content, undefined, arkMetadata);
                 }
+                if (!fullResponse &&
+                    arkMetadata?.completedQuery?.status?.response?.content) {
+                    const responseContent = arkMetadata.completedQuery.status.response.content;
+                    fullResponse = responseContent;
+                    if (onChunk) {
+                        onChunk(responseContent, undefined, arkMetadata);
+                    }
+                }
                 // Handle tool calls
                 if (delta?.tool_calls) {
                     for (const toolCallDelta of delta.tool_calls) {

package/dist/lib/config.js

@@ -33,7 +33,9 @@ export function loadConfig() {
         }
         catch (e) {
             const message = e instanceof Error ? e.message : 'Unknown error';
-            throw new Error(`Invalid YAML in ${userConfigPath}: ${message}`);
+            throw new Error(`Invalid YAML in ${userConfigPath}: ${message}`, {
+                cause: e,
+            });
         }
     }
     // Load project config from current directory
@@ -45,7 +47,9 @@ export function loadConfig() {
         }
         catch (e) {
             const message = e instanceof Error ? e.message : 'Unknown error';
-            throw new Error(`Invalid YAML in ${projectConfigPath}: ${message}`);
+            throw new Error(`Invalid YAML in ${projectConfigPath}: ${message}`, {
+                cause: e,
+            });
         }
     }
     // Apply environment variable overrides
@@ -106,7 +110,8 @@ function mergeConfig(target, source) {
             target.services.reusePortForwards = source.services.reusePortForwards;
         }
         for (const [serviceName, overrides] of Object.entries(source.services)) {
-            if (serviceName !== 'reusePortForwards' && typeof overrides === 'object') {
+            if (serviceName !== 'reusePortForwards' &&
+                typeof overrides === 'object') {
                 target.services[serviceName] = {
                     ...target.services[serviceName],
                     ...overrides,

package/dist/lib/errors.js

@@ -24,6 +24,9 @@ export var ErrorCode;
     ErrorCode["UNKNOWN_ERROR"] = "UNKNOWN_ERROR";
 })(ErrorCode || (ErrorCode = {}));
 export class ArkError extends Error {
+    code;
+    details;
+    suggestions;
     constructor(message, code = ErrorCode.UNKNOWN_ERROR, details, suggestions) {
         super(message);
         this.name = 'ArkError';