rivet-design 0.5.4 → 0.5.7

package/dist/services/AgentBridgeService.js

@@ -1,413 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.AgentBridgeService = void 0;
-const ws_1 = require("ws");
-const logger_1 = require("../utils/logger");
-const ComponentSearchService_1 = require("./ComponentSearchService");
-const FileModificationService_1 = require("./FileModificationService");
-const LocalGitService_1 = require("./LocalGitService");
-const ConfigManager_1 = require("./ConfigManager");
-const promises_1 = require("fs/promises");
-const glob_1 = require("glob");
-const log = (0, logger_1.createLogger)('AgentBridgeService');
-const HEARTBEAT_INTERVAL = 30000; // 30 seconds
-const RECONNECT_DELAY = 5000; // 5 seconds
-/**
- * Connects Rivet CLI to hosted Agent SDK server
- * Executes tools locally and sends results back via WebSocket
- */
-class AgentBridgeService {
-    constructor(serverUrl, authToken, projectPath, sessionService) {
-        this.serverUrl = serverUrl;
-        this.ws = null;
-        this.isConnected = false;
-        this.sseClients = new Set();
-        this.heartbeatInterval = null;
-        this.reconnectTimeout = null;
-        this.configManager = (0, ConfigManager_1.getConfigManager)();
-        this.projectPath = projectPath;
-        this.sessionService = sessionService;
-        this.authToken = authToken;
-        this.proxyUrl = this.configManager.getProxyUrl();
-        this.componentSearch = new ComponentSearchService_1.ComponentSearchService();
-        this.fileModService = new FileModificationService_1.FileModificationService();
-        this.gitService = new LocalGitService_1.LocalGitService();
-    }
-    /**
-     * Refresh access token using refresh token
-     */
-    async refreshAccessToken() {
-        const refreshToken = this.configManager.getRefreshToken();
-        if (!refreshToken) {
-            log.error('No refresh token available');
-            return false;
-        }
-        try {
-            log.info('Refreshing access token...');
-            const response = await fetch(`${this.proxyUrl}/api/auth/refresh`, {
-                method: 'POST',
-                headers: {
-                    'Content-Type': 'application/json',
-                },
-                body: JSON.stringify({ refreshToken }),
-            });
-            if (!response.ok) {
-                log.error('Token refresh failed');
-                return false;
-            }
-            const result = await response.json();
-            if (result.success && result.token) {
-                // Update stored tokens
-                const email = this.configManager.getEmail() || '';
-                this.configManager.setAuth(result.token, email, result.refreshToken);
-                this.authToken = result.token;
-                log.info('Access token refreshed successfully');
-                return true;
-            }
-            return false;
-        }
-        catch (error) {
-            log.error('Error refreshing token:', error);
-            return false;
-        }
-    }
-    /**
-     * Connect to hosted Agent SDK server
-     */
-    async connect(retryCount = 0) {
-        return new Promise((resolve, reject) => {
-            log.info(`Connecting to Agent SDK server: ${this.serverUrl}`);
-            this.ws = new ws_1.WebSocket(this.serverUrl);
-            this.ws.on('open', () => {
-                log.info('WebSocket connected, authenticating...');
-                this.send({
-                    type: 'auth',
-                    token: this.authToken,
-                    clientInfo: {
-                        platform: process.platform,
-                        version: process.version,
-                    },
-                });
-                // Start heartbeat
-                this.startHeartbeat();
-            });
-            this.ws.on('message', async (data) => {
-                try {
-                    const message = JSON.parse(data.toString());
-                    await this.handleMessage(message);
-                    if (message.type === 'auth_success') {
-                        this.isConnected = true;
-                        log.info('Authentication successful');
-                        resolve();
-                    }
-                    else if (message.type === 'auth_error') {
-                        this.isConnected = false;
-                        // If we haven't retried yet, try to refresh token
-                        if (retryCount === 0) {
-                            log.warn('Got auth error, attempting to refresh token...');
-                            this.refreshAccessToken().then((refreshed) => {
-                                if (refreshed) {
-                                    // Close current connection and retry with new token
-                                    this.ws?.close();
-                                    this.connect(retryCount + 1)
-                                        .then(resolve)
-                                        .catch(reject);
-                                }
-                                else {
-                                    reject(new Error('Authentication expired. Please run `rivet logout` and `rivet login` to re-authenticate.'));
-                                }
-                            });
-                        }
-                        else {
-                            reject(new Error('Authentication expired. Please run `rivet logout` and `rivet login` to re-authenticate.'));
-                        }
-                    }
-                }
-                catch (error) {
-                    log.error('Error handling message:', error);
-                }
-            });
-            this.ws.on('error', (error) => {
-                log.error('WebSocket error:', error);
-                reject(error);
-            });
-            this.ws.on('close', () => {
-                log.warn('WebSocket disconnected');
-                this.isConnected = false;
-                this.stopHeartbeat();
-                // Attempt reconnect
-                this.scheduleReconnect();
-            });
-            // Timeout after 10s
-            setTimeout(() => {
-                if (!this.isConnected) {
-                    reject(new Error('Connection timeout'));
-                }
-            }, 10000);
-        });
-    }
-    /**
-     * Handle messages from server
-     */
-    async handleMessage(message) {
-        switch (message.type) {
-            case 'tool_execution':
-                await this.executeToolLocally(message);
-                break;
-            case 'agent_chunk':
-                this.broadcastToSSE('agent_chunk', message.chunk);
-                break;
-            case 'agent_complete':
-                this.broadcastToSSE('agent_complete', message.result);
-                break;
-            case 'agent_error':
-                this.broadcastToSSE('agent_error', { error: message.error });
-                break;
-            case 'ping':
-                this.send({ type: 'pong' });
-                break;
-            case 'auth_success':
-            case 'auth_error':
-                // Handled in connect()
-                break;
-        }
-    }
-    /**
-     * Execute tool locally using Rivet's services
-     */
-    async executeToolLocally(message) {
-        const { requestId, tool, params } = message;
-        log.debug(`Executing tool: ${tool}`, params);
-        try {
-            let result;
-            switch (tool) {
-                case 'search_components':
-                    result = await this.executeSearchComponents(params);
-                    break;
-                case 'read_file':
-                    result = await this.executeReadFile(params);
-                    break;
-                case 'write_file':
-                    result = await this.executeWriteFile(params);
-                    break;
-                case 'glob_files':
-                    result = await this.executeGlobFiles(params);
-                    break;
-                case 'git_status':
-                    result = await this.executeGitStatus();
-                    break;
-                case 'git_diff':
-                    result = await this.executeGitDiff(params);
-                    break;
-                default:
-                    throw new Error(`Unknown tool: ${tool}`);
-            }
-            this.send({
-                type: 'tool_result',
-                requestId,
-                result,
-            });
-        }
-        catch (error) {
-            log.error(`Tool execution error (${tool}):`, error);
-            this.send({
-                type: 'tool_error',
-                requestId,
-                error: error instanceof Error ? error.message : 'Unknown error',
-            });
-        }
-    }
-    /**
-     * Tool implementations
-     */
-    async executeSearchComponents(params) {
-        const matches = await this.componentSearch.findMatchingComponents(this.projectPath, params);
-        return { matches };
-    }
-    async executeReadFile(params) {
-        const content = await (0, promises_1.readFile)(params.filePath, 'utf-8');
-        return {
-            content,
-            filePath: params.filePath,
-        };
-    }
-    async executeWriteFile(params) {
-        const results = await this.fileModService.applyModifications([
-            {
-                filePath: params.filePath,
-                search: params.search,
-                replacement: params.replacement,
-                changes: params.changes,
-            },
-        ]);
-        return results[0];
-    }
-    async executeGlobFiles(params) {
-        const files = await (0, glob_1.glob)(params.pattern, {
-            cwd: this.projectPath,
-            ignore: ['**/node_modules/**', '**/.next/**', '**/dist/**'],
-        });
-        return { files };
-    }
-    async executeGitStatus() {
-        const git = await this.gitService.getGitRootPath(this.projectPath);
-        if (!git) {
-            throw new Error('Not a git repository');
-        }
-        const status = await git.status();
-        return {
-            staged: status.staged,
-            unstaged: status.modified.concat(status.deleted),
-            untracked: status.not_added,
-        };
-    }
-    async executeGitDiff(params) {
-        const git = await this.gitService.getGitRootPath(this.projectPath);
-        if (!git) {
-            throw new Error('Not a git repository');
-        }
-        const diff = params.filePath
-            ? await git.diff([params.filePath])
-            : await git.diff();
-        return {
-            diff,
-            filePath: params.filePath,
-        };
-    }
-    /**
-     * Execute instruction via Agent SDK
-     */
-    async executeTask(instruction, context) {
-        if (!this.isConnected) {
-            throw new Error('Not connected to Agent SDK server');
-        }
-        this.send({
-            type: 'execute_task',
-            instruction,
-            context,
-        });
-    }
-    /**
-     * Register SSE client for streaming updates
-     */
-    registerSSEClient(res) {
-        log.debug('Registering SSE client');
-        res.setHeader('Content-Type', 'text/event-stream');
-        res.setHeader('Cache-Control', 'no-cache');
-        res.setHeader('Connection', 'keep-alive');
-        this.sseClients.add(res);
-        res.on('close', () => {
-            log.debug('SSE client disconnected');
-            this.sseClients.delete(res);
-        });
-        // Send initial connection event
-        this.sendSSE(res, 'connected', { status: 'ok' });
-    }
-    /**
-     * Broadcast event to all SSE clients
-     */
-    broadcastToSSE(event, data) {
-        log.debug(`Broadcasting SSE event: ${event}`);
-        for (const client of this.sseClients) {
-            this.sendSSE(client, event, data);
-        }
-    }
-    /**
-     * Send SSE message to specific client
-     */
-    sendSSE(res, event, data) {
-        try {
-            res.write(`event: ${event}\n`);
-            res.write(`data: ${JSON.stringify(data)}\n\n`);
-        }
-        catch (error) {
-            log.error('Failed to send SSE message:', error);
-            this.sseClients.delete(res);
-        }
-    }
-    /**
-     * Send message to server
-     */
-    send(message) {
-        if (!this.ws || this.ws.readyState !== ws_1.WebSocket.OPEN) {
-            log.warn('WebSocket not connected, cannot send message');
-            return;
-        }
-        this.ws.send(JSON.stringify(message));
-    }
-    /**
-     * Start heartbeat ping/pong
-     */
-    startHeartbeat() {
-        if (this.heartbeatInterval) {
-            clearInterval(this.heartbeatInterval);
-        }
-        this.heartbeatInterval = setInterval(() => {
-            if (this.ws && this.ws.readyState === ws_1.WebSocket.OPEN) {
-                // Server sends ping, we respond with pong (handled in handleMessage)
-            }
-        }, HEARTBEAT_INTERVAL);
-    }
-    /**
-     * Stop heartbeat
-     */
-    stopHeartbeat() {
-        if (this.heartbeatInterval) {
-            clearInterval(this.heartbeatInterval);
-            this.heartbeatInterval = null;
-        }
-    }
-    /**
-     * Schedule reconnection attempt
-     */
-    scheduleReconnect() {
-        if (this.reconnectTimeout) {
-            return;
-        }
-        log.info(`Attempting reconnect in ${RECONNECT_DELAY}ms...`);
-        this.reconnectTimeout = setTimeout(async () => {
-            this.reconnectTimeout = null;
-            try {
-                await this.connect();
-                log.info('Reconnected successfully');
-            }
-            catch (error) {
-                log.error('Reconnection failed:', error);
-                this.scheduleReconnect();
-            }
-        }, RECONNECT_DELAY);
-    }
-    /**
-     * Check if connected to server
-     */
-    getIsConnected() {
-        return this.isConnected;
-    }
-    /**
-     * Cleanup resources
-     */
-    cleanup() {
-        this.stopHeartbeat();
-        if (this.reconnectTimeout) {
-            clearTimeout(this.reconnectTimeout);
-            this.reconnectTimeout = null;
-        }
-        if (this.ws) {
-            this.ws.removeAllListeners();
-            this.ws.close();
-            this.ws = null;
-        }
-        for (const client of this.sseClients) {
-            try {
-                client.end();
-            }
-            catch (error) {
-                log.error('Error closing SSE client:', error);
-            }
-        }
-        this.sseClients.clear();
-        this.isConnected = false;
-    }
-}
-exports.AgentBridgeService = AgentBridgeService;
-//# sourceMappingURL=AgentBridgeService.js.map

package/dist/services/AgentBridgeService.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"AgentBridgeService.js","sourceRoot":"","sources":["../../src/services/AgentBridgeService.ts"],"names":[],"mappings":";;;AAAA,2BAA+B;AAC/B,4CAA+C;AAC/C,qEAAkE;AAClE,uEAAoE;AACpE,uDAAoD;AAEpD,mDAAmD;AAgBnD,0CAAuC;AACvC,+BAA4B;AAE5B,MAAM,GAAG,GAAG,IAAA,qBAAY,EAAC,oBAAoB,CAAC,CAAC;AAE/C,MAAM,kBAAkB,GAAG,KAAK,CAAC,CAAC,aAAa;AAC/C,MAAM,eAAe,GAAG,IAAI,CAAC,CAAC,YAAY;AAE1C;;;GAGG;AACH,MAAa,kBAAkB;IAe7B,YACU,SAAiB,EACzB,SAAiB,EACjB,WAAmB,EACnB,cAAqC;QAH7B,cAAS,GAAT,SAAS,CAAQ;QAfnB,OAAE,GAAqB,IAAI,CAAC;QAM5B,gBAAW,GAAG,KAAK,CAAC;QACpB,eAAU,GAAkB,IAAI,GAAG,EAAE,CAAC;QACtC,sBAAiB,GAA0C,IAAI,CAAC;QAChE,qBAAgB,GAAyC,IAAI,CAAC;QAC9D,kBAAa,GAAG,IAAA,gCAAgB,GAAE,CAAC;QAUzC,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;QACjD,IAAI,CAAC,eAAe,GAAG,IAAI,+CAAsB,EAAE,CAAC;QACpD,IAAI,CAAC,cAAc,GAAG,IAAI,iDAAuB,EAAE,CAAC;QACpD,IAAI,CAAC,UAAU,GAAG,IAAI,iCAAe,EAAE,CAAC;IAC1C,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB;QAC9B,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,eAAe,EAAE,CAAC;QAC1D,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,GAAG,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;YACxC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC;YACH,GAAG,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;YACvC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,QAAQ,mBAAmB,EAAE;gBAChE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,YAAY,EAAE,CAAC;aACvC,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,GAAG,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;gBAClC,OAAO,KAAK,CAAC;YACf,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACrC,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACnC,uBAAuB;gBACvB,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;gBAClD,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;gBACrE,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC;gBAC9B,GAAG,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;gBAChD,OAAO,IAAI,CAAC;YACd,CAAC;YAED,OAAO,KAAK,CAAC;QACf,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;YAC5C,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,aAAqB,CAAC;QAClC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,GAAG,CAAC,IAAI,CAAC,mCAAmC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YAE9D,IAAI,CAAC,EAAE,GAAG,IAAI,cAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAExC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;gBACtB,GAAG,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;gBAEnD,IAAI,CAAC,IAAI,CAAC;oBACR,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,IAAI,CAAC,SAAS;oBACrB,UAAU,EAAE;wBACV,QAAQ,EAAE,OAAO,CAAC,QAAQ;wBAC1B,OAAO,EAAE,OAAO,CAAC,OAAO;qBACzB;iBACF,CAAC,CAAC;gBAEH,kBAAkB;gBAClB,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,IAAY,EAAE,EAAE;gBAC3C,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAA0B,CAAC;oBACrE,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;oBAElC,IAAI,OAAO,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;wBACpC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;wBACxB,GAAG,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;wBACtC,OAAO,EAAE,CAAC;oBACZ,CAAC;yBAAM,IAAI,OAAO,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;wBACzC,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;wBAEzB,kDAAkD;wBAClD,IAAI,UAAU,KAAK,CAAC,EAAE,CAAC;4BACrB,GAAG,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAC;4BAC3D,IAAI,CAAC,kBAAkB,EAAE,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE;gCAC3C,IAAI,SAAS,EAAE,CAAC;oCACd,oDAAoD;oCACpD,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,CAAC;oCACjB,IAAI,CAAC,OAAO,CAAC,UAAU,GAAG,CAAC,CAAC;yCACzB,IAAI,CAAC,OAAO,CAAC;yCACb,KAAK,CAAC,MAAM,CAAC,CAAC;gCACnB,CAAC;qCAAM,CAAC;oCACN,MAAM,CACJ,IAAI,KAAK,CACP,yFAAyF,CAC1F,CACF,CAAC;gCACJ,CAAC;4BACH,CAAC,CAAC,CAAC;wBACL,CAAC;6BAAM,CAAC;4BACN,MAAM,CACJ,IAAI,KAAK,CACP,yFAAyF,CAC1F,CACF,CAAC;wBACJ,CAAC;oBACH,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,GAAG,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;gBAC9C,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBAC5B,GAAG,CAAC,KAAK,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;gBACrC,MAAM,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBACvB,GAAG,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;gBACnC,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;gBACzB,IAAI,CAAC,aAAa,EAAE,CAAC;gBAErB,oBAAoB;gBACpB,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC,CAAC,CAAC;YAEH,oBAAoB;YACpB,UAAU,CAAC,GAAG,EAAE;gBACd,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;oBACtB,MAAM,CAAC,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC,CAAC;gBAC1C,CAAC;YACH,CAAC,EAAE,KAAK,CAAC,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,aAAa,CAAC,OAA8B;QACxD,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,gBAAgB;gBACnB,MAAM,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;gBACvC,MAAM;YAER,KAAK,aAAa;gBAChB,IAAI,CAAC,cAAc,CAAC,aAAa,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;gBAClD,MAAM;YAER,KAAK,gBAAgB;gBACnB,IAAI,CAAC,cAAc,CAAC,gBAAgB,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;gBACtD,MAAM;YAER,KAAK,aAAa;gBAChB,IAAI,CAAC,cAAc,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;gBAC7D,MAAM;YAER,KAAK,MAAM;gBACT,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC5B,MAAM;YAER,KAAK,cAAc,CAAC;YACpB,KAAK,YAAY;gBACf,uBAAuB;gBACvB,MAAM;QACV,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB,CAAC,OAIhC;QACC,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC5C,GAAG,CAAC,KAAK,CAAC,mBAAmB,IAAI,EAAE,EAAE,MAAM,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,IAAI,MAAW,CAAC;YAEhB,QAAQ,IAAiB,EAAE,CAAC;gBAC1B,KAAK,mBAAmB;oBACtB,MAAM,GAAG,MAAM,IAAI,CAAC,uBAAuB,CACzC,MAAyC,CAC1C,CAAC;oBACF,MAAM;gBAER,KAAK,WAAW;oBACd,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CACjC,MAAiC,CAClC,CAAC;oBACF,MAAM;gBAER,KAAK,YAAY;oBACf,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAClC,MAAkC,CACnC,CAAC;oBACF,MAAM;gBAER,KAAK,YAAY;oBACf,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAClC,MAAkC,CACnC,CAAC;oBACF,MAAM;gBAER,KAAK,YAAY;oBACf,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;oBACvC,MAAM;gBAER,KAAK,UAAU;oBACb,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,MAAgC,CAAC,CAAC;oBACrE,MAAM;gBAER;oBACE,MAAM,IAAI,KAAK,CAAC,iBAAiB,IAAI,EAAE,CAAC,CAAC;YAC7C,CAAC;YAED,IAAI,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,aAAa;gBACnB,SAAS;gBACT,MAAM;aACP,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,KAAK,CAAC,yBAAyB,IAAI,IAAI,EAAE,KAAK,CAAC,CAAC;YAEpD,IAAI,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,YAAY;gBAClB,SAAS;gBACT,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IAEK,KAAK,CAAC,uBAAuB,CACnC,MAAuC;QAEvC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,sBAAsB,CAC/D,IAAI,CAAC,WAAW,EAChB,MAAM,CACP,CAAC;QAEF,OAAO,EAAE,OAAO,EAAE,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,MAA+B;QAE/B,MAAM,OAAO,GAAG,MAAM,IAAA,mBAAQ,EAAC,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAEzD,OAAO;YACL,OAAO;YACP,QAAQ,EAAE,MAAM,CAAC,QAAQ;SAC1B,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,MAAgC;QAEhC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC;YAC3D;gBACE,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,WAAW,EAAE,MAAM,CAAC,WAAW;gBAC/B,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB;SACF,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,MAAgC;QAEhC,MAAM,KAAK,GAAG,MAAM,IAAA,WAAI,EAAC,MAAM,CAAC,OAAO,EAAE;YACvC,GAAG,EAAE,IAAI,CAAC,WAAW;YACrB,MAAM,EAAE,CAAC,oBAAoB,EAAE,aAAa,EAAE,YAAY,CAAC;SAC5D,CAAC,CAAC;QAEH,OAAO,EAAE,KAAK,EAAE,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,gBAAgB;QAC5B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACnE,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,CAAC;QAElC,OAAO;YACL,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC;YAChD,SAAS,EAAE,MAAM,CAAC,SAAS;SAC5B,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,MAA8B;QAE9B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACnE,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ;YAC1B,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YACnC,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;QAErB,OAAO;YACL,IAAI;YACJ,QAAQ,EAAE,MAAM,CAAC,QAAQ;SAC1B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,WAAmB,EAAE,OAAa;QAClD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,CAAC,IAAI,CAAC;YACR,IAAI,EAAE,cAAc;YACpB,WAAW;YACX,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,GAAa;QAC7B,GAAG,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAEpC,GAAG,CAAC,SAAS,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;QACnD,GAAG,CAAC,SAAS,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;QAC3C,GAAG,CAAC,SAAS,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;QAE1C,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAEzB,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACnB,GAAG,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;YACrC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,gCAAgC;QAChC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,WAAW,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,KAAa,EAAE,IAAS;QAC7C,GAAG,CAAC,KAAK,CAAC,2BAA2B,KAAK,EAAE,CAAC,CAAC;QAE9C,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACrC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,OAAO,CAAC,GAAa,EAAE,KAAa,EAAE,IAAS;QACrD,IAAI,CAAC;YACH,GAAG,CAAC,KAAK,CAAC,UAAU,KAAK,IAAI,CAAC,CAAC;YAC/B,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;YAChD,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IAED;;OAEG;IACK,IAAI,CAAC,OAA8B;QACzC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,cAAS,CAAC,IAAI,EAAE,CAAC;YACtD,GAAG,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;YACzD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;IACxC,CAAC;IAED;;OAEG;IACK,cAAc;QACpB,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,aAAa,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACxC,CAAC;QAED,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE;YACxC,IAAI,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,cAAS,CAAC,IAAI,EAAE,CAAC;gBACrD,qEAAqE;YACvE,CAAC;QACH,CAAC,EAAE,kBAAkB,CAAC,CAAC;IACzB,CAAC;IAED;;OAEG;IACK,aAAa;QACnB,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,aAAa,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACtC,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;QAChC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,iBAAiB;QACvB,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,OAAO;QACT,CAAC;QAED,GAAG,CAAC,IAAI,CAAC,2BAA2B,eAAe,OAAO,CAAC,CAAC;QAE5D,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAC,KAAK,IAAI,EAAE;YAC5C,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;YAE7B,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;gBACrB,GAAG,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YACvC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,KAAK,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;gBACzC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC;QACH,CAAC,EAAE,eAAe,CAAC,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,OAAO;QACL,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,YAAY,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACpC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC/B,CAAC;QAED,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC,EAAE,CAAC,kBAAkB,EAAE,CAAC;YAC7B,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;QAED,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACrC,IAAI,CAAC;gBACH,MAAM,CAAC,GAAG,EAAE,CAAC;YACf,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;QAED,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACxB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;IAC3B,CAAC;CACF;AAtfD,gDAsfC"}

package/dist/services/AgentModService.d.ts

@@ -1,76 +0,0 @@
-import { SessionService } from './SessionService';
-import { ElementContext } from '@rivet/core';
-export type AgentStep = {
-    type: 'tool_call' | 'tool_result' | 'reasoning' | 'error';
-    timestamp: number;
-    data: unknown;
-};
-export type AgenticProgressEvent = {
-    type: 'step' | 'complete' | 'error';
-    step?: AgentStep;
-    result?: AgenticResult;
-    error?: string;
-};
-export type AgenticResult = {
-    success: boolean;
-    filesModified: string[];
-    reasoning: string[];
-    changes: string;
-};
-/**
- * Service for agentic code modifications using Claude Agent SDK.
- * Agent autonomously searches for files, reads context, and makes modifications.
- */
-export declare class AgentModService {
-    private projectPath;
-    private sessionService?;
-    private configManager;
-    private gatewayUrl;
-    private proxyUrl;
-    constructor(projectPath: string, sessionService?: SessionService, gatewayUrl?: string);
-    /**
-     * Refresh access token using refresh token
-     */
-    private refreshAccessToken;
-    /**
-     * Decode JWT and check if it expires within the next hour
-     */
-    private isTokenExpiringSoon;
-    /**
-     * Execute an agentic modification workflow.
-     * The agent autonomously uses Grep/Glob/Read to find files and Edit to modify them.
-     *
-     * @effect Streams progress events via onProgress callback
-     * @param instruction User's modification instruction
-     * @param initialElements Selected elements from UI with candidate file paths
-     * @param onProgress Callback for streaming progress updates
-     * @param designTokens Optional design system tokens
-     * @param image Optional reference image
-     */
-    executeAgenticModification(instruction: string, initialElements: ElementContext[], onProgress: (event: AgenticProgressEvent) => void, designTokens?: unknown, image?: string, retryCount?: number): Promise<AgenticResult>;
-    /**
-     * Build system prompt for the agent
-     */
-    private buildSystemPrompt;
-    /**
-     * Build the initial agent prompt with context
-     */
-    private buildAgentPrompt;
-    /**
-     * Build initial context string from selected elements and read candidate files
-     *
-     * @effect Reads candidate files from disk
-     */
-    private buildInitialContext;
-    /**
-     * Create a ChangeRecord for git tracking
-     *
-     * @effect Reads modified files from disk
-     */
-    private createChangeRecord;
-    /**
-     * Summarize changes for user display
-     */
-    private summarizeChanges;
-}
-//# sourceMappingURL=AgentModService.d.ts.map

package/dist/services/AgentModService.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"AgentModService.d.ts","sourceRoot":"","sources":["../../src/services/AgentModService.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,cAAc,EAAgB,MAAM,aAAa,CAAC;AAgB3D,MAAM,MAAM,SAAS,GAAG;IACtB,IAAI,EAAE,WAAW,GAAG,aAAa,GAAG,WAAW,GAAG,OAAO,CAAC;IAC1D,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,OAAO,CAAC;CACf,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG;IACjC,IAAI,EAAE,MAAM,GAAG,UAAU,GAAG,OAAO,CAAC;IACpC,IAAI,CAAC,EAAE,SAAS,CAAC;IACjB,MAAM,CAAC,EAAE,aAAa,CAAC;IACvB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,aAAa,EAAE,MAAM,EAAE,CAAC;IACxB,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF;;;GAGG;AACH,qBAAa,eAAe;IAC1B,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,cAAc,CAAC,CAAiB;IACxC,OAAO,CAAC,aAAa,CAAsB;IAC3C,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,QAAQ,CAAS;gBAGvB,WAAW,EAAE,MAAM,EACnB,cAAc,CAAC,EAAE,cAAc,EAC/B,UAAU,CAAC,EAAE,MAAM;IAQrB;;OAEG;YACW,kBAAkB;IAsChC;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAuB3B;;;;;;;;;;OAUG;IACG,0BAA0B,CAC9B,WAAW,EAAE,MAAM,EACnB,eAAe,EAAE,cAAc,EAAE,EACjC,UAAU,EAAE,CAAC,KAAK,EAAE,oBAAoB,KAAK,IAAI,EACjD,YAAY,CAAC,EAAE,OAAO,EACtB,KAAK,CAAC,EAAE,MAAM,EACd,UAAU,GAAE,MAAU,GACrB,OAAO,CAAC,aAAa,CAAC;IAmPzB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IA8CzB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAYxB;;;;OAIG;YACW,mBAAmB;IAsEjC;;;;OAIG;YACW,kBAAkB;IAgChC;;OAEG;IACH,OAAO,CAAC,gBAAgB;CAczB"}