mcp-perplexity-pro 1.1.0

package/dist/streaming-wrapper.js

@@ -0,0 +1,452 @@
+import { PerplexityApiClient } from './perplexity-api.js';
+import { selectOptimalModel } from './models.js';
+/**
+ * Active streaming operations registry
+ */
+const activeStreams = new Map();
+/**
+ * Universal async tool wrapper that creates async jobs and streams progress to Claude Code
+ */
+export class StreamingWrapper {
+    server;
+    config;
+    constructor(server, config) {
+        this.server = server;
+        this.config = config;
+    }
+    /**
+     * Execute a tool call with streaming content updates (not just progress)
+     */
+    async executeWithStreaming(toolName, params, progressToken) {
+        // If no progress token provided, execute synchronously (backwards compatibility)
+        if (!progressToken) {
+            return this.executeSynchronously(toolName, params);
+        }
+        try {
+            // Create async job based on tool type
+            const asyncJob = await this.createAsyncJob(toolName, params);
+            if ('error' in asyncJob) {
+                return asyncJob;
+            }
+            // Register streaming state
+            const streamState = {
+                jobId: asyncJob.id,
+                progressToken,
+                server: this.server,
+                startTime: Date.now(),
+                lastProgress: 0,
+                toolName,
+                params,
+                config: this.config,
+            };
+            activeStreams.set(asyncJob.id, streamState);
+            // Start streaming content (not just progress)
+            this.startContentStreaming(asyncJob.id);
+            // Return initial response indicating streaming has started
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: `🚀 Starting ${toolName}...\n\n`,
+                    },
+                ],
+            };
+        }
+        catch (error) {
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: `Error: ${error instanceof Error ? error.message : String(error)}`,
+                    },
+                ],
+                isError: true,
+            };
+        }
+    }
+    /**
+     * Create async job based on tool type
+     */
+    async createAsyncJob(toolName, params) {
+        const apiClient = new PerplexityApiClient(this.config);
+        // Determine the query/content based on tool type
+        let query;
+        let model;
+        switch (toolName) {
+            case 'ask_perplexity':
+                query = params.query;
+                model = params.model || selectOptimalModel(query, undefined, this.config.default_model);
+                break;
+            case 'chat_perplexity':
+                query = params.message;
+                model = params.model || selectOptimalModel(query, undefined, this.config.default_model);
+                break;
+            case 'research_perplexity':
+                query = params.topic;
+                model = params.model || 'sonar-deep-research';
+                break;
+            default:
+                throw new Error(`Unsupported tool for streaming: ${toolName}`);
+        }
+        // Create the async request
+        const request = {
+            model,
+            messages: [{ role: 'user', content: query }],
+            temperature: params.temperature ?? 0.2,
+            ...(params.max_tokens && { max_tokens: params.max_tokens }),
+            ...(params.search_domain_filter && { search_domain_filter: params.search_domain_filter }),
+            ...(params.return_images && { return_images: params.return_images }),
+            ...(params.return_related_questions && {
+                return_related_questions: params.return_related_questions,
+            }),
+        };
+        return await apiClient.createAsyncChatCompletion(request);
+    }
+    /**
+     * Execute tool synchronously (fallback for no progress token)
+     */
+    async executeSynchronously(toolName, params) {
+        // Import handlers dynamically to avoid circular dependencies
+        const { handleAskPerplexity, handleResearchPerplexity } = await import('./tools/query.js');
+        const { handleChatPerplexity } = await import('./tools/chat.js');
+        switch (toolName) {
+            case 'ask_perplexity':
+                return await handleAskPerplexity(params, this.config);
+            case 'chat_perplexity': {
+                const result = await handleChatPerplexity(params, this.config);
+                return { content: [{ type: 'text', text: JSON.stringify(result, null, 2) }] };
+            }
+            case 'research_perplexity': {
+                const researchResult = await handleResearchPerplexity(params, this.config);
+                return { content: [{ type: 'text', text: JSON.stringify(researchResult, null, 2) }] };
+            }
+            default:
+                throw new Error(`Unknown tool: ${toolName}`);
+        }
+    }
+    /**
+     * Start streaming actual content from async job (not just progress)
+     */
+    async startContentStreaming(jobId) {
+        const state = activeStreams.get(jobId);
+        if (!state)
+            return;
+        const apiClient = new PerplexityApiClient(this.config);
+        let pollCount = 0;
+        let lastContentLength = 0;
+        const maxPolls = 60; // 5 minutes max (5s intervals)
+        const pollAndStreamContent = async () => {
+            try {
+                const jobStatus = await apiClient.getAsyncJob(jobId);
+                if (!activeStreams.has(jobId)) {
+                    // Stream was cancelled
+                    return;
+                }
+                pollCount++;
+                // Check if we have new content to stream
+                const currentContent = jobStatus.choices?.[0]?.message?.content || '';
+                if (currentContent.length > lastContentLength) {
+                    // We have new content! Stream the delta
+                    const newContent = currentContent.slice(lastContentLength);
+                    await this.streamContentChunk(state, newContent, jobStatus.status);
+                    lastContentLength = currentContent.length;
+                }
+                // Send status updates for major state changes
+                if (jobStatus.status === 'STARTED' && state.lastProgress < 25) {
+                    await this.streamContentChunk(state, '📡 Processing your request...\n\n', jobStatus.status);
+                    state.lastProgress = 25;
+                }
+                // Handle completion
+                if (jobStatus.status === 'COMPLETED') {
+                    // Stream any remaining content
+                    if (currentContent.length > lastContentLength) {
+                        const finalContent = currentContent.slice(lastContentLength);
+                        await this.streamContentChunk(state, finalContent, jobStatus.status);
+                    }
+                    // Process final job result (save report, etc.)
+                    await this.handleJobCompletion(jobId, jobStatus);
+                    // Stream completion message
+                    await this.streamContentChunk(state, '\n\n✅ Complete!', 'COMPLETED');
+                    activeStreams.delete(jobId);
+                    return;
+                }
+                else if (jobStatus.status === 'FAILED') {
+                    await this.streamContentChunk(state, `\n\n❌ Job failed: ${jobStatus.error || 'Unknown error'}`, 'FAILED');
+                    activeStreams.delete(jobId);
+                    return;
+                }
+                // Continue polling if job is still running
+                if (pollCount < maxPolls &&
+                    (jobStatus.status === 'CREATED' || jobStatus.status === 'STARTED')) {
+                    setTimeout(pollAndStreamContent, 2000); // Poll every 2 seconds for more responsive streaming
+                }
+                else if (pollCount >= maxPolls) {
+                    // Timeout reached
+                    await this.streamContentChunk(state, '\n\n⏰ Request timed out after 5 minutes', 'FAILED');
+                    activeStreams.delete(jobId);
+                }
+            }
+            catch (error) {
+                console.error(`Error polling job ${jobId}:`, error);
+                await this.streamContentChunk(state, `\n\n❌ Error: ${error instanceof Error ? error.message : String(error)}`, 'FAILED');
+                activeStreams.delete(jobId);
+            }
+        };
+        // Start polling immediately
+        setTimeout(pollAndStreamContent, 500); // Start after 500ms
+    }
+    /**
+     * Stream a chunk of content to Claude Code
+     */
+    async streamContentChunk(state, content, status) {
+        try {
+            // Use MCP's streaming capabilities to send partial content
+            // This creates the illusion of real-time streaming by sending content in chunks
+            await state.server.notification({
+                method: 'notifications/message',
+                params: {
+                    level: 'info',
+                    logger: `mcp-perplexity-${state.toolName}`,
+                    data: {
+                        type: 'content_chunk',
+                        content: content,
+                        job_id: state.jobId,
+                        status: status,
+                        timestamp: Date.now(),
+                    },
+                },
+            });
+            // Also send to stdout for CLI clients that capture output
+            process.stdout.write(content);
+        }
+        catch (error) {
+            console.error('Error streaming content chunk:', error);
+        }
+    }
+    /**
+     * Legacy method - keeping for backwards compatibility
+     */
+    async startProgressStreaming(jobId) {
+        const state = activeStreams.get(jobId);
+        if (!state)
+            return;
+        const apiClient = new PerplexityApiClient(this.config);
+        let pollCount = 0;
+        const maxPolls = 60; // 5 minutes max (5s intervals)
+        const pollProgress = async () => {
+            try {
+                const jobStatus = await apiClient.getAsyncJob(jobId);
+                if (!activeStreams.has(jobId)) {
+                    // Stream was cancelled
+                    return;
+                }
+                pollCount++;
+                // Calculate progress based on status and elapsed time
+                let progress = state.lastProgress;
+                let total = 100;
+                switch (jobStatus.status) {
+                    case 'CREATED':
+                        progress = Math.min(10 + pollCount * 2, 20);
+                        break;
+                    case 'STARTED':
+                        progress = Math.min(25 + pollCount * 3, 80);
+                        break;
+                    case 'COMPLETED':
+                        progress = 100;
+                        break;
+                    case 'FAILED':
+                        progress = 0;
+                        total = 0;
+                        break;
+                }
+                // Only send progress update if progress increased
+                if (progress > state.lastProgress ||
+                    jobStatus.status === 'COMPLETED' ||
+                    jobStatus.status === 'FAILED') {
+                    await this.sendProgressNotification(state, progress, total, jobStatus);
+                    state.lastProgress = progress;
+                }
+                // Handle completion or failure
+                if (jobStatus.status === 'COMPLETED') {
+                    await this.handleJobCompletion(jobId, jobStatus);
+                    activeStreams.delete(jobId);
+                    return;
+                }
+                else if (jobStatus.status === 'FAILED') {
+                    await this.handleJobFailure(jobId, jobStatus);
+                    activeStreams.delete(jobId);
+                    return;
+                }
+                // Continue polling if job is still running and we haven't exceeded max polls
+                if (pollCount < maxPolls &&
+                    (jobStatus.status === 'CREATED' || jobStatus.status === 'STARTED')) {
+                    setTimeout(pollProgress, 5000); // Poll every 5 seconds
+                }
+                else if (pollCount >= maxPolls) {
+                    // Timeout reached
+                    await this.sendProgressNotification(state, 0, 0, {
+                        ...jobStatus,
+                        status: 'FAILED',
+                        error: 'Job timed out after 5 minutes',
+                    });
+                    activeStreams.delete(jobId);
+                }
+            }
+            catch (error) {
+                console.error(`Error polling job ${jobId}:`, error);
+                // Send error notification
+                await this.sendProgressNotification(state, 0, 0, {
+                    id: jobId,
+                    status: 'FAILED',
+                    error: error instanceof Error ? error.message : String(error),
+                    created_at: Math.floor(Date.now() / 1000),
+                    model: 'unknown',
+                });
+                activeStreams.delete(jobId);
+            }
+        };
+        // Start polling
+        setTimeout(pollProgress, 1000); // Start after 1 second
+    }
+    /**
+     * Send progress notification to Claude Code
+     */
+    async sendProgressNotification(state, progress, total, jobStatus) {
+        try {
+            const elapsedTime = Date.now() - state.startTime;
+            const statusMessage = this.getStatusMessage(jobStatus.status, progress, elapsedTime);
+            // Send MCP progress notification
+            await state.server.notification({
+                method: 'notifications/progress',
+                params: {
+                    progressToken: state.progressToken,
+                    progress,
+                    total,
+                    _meta: {
+                        status: jobStatus.status,
+                        job_id: jobStatus.id,
+                        tool_name: state.toolName,
+                        elapsed_time_ms: elapsedTime,
+                        message: statusMessage,
+                        ...(jobStatus.error && { error: jobStatus.error }),
+                    },
+                },
+            });
+        }
+        catch (error) {
+            console.error('Error sending progress notification:', error);
+        }
+    }
+    /**
+     * Handle job completion and send final result
+     */
+    async handleJobCompletion(jobId, jobStatus) {
+        const state = activeStreams.get(jobId);
+        if (!state)
+            return;
+        try {
+            // Process the completed result based on tool type
+            const finalResult = await this.processCompletedJob(state.toolName, state.params, jobStatus);
+            // Send final progress notification with result
+            await state.server.notification({
+                method: 'notifications/progress',
+                params: {
+                    progressToken: state.progressToken,
+                    progress: 100,
+                    total: 100,
+                    _meta: {
+                        status: 'COMPLETED',
+                        job_id: jobId,
+                        tool_name: state.toolName,
+                        elapsed_time_ms: Date.now() - state.startTime,
+                        message: 'Completed successfully',
+                        final_result: finalResult,
+                    },
+                },
+            });
+        }
+        catch (error) {
+            console.error(`Error handling job completion for ${jobId}:`, error);
+        }
+    }
+    /**
+     * Handle job failure
+     */
+    async handleJobFailure(jobId, jobStatus) {
+        const state = activeStreams.get(jobId);
+        if (!state)
+            return;
+        await this.sendProgressNotification(state, 0, 0, jobStatus);
+    }
+    /**
+     * Process completed job based on tool type
+     */
+    async processCompletedJob(toolName, params, jobStatus) {
+        // Save report if requested
+        if (params.save_report && jobStatus.choices?.[0]?.message?.content) {
+            const { StorageManager } = await import('./storage.js');
+            const { detectProjectWithSuggestions } = await import('./tools/projects.js');
+            const projectName = params.project_name || (await detectProjectWithSuggestions(undefined, this.config));
+            const toolSubdir = toolName.replace('_perplexity', '');
+            const projectConfig = {
+                ...this.config,
+                storage_path: `projects/${projectName}/${toolSubdir}`,
+            };
+            const storageManager = new StorageManager(projectConfig);
+            let reportContent;
+            switch (toolName) {
+                case 'ask_perplexity':
+                    reportContent = `# Ask Perplexity Report\n\n**Query:** ${params.query}\n**Model:** ${jobStatus.model}\n**Timestamp:** ${new Date().toISOString()}\n\n## Response\n\n${jobStatus.choices[0].message.content}`;
+                    break;
+                case 'research_perplexity':
+                    reportContent = `# Research Report\n\n**Topic:** ${params.topic}\n**Model:** ${jobStatus.model}\n**Timestamp:** ${new Date().toISOString()}\n\n## Research Results\n\n${jobStatus.choices[0].message.content}`;
+                    break;
+                case 'chat_perplexity':
+                    reportContent = `# Chat Export\n\n**Message:** ${params.message}\n**Model:** ${jobStatus.model}\n**Timestamp:** ${new Date().toISOString()}\n\n## Response\n\n${jobStatus.choices[0].message.content}`;
+                    break;
+                default:
+                    reportContent = `# ${toolName} Report\n\n${jobStatus.choices[0].message.content}`;
+            }
+            await storageManager.saveReport(reportContent, `${toolName}-${Date.now()}`);
+        }
+        return {
+            job_id: jobStatus.id,
+            status: jobStatus.status,
+            model: jobStatus.model,
+            response: jobStatus.choices?.[0]?.message?.content || 'No response content',
+            usage: jobStatus.usage,
+            ...(params.save_report && { report_saved: true }),
+        };
+    }
+    /**
+     * Get status message for progress updates
+     */
+    getStatusMessage(status, progress, elapsedTime) {
+        const elapsedSeconds = Math.floor(elapsedTime / 1000);
+        switch (status) {
+            case 'CREATED':
+                return `Job created, waiting to start... (${elapsedSeconds}s)`;
+            case 'STARTED':
+                return `Processing... ${progress}% complete (${elapsedSeconds}s)`;
+            case 'COMPLETED':
+                return `Completed successfully! (${elapsedSeconds}s total)`;
+            case 'FAILED':
+                return `Job failed (${elapsedSeconds}s)`;
+            default:
+                return `Status: ${status} (${elapsedSeconds}s)`;
+        }
+    }
+    /**
+     * Cancel streaming for a job
+     */
+    cancelStreaming(jobId) {
+        activeStreams.delete(jobId);
+    }
+    /**
+     * Get active streaming jobs
+     */
+    getActiveStreams() {
+        return Array.from(activeStreams.keys());
+    }
+}
+//# sourceMappingURL=streaming-wrapper.js.map

package/dist/streaming-wrapper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming-wrapper.js","sourceRoot":"","sources":["../src/streaming-wrapper.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAiBjD;;GAEG;AACH,MAAM,aAAa,GAAG,IAAI,GAAG,EAA0B,CAAC;AAExD;;GAEG;AACH,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAS;IACf,MAAM,CAAS;IAEvB,YAAY,MAAc,EAAE,MAAc;QACxC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,oBAAoB,CACxB,QAAgB,EAChB,MAAW,EACX,aAA+B;QAE/B,iFAAiF;QACjF,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,IAAI,CAAC,oBAAoB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,CAAC;YACH,sCAAsC;YACtC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAE7D,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;gBACxB,OAAO,QAAQ,CAAC;YAClB,CAAC;YAED,2BAA2B;YAC3B,MAAM,WAAW,GAAmB;gBAClC,KAAK,EAAE,QAAQ,CAAC,EAAE;gBAClB,aAAa;gBACb,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,YAAY,EAAE,CAAC;gBACf,QAAQ;gBACR,MAAM;gBACN,MAAM,EAAE,IAAI,CAAC,MAAM;aACpB,CAAC;YAEF,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;YAE5C,8CAA8C;YAC9C,IAAI,CAAC,qBAAqB,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAExC,2DAA2D;YAC3D,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,eAAe,QAAQ,SAAS;qBACvC;iBACF;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,UAAU,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;qBACzE;iBACF;gBACD,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,cAAc,CAAC,QAAgB,EAAE,MAAW;QACxD,MAAM,SAAS,GAAG,IAAI,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEvD,iDAAiD;QACjD,IAAI,KAAa,CAAC;QAClB,IAAI,KAAa,CAAC;QAElB,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,gBAAgB;gBACnB,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;gBACrB,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,kBAAkB,CAAC,KAAK,EAAE,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;gBACxF,MAAM;YAER,KAAK,iBAAiB;gBACpB,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC;gBACvB,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,kBAAkB,CAAC,KAAK,EAAE,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;gBACxF,MAAM;YAER,KAAK,qBAAqB;gBACxB,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;gBACrB,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,qBAAqB,CAAC;gBAC9C,MAAM;YAER;gBACE,MAAM,IAAI,KAAK,CAAC,mCAAmC,QAAQ,EAAE,CAAC,CAAC;QACnE,CAAC;QAED,2BAA2B;QAC3B,MAAM,OAAO,GAAG;YACd,KAAK;YACL,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YACrD,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,GAAG;YACtC,GAAG,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC;YAC3D,GAAG,CAAC,MAAM,CAAC,oBAAoB,IAAI,EAAE,oBAAoB,EAAE,MAAM,CAAC,oBAAoB,EAAE,CAAC;YACzF,GAAG,CAAC,MAAM,CAAC,aAAa,IAAI,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC;YACpE,GAAG,CAAC,MAAM,CAAC,wBAAwB,IAAI;gBACrC,wBAAwB,EAAE,MAAM,CAAC,wBAAwB;aAC1D,CAAC;SACH,CAAC;QAEF,OAAO,MAAM,SAAS,CAAC,yBAAyB,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,oBAAoB,CAAC,QAAgB,EAAE,MAAW;QAC9D,6DAA6D;QAC7D,MAAM,EAAE,mBAAmB,EAAE,wBAAwB,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC3F,MAAM,EAAE,oBAAoB,EAAE,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;QAEjE,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,gBAAgB;gBACnB,OAAO,MAAM,mBAAmB,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YAExD,KAAK,iBAAiB,CAAC,CAAC,CAAC;gBACvB,MAAM,MAAM,GAAG,MAAM,oBAAoB,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC/D,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;YAChF,CAAC;YAED,KAAK,qBAAqB,CAAC,CAAC,CAAC;gBAC3B,MAAM,cAAc,GAAG,MAAM,wBAAwB,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC3E,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;YACxF,CAAC;YAED;gBACE,MAAM,IAAI,KAAK,CAAC,iBAAiB,QAAQ,EAAE,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,qBAAqB,CAAC,KAAa;QAC/C,MAAM,KAAK,GAAG,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,MAAM,SAAS,GAAG,IAAI,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvD,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,IAAI,iBAAiB,GAAG,CAAC,CAAC;QAC1B,MAAM,QAAQ,GAAG,EAAE,CAAC,CAAC,+BAA+B;QAEpD,MAAM,oBAAoB,GAAG,KAAK,IAAI,EAAE;YACtC,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;gBAErD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC9B,uBAAuB;oBACvB,OAAO;gBACT,CAAC;gBAED,SAAS,EAAE,CAAC;gBAEZ,yCAAyC;gBACzC,MAAM,cAAc,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;gBAEtE,IAAI,cAAc,CAAC,MAAM,GAAG,iBAAiB,EAAE,CAAC;oBAC9C,wCAAwC;oBACxC,MAAM,UAAU,GAAG,cAAc,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;oBAC3D,MAAM,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,UAAU,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;oBACnE,iBAAiB,GAAG,cAAc,CAAC,MAAM,CAAC;gBAC5C,CAAC;gBAED,8CAA8C;gBAC9C,IAAI,SAAS,CAAC,MAAM,KAAK,SAAS,IAAI,KAAK,CAAC,YAAY,GAAG,EAAE,EAAE,CAAC;oBAC9D,MAAM,IAAI,CAAC,kBAAkB,CAC3B,KAAK,EACL,mCAAmC,EACnC,SAAS,CAAC,MAAM,CACjB,CAAC;oBACF,KAAK,CAAC,YAAY,GAAG,EAAE,CAAC;gBAC1B,CAAC;gBAED,oBAAoB;gBACpB,IAAI,SAAS,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;oBACrC,+BAA+B;oBAC/B,IAAI,cAAc,CAAC,MAAM,GAAG,iBAAiB,EAAE,CAAC;wBAC9C,MAAM,YAAY,GAAG,cAAc,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;wBAC7D,MAAM,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,YAAY,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;oBACvE,CAAC;oBAED,+CAA+C;oBAC/C,MAAM,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;oBAEjD,4BAA4B;oBAC5B,MAAM,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,iBAAiB,EAAE,WAAW,CAAC,CAAC;oBAErE,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBAC5B,OAAO;gBACT,CAAC;qBAAM,IAAI,SAAS,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;oBACzC,MAAM,IAAI,CAAC,kBAAkB,CAC3B,KAAK,EACL,qBAAqB,SAAS,CAAC,KAAK,IAAI,eAAe,EAAE,EACzD,QAAQ,CACT,CAAC;oBACF,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBAC5B,OAAO;gBACT,CAAC;gBAED,2CAA2C;gBAC3C,IACE,SAAS,GAAG,QAAQ;oBACpB,CAAC,SAAS,CAAC,MAAM,KAAK,SAAS,IAAI,SAAS,CAAC,MAAM,KAAK,SAAS,CAAC,EAClE,CAAC;oBACD,UAAU,CAAC,oBAAoB,EAAE,IAAI,CAAC,CAAC,CAAC,qDAAqD;gBAC/F,CAAC;qBAAM,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;oBACjC,kBAAkB;oBAClB,MAAM,IAAI,CAAC,kBAAkB,CAC3B,KAAK,EACL,yCAAyC,EACzC,QAAQ,CACT,CAAC;oBACF,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,qBAAqB,KAAK,GAAG,EAAE,KAAK,CAAC,CAAC;gBACpD,MAAM,IAAI,CAAC,kBAAkB,CAC3B,KAAK,EACL,gBAAgB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EACxE,QAAQ,CACT,CAAC;gBACF,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC,CAAC;QAEF,4BAA4B;QAC5B,UAAU,CAAC,oBAAoB,EAAE,GAAG,CAAC,CAAC,CAAC,oBAAoB;IAC7D,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB,CAC9B,KAAqB,EACrB,OAAe,EACf,MAAc;QAEd,IAAI,CAAC;YACH,2DAA2D;YAC3D,gFAAgF;YAChF,MAAM,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC;gBAC9B,MAAM,EAAE,uBAAuB;gBAC/B,MAAM,EAAE;oBACN,KAAK,EAAE,MAAM;oBACb,MAAM,EAAE,kBAAkB,KAAK,CAAC,QAAQ,EAAE;oBAC1C,IAAI,EAAE;wBACJ,IAAI,EAAE,eAAe;wBACrB,OAAO,EAAE,OAAO;wBAChB,MAAM,EAAE,KAAK,CAAC,KAAK;wBACnB,MAAM,EAAE,MAAM;wBACd,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;qBACtB;iBACF;aACK,CAAC,CAAC;YAEV,0DAA0D;YAC1D,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,sBAAsB,CAAC,KAAa;QAChD,MAAM,KAAK,GAAG,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,MAAM,SAAS,GAAG,IAAI,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvD,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,MAAM,QAAQ,GAAG,EAAE,CAAC,CAAC,+BAA+B;QAEpD,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE;YAC9B,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;gBAErD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC9B,uBAAuB;oBACvB,OAAO;gBACT,CAAC;gBAED,SAAS,EAAE,CAAC;gBAEZ,sDAAsD;gBACtD,IAAI,QAAQ,GAAG,KAAK,CAAC,YAAY,CAAC;gBAClC,IAAI,KAAK,GAAG,GAAG,CAAC;gBAEhB,QAAQ,SAAS,CAAC,MAAM,EAAE,CAAC;oBACzB,KAAK,SAAS;wBACZ,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,SAAS,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC5C,MAAM;oBACR,KAAK,SAAS;wBACZ,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,SAAS,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC5C,MAAM;oBACR,KAAK,WAAW;wBACd,QAAQ,GAAG,GAAG,CAAC;wBACf,MAAM;oBACR,KAAK,QAAQ;wBACX,QAAQ,GAAG,CAAC,CAAC;wBACb,KAAK,GAAG,CAAC,CAAC;wBACV,MAAM;gBACV,CAAC;gBAED,kDAAkD;gBAClD,IACE,QAAQ,GAAG,KAAK,CAAC,YAAY;oBAC7B,SAAS,CAAC,MAAM,KAAK,WAAW;oBAChC,SAAS,CAAC,MAAM,KAAK,QAAQ,EAC7B,CAAC;oBACD,MAAM,IAAI,CAAC,wBAAwB,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;oBACvE,KAAK,CAAC,YAAY,GAAG,QAAQ,CAAC;gBAChC,CAAC;gBAED,+BAA+B;gBAC/B,IAAI,SAAS,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;oBACrC,MAAM,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;oBACjD,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBAC5B,OAAO;gBACT,CAAC;qBAAM,IAAI,SAAS,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;oBACzC,MAAM,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;oBAC9C,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBAC5B,OAAO;gBACT,CAAC;gBAED,6EAA6E;gBAC7E,IACE,SAAS,GAAG,QAAQ;oBACpB,CAAC,SAAS,CAAC,MAAM,KAAK,SAAS,IAAI,SAAS,CAAC,MAAM,KAAK,SAAS,CAAC,EAClE,CAAC;oBACD,UAAU,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC,CAAC,uBAAuB;gBACzD,CAAC;qBAAM,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;oBACjC,kBAAkB;oBAClB,MAAM,IAAI,CAAC,wBAAwB,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE;wBAC/C,GAAG,SAAS;wBACZ,MAAM,EAAE,QAAiB;wBACzB,KAAK,EAAE,+BAA+B;qBACvC,CAAC,CAAC;oBACH,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,qBAAqB,KAAK,GAAG,EAAE,KAAK,CAAC,CAAC;gBACpD,0BAA0B;gBAC1B,MAAM,IAAI,CAAC,wBAAwB,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE;oBAC/C,EAAE,EAAE,KAAK;oBACT,MAAM,EAAE,QAAiB;oBACzB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;oBAC7D,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;oBACzC,KAAK,EAAE,SAAS;iBACjB,CAAC,CAAC;gBACH,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC,CAAC;QAEF,gBAAgB;QAChB,UAAU,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC,CAAC,uBAAuB;IACzD,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,wBAAwB,CACpC,KAAqB,EACrB,QAAgB,EAChB,KAAa,EACb,SAAmB;QAEnB,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,SAAS,CAAC;YACjD,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,MAAM,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC;YAErF,iCAAiC;YACjC,MAAM,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC;gBAC9B,MAAM,EAAE,wBAAwB;gBAChC,MAAM,EAAE;oBACN,aAAa,EAAE,KAAK,CAAC,aAAa;oBAClC,QAAQ;oBACR,KAAK;oBACL,KAAK,EAAE;wBACL,MAAM,EAAE,SAAS,CAAC,MAAM;wBACxB,MAAM,EAAE,SAAS,CAAC,EAAE;wBACpB,SAAS,EAAE,KAAK,CAAC,QAAQ;wBACzB,eAAe,EAAE,WAAW;wBAC5B,OAAO,EAAE,aAAa;wBACtB,GAAG,CAAC,SAAS,CAAC,KAAK,IAAI,EAAE,KAAK,EAAE,SAAS,CAAC,KAAK,EAAE,CAAC;qBACnD;iBACF;aACK,CAAC,CAAC;QACZ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,sCAAsC,EAAE,KAAK,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,mBAAmB,CAAC,KAAa,EAAE,SAAmB;QAClE,MAAM,KAAK,GAAG,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,IAAI,CAAC;YACH,kDAAkD;YAClD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAE5F,+CAA+C;YAC/C,MAAM,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC;gBAC9B,MAAM,EAAE,wBAAwB;gBAChC,MAAM,EAAE;oBACN,aAAa,EAAE,KAAK,CAAC,aAAa;oBAClC,QAAQ,EAAE,GAAG;oBACb,KAAK,EAAE,GAAG;oBACV,KAAK,EAAE;wBACL,MAAM,EAAE,WAAW;wBACnB,MAAM,EAAE,KAAK;wBACb,SAAS,EAAE,KAAK,CAAC,QAAQ;wBACzB,eAAe,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,SAAS;wBAC7C,OAAO,EAAE,wBAAwB;wBACjC,YAAY,EAAE,WAAW;qBAC1B;iBACF;aACK,CAAC,CAAC;QACZ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,qCAAqC,KAAK,GAAG,EAAE,KAAK,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,gBAAgB,CAAC,KAAa,EAAE,SAAmB;QAC/D,MAAM,KAAK,GAAG,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,MAAM,IAAI,CAAC,wBAAwB,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;IAC9D,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,mBAAmB,CAC/B,QAAgB,EAChB,MAAW,EACX,SAAmB;QAEnB,2BAA2B;QAC3B,IAAI,MAAM,CAAC,WAAW,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;YACnE,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC;YACxD,MAAM,EAAE,4BAA4B,EAAE,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAAC,CAAC;YAE7E,MAAM,WAAW,GACf,MAAM,CAAC,YAAY,IAAI,CAAC,MAAM,4BAA4B,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;YACtF,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;YAEvD,MAAM,aAAa,GAAG;gBACpB,GAAG,IAAI,CAAC,MAAM;gBACd,YAAY,EAAE,YAAY,WAAW,IAAI,UAAU,EAAE;aACtD,CAAC;YAEF,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC,aAAa,CAAC,CAAC;YAEzD,IAAI,aAAqB,CAAC;YAE1B,QAAQ,QAAQ,EAAE,CAAC;gBACjB,KAAK,gBAAgB;oBACnB,aAAa,GAAG,yCAAyC,MAAM,CAAC,KAAK,gBAAgB,SAAS,CAAC,KAAK,oBAAoB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,sBAAsB,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;oBAC7M,MAAM;gBAER,KAAK,qBAAqB;oBACxB,aAAa,GAAG,mCAAmC,MAAM,CAAC,KAAK,gBAAgB,SAAS,CAAC,KAAK,oBAAoB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,8BAA8B,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;oBAC/M,MAAM;gBAER,KAAK,iBAAiB;oBACpB,aAAa,GAAG,iCAAiC,MAAM,CAAC,OAAO,gBAAgB,SAAS,CAAC,KAAK,oBAAoB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,sBAAsB,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;oBACvM,MAAM;gBAER;oBACE,aAAa,GAAG,KAAK,QAAQ,cAAc,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YACtF,CAAC;YAED,MAAM,cAAc,CAAC,UAAU,CAAC,aAAa,EAAE,GAAG,QAAQ,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC9E,CAAC;QAED,OAAO;YACL,MAAM,EAAE,SAAS,CAAC,EAAE;YACpB,MAAM,EAAE,SAAS,CAAC,MAAM;YACxB,KAAK,EAAE,SAAS,CAAC,KAAK;YACtB,QAAQ,EAAE,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,qBAAqB;YAC3E,KAAK,EAAE,SAAS,CAAC,KAAK;YACtB,GAAG,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;SAClD,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,MAAc,EAAE,QAAgB,EAAE,WAAmB;QAC5E,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,CAAC;QAEtD,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,SAAS;gBACZ,OAAO,qCAAqC,cAAc,IAAI,CAAC;YACjE,KAAK,SAAS;gBACZ,OAAO,iBAAiB,QAAQ,eAAe,cAAc,IAAI,CAAC;YACpE,KAAK,WAAW;gBACd,OAAO,4BAA4B,cAAc,UAAU,CAAC;YAC9D,KAAK,QAAQ;gBACX,OAAO,eAAe,cAAc,IAAI,CAAC;YAC3C;gBACE,OAAO,WAAW,MAAM,KAAK,cAAc,IAAI,CAAC;QACpD,CAAC;IACH,CAAC;IAED;;OAEG;IACI,eAAe,CAAC,KAAa;QAClC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC9B,CAAC;IAED;;OAEG;IACI,gBAAgB;QACrB,OAAO,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,CAAC;IAC1C,CAAC;CACF"}

package/dist/tools/async.d.ts

@@ -0,0 +1,28 @@
+import type { AsyncPerplexityParams, CheckAsyncParams, Config, AsyncJob, ErrorResponse } from '../types.js';
+/**
+ * Handles the async_perplexity tool - creates async jobs for long-running queries
+ */
+export declare function handleAsyncPerplexity(params: AsyncPerplexityParams, config: Config): Promise<(AsyncJob & {
+    selected_model: string;
+    model_selection_reason: string;
+    estimated_completion: string;
+}) | ErrorResponse>;
+/**
+ * Handles the check_async_perplexity tool - checks status of async jobs
+ */
+export declare function handleCheckAsync(params: CheckAsyncParams, config: Config): Promise<(AsyncJob & {
+    completion_percentage?: number;
+    next_check_recommended?: string;
+}) | ErrorResponse>;
+/**
+ * Handles the list_async_jobs tool - lists all async jobs
+ */
+export declare function handleListAsyncJobs(config: Config, limit?: number, nextToken?: string): Promise<{
+    jobs: (AsyncJob & {
+        time_since_created: string;
+        estimated_time_remaining?: string;
+    })[];
+    next_token?: string;
+    total_jobs: number;
+} | ErrorResponse>;
+//# sourceMappingURL=async.d.ts.map

package/dist/tools/async.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"async.d.ts","sourceRoot":"","sources":["../../src/tools/async.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,qBAAqB,EACrB,gBAAgB,EAChB,MAAM,EACN,QAAQ,EACR,aAAa,EACd,MAAM,aAAa,CAAC;AAIrB;;GAEG;AACH,wBAAsB,qBAAqB,CACzC,MAAM,EAAE,qBAAqB,EAC7B,MAAM,EAAE,MAAM,GACb,OAAO,CACN,CAAC,QAAQ,GAAG;IACV,cAAc,EAAE,MAAM,CAAC;IACvB,sBAAsB,EAAE,MAAM,CAAC;IAC/B,oBAAoB,EAAE,MAAM,CAAC;CAC9B,CAAC,GACF,aAAa,CAChB,CAyCA;AAED;;GAEG;AACH,wBAAsB,gBAAgB,CACpC,MAAM,EAAE,gBAAgB,EACxB,MAAM,EAAE,MAAM,GACb,OAAO,CACN,CAAC,QAAQ,GAAG;IACV,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,sBAAsB,CAAC,EAAE,MAAM,CAAC;CACjC,CAAC,GACF,aAAa,CAChB,CA0CA;AAED;;GAEG;AACH,wBAAsB,mBAAmB,CACvC,MAAM,EAAE,MAAM,EACd,KAAK,SAAK,EACV,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CACN;IACE,IAAI,EAAE,CAAC,QAAQ,GAAG;QAAE,kBAAkB,EAAE,MAAM,CAAC;QAAC,wBAAwB,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,EAAE,CAAC;IACvF,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;CACpB,GACD,aAAa,CAChB,CAsCA"}

package/dist/tools/async.js

@@ -0,0 +1,167 @@
+import { PerplexityApiClient } from '../perplexity-api.js';
+import { selectOptimalModel } from '../models.js';
+/**
+ * Handles the async_perplexity tool - creates async jobs for long-running queries
+ */
+export async function handleAsyncPerplexity(params, config) {
+    try {
+        const apiClient = new PerplexityApiClient(config);
+        // Detect project for potential report saving (not currently used)
+        // const { detectProjectWithSuggestions } = await import('./projects.js');
+        // const projectName = await detectProjectWithSuggestions(undefined, config);
+        // Select optimal model based on query or use explicit model
+        const selectedModel = selectOptimalModel(params.query, params.model, config.default_model);
+        // Prepare the async request
+        const request = {
+            model: selectedModel,
+            messages: [{ role: 'user', content: params.query }],
+            temperature: params.temperature ?? 0.2,
+            ...(params.max_tokens && { max_tokens: params.max_tokens }),
+        };
+        const response = await apiClient.createAsyncChatCompletion(request);
+        // Estimate completion time based on model type and complexity
+        const estimatedMinutes = getEstimatedCompletionTime(selectedModel, params.query);
+        const estimatedCompletion = new Date(Date.now() + estimatedMinutes * 60 * 1000).toISOString();
+        return {
+            ...response,
+            selected_model: selectedModel,
+            model_selection_reason: params.model ? 'user_specified' : 'auto_selected',
+            estimated_completion: estimatedCompletion,
+        };
+    }
+    catch (error) {
+        return PerplexityApiClient.handleError(error, {
+            model: params.model || config.default_model,
+            query: params.query,
+        });
+    }
+}
+/**
+ * Handles the check_async_perplexity tool - checks status of async jobs
+ */
+export async function handleCheckAsync(params, config) {
+    try {
+        const apiClient = new PerplexityApiClient(config);
+        // Note: async job checking doesn't need project-aware config since
+        // job IDs are global and not project-specific
+        const response = await apiClient.getAsyncJob(params.job_id);
+        // Add helpful metadata for job monitoring
+        let completionPercentage;
+        let nextCheckRecommended;
+        switch (response.status) {
+            case 'CREATED':
+                completionPercentage = 10;
+                nextCheckRecommended = new Date(Date.now() + 30 * 1000).toISOString(); // 30 seconds
+                break;
+            case 'STARTED':
+                completionPercentage = 50;
+                nextCheckRecommended = new Date(Date.now() + 60 * 1000).toISOString(); // 1 minute
+                break;
+            case 'COMPLETED':
+                completionPercentage = 100;
+                break;
+            case 'FAILED':
+                completionPercentage = 0;
+                break;
+        }
+        return {
+            ...response,
+            ...(completionPercentage !== undefined && { completion_percentage: completionPercentage }),
+            ...(nextCheckRecommended && { next_check_recommended: nextCheckRecommended }),
+        };
+    }
+    catch (error) {
+        return PerplexityApiClient.handleError(error, {
+            query: `Async job check: ${params.job_id}`,
+        });
+    }
+}
+/**
+ * Handles the list_async_jobs tool - lists all async jobs
+ */
+export async function handleListAsyncJobs(config, limit = 20, nextToken) {
+    try {
+        const apiClient = new PerplexityApiClient(config);
+        // Note: listing async jobs doesn't need project-aware config since
+        // jobs are listed globally, not per project
+        const response = await apiClient.listAsyncJobs(limit, nextToken);
+        // Add helpful metadata to each job
+        const enrichedJobs = (response.jobs || []).map(job => {
+            const timeSinceCreated = formatTimeDuration(Date.now() - job.created_at * 1000);
+            let estimatedTimeRemaining;
+            // Estimate remaining time for active jobs
+            if (job.status === 'CREATED' || job.status === 'STARTED') {
+                const elapsedMinutes = (Date.now() - job.created_at * 1000) / (1000 * 60);
+                const estimatedTotalMinutes = getEstimatedCompletionTime(job.model, 'complex query');
+                const remainingMinutes = Math.max(0, estimatedTotalMinutes - elapsedMinutes);
+                estimatedTimeRemaining = `${Math.ceil(remainingMinutes)} minutes`;
+            }
+            return {
+                ...job,
+                time_since_created: timeSinceCreated,
+                ...(estimatedTimeRemaining && { estimated_time_remaining: estimatedTimeRemaining }),
+            };
+        });
+        return {
+            jobs: enrichedJobs,
+            ...(response.next_token && { next_token: response.next_token }),
+            total_jobs: (response.jobs || []).length,
+        };
+    }
+    catch (error) {
+        return PerplexityApiClient.handleError(error, {
+            query: 'List async jobs',
+        });
+    }
+}
+/**
+ * Estimates completion time based on model and query complexity
+ */
+function getEstimatedCompletionTime(model, query) {
+    // Base times in minutes
+    const baseTimes = {
+        sonar: 0.5,
+        'sonar-pro': 1,
+        'sonar-reasoning': 1.5,
+        'sonar-reasoning-pro': 3,
+        'sonar-deep-research': 5,
+    };
+    let baseTime = baseTimes[model] || 2;
+    // Adjust based on query complexity
+    const queryLength = query.length;
+    const complexityKeywords = [
+        'comprehensive',
+        'detailed',
+        'analysis',
+        'research',
+        'compare',
+        'evaluate',
+        'investigate',
+    ];
+    // Add time for query length
+    if (queryLength > 500)
+        baseTime *= 1.5;
+    else if (queryLength > 200)
+        baseTime *= 1.2;
+    // Add time for complexity keywords
+    const complexityScore = complexityKeywords.reduce((score, keyword) => score + (query.toLowerCase().includes(keyword) ? 1 : 0), 0);
+    baseTime *= 1 + complexityScore * 0.3;
+    return Math.ceil(baseTime);
+}
+/**
+ * Formats time duration in a human-readable format
+ */
+function formatTimeDuration(milliseconds) {
+    const seconds = Math.floor(milliseconds / 1000);
+    const minutes = Math.floor(seconds / 60);
+    const hours = Math.floor(minutes / 60);
+    const days = Math.floor(hours / 24);
+    if (days > 0)
+        return `${days}d ${hours % 24}h`;
+    if (hours > 0)
+        return `${hours}h ${minutes % 60}m`;
+    if (minutes > 0)
+        return `${minutes}m ${seconds % 60}s`;
+    return `${seconds}s`;
+}
+//# sourceMappingURL=async.js.map