jiva-core 0.1.0

package/dist/mcp/server-manager.js

@@ -0,0 +1,135 @@
+/**
+ * MCP Server Manager
+ *
+ * Manages lifecycle of MCP servers based on configuration
+ */
+import { MCPClient } from './client.js';
+import { logger } from '../utils/logger.js';
+import { MCPError } from '../utils/errors.js';
+export class MCPServerManager {
+    mcpClient;
+    serverConfigs = new Map();
+    constructor() {
+        this.mcpClient = new MCPClient();
+    }
+    /**
+     * Initialize servers from configuration
+     */
+    async initialize(servers) {
+        logger.info('Initializing MCP servers...');
+        const serverEntries = Object.entries(servers);
+        const enabledServers = serverEntries.filter(([_, config]) => config.enabled);
+        logger.info(`Found ${enabledServers.length} enabled MCP servers`);
+        // Store configs
+        for (const [name, config] of serverEntries) {
+            this.serverConfigs.set(name, config);
+        }
+        // Connect to enabled servers
+        const connectionPromises = enabledServers.map(async ([name, config]) => {
+            try {
+                await this.connectServer(name, config);
+                return { name, success: true };
+            }
+            catch (error) {
+                logger.warn(`Failed to connect to MCP server '${name}': ${error instanceof Error ? error.message : String(error)}`);
+                // Continue with other servers even if one fails
+                return { name, success: false };
+            }
+        });
+        const results = await Promise.allSettled(connectionPromises);
+        const connectedServers = this.mcpClient.getConnectedServers();
+        if (connectedServers.length > 0) {
+            logger.success(`MCP servers connected: ${connectedServers.join(', ')}`);
+        }
+        else if (enabledServers.length > 0) {
+            logger.warn('No MCP servers connected. Agent will run without external tools.');
+        }
+        else {
+            logger.info('No MCP servers enabled. Agent will run without external tools.');
+        }
+    }
+    /**
+     * Connect to a specific server
+     */
+    async connectServer(name, config) {
+        await this.mcpClient.connect(name, config.command, config.args || [], config.env);
+    }
+    /**
+     * Add and connect to a new server
+     */
+    async addServer(name, config) {
+        if (this.serverConfigs.has(name)) {
+            throw new MCPError(`Server '${name}' already exists`, name);
+        }
+        this.serverConfigs.set(name, config);
+        if (config.enabled) {
+            await this.connectServer(name, config);
+        }
+    }
+    /**
+     * Remove and disconnect from a server
+     */
+    async removeServer(name) {
+        await this.mcpClient.disconnect(name);
+        this.serverConfigs.delete(name);
+    }
+    /**
+     * Enable a server and connect
+     */
+    async enableServer(name) {
+        const config = this.serverConfigs.get(name);
+        if (!config) {
+            throw new MCPError(`Server '${name}' not found in configuration`, name);
+        }
+        config.enabled = true;
+        if (!this.mcpClient.isConnected(name)) {
+            await this.connectServer(name, config);
+        }
+    }
+    /**
+     * Disable a server and disconnect
+     */
+    async disableServer(name) {
+        const config = this.serverConfigs.get(name);
+        if (!config) {
+            throw new MCPError(`Server '${name}' not found in configuration`, name);
+        }
+        config.enabled = false;
+        await this.mcpClient.disconnect(name);
+    }
+    /**
+     * Get the MCP client instance
+     */
+    getClient() {
+        return this.mcpClient;
+    }
+    /**
+     * Get all server configurations
+     */
+    getServerConfigs() {
+        return new Map(this.serverConfigs);
+    }
+    /**
+     * Cleanup all servers
+     */
+    async cleanup() {
+        logger.info('Cleaning up MCP servers...');
+        await this.mcpClient.disconnectAll();
+    }
+    /**
+     * Get server status
+     */
+    getServerStatus() {
+        return Array.from(this.serverConfigs.entries()).map(([name, config]) => {
+            const connected = this.mcpClient.isConnected(name);
+            const tools = connected ? this.mcpClient.getServerTools(name) : [];
+            return {
+                name,
+                enabled: config.enabled,
+                connected,
+                toolCount: tools.length,
+            };
+        });
+    }
+}
+//# sourceMappingURL=server-manager.js.map

package/dist/mcp/server-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server-manager.js","sourceRoot":"","sources":["../../src/mcp/server-manager.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAExC,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAE9C,MAAM,OAAO,gBAAgB;IACnB,SAAS,CAAY;IACrB,aAAa,GAAiC,IAAI,GAAG,EAAE,CAAC;IAEhE;QACE,IAAI,CAAC,SAAS,GAAG,IAAI,SAAS,EAAE,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,OAAwC;QACvD,MAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;QAE3C,MAAM,aAAa,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC9C,MAAM,cAAc,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAE7E,MAAM,CAAC,IAAI,CAAC,SAAS,cAAc,CAAC,MAAM,sBAAsB,CAAC,CAAC;QAElE,gBAAgB;QAChB,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;YAC3C,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACvC,CAAC;QAED,6BAA6B;QAC7B,MAAM,kBAAkB,GAAG,cAAc,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,EAAE;YACrE,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;gBACvC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACjC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,CAAC,oCAAoC,IAAI,MAAM,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBACpH,gDAAgD;gBAChD,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAClC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;QAE7D,MAAM,gBAAgB,GAAG,IAAI,CAAC,SAAS,CAAC,mBAAmB,EAAE,CAAC;QAE9D,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,MAAM,CAAC,OAAO,CAAC,0BAA0B,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC1E,CAAC;aAAM,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrC,MAAM,CAAC,IAAI,CAAC,kEAAkE,CAAC,CAAC;QAClF,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CAAC,gEAAgE,CAAC,CAAC;QAChF,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,aAAa,CAAC,IAAY,EAAE,MAAuB;QAC/D,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAC1B,IAAI,EACJ,MAAM,CAAC,OAAO,EACd,MAAM,CAAC,IAAI,IAAI,EAAE,EACjB,MAAM,CAAC,GAAG,CACX,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,IAAY,EAAE,MAAuB;QACnD,IAAI,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,QAAQ,CAAC,WAAW,IAAI,kBAAkB,EAAE,IAAI,CAAC,CAAC;QAC9D,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAErC,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,IAAY;QAC7B,MAAM,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,IAAY;QAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,QAAQ,CAAC,WAAW,IAAI,8BAA8B,EAAE,IAAI,CAAC,CAAC;QAC1E,CAAC;QAED,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC;QAEtB,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;YACtC,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa,CAAC,IAAY;QAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,QAAQ,CAAC,WAAW,IAAI,8BAA8B,EAAE,IAAI,CAAC,CAAC;QAC1E,CAAC;QAED,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC;QACvB,MAAM,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IACxC,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO;QACX,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QAC1C,MAAM,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,eAAe;QAMb,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,EAAE;YACrE,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACnD,MAAM,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAEnE,OAAO;gBACL,IAAI;gBACJ,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,SAAS;gBACT,SAAS,EAAE,KAAK,CAAC,MAAM;aACxB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/models/base.d.ts

@@ -0,0 +1,57 @@
+/**
+ * Base model interfaces and types
+ */
+export interface Message {
+    role: 'system' | 'developer' | 'user' | 'assistant' | 'tool';
+    content: string | MessageContent[];
+    name?: string;
+    tool_call_id?: string;
+}
+export interface MessageContent {
+    type: 'text' | 'image_url';
+    text?: string;
+    image_url?: {
+        url: string;
+    };
+}
+export interface ModelResponse {
+    content: string;
+    toolCalls?: ToolCall[];
+    usage?: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+    raw?: any;
+}
+export interface ToolCall {
+    id: string;
+    type: 'function';
+    function: {
+        name: string;
+        arguments: string;
+    };
+}
+export interface Tool {
+    name: string;
+    description: string;
+    parameters: {
+        type: 'object';
+        properties: Record<string, any>;
+        required?: string[];
+    };
+}
+export interface ChatCompletionOptions {
+    model: string;
+    messages: Message[];
+    tools?: Tool[];
+    temperature?: number;
+    maxTokens?: number;
+    stream?: boolean;
+}
+export interface IModel {
+    chat(options: ChatCompletionOptions): Promise<ModelResponse>;
+    supportsVision(): boolean;
+    supportsToolCalling(): boolean;
+}
+//# sourceMappingURL=base.d.ts.map

package/dist/models/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../src/models/base.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,QAAQ,GAAG,WAAW,GAAG,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;IAC7D,OAAO,EAAE,MAAM,GAAG,cAAc,EAAE,CAAC;IACnC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;IAC3B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE;QACV,GAAG,EAAE,MAAM,CAAC;KACb,CAAC;CACH;AAED,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC;IACvB,KAAK,CAAC,EAAE;QACN,YAAY,EAAE,MAAM,CAAC;QACrB,gBAAgB,EAAE,MAAM,CAAC;QACzB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;IACF,GAAG,CAAC,EAAE,GAAG,CAAC;CACX;AAED,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;CACH;AAED,MAAM,WAAW,IAAI;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ,CAAC;QACf,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAChC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;KACrB,CAAC;CACH;AAED,MAAM,WAAW,qBAAqB;IACpC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,MAAM;IACrB,IAAI,CAAC,OAAO,EAAE,qBAAqB,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC;IAC7D,cAAc,IAAI,OAAO,CAAC;IAC1B,mBAAmB,IAAI,OAAO,CAAC;CAChC"}

package/dist/models/base.js

@@ -0,0 +1,5 @@
+/**
+ * Base model interfaces and types
+ */
+export {};
+//# sourceMappingURL=base.js.map

package/dist/models/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/models/base.ts"],"names":[],"mappings":"AAAA;;GAEG"}

package/dist/models/harmony.d.ts

@@ -0,0 +1,78 @@
+/**
+ * Harmony Response Format Handler
+ *
+ * Implements the Harmony format required by gpt-oss-120b models.
+ * Handles multi-channel output, tool calling, and structured message formatting.
+ *
+ * Reference: https://github.com/openai/harmony
+ */
+export interface HarmonyMessage {
+    role: 'system' | 'developer' | 'user' | 'assistant' | 'tool';
+    content: string | HarmonyContent[];
+    name?: string;
+    tool_call_id?: string;
+}
+export interface HarmonyContent {
+    type: 'text' | 'image_url';
+    text?: string;
+    image_url?: {
+        url: string;
+    };
+}
+export interface HarmonyToolDefinition {
+    name: string;
+    description: string;
+    parameters: {
+        type: 'object';
+        properties: Record<string, any>;
+        required?: string[];
+    };
+}
+export interface HarmonyToolCall {
+    id: string;
+    type: 'function';
+    function: {
+        name: string;
+        arguments: string;
+    };
+}
+export interface ParsedHarmonyResponse {
+    analysis?: string;
+    commentary?: string;
+    final?: string;
+    toolCalls: HarmonyToolCall[];
+    rawResponse: string;
+}
+/**
+ * Formats tools into Harmony's TypeScript-like syntax for the developer message
+ */
+export declare function formatToolsForHarmony(tools: HarmonyToolDefinition[]): string;
+/**
+ * Formats messages into Harmony format
+ */
+export declare function formatMessagesForHarmony(messages: HarmonyMessage[], tools?: HarmonyToolDefinition[]): HarmonyMessage[];
+/**
+ * Parses Harmony response with multi-channel support
+ *
+ * Expected format:
+ * <|channel|>analysis
+ * [chain of thought content]
+ * <|channel|>final
+ * [final response]
+ * <|call|>function_name({"param": "value"})<|return|>
+ */
+export declare function parseHarmonyResponse(response: string): ParsedHarmonyResponse;
+/**
+ * Extracts assistant message with malformed tool call detection
+ * This handles cases where gpt-oss-120b generates incorrect tool call formats
+ */
+export declare function extractAssistantMessage(response: string): string;
+/**
+ * Validates tool call against available tools
+ */
+export declare function validateToolCall(toolCall: HarmonyToolCall, availableTools: HarmonyToolDefinition[]): boolean;
+/**
+ * Formats tool result for Harmony format
+ */
+export declare function formatToolResult(toolCallId: string, toolName: string, result: any): HarmonyMessage;
+//# sourceMappingURL=harmony.d.ts.map

package/dist/models/harmony.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"harmony.d.ts","sourceRoot":"","sources":["../../src/models/harmony.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAKH,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,QAAQ,GAAG,WAAW,GAAG,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;IAC7D,OAAO,EAAE,MAAM,GAAG,cAAc,EAAE,CAAC;IACnC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;IAC3B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE;QACV,GAAG,EAAE,MAAM,CAAC;KACb,CAAC;CACH;AAED,MAAM,WAAW,qBAAqB;IACpC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ,CAAC;QACf,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAChC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;KACrB,CAAC;CACH;AAED,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;CACH;AAED,MAAM,WAAW,qBAAqB;IACpC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,eAAe,EAAE,CAAC;IAC7B,WAAW,EAAE,MAAM,CAAC;CACrB;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,qBAAqB,EAAE,GAAG,MAAM,CAoD5E;AAED;;GAEG;AACH,wBAAgB,wBAAwB,CACtC,QAAQ,EAAE,cAAc,EAAE,EAC1B,KAAK,CAAC,EAAE,qBAAqB,EAAE,GAC9B,cAAc,EAAE,CAmBlB;AAED;;;;;;;;;GASG;AACH,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,GAAG,qBAAqB,CAmF5E;AAED;;;GAGG;AACH,wBAAgB,uBAAuB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAchE;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAC9B,QAAQ,EAAE,eAAe,EACzB,cAAc,EAAE,qBAAqB,EAAE,GACtC,OAAO,CAyBT;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAC9B,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,GAAG,GACV,cAAc,CAOhB"}

package/dist/models/harmony.js

@@ -0,0 +1,226 @@
+/**
+ * Harmony Response Format Handler
+ *
+ * Implements the Harmony format required by gpt-oss-120b models.
+ * Handles multi-channel output, tool calling, and structured message formatting.
+ *
+ * Reference: https://github.com/openai/harmony
+ */
+import { logger } from '../utils/logger.js';
+import { ToolCallError } from '../utils/errors.js';
+/**
+ * Formats tools into Harmony's TypeScript-like syntax for the developer message
+ */
+export function formatToolsForHarmony(tools) {
+    if (tools.length === 0)
+        return '';
+    const toolDefinitions = tools.map(tool => {
+        const params = Object.entries(tool.parameters.properties || {})
+            .map(([name, schema]) => {
+            const required = tool.parameters.required?.includes(name) ? '' : '?';
+            const type = schema.type || 'any';
+            const description = schema.description ? ` // ${schema.description}` : '';
+            return `  ${name}${required}: ${type};${description}`;
+        })
+            .join('\n');
+        return `/**
+ * ${tool.description}
+ */
+function ${tool.name}(params: {
+${params}
+}): void;`;
+    }).join('\n\n');
+    return `# Available Tools
+
+You have access to the following tools:
+
+<namespace name="functions">
+${toolDefinitions}
+</namespace>
+
+## CRITICAL: How to Use Tools
+
+To execute a tool, you MUST use this EXACT format:
+
+<|call|>function_name({"param": "value"})<|return|>
+
+Example:
+<|call|>read_file({"path": "/path/to/file"})<|return|>
+
+## Rules:
+1. Use the EXACT function names from above (e.g., filesystem__read_file)
+2. Parameters MUST be valid JSON
+3. Do NOT output markdown code blocks with tool calls
+4. Do NOT explain what you're doing - just call the tool
+5. Output the tool call directly with the <|call|> and <|return|> markers
+
+WRONG - Do not do this:
+\`\`\`json
+{"action": "read_file", "parameters": {...}}
+\`\`\`
+
+CORRECT - Do this:
+<|call|>filesystem__read_file({"path": "/path"})<|return|>`;
+}
+/**
+ * Formats messages into Harmony format
+ */
+export function formatMessagesForHarmony(messages, tools) {
+    const formattedMessages = [];
+    // Process messages based on role hierarchy
+    for (const msg of messages) {
+        if (msg.role === 'developer' && tools && tools.length > 0) {
+            // Inject tool definitions into developer message
+            const toolSection = formatToolsForHarmony(tools);
+            const existingContent = typeof msg.content === 'string' ? msg.content : '';
+            formattedMessages.push({
+                ...msg,
+                content: `${existingContent}\n\n${toolSection}`,
+            });
+        }
+        else {
+            formattedMessages.push(msg);
+        }
+    }
+    return formattedMessages;
+}
+/**
+ * Parses Harmony response with multi-channel support
+ *
+ * Expected format:
+ * <|channel|>analysis
+ * [chain of thought content]
+ * <|channel|>final
+ * [final response]
+ * <|call|>function_name({"param": "value"})<|return|>
+ */
+export function parseHarmonyResponse(response) {
+    const result = {
+        toolCalls: [],
+        rawResponse: response,
+    };
+    // Parse channels
+    const channelRegex = /<\|channel\|>(\w+)\s*([\s\S]*?)(?=<\|channel\|>|<\|call\|>|$)/g;
+    let match;
+    while ((match = channelRegex.exec(response)) !== null) {
+        const channelName = match[1];
+        const channelContent = match[2].trim();
+        if (channelName === 'analysis') {
+            result.analysis = channelContent;
+        }
+        else if (channelName === 'commentary') {
+            result.commentary = channelContent;
+        }
+        else if (channelName === 'final') {
+            result.final = channelContent;
+        }
+    }
+    // Parse tool calls
+    const toolCallRegex = /<\|call\|>([\s\S]*?)<\|return\|>/g;
+    let toolMatch;
+    let callId = 0;
+    while ((toolMatch = toolCallRegex.exec(response)) !== null) {
+        const toolCallContent = toolMatch[1].trim();
+        try {
+            // Parse function call: function_name({"param": "value"})
+            const functionMatch = /^(\w+)\(([\s\S]*)\)$/.exec(toolCallContent);
+            if (functionMatch) {
+                const functionName = functionMatch[1];
+                const argsString = functionMatch[2];
+                // Validate JSON
+                let parsedArgs;
+                try {
+                    parsedArgs = JSON.parse(argsString);
+                }
+                catch (e) {
+                    logger.warn(`Failed to parse tool call arguments for ${functionName}: ${argsString}`);
+                    // Try to fix common JSON issues
+                    const fixedArgs = argsString
+                        .replace(/'/g, '"') // Replace single quotes
+                        .replace(/(\w+):/g, '"$1":'); // Quote unquoted keys
+                    try {
+                        parsedArgs = JSON.parse(fixedArgs);
+                        logger.debug(`Successfully fixed and parsed arguments: ${fixedArgs}`);
+                    }
+                    catch (e2) {
+                        throw new ToolCallError(`Invalid JSON in tool call arguments: ${argsString}`, functionName);
+                    }
+                }
+                result.toolCalls.push({
+                    id: `call_${callId++}`,
+                    type: 'function',
+                    function: {
+                        name: functionName,
+                        arguments: JSON.stringify(parsedArgs),
+                    },
+                });
+            }
+            else {
+                logger.warn(`Failed to parse tool call format: ${toolCallContent}`);
+            }
+        }
+        catch (error) {
+            logger.error('Error parsing tool call', error);
+        }
+    }
+    // If no channels found, treat entire response as final
+    if (!result.analysis && !result.commentary && !result.final && result.toolCalls.length === 0) {
+        result.final = response.trim();
+    }
+    return result;
+}
+/**
+ * Extracts assistant message with malformed tool call detection
+ * This handles cases where gpt-oss-120b generates incorrect tool call formats
+ */
+export function extractAssistantMessage(response) {
+    // Remove channel markers and tool calls to get clean message
+    let cleaned = response
+        .replace(/<\|channel\|>\w+/g, '')
+        .replace(/<\|call\|>[\s\S]*?<\|return\|>/g, '')
+        .replace(/<\|start\|>/g, '')
+        .replace(/<\|end\|>/g, '')
+        .replace(/<\|message\|>/g, '')
+        .trim();
+    // Handle malformed patterns like "assistant<|channel|>analysis"
+    cleaned = cleaned.replace(/assistant<\|channel\|>\w+/g, '');
+    return cleaned;
+}
+/**
+ * Validates tool call against available tools
+ */
+export function validateToolCall(toolCall, availableTools) {
+    const tool = availableTools.find(t => t.name === toolCall.function.name);
+    if (!tool) {
+        logger.warn(`Tool not found: ${toolCall.function.name}`);
+        return false;
+    }
+    try {
+        const args = JSON.parse(toolCall.function.arguments);
+        const required = tool.parameters.required || [];
+        // Check required parameters
+        for (const param of required) {
+            if (!(param in args)) {
+                logger.warn(`Missing required parameter '${param}' for tool ${tool.name}`);
+                return false;
+            }
+        }
+        return true;
+    }
+    catch (error) {
+        logger.error(`Invalid arguments for tool ${tool.name}`, error);
+        return false;
+    }
+}
+/**
+ * Formats tool result for Harmony format
+ */
+export function formatToolResult(toolCallId, toolName, result) {
+    return {
+        role: 'tool',
+        name: toolName,
+        tool_call_id: toolCallId,
+        content: typeof result === 'string' ? result : JSON.stringify(result, null, 2),
+    };
+}
+//# sourceMappingURL=harmony.js.map

package/dist/models/harmony.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"harmony.js","sourceRoot":"","sources":["../../src/models/harmony.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AA4CnD;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,KAA8B;IAClE,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAElC,MAAM,eAAe,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;QACvC,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,IAAI,EAAE,CAAC;aAC5D,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,MAAM,CAAgB,EAAE,EAAE;YACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;YACrE,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC;YAClC,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1E,OAAO,KAAK,IAAI,GAAG,QAAQ,KAAK,IAAI,IAAI,WAAW,EAAE,CAAC;QACxD,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,OAAO;KACN,IAAI,CAAC,WAAW;;WAEV,IAAI,CAAC,IAAI;EAClB,MAAM;UACE,CAAC;IACT,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAEhB,OAAO;;;;;EAKP,eAAe;;;;;;;;;;;;;;;;;;;;;;;;;2DAyB0C,CAAC;AAC5D,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,wBAAwB,CACtC,QAA0B,EAC1B,KAA+B;IAE/B,MAAM,iBAAiB,GAAqB,EAAE,CAAC;IAE/C,2CAA2C;IAC3C,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1D,iDAAiD;YACjD,MAAM,WAAW,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;YACjD,MAAM,eAAe,GAAG,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3E,iBAAiB,CAAC,IAAI,CAAC;gBACrB,GAAG,GAAG;gBACN,OAAO,EAAE,GAAG,eAAe,OAAO,WAAW,EAAE;aAChD,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,iBAAiB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IAED,OAAO,iBAAiB,CAAC;AAC3B,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,UAAU,oBAAoB,CAAC,QAAgB;IACnD,MAAM,MAAM,GAA0B;QACpC,SAAS,EAAE,EAAE;QACb,WAAW,EAAE,QAAQ;KACtB,CAAC;IAEF,iBAAiB;IACjB,MAAM,YAAY,GAAG,gEAAgE,CAAC;IACtF,IAAI,KAAK,CAAC;IAEV,OAAO,CAAC,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACtD,MAAM,WAAW,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC7B,MAAM,cAAc,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAEvC,IAAI,WAAW,KAAK,UAAU,EAAE,CAAC;YAC/B,MAAM,CAAC,QAAQ,GAAG,cAAc,CAAC;QACnC,CAAC;aAAM,IAAI,WAAW,KAAK,YAAY,EAAE,CAAC;YACxC,MAAM,CAAC,UAAU,GAAG,cAAc,CAAC;QACrC,CAAC;aAAM,IAAI,WAAW,KAAK,OAAO,EAAE,CAAC;YACnC,MAAM,CAAC,KAAK,GAAG,cAAc,CAAC;QAChC,CAAC;IACH,CAAC;IAED,mBAAmB;IACnB,MAAM,aAAa,GAAG,mCAAmC,CAAC;IAC1D,IAAI,SAAS,CAAC;IACd,IAAI,MAAM,GAAG,CAAC,CAAC;IAEf,OAAO,CAAC,SAAS,GAAG,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QAC3D,MAAM,eAAe,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAE5C,IAAI,CAAC;YACH,yDAAyD;YACzD,MAAM,aAAa,GAAG,sBAAsB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAEnE,IAAI,aAAa,EAAE,CAAC;gBAClB,MAAM,YAAY,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;gBACtC,MAAM,UAAU,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;gBAEpC,gBAAgB;gBAChB,IAAI,UAAU,CAAC;gBACf,IAAI,CAAC;oBACH,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;gBACtC,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,CAAC,IAAI,CAAC,2CAA2C,YAAY,KAAK,UAAU,EAAE,CAAC,CAAC;oBACtF,gCAAgC;oBAChC,MAAM,SAAS,GAAG,UAAU;yBACzB,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAE,wBAAwB;yBAC5C,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAE,sBAAsB;oBAEvD,IAAI,CAAC;wBACH,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;wBACnC,MAAM,CAAC,KAAK,CAAC,4CAA4C,SAAS,EAAE,CAAC,CAAC;oBACxE,CAAC;oBAAC,OAAO,EAAE,EAAE,CAAC;wBACZ,MAAM,IAAI,aAAa,CACrB,wCAAwC,UAAU,EAAE,EACpD,YAAY,CACb,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC;oBACpB,EAAE,EAAE,QAAQ,MAAM,EAAE,EAAE;oBACtB,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE;wBACR,IAAI,EAAE,YAAY;wBAClB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;qBACtC;iBACF,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,IAAI,CAAC,qCAAqC,eAAe,EAAE,CAAC,CAAC;YACtE,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;IAED,uDAAuD;IACvD,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7F,MAAM,CAAC,KAAK,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;IACjC,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,uBAAuB,CAAC,QAAgB;IACtD,6DAA6D;IAC7D,IAAI,OAAO,GAAG,QAAQ;SACnB,OAAO,CAAC,mBAAmB,EAAE,EAAE,CAAC;SAChC,OAAO,CAAC,iCAAiC,EAAE,EAAE,CAAC;SAC9C,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC;SAC3B,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC;SACzB,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC;SAC7B,IAAI,EAAE,CAAC;IAEV,gEAAgE;IAChE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,4BAA4B,EAAE,EAAE,CAAC,CAAC;IAE5D,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAC9B,QAAyB,EACzB,cAAuC;IAEvC,MAAM,IAAI,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAEzE,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,MAAM,CAAC,IAAI,CAAC,mBAAmB,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;QACzD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,IAAI,EAAE,CAAC;QAEhD,4BAA4B;QAC5B,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;YAC7B,IAAI,CAAC,CAAC,KAAK,IAAI,IAAI,CAAC,EAAE,CAAC;gBACrB,MAAM,CAAC,IAAI,CAAC,+BAA+B,KAAK,cAAc,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC3E,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,8BAA8B,IAAI,CAAC,IAAI,EAAE,EAAE,KAAK,CAAC,CAAC;QAC/D,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAC9B,UAAkB,EAClB,QAAgB,EAChB,MAAW;IAEX,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,QAAQ;QACd,YAAY,EAAE,UAAU;QACxB,OAAO,EAAE,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;KAC/E,CAAC;AACJ,CAAC"}

package/dist/models/krutrim.d.ts

@@ -0,0 +1,30 @@
+/**
+ * Krutrim API Client
+ *
+ * Handles communication with Krutrim Cloud API for both gpt-oss-120b and Llama-4-Maverick-17B models.
+ * Implements Harmony format for gpt-oss-120b and standard format for multimodal model.
+ */
+import { IModel, ChatCompletionOptions, ModelResponse } from './base.js';
+export interface KrutrimConfig {
+    endpoint: string;
+    apiKey: string;
+    model: string;
+    type: 'reasoning' | 'multimodal';
+}
+export declare class KrutrimModel implements IModel {
+    private config;
+    constructor(config: KrutrimConfig);
+    supportsVision(): boolean;
+    supportsToolCalling(): boolean;
+    chat(options: ChatCompletionOptions): Promise<ModelResponse>;
+    private attemptChat;
+    /**
+     * Helper method for vision tasks using multimodal model
+     */
+    describeImage(imageUrl: string, prompt?: string): Promise<string>;
+}
+/**
+ * Factory function to create Krutrim model instances
+ */
+export declare function createKrutrimModel(config: KrutrimConfig): KrutrimModel;
+//# sourceMappingURL=krutrim.d.ts.map

package/dist/models/krutrim.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"krutrim.d.ts","sourceRoot":"","sources":["../../src/models/krutrim.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,MAAM,EAAE,qBAAqB,EAAE,aAAa,EAAW,MAAM,WAAW,CAAC;AAUlF,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,WAAW,GAAG,YAAY,CAAC;CAClC;AAED,qBAAa,YAAa,YAAW,MAAM;IACzC,OAAO,CAAC,MAAM,CAAgB;gBAElB,MAAM,EAAE,aAAa;IAIjC,cAAc,IAAI,OAAO;IAIzB,mBAAmB,IAAI,OAAO;IAKxB,IAAI,CAAC,OAAO,EAAE,qBAAqB,GAAG,OAAO,CAAC,aAAa,CAAC;YAmCpD,WAAW;IAuHzB;;OAEG;IACG,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CA+BxE;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,aAAa,GAAG,YAAY,CAEtE"}