@defai.digital/ax-cli 0.0.34

package/dist/agent/index.d.ts

@@ -0,0 +1,14 @@
+import { ToolResult, AgentState } from '../types/index.js';
+export declare class Agent {
+    private textEditor;
+    private bash;
+    private state;
+    constructor();
+    processCommand(input: string): Promise<ToolResult>;
+    private parseViewCommand;
+    private parseStrReplaceCommand;
+    private parseCreateCommand;
+    private parseInsertCommand;
+    private getHelp;
+    getCurrentState(): AgentState;
+}

package/dist/agent/index.js

@@ -0,0 +1,136 @@
+import { TextEditorTool, BashTool } from '../tools/index.js';
+export class Agent {
+    textEditor;
+    bash;
+    state;
+    constructor() {
+        this.textEditor = new TextEditorTool();
+        this.bash = new BashTool();
+        this.state = {
+            currentDirectory: process.cwd(),
+            editHistory: [],
+            tools: []
+        };
+    }
+    async processCommand(input) {
+        const trimmedInput = input.trim();
+        if (trimmedInput.startsWith('view ')) {
+            const args = this.parseViewCommand(trimmedInput);
+            return this.textEditor.view(args.path, args.range);
+        }
+        if (trimmedInput.startsWith('str_replace ')) {
+            const args = this.parseStrReplaceCommand(trimmedInput);
+            if (!args) {
+                return { success: false, error: 'Invalid str_replace command format' };
+            }
+            return this.textEditor.strReplace(args.path, args.oldStr, args.newStr);
+        }
+        if (trimmedInput.startsWith('create ')) {
+            const args = this.parseCreateCommand(trimmedInput);
+            if (!args) {
+                return { success: false, error: 'Invalid create command format' };
+            }
+            return this.textEditor.create(args.path, args.content);
+        }
+        if (trimmedInput.startsWith('insert ')) {
+            const args = this.parseInsertCommand(trimmedInput);
+            if (!args) {
+                return { success: false, error: 'Invalid insert command format' };
+            }
+            return this.textEditor.insert(args.path, args.line, args.content);
+        }
+        if (trimmedInput === 'undo_edit') {
+            return this.textEditor.undoEdit();
+        }
+        if (trimmedInput.startsWith('bash ') || trimmedInput.startsWith('$ ')) {
+            const command = trimmedInput.startsWith('bash ')
+                ? trimmedInput.substring(5)
+                : trimmedInput.substring(2);
+            return this.bash.execute(command);
+        }
+        if (trimmedInput === 'pwd') {
+            return {
+                success: true,
+                output: this.bash.getCurrentDirectory()
+            };
+        }
+        if (trimmedInput === 'history') {
+            const history = this.textEditor.getEditHistory();
+            return {
+                success: true,
+                output: history.length > 0
+                    ? JSON.stringify(history, null, 2)
+                    : 'No edit history'
+            };
+        }
+        if (trimmedInput === 'help') {
+            return this.getHelp();
+        }
+        return this.bash.execute(trimmedInput);
+    }
+    parseViewCommand(input) {
+        const parts = input.split(' ');
+        const path = parts[1];
+        if (parts.length > 2) {
+            const rangePart = parts[2];
+            if (rangePart.includes('-')) {
+                const [start, end] = rangePart.split('-').map(Number);
+                return { path, range: [start, end] };
+            }
+        }
+        return { path };
+    }
+    parseStrReplaceCommand(input) {
+        const match = input.match(/str_replace\s+(\S+)\s+"([^"]+)"\s+"([^"]*)"/);
+        if (!match)
+            return null;
+        return {
+            path: match[1],
+            oldStr: match[2],
+            newStr: match[3]
+        };
+    }
+    parseCreateCommand(input) {
+        const match = input.match(/create\s+(\S+)\s+"([^"]*)"/);
+        if (!match)
+            return null;
+        return {
+            path: match[1],
+            content: match[2]
+        };
+    }
+    parseInsertCommand(input) {
+        const match = input.match(/insert\s+(\S+)\s+(\d+)\s+"([^"]*)"/);
+        if (!match)
+            return null;
+        return {
+            path: match[1],
+            line: parseInt(match[2]),
+            content: match[3]
+        };
+    }
+    getHelp() {
+        return {
+            success: true,
+            output: `Available commands:
+  view <path> [start-end]     - View file contents or directory
+  str_replace <path> "old" "new" - Replace text in file
+  create <path> "content"     - Create new file with content
+  insert <path> <line> "text" - Insert text at specific line
+  undo_edit                   - Undo last edit operation
+  bash <command>              - Execute bash command
+  $ <command>                 - Execute bash command (shorthand)
+  pwd                         - Show current directory
+  history                     - Show edit history
+  help                        - Show this help message`
+        };
+    }
+    getCurrentState() {
+        return {
+            ...this.state,
+            currentDirectory: this.bash.getCurrentDirectory(),
+            editHistory: this.textEditor.getEditHistory()
+        };
+    }
+}
+//# sourceMappingURL=index.js.map

package/dist/agent/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/agent/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAG7D,MAAM,OAAO,KAAK;IACR,UAAU,CAAiB;IAC3B,IAAI,CAAW;IACf,KAAK,CAAa;IAE1B;QACE,IAAI,CAAC,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;QACvC,IAAI,CAAC,IAAI,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,IAAI,CAAC,KAAK,GAAG;YACX,gBAAgB,EAAE,OAAO,CAAC,GAAG,EAAE;YAC/B,WAAW,EAAE,EAAE;YACf,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAa;QAChC,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QAElC,IAAI,YAAY,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YACrC,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;YACjD,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,YAAY,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;YAC5C,MAAM,IAAI,GAAG,IAAI,CAAC,sBAAsB,CAAC,YAAY,CAAC,CAAC;YACvD,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,oCAAoC,EAAE,CAAC;YACzE,CAAC;YACD,OAAO,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QACzE,CAAC;QAED,IAAI,YAAY,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YACvC,MAAM,IAAI,GAAG,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;YACnD,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,+BAA+B,EAAE,CAAC;YACpE,CAAC;YACD,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACzD,CAAC;QAED,IAAI,YAAY,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YACvC,MAAM,IAAI,GAAG,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;YACnD,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,+BAA+B,EAAE,CAAC;YACpE,CAAC;YACD,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACpE,CAAC;QAED,IAAI,YAAY,KAAK,WAAW,EAAE,CAAC;YACjC,OAAO,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;QACpC,CAAC;QAED,IAAI,YAAY,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YACtE,MAAM,OAAO,GAAG,YAAY,CAAC,UAAU,CAAC,OAAO,CAAC;gBAC9C,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC;gBAC3B,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YAC9B,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACpC,CAAC;QAED,IAAI,YAAY,KAAK,KAAK,EAAE,CAAC;YAC3B,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE;aACxC,CAAC;QACJ,CAAC;QAED,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;YAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC;YACjD,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC;oBACxB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;oBAClC,CAAC,CAAC,iBAAiB;aACtB,CAAC;QACJ,CAAC;QAED,IAAI,YAAY,KAAK,MAAM,EAAE,CAAC;YAC5B,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;QACxB,CAAC;QAED,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IACzC,CAAC;IAEO,gBAAgB,CAAC,KAAa;QACpC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC/B,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAEtB,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YAC3B,IAAI,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC5B,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBACtD,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC;YACvC,CAAC;QACH,CAAC;QAED,OAAO,EAAE,IAAI,EAAE,CAAC;IAClB,CAAC;IAEO,sBAAsB,CAAC,KAAa;QAC1C,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACzE,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAExB,OAAO;YACL,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;YACd,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;YAChB,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;SACjB,CAAC;IACJ,CAAC;IAEO,kBAAkB,CAAC,KAAa;QACtC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;QACxD,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAExB,OAAO;YACL,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;YACd,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;SAClB,CAAC;IACJ,CAAC;IAEO,kBAAkB,CAAC,KAAa;QACtC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC;QAChE,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAExB,OAAO;YACL,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;YACd,IAAI,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACxB,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;SAClB,CAAC;IACJ,CAAC;IAEO,OAAO;QACb,OAAO;YACL,OAAO,EAAE,IAAI;YACb,MAAM,EAAE;;;;;;;;;;uDAUyC;SAClD,CAAC;IACJ,CAAC;IAED,eAAe;QACb,OAAO;YACL,GAAG,IAAI,CAAC,KAAK;YACb,gBAAgB,EAAE,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE;YACjD,WAAW,EAAE,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE;SAC9C,CAAC;IACJ,CAAC;CACF"}

package/dist/commands/mcp.d.ts

@@ -0,0 +1,2 @@
+import { Command } from 'commander';
+export declare function createMCPCommand(): Command;

package/dist/commands/mcp.js

@@ -0,0 +1,239 @@
+import { Command } from 'commander';
+import { addMCPServer, removeMCPServer, loadMCPConfig, PREDEFINED_SERVERS } from '../mcp/config.js';
+import { getMCPManager } from '../grok/tools.js';
+import chalk from 'chalk';
+export function createMCPCommand() {
+    const mcpCommand = new Command('mcp');
+    mcpCommand.description('Manage MCP (Model Context Protocol) servers');
+    // Add server command
+    mcpCommand
+        .command('add <name>')
+        .description('Add an MCP server')
+        .option('-t, --transport <type>', 'Transport type (stdio, http, sse, streamable_http)', 'stdio')
+        .option('-c, --command <command>', 'Command to run the server (for stdio transport)')
+        .option('-a, --args [args...]', 'Arguments for the server command (for stdio transport)', [])
+        .option('-u, --url <url>', 'URL for HTTP/SSE transport')
+        .option('-h, --headers [headers...]', 'HTTP headers (key=value format)', [])
+        .option('-e, --env [env...]', 'Environment variables (key=value format)', [])
+        .action(async (name, options) => {
+        try {
+            // Check if it's a predefined server
+            if (PREDEFINED_SERVERS[name]) {
+                const config = PREDEFINED_SERVERS[name];
+                addMCPServer(config);
+                console.log(chalk.green(`✓ Added predefined MCP server: ${name}`));
+                // Try to connect immediately
+                const manager = getMCPManager();
+                await manager.addServer(config);
+                console.log(chalk.green(`✓ Connected to MCP server: ${name}`));
+                const tools = manager.getTools().filter(t => t.serverName === name);
+                console.log(chalk.blue(`  Available tools: ${tools.length}`));
+                return;
+            }
+            // Custom server
+            const transportType = options.transport.toLowerCase();
+            if (transportType === 'stdio') {
+                if (!options.command) {
+                    console.error(chalk.red('Error: --command is required for stdio transport'));
+                    process.exit(1);
+                }
+            }
+            else if (transportType === 'http' || transportType === 'sse' || transportType === 'streamable_http') {
+                if (!options.url) {
+                    console.error(chalk.red(`Error: --url is required for ${transportType} transport`));
+                    process.exit(1);
+                }
+            }
+            else {
+                console.error(chalk.red('Error: Transport type must be stdio, http, sse, or streamable_http'));
+                process.exit(1);
+            }
+            // Parse environment variables
+            const env = {};
+            for (const envVar of options.env || []) {
+                const [key, value] = envVar.split('=', 2);
+                if (key && value) {
+                    env[key] = value;
+                }
+            }
+            // Parse headers
+            const headers = {};
+            for (const header of options.headers || []) {
+                const [key, value] = header.split('=', 2);
+                if (key && value) {
+                    headers[key] = value;
+                }
+            }
+            const config = {
+                name,
+                transport: {
+                    type: transportType,
+                    command: options.command,
+                    args: options.args || [],
+                    url: options.url,
+                    env,
+                    headers: Object.keys(headers).length > 0 ? headers : undefined
+                }
+            };
+            addMCPServer(config);
+            console.log(chalk.green(`✓ Added MCP server: ${name}`));
+            // Try to connect immediately
+            const manager = getMCPManager();
+            await manager.addServer(config);
+            console.log(chalk.green(`✓ Connected to MCP server: ${name}`));
+            const tools = manager.getTools().filter(t => t.serverName === name);
+            console.log(chalk.blue(`  Available tools: ${tools.length}`));
+        }
+        catch (error) {
+            console.error(chalk.red(`Error adding MCP server: ${error.message}`));
+            process.exit(1);
+        }
+    });
+    // Add server from JSON command
+    mcpCommand
+        .command('add-json <name> <json>')
+        .description('Add an MCP server from JSON configuration')
+        .action(async (name, jsonConfig) => {
+        try {
+            let config;
+            try {
+                config = JSON.parse(jsonConfig);
+            }
+            catch (error) {
+                console.error(chalk.red('Error: Invalid JSON configuration'));
+                process.exit(1);
+            }
+            const serverConfig = {
+                name,
+                transport: {
+                    type: 'stdio', // default
+                    command: config.command,
+                    args: config.args || [],
+                    env: config.env || {},
+                    url: config.url,
+                    headers: config.headers
+                }
+            };
+            // Override transport type if specified
+            if (config.transport) {
+                if (typeof config.transport === 'string') {
+                    serverConfig.transport.type = config.transport;
+                }
+                else if (typeof config.transport === 'object') {
+                    serverConfig.transport = { ...serverConfig.transport, ...config.transport };
+                }
+            }
+            addMCPServer(serverConfig);
+            console.log(chalk.green(`✓ Added MCP server: ${name}`));
+            // Try to connect immediately
+            const manager = getMCPManager();
+            await manager.addServer(serverConfig);
+            console.log(chalk.green(`✓ Connected to MCP server: ${name}`));
+            const tools = manager.getTools().filter(t => t.serverName === name);
+            console.log(chalk.blue(`  Available tools: ${tools.length}`));
+        }
+        catch (error) {
+            console.error(chalk.red(`Error adding MCP server: ${error.message}`));
+            process.exit(1);
+        }
+    });
+    // Remove server command
+    mcpCommand
+        .command('remove <name>')
+        .description('Remove an MCP server')
+        .action(async (name) => {
+        try {
+            const manager = getMCPManager();
+            await manager.removeServer(name);
+            removeMCPServer(name);
+            console.log(chalk.green(`✓ Removed MCP server: ${name}`));
+        }
+        catch (error) {
+            console.error(chalk.red(`Error removing MCP server: ${error.message}`));
+            process.exit(1);
+        }
+    });
+    // List servers command
+    mcpCommand
+        .command('list')
+        .description('List configured MCP servers')
+        .action(() => {
+        const config = loadMCPConfig();
+        const manager = getMCPManager();
+        if (config.servers.length === 0) {
+            console.log(chalk.yellow('No MCP servers configured'));
+            return;
+        }
+        console.log(chalk.bold('Configured MCP servers:'));
+        console.log();
+        for (const server of config.servers) {
+            const isConnected = manager.getServers().includes(server.name);
+            const status = isConnected
+                ? chalk.green('✓ Connected')
+                : chalk.red('✗ Disconnected');
+            console.log(`${chalk.bold(server.name)}: ${status}`);
+            // Display transport information
+            if (server.transport) {
+                console.log(`  Transport: ${server.transport.type}`);
+                if (server.transport.type === 'stdio') {
+                    console.log(`  Command: ${server.transport.command} ${(server.transport.args || []).join(' ')}`);
+                }
+                else if (server.transport.type === 'http' || server.transport.type === 'sse') {
+                    console.log(`  URL: ${server.transport.url}`);
+                }
+            }
+            else if (server.command) {
+                // Legacy format
+                console.log(`  Command: ${server.command} ${(server.args || []).join(' ')}`);
+            }
+            if (isConnected) {
+                const transportType = manager.getTransportType(server.name);
+                if (transportType) {
+                    console.log(`  Active Transport: ${transportType}`);
+                }
+                const tools = manager.getTools().filter(t => t.serverName === server.name);
+                console.log(`  Tools: ${tools.length}`);
+                if (tools.length > 0) {
+                    tools.forEach(tool => {
+                        const displayName = tool.name.replace(`mcp__${server.name}__`, '');
+                        console.log(`    - ${displayName}: ${tool.description}`);
+                    });
+                }
+            }
+            console.log();
+        }
+    });
+    // Test server command
+    mcpCommand
+        .command('test <name>')
+        .description('Test connection to an MCP server')
+        .action(async (name) => {
+        try {
+            const config = loadMCPConfig();
+            const serverConfig = config.servers.find(s => s.name === name);
+            if (!serverConfig) {
+                console.error(chalk.red(`Server ${name} not found`));
+                process.exit(1);
+            }
+            console.log(chalk.blue(`Testing connection to ${name}...`));
+            const manager = getMCPManager();
+            await manager.addServer(serverConfig);
+            const tools = manager.getTools().filter(t => t.serverName === name);
+            console.log(chalk.green(`✓ Successfully connected to ${name}`));
+            console.log(chalk.blue(`  Available tools: ${tools.length}`));
+            if (tools.length > 0) {
+                console.log('  Tools:');
+                tools.forEach(tool => {
+                    const displayName = tool.name.replace(`mcp__${name}__`, '');
+                    console.log(`    - ${displayName}: ${tool.description}`);
+                });
+            }
+        }
+        catch (error) {
+            console.error(chalk.red(`✗ Failed to connect to ${name}: ${error.message}`));
+            process.exit(1);
+        }
+    });
+    return mcpCommand;
+}
+//# sourceMappingURL=mcp.js.map

package/dist/commands/mcp.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp.js","sourceRoot":"","sources":["../../src/commands/mcp.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,YAAY,EAAE,eAAe,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACpG,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAEjD,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,MAAM,UAAU,gBAAgB;IAC9B,MAAM,UAAU,GAAG,IAAI,OAAO,CAAC,KAAK,CAAC,CAAC;IACtC,UAAU,CAAC,WAAW,CAAC,6CAA6C,CAAC,CAAC;IAEtE,qBAAqB;IACrB,UAAU;SACP,OAAO,CAAC,YAAY,CAAC;SACrB,WAAW,CAAC,mBAAmB,CAAC;SAChC,MAAM,CAAC,wBAAwB,EAAE,oDAAoD,EAAE,OAAO,CAAC;SAC/F,MAAM,CAAC,yBAAyB,EAAE,iDAAiD,CAAC;SACpF,MAAM,CAAC,sBAAsB,EAAE,wDAAwD,EAAE,EAAE,CAAC;SAC5F,MAAM,CAAC,iBAAiB,EAAE,4BAA4B,CAAC;SACvD,MAAM,CAAC,4BAA4B,EAAE,iCAAiC,EAAE,EAAE,CAAC;SAC3E,MAAM,CAAC,oBAAoB,EAAE,0CAA0C,EAAE,EAAE,CAAC;SAC5E,MAAM,CAAC,KAAK,EAAE,IAAY,EAAE,OAAO,EAAE,EAAE;QACtC,IAAI,CAAC;YACH,oCAAoC;YACpC,IAAI,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC7B,MAAM,MAAM,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC;gBACxC,YAAY,CAAC,MAAM,CAAC,CAAC;gBACrB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,kCAAkC,IAAI,EAAE,CAAC,CAAC,CAAC;gBAEnE,6BAA6B;gBAC7B,MAAM,OAAO,GAAG,aAAa,EAAE,CAAC;gBAChC,MAAM,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;gBAChC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,8BAA8B,IAAI,EAAE,CAAC,CAAC,CAAC;gBAE/D,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,IAAI,CAAC,CAAC;gBACpE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,sBAAsB,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;gBAE9D,OAAO;YACT,CAAC;YAED,gBAAgB;YAChB,MAAM,aAAa,GAAG,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC;YAEtD,IAAI,aAAa,KAAK,OAAO,EAAE,CAAC;gBAC9B,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;oBACrB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC,CAAC;oBAC7E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBAClB,CAAC;YACH,CAAC;iBAAM,IAAI,aAAa,KAAK,MAAM,IAAI,aAAa,KAAK,KAAK,IAAI,aAAa,KAAK,iBAAiB,EAAE,CAAC;gBACtG,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;oBACjB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,gCAAgC,aAAa,YAAY,CAAC,CAAC,CAAC;oBACpF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBAClB,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,oEAAoE,CAAC,CAAC,CAAC;gBAC/F,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAClB,CAAC;YAED,8BAA8B;YAC9B,MAAM,GAAG,GAA2B,EAAE,CAAC;YACvC,KAAK,MAAM,MAAM,IAAI,OAAO,CAAC,GAAG,IAAI,EAAE,EAAE,CAAC;gBACvC,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;gBAC1C,IAAI,GAAG,IAAI,KAAK,EAAE,CAAC;oBACjB,GAAG,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;gBACnB,CAAC;YACH,CAAC;YAED,gBAAgB;YAChB,MAAM,OAAO,GAA2B,EAAE,CAAC;YAC3C,KAAK,MAAM,MAAM,IAAI,OAAO,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;gBAC3C,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;gBAC1C,IAAI,GAAG,IAAI,KAAK,EAAE,CAAC;oBACjB,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;gBACvB,CAAC;YACH,CAAC;YAED,MAAM,MAAM,GAAG;gBACb,IAAI;gBACJ,SAAS,EAAE;oBACT,IAAI,EAAE,aAA6D;oBACnE,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE;oBACxB,GAAG,EAAE,OAAO,CAAC,GAAG;oBAChB,GAAG;oBACH,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;iBAC/D;aACF,CAAC;YAEF,YAAY,CAAC,MAAM,CAAC,CAAC;YACrB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,uBAAuB,IAAI,EAAE,CAAC,CAAC,CAAC;YAExD,6BAA6B;YAC7B,MAAM,OAAO,GAAG,aAAa,EAAE,CAAC;YAChC,MAAM,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAChC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,8BAA8B,IAAI,EAAE,CAAC,CAAC,CAAC;YAE/D,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,IAAI,CAAC,CAAC;YACpE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,sBAAsB,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAEhE,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,4BAA4B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACtE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,+BAA+B;IAC/B,UAAU;SACP,OAAO,CAAC,wBAAwB,CAAC;SACjC,WAAW,CAAC,2CAA2C,CAAC;SACxD,MAAM,CAAC,KAAK,EAAE,IAAY,EAAE,UAAkB,EAAE,EAAE;QACjD,IAAI,CAAC;YACH,IAAI,MAAM,CAAC;YACX,IAAI,CAAC;gBACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAClC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC,CAAC;gBAC9D,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAClB,CAAC;YAED,MAAM,YAAY,GAAoB;gBACpC,IAAI;gBACJ,SAAS,EAAE;oBACT,IAAI,EAAE,OAAO,EAAE,UAAU;oBACzB,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,EAAE;oBACvB,GAAG,EAAE,MAAM,CAAC,GAAG,IAAI,EAAE;oBACrB,GAAG,EAAE,MAAM,CAAC,GAAG;oBACf,OAAO,EAAE,MAAM,CAAC,OAAO;iBACxB;aACF,CAAC;YAEF,uCAAuC;YACvC,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;gBACrB,IAAI,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;oBACzC,YAAY,CAAC,SAAS,CAAC,IAAI,GAAG,MAAM,CAAC,SAAqC,CAAC;gBAC7E,CAAC;qBAAM,IAAI,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;oBAChD,YAAY,CAAC,SAAS,GAAG,EAAE,GAAG,YAAY,CAAC,SAAS,EAAE,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;gBAC9E,CAAC;YACH,CAAC;YAED,YAAY,CAAC,YAAY,CAAC,CAAC;YAC3B,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,uBAAuB,IAAI,EAAE,CAAC,CAAC,CAAC;YAExD,6BAA6B;YAC7B,MAAM,OAAO,GAAG,aAAa,EAAE,CAAC;YAChC,MAAM,OAAO,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YACtC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,8BAA8B,IAAI,EAAE,CAAC,CAAC,CAAC;YAE/D,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,IAAI,CAAC,CAAC;YACpE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,sBAAsB,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAEhE,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,4BAA4B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACtE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,wBAAwB;IACxB,UAAU;SACP,OAAO,CAAC,eAAe,CAAC;SACxB,WAAW,CAAC,sBAAsB,CAAC;SACnC,MAAM,CAAC,KAAK,EAAE,IAAY,EAAE,EAAE;QAC7B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,aAAa,EAAE,CAAC;YAChC,MAAM,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YACjC,eAAe,CAAC,IAAI,CAAC,CAAC;YACtB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC,CAAC;QAC5D,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,8BAA8B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACxE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,uBAAuB;IACvB,UAAU;SACP,OAAO,CAAC,MAAM,CAAC;SACf,WAAW,CAAC,6BAA6B,CAAC;SAC1C,MAAM,CAAC,GAAG,EAAE;QACX,MAAM,MAAM,GAAG,aAAa,EAAE,CAAC;QAC/B,MAAM,OAAO,GAAG,aAAa,EAAE,CAAC;QAEhC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,2BAA2B,CAAC,CAAC,CAAC;YACvD,OAAO;QACT,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC;QACnD,OAAO,CAAC,GAAG,EAAE,CAAC;QAEd,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACpC,MAAM,WAAW,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC/D,MAAM,MAAM,GAAG,WAAW;gBACxB,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,CAAC;gBAC5B,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;YAEhC,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,MAAM,EAAE,CAAC,CAAC;YAErD,gCAAgC;YAChC,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;gBACrB,OAAO,CAAC,GAAG,CAAC,gBAAgB,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;gBACrD,IAAI,MAAM,CAAC,SAAS,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;oBACtC,OAAO,CAAC,GAAG,CAAC,cAAc,MAAM,CAAC,SAAS,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACnG,CAAC;qBAAM,IAAI,MAAM,CAAC,SAAS,CAAC,IAAI,KAAK,MAAM,IAAI,MAAM,CAAC,SAAS,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;oBAC/E,OAAO,CAAC,GAAG,CAAC,UAAU,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC,CAAC;gBAChD,CAAC;YACH,CAAC;iBAAM,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBAC1B,gBAAgB;gBAChB,OAAO,CAAC,GAAG,CAAC,cAAc,MAAM,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC/E,CAAC;YAED,IAAI,WAAW,EAAE,CAAC;gBAChB,MAAM,aAAa,GAAG,OAAO,CAAC,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC5D,IAAI,aAAa,EAAE,CAAC;oBAClB,OAAO,CAAC,GAAG,CAAC,uBAAuB,aAAa,EAAE,CAAC,CAAC;gBACtD,CAAC;gBAED,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC3E,OAAO,CAAC,GAAG,CAAC,YAAY,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;gBACxC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACrB,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;wBACnB,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,MAAM,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;wBACnE,OAAO,CAAC,GAAG,CAAC,SAAS,WAAW,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;oBAC3D,CAAC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,OAAO,CAAC,GAAG,EAAE,CAAC;QAChB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,sBAAsB;IACtB,UAAU;SACP,OAAO,CAAC,aAAa,CAAC;SACtB,WAAW,CAAC,kCAAkC,CAAC;SAC/C,MAAM,CAAC,KAAK,EAAE,IAAY,EAAE,EAAE;QAC7B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,aAAa,EAAE,CAAC;YAC/B,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;YAE/D,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,IAAI,YAAY,CAAC,CAAC,CAAC;gBACrD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAClB,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,yBAAyB,IAAI,KAAK,CAAC,CAAC,CAAC;YAE5D,MAAM,OAAO,GAAG,aAAa,EAAE,CAAC;YAChC,MAAM,OAAO,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YAEtC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,IAAI,CAAC,CAAC;YACpE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,+BAA+B,IAAI,EAAE,CAAC,CAAC,CAAC;YAChE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,sBAAsB,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAE9D,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACrB,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBACxB,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;oBACnB,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;oBAC5D,OAAO,CAAC,GAAG,CAAC,SAAS,WAAW,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;gBAC3D,CAAC,CAAC,CAAC;YACL,CAAC;QAEH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,0BAA0B,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAC7E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,OAAO,UAAU,CAAC;AACpB,CAAC"}

package/dist/grok/client.d.ts

@@ -0,0 +1,49 @@
+import type { ChatCompletionMessageParam } from "openai/resources/chat";
+export type GrokMessage = ChatCompletionMessageParam;
+export interface GrokTool {
+    type: "function";
+    function: {
+        name: string;
+        description: string;
+        parameters: {
+            type: "object";
+            properties: Record<string, any>;
+            required: string[];
+        };
+    };
+}
+export interface GrokToolCall {
+    id: string;
+    type: "function";
+    function: {
+        name: string;
+        arguments: string;
+    };
+}
+export interface SearchParameters {
+    mode?: "auto" | "on" | "off";
+}
+export interface SearchOptions {
+    search_parameters?: SearchParameters;
+}
+export interface GrokResponse {
+    choices: Array<{
+        message: {
+            role: string;
+            content: string | null;
+            tool_calls?: GrokToolCall[];
+        };
+        finish_reason: string;
+    }>;
+}
+export declare class GrokClient {
+    private client;
+    private currentModel;
+    private defaultMaxTokens;
+    constructor(apiKey: string, model?: string, baseURL?: string);
+    setModel(model: string): void;
+    getCurrentModel(): string;
+    chat(messages: GrokMessage[], tools?: GrokTool[], model?: string, searchOptions?: SearchOptions): Promise<GrokResponse>;
+    chatStream(messages: GrokMessage[], tools?: GrokTool[], model?: string, searchOptions?: SearchOptions): AsyncGenerator<any, void, unknown>;
+    search(query: string, searchParameters?: SearchParameters): Promise<GrokResponse>;
+}

package/dist/grok/client.js

@@ -0,0 +1,80 @@
+import OpenAI from "openai";
+export class GrokClient {
+    client;
+    currentModel = "grok-code-fast-1";
+    defaultMaxTokens;
+    constructor(apiKey, model, baseURL) {
+        this.client = new OpenAI({
+            apiKey,
+            baseURL: baseURL || process.env.GROK_BASE_URL || "https://api.x.ai/v1",
+            timeout: 360000,
+        });
+        const envMax = Number(process.env.GROK_MAX_TOKENS);
+        this.defaultMaxTokens = Number.isFinite(envMax) && envMax > 0 ? envMax : 1536;
+        if (model) {
+            this.currentModel = model;
+        }
+    }
+    setModel(model) {
+        this.currentModel = model;
+    }
+    getCurrentModel() {
+        return this.currentModel;
+    }
+    async chat(messages, tools, model, searchOptions) {
+        try {
+            const requestPayload = {
+                model: model || this.currentModel,
+                messages,
+                tools: tools || [],
+                tool_choice: tools && tools.length > 0 ? "auto" : undefined,
+                temperature: 0.7,
+                max_tokens: this.defaultMaxTokens,
+            };
+            // Add search parameters if specified
+            if (searchOptions?.search_parameters) {
+                requestPayload.search_parameters = searchOptions.search_parameters;
+            }
+            const response = await this.client.chat.completions.create(requestPayload);
+            return response;
+        }
+        catch (error) {
+            throw new Error(`Grok API error: ${error.message}`);
+        }
+    }
+    async *chatStream(messages, tools, model, searchOptions) {
+        try {
+            const requestPayload = {
+                model: model || this.currentModel,
+                messages,
+                tools: tools || [],
+                tool_choice: tools && tools.length > 0 ? "auto" : undefined,
+                temperature: 0.7,
+                max_tokens: this.defaultMaxTokens,
+                stream: true,
+            };
+            // Add search parameters if specified
+            if (searchOptions?.search_parameters) {
+                requestPayload.search_parameters = searchOptions.search_parameters;
+            }
+            const stream = (await this.client.chat.completions.create(requestPayload));
+            for await (const chunk of stream) {
+                yield chunk;
+            }
+        }
+        catch (error) {
+            throw new Error(`Grok API error: ${error.message}`);
+        }
+    }
+    async search(query, searchParameters) {
+        const searchMessage = {
+            role: "user",
+            content: query,
+        };
+        const searchOptions = {
+            search_parameters: searchParameters || { mode: "on" },
+        };
+        return this.chat([searchMessage], [], undefined, searchOptions);
+    }
+}
+//# sourceMappingURL=client.js.map

package/dist/grok/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/grok/client.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AA+C5B,MAAM,OAAO,UAAU;IACb,MAAM,CAAS;IACf,YAAY,GAAW,kBAAkB,CAAC;IAC1C,gBAAgB,CAAS;IAEjC,YAAY,MAAc,EAAE,KAAc,EAAE,OAAgB;QAC1D,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,MAAM;YACN,OAAO,EAAE,OAAO,IAAI,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,qBAAqB;YACtE,OAAO,EAAE,MAAM;SAChB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QACnD,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC;QAC9E,IAAI,KAAK,EAAE,CAAC;YACV,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,KAAa;QACpB,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;IAC5B,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,IAAI,CACR,QAAuB,EACvB,KAAkB,EAClB,KAAc,EACd,aAA6B;QAE7B,IAAI,CAAC;YACH,MAAM,cAAc,GAAQ;gBAC1B,KAAK,EAAE,KAAK,IAAI,IAAI,CAAC,YAAY;gBACjC,QAAQ;gBACR,KAAK,EAAE,KAAK,IAAI,EAAE;gBAClB,WAAW,EAAE,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;gBAC3D,WAAW,EAAE,GAAG;gBAChB,UAAU,EAAE,IAAI,CAAC,gBAAgB;aAClC,CAAC;YAEF,qCAAqC;YACrC,IAAI,aAAa,EAAE,iBAAiB,EAAE,CAAC;gBACrC,cAAc,CAAC,iBAAiB,GAAG,aAAa,CAAC,iBAAiB,CAAC;YACrE,CAAC;YAED,MAAM,QAAQ,GACZ,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;YAE5D,OAAO,QAAwB,CAAC;QAClC,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,mBAAmB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,UAAU,CACf,QAAuB,EACvB,KAAkB,EAClB,KAAc,EACd,aAA6B;QAE7B,IAAI,CAAC;YACH,MAAM,cAAc,GAAQ;gBAC1B,KAAK,EAAE,KAAK,IAAI,IAAI,CAAC,YAAY;gBACjC,QAAQ;gBACR,KAAK,EAAE,KAAK,IAAI,EAAE;gBAClB,WAAW,EAAE,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;gBAC3D,WAAW,EAAE,GAAG;gBAChB,UAAU,EAAE,IAAI,CAAC,gBAAgB;gBACjC,MAAM,EAAE,IAAI;aACb,CAAC;YAEF,qCAAqC;YACrC,IAAI,aAAa,EAAE,iBAAiB,EAAE,CAAC;gBACrC,cAAc,CAAC,iBAAiB,GAAG,aAAa,CAAC,iBAAiB,CAAC;YACrE,CAAC;YAED,MAAM,MAAM,GAAG,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CACvD,cAAc,CACf,CAAQ,CAAC;YAEV,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,mBAAmB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CACV,KAAa,EACb,gBAAmC;QAEnC,MAAM,aAAa,GAAgB;YACjC,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,KAAK;SACf,CAAC;QAEF,MAAM,aAAa,GAAkB;YACnC,iBAAiB,EAAE,gBAAgB,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE;SACtD,CAAC;QAEF,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IAClE,CAAC;CACF"}

package/dist/grok/tools.d.ts

@@ -0,0 +1,8 @@
+import { GrokTool } from "./client.js";
+import { MCPManager, MCPTool } from "../mcp/client.js";
+export declare const GROK_TOOLS: GrokTool[];
+export declare function getMCPManager(): MCPManager;
+export declare function initializeMCPServers(): Promise<void>;
+export declare function convertMCPToolToGrokTool(mcpTool: MCPTool): GrokTool;
+export declare function addMCPToolsToGrokTools(baseTools: GrokTool[]): GrokTool[];
+export declare function getAllGrokTools(): Promise<GrokTool[]>;