@exreve/exk 1.0.39 → 1.0.40

package/dist/agentSession.js

@@ -197,6 +197,18 @@ function loadAiConfig() {
 export function getOpenrouterApiKey() {
     return loadAiConfig().openrouterApiKey || process.env.OPENROUTER_API_KEY || '';
 }
+/** Get the backend API URL from config file */
+export function getApiUrl() {
+    const configPath = path.join(os.homedir(), '.talk-to-code', 'config.json');
+    try {
+        const data = readFileSync(configPath, 'utf-8');
+        const config = JSON.parse(data);
+        return config.apiUrl || 'https://api.talk-to-code.com';
+    }
+    catch {
+        return process.env.API_URL || 'https://api.talk-to-code.com';
+    }
+}
 /** Create (or reuse) an empty directory to use as CLAUDE_CONFIG_DIR.
  *  Setting this prevents the spawned Claude CLI from reading ~/.claude/settings.json,
  *  which may contain env.ANTHROPIC_BASE_URL pointing to z.ai and would override our
@@ -1244,6 +1256,7 @@ export class AgentSessionManager {
         console.log(`[buildMcpServer] Session ${sessionId}: attachmentDir=${attachmentDir || 'none'}`);
         return createModuleMcpServer({
             attachmentDir,
+            sessionId,
         });
     }
 }

package/dist/moduleMcpServer.js

@@ -9,7 +9,7 @@ import { z } from 'zod';
 import * as fs from 'fs';
 import * as path from 'path';
 import * as os from 'os';
-import { getOpenrouterApiKey } from './agentSession.js';
+import { getOpenrouterApiKey, getApiUrl } from './agentSession.js';
 const MAX_FILE_SIZE = 10 * 1024 * 1024; // 10 MB
 /** Comprehensive MIME type map for file extension detection */
 const MIME_MAP = {
@@ -175,6 +175,92 @@ function createSendFileTool(attachmentDir) {
         }
     });
 }
+/**
+ * Create the browser_query tool for web automation via the backend.
+ * The agent can fire multiple queries concurrently — screenshots stream
+ * to the frontend independently while the tool blocks until completion.
+ */
+function createBrowserQueryTool(config) {
+    return tool('browser_query', 'Launch a headless browser to automate web tasks such as searching, reading pages, filling forms, extracting data, etc. ' +
+        'Returns the answer, optionally structured data, and step count. ' +
+        'IMPORTANT: This tool is slow (30-120 seconds per query). You CAN and SHOULD call browser_query multiple times concurrently ' +
+        '— the browser handles each in a separate session. While waiting for results, continue with other work (file edits, analysis, etc.). ' +
+        'Do NOT wait for one browser query to finish before starting another if you need multiple lookups.', {
+        query: z.string().describe('Natural language task for the browser agent (e.g. "Go to google.com and search for the price of Bitcoin")'),
+        schema: z.string().optional().describe('JSON schema for structured output, as a JSON string (e.g. \'{"type":"object","properties":{"price":{"type":"number"}}}\')'),
+        maxSteps: z.number().optional().describe('Max automation steps, default 20. Use lower values for simple tasks.'),
+        country: z.string().optional().describe('2-letter country code for proxy and locale (e.g. "US", "GB", "DE"). Uses direct connection if omitted.'),
+    }, async (args) => {
+        const apiUrl = getApiUrl();
+        // Read device ID for CLI auth
+        let deviceId = '';
+        try {
+            const deviceIdPath = path.join(os.homedir(), '.talk-to-code', 'device-id.json');
+            const data = fs.readFileSync(deviceIdPath, 'utf-8');
+            deviceId = JSON.parse(data).deviceId || '';
+        }
+        catch {
+            // No device ID file — will still work if backend has relaxed auth
+        }
+        try {
+            const body = {
+                query: args.query,
+                maxSteps: args.maxSteps || 20,
+            };
+            if (args.schema) {
+                try {
+                    body.schema = JSON.parse(args.schema);
+                }
+                catch {
+                    body.schema = args.schema;
+                }
+            }
+            if (args.country)
+                body.country = args.country;
+            if (config.sessionId)
+                body.sessionId = config.sessionId;
+            if (config.promptId)
+                body.promptId = config.promptId;
+            const res = await fetch(`${apiUrl}/api/browser/query`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    ...(deviceId ? { 'X-Device-ID': deviceId } : {}),
+                },
+                body: JSON.stringify(body),
+                signal: AbortSignal.timeout(10 * 60 * 1000), // 10 min timeout
+            });
+            const raw = await res.text();
+            if (!res.ok) {
+                return {
+                    content: [{ type: 'text', text: `Error from browser agent (${res.status}): ${raw.slice(0, 500)}` }],
+                    isError: true,
+                };
+            }
+            const result = JSON.parse(raw);
+            // Format a nice summary for the agent
+            const summary = [
+                `Browser query completed in ${result.steps} steps.`,
+                result.answer ? `\n\n**Answer:** ${result.answer}` : '',
+                result.data ? `\n\n**Structured Data:**\n\`\`\`json\n${JSON.stringify(result.data, null, 2)}\n\`\`\`` : '',
+                result.logs?.length ? `\n\n**Log:**\n${result.logs.slice(-5).join('\n')}` : '',
+            ].join('');
+            return { content: [{ type: 'text', text: summary }] };
+        }
+        catch (error) {
+            if (error.name === 'TimeoutError') {
+                return {
+                    content: [{ type: 'text', text: 'Browser query timed out after 10 minutes. Try reducing maxSteps or simplifying the query.' }],
+                    isError: true,
+                };
+            }
+            return {
+                content: [{ type: 'text', text: `Error running browser query: ${error.message}` }],
+                isError: true,
+            };
+        }
+    });
+}
 /**
  * Create the MCP server with built-in tools
  */
@@ -184,6 +270,8 @@ export function createModuleMcpServer(config) {
     tools.push(createAnalyzeImageTool(config.attachmentDir));
     // Add send_file tool for displaying files to the user in chat
     tools.push(createSendFileTool(config.attachmentDir));
+    // Add browser_query tool for web automation
+    tools.push(createBrowserQueryTool(config));
     const server = createSdkMcpServer({
         name: 'claude-voice-modules',
         version: '1.0.0',

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@exreve/exk",
-  "version": "1.0.39",
+  "version": "1.0.40",
   "description": "exk - Control Claude CLI with voice and programmable interfaces",
   "type": "module",
   "bin": {