Kusto AI Chat Extension

An intelligent AI-powered VS Code extension for Azure Data Explorer (Kusto) query optimization and assistance. This extension provides a specialized interface for interacting with Kusto databases through AI-powered query generation and optimization.

Features

• Intelligent Query Processing: 5-step AI analysis for optimal KQL query generation
• Multi-Configuration Management: Support for multiple Kusto and OpenAI configurations
• Advanced Table Statistics: Real-time analysis of column statistics, compression ratios, and row counts
• Smart Query Optimization: Uses comprehensive database statistics for performance optimization
• Interactive Chat Interface: Dynamic message sizing and proper code formatting
• Kusto Context Detection: Automatically switches between intelligent and basic modes based on message content
• Configuration Management: Easy setup and management of Kusto cluster connections

Requirements

• VS Code 1.60.0 or higher
• Access to Azure Data Explorer cluster
• OpenAI API key or Azure OpenAI service

Extension Settings

This extension contributes the following settings:

• kustoAIChat.clusterUrl: Kusto cluster URL
• kustoAIChat.database: Default database name
• kustoAIChat.authType: Authentication method (Service Principal or Interactive)
• kustoAIChat.openaiProvider: OpenAI service provider (OpenAI or Azure OpenAI)

Commands

• Kusto AI Chat: Configure Kusto Connection: Set up Kusto cluster connection
• Kusto AI Chat: Configure OpenAI Settings: Configure AI service settings
• Kusto AI Chat: Clear Authentication Token Cache: Clear cached authentication tokens

Usage

1. Install the extension from the VS Code marketplace
2. Configure your Kusto connection using the Configuration Manager panel
3. Set up your OpenAI or Azure OpenAI credentials
4. Start asking Kusto-related questions in the AI Chat panel

Available Commands in Chat

• !check-config: View current configuration status
• !test-connection: Test connectivity to Kusto and OpenAI services
• !help: Show help information
• !logs: View extension logging information

5-Step Intelligent Processing

When you ask Kusto-related questions, the extension uses advanced processing:

1. Requirement Analysis: Understanding your data needs
2. Table Discovery: Finding all available tables in your database
3. Relevance Matching: Identifying tables relevant to your query
4. Statistics Analysis: Gathering detailed table and column statistics
5. Query Optimization: Generating optimal KQL queries based on comprehensive analysis

Known Issues

• Large databases may take longer during the table discovery phase
• Interactive authentication may require browser interaction

Release Notes

0.1.0

Initial release of Kusto AI Chat Extension with intelligent query processing and multi-configuration support.

Contributing

This extension is part of the Kusto AI Optimizer project. Contributions are welcome!

License

This project is licensed under the MIT License. See the LICENSE file for details.