@sqlrooms/ai 0.25.0-rc.0 → 0.26.0-rc.1

package/README.md

@@ -60,7 +60,7 @@ npm install ollama-ai-provider-v2
 ### Setting Up AI Integration
 
 ```tsx
-import {createAiSlice, createDefaultAiConfig} from '@sqlrooms/ai';
+import {createAiSlice} from '@sqlrooms/ai';
 import {createRoomStore} from '@sqlrooms/room-shell';
 
 // Create a room store with AI capabilities
@@ -69,7 +69,6 @@ const {roomStore, useRoomStore} = createRoomStore({
   ...createRoomShellSlice({
     config: {
       // Your room configuration
-      ...createDefaultAiConfig(), // Default AI configuration
     },
   }),
   // Add AI slice
@@ -104,11 +103,7 @@ function MyApp() {
 For more complex applications, you can combine multiple slices:
 
 ```tsx
-import {
-  createAiSlice,
-  createDefaultAiConfig,
-  AiSliceConfig,
-} from '@sqlrooms/ai';
+import {createAiSlice} from '@sqlrooms/ai';
 import {
   createSqlEditorSlice,
   createDefaultSqlEditorConfig,
@@ -126,26 +121,25 @@ export const {roomStore, useRoomStore} = createRoomStore<RoomConfig, RoomState>(
     // Base room slice
     ...createRoomShellSlice({
       config: {
-        // Your base configuration
-        ...createDefaultAiConfig({
-          // Optional: Pre-configured AI sessions
-          sessions: [
-            {
-              id: 'default-session',
-              name: 'Default Analysis',
-              modelProvider: 'openai',
-              model: 'gpt-4o',
-              analysisResults: [],
-              createdAt: new Date(),
-            },
-          ],
-          currentSessionId: 'default-session',
-        }),
         ...createDefaultSqlEditorConfig(),
       },
     }),
     // AI slice
     ...createAiSlice({
+      config: {
+        // Optional: Pre-configured AI sessions
+        sessions: [
+          {
+            id: 'default-session',
+            name: 'Default Analysis',
+            modelProvider: 'openai',
+            model: 'gpt-4o',
+            analysisResults: [],
+            createdAt: new Date(),
+          },
+        ],
+        currentSessionId: 'default-session',
+      }
       getApiKey: (modelProvider) => {
         // Return API key based on provider
         return apiKeys[modelProvider] || '';
@@ -505,7 +499,6 @@ This package now includes comprehensive AI settings components. These components
 - **ModelsSettings**: Component for managing available models and their parameters
 - **ModelParametersSettings**: Component for configuring model parameters like max steps and system instructions
 - **ModelSelector**: Standalone model selector component for quick model switching
-- **extractModelsFromSettings**: Utility function to extract models from configuration for use in selectors
 
 ### AI Settings Usage
 
@@ -515,25 +508,18 @@ This package now includes comprehensive AI settings components. These components
 import {
   AiSettingsPanel,
   ModelSelector,
-  extractModelsFromSettings,
 } from '@sqlrooms/ai';
 import {useRoomStore} from '../store';
 
-// Access AI model configuration from store
-const aiSettings = useRoomStore((s) => s.aiSettings.config);
-const models = extractModelsFromSettings(aiSettings);
-
 // Main configuration panel with sub-components
 <AiSettingsPanel isOpen={isConfigOpen} setIsOpen={setIsConfigOpen}>
   <AiSettingsPanel.ProvidersSettings />
   <AiSettingsPanel.ModelsSettings />
-  <AiSettingsPanel.ModelParametersSettings
-    getDefaultInstructions={getDefaultInstructionsWrapper}
-  />
+  <AiSettingsPanel.ModelParametersSettings  />
 </AiSettingsPanel>
 
 // Standalone model selector
-<ModelSelector models={models} />
+<ModelSelector />
 ```
 
 ### AI Settings API Reference
@@ -545,7 +531,6 @@ const models = extractModelsFromSettings(aiSettings);
   - `AiSettingsPanel.ModelsSettings`: Manage available models and their parameters
   - `AiSettingsPanel.ModelParametersSettings`: Configure model parameters and instructions
 - **`ModelSelector`**: Standalone model selector for quick switching
-- **`extractModelsFromSettings(config)`**: Utility to extract models from configuration
 
 #### Slice Configuration
 

package/dist/index.d.ts

@@ -2,31 +2,12 @@
  * {@include ../README.md}
  * @packageDocumentation
  */
-export { AiSliceConfig, createAiSlice, useStoreWithAi, createDefaultAiConfig, } from './AiSlice';
-export type { AiSliceState } from './AiSlice';
-export { QueryControls } from './components/QueryControls';
-export { AnalysisResultsContainer } from './components/AnalysisResultsContainer';
-export { AnalysisResult } from './components/AnalysisResult';
-export { useScrollToBottom } from './hooks/useScrollToBottom';
-export type { AiSliceTool } from './AiSlice';
-export { QueryToolResult } from './components/tools/QueryToolResult';
-export { getDefaultInstructions } from './analysis';
-export * from './components/ModelSelector';
-export * from './components/SessionControls';
-export * from './components/QueryControls';
-export * from './components/session/DeleteSessionDialog';
-export * from './components/session/SessionActions';
-export * from './components/session/SessionDropdown';
-export * from './components/session/SessionTitle';
-export * from './components/session/SessionType';
-export * from './components/tools/ToolErrorMessage';
-export { createAiSettingsSlice, useStoreWithAiSettings, AiSettingsSliceConfig, createDefaultAiSettingsConfig, } from './AiSettingsSlice';
-export type { AiSettingsSliceState } from './AiSettingsSlice';
-export { AiSettingsPanel } from './components/settings/AiSettingsPanel';
-export { AiProvidersSettings } from './components/settings/AiProvidersSettings';
-export { AiModelsSettings } from './components/settings/AiModelsSettings';
-export { AiModelParameters } from './components/settings/AiModelParameters';
-export { AiModelUsage } from './components/settings/AiModelUsage';
-export type { ModelUsageData } from './components/settings/AiModelUsage';
-export { extractModelsFromSettings } from './utils';
+export { QueryToolResult } from './tools/query/QueryToolResult';
+export { QueryToolParameters } from './tools/query/queryTool';
+export { createDefaultAiTools } from './tools/defaultTools';
+export type { DefaultToolsOptions } from './tools/defaultTools';
+export { createDefaultAiInstructions } from './tools/defaultInstructions';
+export * from '@sqlrooms/ai-core';
+export * from '@sqlrooms/ai-config';
+export * from '@sqlrooms/ai-settings';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EACL,aAAa,EACb,aAAa,EACb,cAAc,EACd,qBAAqB,GACtB,MAAM,WAAW,CAAC;AAEnB,YAAY,EAAC,YAAY,EAAC,MAAM,WAAW,CAAC;AAC5C,OAAO,EAAC,aAAa,EAAC,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAC,wBAAwB,EAAC,MAAM,uCAAuC,CAAC;AAC/E,OAAO,EAAC,cAAc,EAAC,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAC,iBAAiB,EAAC,MAAM,2BAA2B,CAAC;AAC5D,YAAY,EAAC,WAAW,EAAC,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAC,eAAe,EAAC,MAAM,oCAAoC,CAAC;AACnE,OAAO,EAAC,sBAAsB,EAAC,MAAM,YAAY,CAAC;AAElD,cAAc,4BAA4B,CAAC;AAC3C,cAAc,8BAA8B,CAAC;AAC7C,cAAc,4BAA4B,CAAC;AAC3C,cAAc,0CAA0C,CAAC;AACzD,cAAc,qCAAqC,CAAC;AACpD,cAAc,sCAAsC,CAAC;AACrD,cAAc,mCAAmC,CAAC;AAClD,cAAc,kCAAkC,CAAC;AACjD,cAAc,qCAAqC,CAAC;AAEpD,OAAO,EACL,qBAAqB,EACrB,sBAAsB,EACtB,qBAAqB,EACrB,6BAA6B,GAC9B,MAAM,mBAAmB,CAAC;AAE3B,YAAY,EAAC,oBAAoB,EAAC,MAAM,mBAAmB,CAAC;AAE5D,OAAO,EAAC,eAAe,EAAC,MAAM,uCAAuC,CAAC;AACtE,OAAO,EAAC,mBAAmB,EAAC,MAAM,2CAA2C,CAAC;AAC9E,OAAO,EAAC,gBAAgB,EAAC,MAAM,wCAAwC,CAAC;AACxE,OAAO,EAAC,iBAAiB,EAAC,MAAM,yCAAyC,CAAC;AAC1E,OAAO,EAAC,YAAY,EAAC,MAAM,oCAAoC,CAAC;AAChE,YAAY,EAAC,cAAc,EAAC,MAAM,oCAAoC,CAAC;AACvE,OAAO,EAAC,yBAAyB,EAAC,MAAM,SAAS,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAC,eAAe,EAAC,MAAM,+BAA+B,CAAC;AAC9D,OAAO,EAAC,mBAAmB,EAAC,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAC,oBAAoB,EAAC,MAAM,sBAAsB,CAAC;AAC1D,YAAY,EAAC,mBAAmB,EAAC,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAC,2BAA2B,EAAC,MAAM,6BAA6B,CAAC;AAExE,cAAc,mBAAmB,CAAC;AAClC,cAAc,qBAAqB,CAAC;AACpC,cAAc,uBAAuB,CAAC"}

package/dist/index.js

@@ -2,27 +2,11 @@
  * {@include ../README.md}
  * @packageDocumentation
  */
-export { AiSliceConfig, createAiSlice, useStoreWithAi, createDefaultAiConfig, } from './AiSlice';
-export { QueryControls } from './components/QueryControls';
-export { AnalysisResultsContainer } from './components/AnalysisResultsContainer';
-export { AnalysisResult } from './components/AnalysisResult';
-export { useScrollToBottom } from './hooks/useScrollToBottom';
-export { QueryToolResult } from './components/tools/QueryToolResult';
-export { getDefaultInstructions } from './analysis';
-export * from './components/ModelSelector';
-export * from './components/SessionControls';
-export * from './components/QueryControls';
-export * from './components/session/DeleteSessionDialog';
-export * from './components/session/SessionActions';
-export * from './components/session/SessionDropdown';
-export * from './components/session/SessionTitle';
-export * from './components/session/SessionType';
-export * from './components/tools/ToolErrorMessage';
-export { createAiSettingsSlice, useStoreWithAiSettings, AiSettingsSliceConfig, createDefaultAiSettingsConfig, } from './AiSettingsSlice';
-export { AiSettingsPanel } from './components/settings/AiSettingsPanel';
-export { AiProvidersSettings } from './components/settings/AiProvidersSettings';
-export { AiModelsSettings } from './components/settings/AiModelsSettings';
-export { AiModelParameters } from './components/settings/AiModelParameters';
-export { AiModelUsage } from './components/settings/AiModelUsage';
-export { extractModelsFromSettings } from './utils';
+export { QueryToolResult } from './tools/query/QueryToolResult';
+export { QueryToolParameters } from './tools/query/queryTool';
+export { createDefaultAiTools } from './tools/defaultTools';
+export { createDefaultAiInstructions } from './tools/defaultInstructions';
+export * from '@sqlrooms/ai-core';
+export * from '@sqlrooms/ai-config';
+export * from '@sqlrooms/ai-settings';
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EACL,aAAa,EACb,aAAa,EACb,cAAc,EACd,qBAAqB,GACtB,MAAM,WAAW,CAAC;AAGnB,OAAO,EAAC,aAAa,EAAC,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAC,wBAAwB,EAAC,MAAM,uCAAuC,CAAC;AAC/E,OAAO,EAAC,cAAc,EAAC,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAC,iBAAiB,EAAC,MAAM,2BAA2B,CAAC;AAE5D,OAAO,EAAC,eAAe,EAAC,MAAM,oCAAoC,CAAC;AACnE,OAAO,EAAC,sBAAsB,EAAC,MAAM,YAAY,CAAC;AAElD,cAAc,4BAA4B,CAAC;AAC3C,cAAc,8BAA8B,CAAC;AAC7C,cAAc,4BAA4B,CAAC;AAC3C,cAAc,0CAA0C,CAAC;AACzD,cAAc,qCAAqC,CAAC;AACpD,cAAc,sCAAsC,CAAC;AACrD,cAAc,mCAAmC,CAAC;AAClD,cAAc,kCAAkC,CAAC;AACjD,cAAc,qCAAqC,CAAC;AAEpD,OAAO,EACL,qBAAqB,EACrB,sBAAsB,EACtB,qBAAqB,EACrB,6BAA6B,GAC9B,MAAM,mBAAmB,CAAC;AAI3B,OAAO,EAAC,eAAe,EAAC,MAAM,uCAAuC,CAAC;AACtE,OAAO,EAAC,mBAAmB,EAAC,MAAM,2CAA2C,CAAC;AAC9E,OAAO,EAAC,gBAAgB,EAAC,MAAM,wCAAwC,CAAC;AACxE,OAAO,EAAC,iBAAiB,EAAC,MAAM,yCAAyC,CAAC;AAC1E,OAAO,EAAC,YAAY,EAAC,MAAM,oCAAoC,CAAC;AAEhE,OAAO,EAAC,yBAAyB,EAAC,MAAM,SAAS,CAAC","sourcesContent":["/**\n * {@include ../README.md}\n * @packageDocumentation\n */\n\nexport {\n  AiSliceConfig,\n  createAiSlice,\n  useStoreWithAi,\n  createDefaultAiConfig,\n} from './AiSlice';\n\nexport type {AiSliceState} from './AiSlice';\nexport {QueryControls} from './components/QueryControls';\nexport {AnalysisResultsContainer} from './components/AnalysisResultsContainer';\nexport {AnalysisResult} from './components/AnalysisResult';\nexport {useScrollToBottom} from './hooks/useScrollToBottom';\nexport type {AiSliceTool} from './AiSlice';\nexport {QueryToolResult} from './components/tools/QueryToolResult';\nexport {getDefaultInstructions} from './analysis';\n\nexport * from './components/ModelSelector';\nexport * from './components/SessionControls';\nexport * from './components/QueryControls';\nexport * from './components/session/DeleteSessionDialog';\nexport * from './components/session/SessionActions';\nexport * from './components/session/SessionDropdown';\nexport * from './components/session/SessionTitle';\nexport * from './components/session/SessionType';\nexport * from './components/tools/ToolErrorMessage';\n\nexport {\n  createAiSettingsSlice,\n  useStoreWithAiSettings,\n  AiSettingsSliceConfig,\n  createDefaultAiSettingsConfig,\n} from './AiSettingsSlice';\n\nexport type {AiSettingsSliceState} from './AiSettingsSlice';\n\nexport {AiSettingsPanel} from './components/settings/AiSettingsPanel';\nexport {AiProvidersSettings} from './components/settings/AiProvidersSettings';\nexport {AiModelsSettings} from './components/settings/AiModelsSettings';\nexport {AiModelParameters} from './components/settings/AiModelParameters';\nexport {AiModelUsage} from './components/settings/AiModelUsage';\nexport type {ModelUsageData} from './components/settings/AiModelUsage';\nexport {extractModelsFromSettings} from './utils';\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAC,eAAe,EAAC,MAAM,+BAA+B,CAAC;AAC9D,OAAO,EAAC,mBAAmB,EAAC,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAC,oBAAoB,EAAC,MAAM,sBAAsB,CAAC;AAE1D,OAAO,EAAC,2BAA2B,EAAC,MAAM,6BAA6B,CAAC;AAExE,cAAc,mBAAmB,CAAC;AAClC,cAAc,qBAAqB,CAAC;AACpC,cAAc,uBAAuB,CAAC","sourcesContent":["/**\n * {@include ../README.md}\n * @packageDocumentation\n */\n\nexport {QueryToolResult} from './tools/query/QueryToolResult';\nexport {QueryToolParameters} from './tools/query/queryTool';\nexport {createDefaultAiTools} from './tools/defaultTools';\nexport type {DefaultToolsOptions} from './tools/defaultTools';\nexport {createDefaultAiInstructions} from './tools/defaultInstructions';\n\nexport * from '@sqlrooms/ai-core';\nexport * from '@sqlrooms/ai-config';\nexport * from '@sqlrooms/ai-settings';\n"]}

package/dist/tools/defaultInstructions.d.ts

@@ -0,0 +1,8 @@
+import { AiSliceState } from '@sqlrooms/ai-core';
+import { DuckDbSliceState } from '@sqlrooms/duckdb';
+import { StoreApi } from '@sqlrooms/room-shell';
+/**
+ * Returns the default system instructions for the AI assistant
+ */
+export declare function createDefaultAiInstructions(store: StoreApi<AiSliceState & DuckDbSliceState>): string;
+//# sourceMappingURL=defaultInstructions.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"defaultInstructions.d.ts","sourceRoot":"","sources":["../../src/tools/defaultInstructions.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,YAAY,EAAC,MAAM,mBAAmB,CAAC;AAC/C,OAAO,EAAC,gBAAgB,EAAC,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAC,QAAQ,EAAC,MAAM,sBAAsB,CAAC;AA2D9C;;GAEG;AACH,wBAAgB,2BAA2B,CACzC,KAAK,EAAE,QAAQ,CAAC,YAAY,GAAG,gBAAgB,CAAC,GAC/C,MAAM,CAGR"}

package/dist/tools/defaultInstructions.js

@@ -0,0 +1,64 @@
+/**
+ * System prompt template for the AI assistant that provides instructions for:
+ * - Using DuckDB-specific SQL syntax and functions
+ * - Handling query results and error cases
+ * - Creating visualizations with VegaLite
+ * - Formatting final answers
+ */
+const DEFAULT_INSTRUCTIONS = `
+You are analyzing tables in DuckDB database in the context of a room.
+
+Instructions for analysis:
+- When using 'query' tool, please assign parameter 'type' with value 'query'
+- Use DuckDB-specific SQL syntax and functions (not Oracle, PostgreSQL, or other SQL dialects)
+- Some key DuckDB-specific functions to use:
+  * regexp_matches() for regex (not regexp_like)
+  * strftime() for date formatting (not to_char)
+  * list_aggregate() for array operations
+  * unnest() for array expansion
+  * regr_sxy()
+  * corr()
+  * skewness()
+- Please always try to use SQL queries to answer users questions
+- Please run tool calls sequentially, don't run multiple tool calls in parallel
+- IMPORTANT: Do not list out raw query results in your response. Instead:
+  * Describe the results in natural language
+  * Provide summary statistics
+  * Use comparisons and relative terms
+  * Include only the most relevant values if necessary
+- Break down complex problems into smaller steps
+- Use "SUMMARIZE table_name"for quick overview of the table
+- Please don't modify data
+- IMPORTANT: When you receive an error response from a tool call (where success: false):
+  * Stop making any further tool calls immediately
+  * Return a final answer that includes the error message
+  * Explain what went wrong and suggest possible fixes if applicable
+
+When creating visualizations:
+- Follow VegaLite syntax
+- Choose appropriate chart types based on the data and analysis goals
+- Use clear titles and axis labels
+- Consider color schemes for better readability
+- Add meaningful tooltips when relevant
+- Format numbers and dates appropriately
+- Use aggregations when dealing with large datasets
+
+For your final answer:
+- Provide an explanation for how you got it
+- Explain your reasoning step by step
+- Include relevant statistics or metrics
+- For each prompt, please always provide the final answer.
+- IMPORTANT: Query tool results may include sample rows (firstRows) or may be empty:
+  * If no sample rows provided: Never fabricate data. Direct users to the table component for actual results.
+  * If sample rows provided: Use them to enhance your analysis, but always direct users to the table component for complete results.
+
+Please use the following schema for the tables:
+`;
+/**
+ * Returns the default system instructions for the AI assistant
+ */
+export function createDefaultAiInstructions(store) {
+    const tables = store.getState().db.tables;
+    return `${DEFAULT_INSTRUCTIONS}\n${JSON.stringify(tables)}`;
+}
+//# sourceMappingURL=defaultInstructions.js.map

package/dist/tools/defaultInstructions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"defaultInstructions.js","sourceRoot":"","sources":["../../src/tools/defaultInstructions.ts"],"names":[],"mappings":"AAIA;;;;;;GAMG;AACH,MAAM,oBAAoB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAgD5B,CAAC;AAEF;;GAEG;AACH,MAAM,UAAU,2BAA2B,CACzC,KAAgD;IAEhD,MAAM,MAAM,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC,MAAM,CAAC;IAC1C,OAAO,GAAG,oBAAoB,KAAK,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;AAC9D,CAAC","sourcesContent":["import {AiSliceState} from '@sqlrooms/ai-core';\nimport {DuckDbSliceState} from '@sqlrooms/duckdb';\nimport {StoreApi} from '@sqlrooms/room-shell';\n\n/**\n * System prompt template for the AI assistant that provides instructions for:\n * - Using DuckDB-specific SQL syntax and functions\n * - Handling query results and error cases\n * - Creating visualizations with VegaLite\n * - Formatting final answers\n */\nconst DEFAULT_INSTRUCTIONS = `\nYou are analyzing tables in DuckDB database in the context of a room.\n\nInstructions for analysis:\n- When using 'query' tool, please assign parameter 'type' with value 'query'\n- Use DuckDB-specific SQL syntax and functions (not Oracle, PostgreSQL, or other SQL dialects)\n- Some key DuckDB-specific functions to use:\n  * regexp_matches() for regex (not regexp_like)\n  * strftime() for date formatting (not to_char)\n  * list_aggregate() for array operations\n  * unnest() for array expansion\n  * regr_sxy()\n  * corr()\n  * skewness()\n- Please always try to use SQL queries to answer users questions\n- Please run tool calls sequentially, don't run multiple tool calls in parallel\n- IMPORTANT: Do not list out raw query results in your response. Instead:\n  * Describe the results in natural language\n  * Provide summary statistics\n  * Use comparisons and relative terms\n  * Include only the most relevant values if necessary\n- Break down complex problems into smaller steps\n- Use \"SUMMARIZE table_name\"for quick overview of the table\n- Please don't modify data\n- IMPORTANT: When you receive an error response from a tool call (where success: false):\n  * Stop making any further tool calls immediately\n  * Return a final answer that includes the error message\n  * Explain what went wrong and suggest possible fixes if applicable\n\nWhen creating visualizations:\n- Follow VegaLite syntax\n- Choose appropriate chart types based on the data and analysis goals\n- Use clear titles and axis labels\n- Consider color schemes for better readability\n- Add meaningful tooltips when relevant\n- Format numbers and dates appropriately\n- Use aggregations when dealing with large datasets\n\nFor your final answer:\n- Provide an explanation for how you got it\n- Explain your reasoning step by step\n- Include relevant statistics or metrics\n- For each prompt, please always provide the final answer.\n- IMPORTANT: Query tool results may include sample rows (firstRows) or may be empty:\n  * If no sample rows provided: Never fabricate data. Direct users to the table component for actual results.\n  * If sample rows provided: Use them to enhance your analysis, but always direct users to the table component for complete results.\n\nPlease use the following schema for the tables:\n`;\n\n/**\n * Returns the default system instructions for the AI assistant\n */\nexport function createDefaultAiInstructions(\n  store: StoreApi<AiSliceState & DuckDbSliceState>,\n): string {\n  const tables = store.getState().db.tables;\n  return `${DEFAULT_INSTRUCTIONS}\\n${JSON.stringify(tables)}`;\n}\n"]}

package/dist/tools/defaultTools.d.ts

@@ -0,0 +1,14 @@
+import { AiSliceState, AiSliceTool } from '@sqlrooms/ai-core';
+import { DuckDbSliceState } from '@sqlrooms/duckdb';
+import { StoreApi } from '@sqlrooms/room-shell';
+import { QueryToolOptions } from './query/queryTool';
+export type DefaultToolsOptions = {
+    query?: QueryToolOptions;
+};
+/**
+ * Default tools available to the AI assistant for data analysis
+ * Includes:
+ * - query: Executes SQL queries against DuckDB
+ */
+export declare function createDefaultAiTools(store: StoreApi<AiSliceState & DuckDbSliceState>, options?: DefaultToolsOptions): Record<string, AiSliceTool>;
+//# sourceMappingURL=defaultTools.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"defaultTools.d.ts","sourceRoot":"","sources":["../../src/tools/defaultTools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,YAAY,EAAE,WAAW,EAAC,MAAM,mBAAmB,CAAC;AAC5D,OAAO,EAAC,gBAAgB,EAAC,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAC,QAAQ,EAAC,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAAkB,gBAAgB,EAAC,MAAM,mBAAmB,CAAC;AAEpE,MAAM,MAAM,mBAAmB,GAAG;IAChC,KAAK,CAAC,EAAE,gBAAgB,CAAC;CAC1B,CAAC;AAEF;;;;GAIG;AACH,wBAAgB,oBAAoB,CAClC,KAAK,EAAE,QAAQ,CAAC,YAAY,GAAG,gBAAgB,CAAC,EAChD,OAAO,CAAC,EAAE,mBAAmB,GAC5B,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAK7B"}

package/dist/tools/defaultTools.js

@@ -0,0 +1,13 @@
+import { createQueryTool } from './query/queryTool';
+/**
+ * Default tools available to the AI assistant for data analysis
+ * Includes:
+ * - query: Executes SQL queries against DuckDB
+ */
+export function createDefaultAiTools(store, options) {
+    const { query } = options || {};
+    return {
+        query: createQueryTool(store, query),
+    };
+}
+//# sourceMappingURL=defaultTools.js.map

package/dist/tools/defaultTools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"defaultTools.js","sourceRoot":"","sources":["../../src/tools/defaultTools.ts"],"names":[],"mappings":"AAGA,OAAO,EAAC,eAAe,EAAmB,MAAM,mBAAmB,CAAC;AAMpE;;;;GAIG;AACH,MAAM,UAAU,oBAAoB,CAClC,KAAgD,EAChD,OAA6B;IAE7B,MAAM,EAAC,KAAK,EAAC,GAAG,OAAO,IAAI,EAAE,CAAC;IAC9B,OAAO;QACL,KAAK,EAAE,eAAe,CAAC,KAAK,EAAE,KAAK,CAAC;KACrC,CAAC;AACJ,CAAC","sourcesContent":["import {AiSliceState, AiSliceTool} from '@sqlrooms/ai-core';\nimport {DuckDbSliceState} from '@sqlrooms/duckdb';\nimport {StoreApi} from '@sqlrooms/room-shell';\nimport {createQueryTool, QueryToolOptions} from './query/queryTool';\n\nexport type DefaultToolsOptions = {\n  query?: QueryToolOptions;\n};\n\n/**\n * Default tools available to the AI assistant for data analysis\n * Includes:\n * - query: Executes SQL queries against DuckDB\n */\nexport function createDefaultAiTools(\n  store: StoreApi<AiSliceState & DuckDbSliceState>,\n  options?: DefaultToolsOptions,\n): Record<string, AiSliceTool> {\n  const {query} = options || {};\n  return {\n    query: createQueryTool(store, query),\n  };\n}\n"]}

package/dist/tools/query/QueryToolResult.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryToolResult.d.ts","sourceRoot":"","sources":["../../../src/tools/query/QueryToolResult.tsx"],"names":[],"mappings":"AAEA,OAAO,KAAK,KAAK,MAAM,cAAc,CAAC;AAGtC,KAAK,oBAAoB,GAAG;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;IACjB,kEAAkE;IAClE,UAAU,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC;CAC1B,CAAC;AAEF,wBAAgB,eAAe,CAAC,KAAK,EAAE,oBAAoB,2CA2C1D"}

package/dist/tools/query/QueryToolResult.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryToolResult.js","sourceRoot":"","sources":["../../../src/tools/query/QueryToolResult.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAC,cAAc,EAAC,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAC,MAAM,EAAE,UAAU,EAAE,aAAa,EAAC,MAAM,cAAc,CAAC;AAE/D,OAAO,EAAC,SAAS,EAAC,MAAM,cAAc,CAAC;AASvC,MAAM,UAAU,eAAe,CAAC,KAA2B;IACzD,MAAM,EAAC,KAAK,EAAE,QAAQ,EAAC,GAAG,KAAK,CAAC;IAChC,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IACnC,OAAO,CACL,8BACE,eAAK,SAAS,EAAC,6GAA6G,aAC1H,cAAK,SAAS,EAAC,sCAAsC,YAAE,QAAQ,GAAO,EACtE,cAAK,SAAS,EAAC,wBAAwB,YACrC,KAAC,UAAU,IACT,IAAI,EAAE,QAAQ,EACd,OAAO,EAAC,OAAO,EACf,IAAI,EAAC,MAAM,EACX,SAAS,EAAC,SAAS,EACnB,SAAS,EAAC,UAAU,GACpB,GACE,IACF,EACN,cAAK,SAAS,EAAC,yBAAyB,YACtC,MAAC,MAAM,IAAC,OAAO,EAAC,OAAO,EAAC,IAAI,EAAC,IAAI,EAAC,OAAO,EAAE,UAAU,CAAC,MAAM,aAC1D,KAAC,SAAS,IAAC,SAAS,EAAC,SAAS,GAAG,EACjC,aAAI,SAAS,EAAC,SAAS,kCAAuB,IACvC,GACL,EAEL,YAAY,IAAI,KAAK,CAAC,CAAC,CAAC,CACvB,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CACjB,KAAC,cAAc,IACb,KAAK,EAAE,KAAK,EACZ,UAAU,EAAE,KAAK,CAAC,UAAU,EAC5B,UAAU,EAAE,UAAU,GACtB,CACH,CAAC,CAAC,CAAC,CACF,cAAK,SAAS,EAAC,aAAa,wBAAc,CAC3C,CACF,CAAC,CAAC,CAAC,CACF,KAAC,cAAc,IACb,KAAK,EAAE,KAAK,EACZ,KAAK,EAAE,QAAQ,EACf,UAAU,EAAE,UAAU,GACtB,CACH,IACA,CACJ,CAAC;AACJ,CAAC","sourcesContent":["import {DataTableModal} from '@sqlrooms/data-table';\nimport {Button, CopyButton, useDisclosure} from '@sqlrooms/ui';\nimport * as arrow from 'apache-arrow';\nimport {TableIcon} from 'lucide-react';\n\ntype QueryToolResultProps = {\n  title: string;\n  sqlQuery: string;\n  /** Provided in case the query result is already an arrow table */\n  arrowTable?: arrow.Table;\n};\n\nexport function QueryToolResult(props: QueryToolResultProps) {\n  const {title, sqlQuery} = props;\n  const tableModal = useDisclosure();\n  return (\n    <>\n      <div className=\"text-muted-foreground bg-muted relative max-h-[150px] w-full overflow-auto rounded-md p-2 font-mono text-xs\">\n        <pre className=\"whitespace-pre-wrap break-words pr-8\">{sqlQuery}</pre>\n        <div className=\"absolute right-1 top-1\">\n          <CopyButton\n            text={sqlQuery}\n            variant=\"ghost\"\n            size=\"icon\"\n            className=\"h-6 w-6\"\n            ariaLabel=\"Copy SQL\"\n          />\n        </div>\n      </div>\n      <div className=\"flex items-center gap-2\">\n        <Button variant=\"ghost\" size=\"xs\" onClick={tableModal.onOpen}>\n          <TableIcon className=\"h-4 w-4\" />\n          <h3 className=\"text-xs\">Show Query Result</h3>\n        </Button>\n      </div>\n\n      {'arrowTable' in props ? (\n        props.arrowTable ? (\n          <DataTableModal\n            title={title}\n            arrowTable={props.arrowTable}\n            tableModal={tableModal}\n          />\n        ) : (\n          <div className=\"p-4 text-xs\">No data</div>\n        )\n      ) : (\n        <DataTableModal\n          title={title}\n          query={sqlQuery}\n          tableModal={tableModal}\n        />\n      )}\n    </>\n  );\n}\n"]}

package/dist/tools/query/queryTool.d.ts

@@ -0,0 +1,25 @@
+import { AiSliceState, AiSliceTool } from '@sqlrooms/ai-core';
+import { DuckDbSliceState } from '@sqlrooms/duckdb';
+import type { StoreApi } from '@sqlrooms/room-shell';
+import { z } from 'zod';
+export declare const QueryToolParameters: z.ZodObject<{
+    type: z.ZodLiteral<"query">;
+    sqlQuery: z.ZodString;
+    reasoning: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    sqlQuery: string;
+    type: "query";
+    reasoning: string;
+}, {
+    sqlQuery: string;
+    type: "query";
+    reasoning: string;
+}>;
+export type QueryToolParameters = z.infer<typeof QueryToolParameters>;
+export type QueryToolOptions = {
+    readOnly?: boolean;
+    autoSummary?: boolean;
+    numberOfRowsToShareWithLLM?: number;
+};
+export declare function createQueryTool(store: StoreApi<AiSliceState & DuckDbSliceState>, options?: QueryToolOptions): AiSliceTool;
+//# sourceMappingURL=queryTool.d.ts.map

package/dist/tools/query/queryTool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"queryTool.d.ts","sourceRoot":"","sources":["../../../src/tools/query/queryTool.ts"],"names":[],"mappings":"AACA,OAAO,EAAC,YAAY,EAAE,WAAW,EAAC,MAAM,mBAAmB,CAAC;AAC5D,OAAO,EAGL,gBAAgB,EAEjB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EAAC,QAAQ,EAAC,MAAM,sBAAsB,CAAC;AACnD,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAGtB,eAAO,MAAM,mBAAmB;;;;;;;;;;;;EAI9B,CAAC;AACH,MAAM,MAAM,mBAAmB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAEtE,MAAM,MAAM,gBAAgB,GAAG;IAC7B,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,0BAA0B,CAAC,EAAE,MAAM,CAAC;CACrC,CAAC;AAEF,wBAAgB,eAAe,CAC7B,KAAK,EAAE,QAAQ,CAAC,YAAY,GAAG,gBAAgB,CAAC,EAChD,OAAO,CAAC,EAAE,gBAAgB,GACzB,WAAW,CA6Fb"}

package/dist/tools/query/queryTool.js

@@ -0,0 +1,108 @@
+import { extendedTool } from '@openassistant/utils';
+import { arrowTableToJson, splitSqlStatements, } from '@sqlrooms/duckdb';
+import { z } from 'zod';
+import { QueryToolResult } from './QueryToolResult';
+export const QueryToolParameters = z.object({
+    type: z.literal('query'),
+    sqlQuery: z.string(),
+    reasoning: z.string(),
+});
+export function createQueryTool(store, options) {
+    const { readOnly = true, autoSummary = false, numberOfRowsToShareWithLLM = 0, } = options || {};
+    return extendedTool({
+        description: `A tool for running SQL queries on the tables in the database.
+Please only run one query at a time.
+If a query fails, please don't try to run it again with the same syntax.`,
+        parameters: QueryToolParameters,
+        execute: async ({ type, sqlQuery }) => {
+            try {
+                const connector = await store.getState().db.getConnector();
+                // TODO use options.abortSignal: maybe call db.cancelPendingQuery
+                const result = await connector.query(sqlQuery);
+                const parsedQuery = await store.getState().db.sqlSelectToJson(sqlQuery);
+                if (parsedQuery.error &&
+                    // Only SELECT statements can be serialized to json, so we ignore not implemented errors
+                    parsedQuery.error_type !== 'not implemented') {
+                    throw new Error(parsedQuery.error_message);
+                }
+                if (readOnly) {
+                    if (parsedQuery.error) {
+                        throw new Error(`Query is not a valid SELECT statement: ${parsedQuery.error_message}`);
+                    }
+                    if (parsedQuery.statements.length !== 1 || // only one statement allowed
+                        parsedQuery.statements[0]?.node.type !== 'SELECT_NODE' // only SELECT statements allowed
+                    ) {
+                        throw new Error('Query is not a valid SELECT statement');
+                    }
+                }
+                const summaryData = await (async () => {
+                    if (!autoSummary)
+                        return null;
+                    if (parsedQuery.error)
+                        return null;
+                    const lastNode = parsedQuery.statements[parsedQuery.statements.length - 1]?.node;
+                    // Only get summary if the last statement isn't already a SUMMARIZE query
+                    if (lastNode?.type === 'SELECT_NODE' &&
+                        lastNode?.from_table?.show_type === 'SUMMARY') {
+                        return arrowTableToJson(result);
+                    }
+                    const statements = splitSqlStatements(sqlQuery);
+                    const lastStatement = statements[statements.length - 1];
+                    if (!lastStatement)
+                        return null;
+                    return await getQuerySummary(connector, lastStatement);
+                })();
+                // Conditionally get rows of the result as a json object based on numberOfRowsToShareWithLLM
+                const firstRows = numberOfRowsToShareWithLLM > 0
+                    ? arrowTableToJson(result.slice(0, numberOfRowsToShareWithLLM))
+                    : [];
+                return {
+                    llmResult: {
+                        success: true,
+                        data: {
+                            type,
+                            summary: summaryData,
+                            ...(numberOfRowsToShareWithLLM > 0 ? { firstRows } : {}),
+                        },
+                    },
+                    additionalData: {
+                        title: 'Query Result',
+                        sqlQuery,
+                    },
+                };
+            }
+            catch (error) {
+                return {
+                    llmResult: {
+                        success: false,
+                        details: 'Query execution failed.',
+                        errorMessage: error instanceof Error ? error.message : 'Unknown error',
+                    },
+                };
+            }
+        },
+        component: QueryToolResult,
+    });
+}
+/**
+ * Generates summary statistics for a SQL query result
+ * @param connector - DuckDB connection instance
+ * @param sqlQuery - SQL SELECT query to analyze
+ * @returns Summary statistics as JSON object, or null if the query is not a SELECT statement or if summary generation fails
+ */
+async function getQuerySummary(connector, sqlQuery) {
+    if (!sqlQuery.toLowerCase().trim().startsWith('select')) {
+        return null;
+    }
+    try {
+        const summaryResult = await connector.query(`SUMMARIZE (
+      ${sqlQuery}
+    )`);
+        return arrowTableToJson(summaryResult);
+    }
+    catch (error) {
+        console.warn('Failed to get summary for query. Error:', error);
+        return null;
+    }
+}
+//# sourceMappingURL=queryTool.js.map

package/dist/tools/query/queryTool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"queryTool.js","sourceRoot":"","sources":["../../../src/tools/query/queryTool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,YAAY,EAAC,MAAM,sBAAsB,CAAC;AAElD,OAAO,EACL,gBAAgB,EAGhB,kBAAkB,GACnB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EAAC,eAAe,EAAC,MAAM,mBAAmB,CAAC;AAElD,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC1C,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;IACxB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAC,CAAC;AASH,MAAM,UAAU,eAAe,CAC7B,KAAgD,EAChD,OAA0B;IAE1B,MAAM,EACJ,QAAQ,GAAG,IAAI,EACf,WAAW,GAAG,KAAK,EACnB,0BAA0B,GAAG,CAAC,GAC/B,GAAG,OAAO,IAAI,EAAE,CAAC;IAClB,OAAO,YAAY,CAAC;QAClB,WAAW,EAAE;;yEAEwD;QACrE,UAAU,EAAE,mBAAmB;QAC/B,OAAO,EAAE,KAAK,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC,EAAE,EAAE;YAClC,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC;gBAC3D,iEAAiE;gBACjE,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;gBAE/C,MAAM,WAAW,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;gBAExE,IACE,WAAW,CAAC,KAAK;oBACjB,wFAAwF;oBACxF,WAAW,CAAC,UAAU,KAAK,iBAAiB,EAC5C,CAAC;oBACD,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;gBAC7C,CAAC;gBAED,IAAI,QAAQ,EAAE,CAAC;oBACb,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC;wBACtB,MAAM,IAAI,KAAK,CACb,0CAA0C,WAAW,CAAC,aAAa,EAAE,CACtE,CAAC;oBACJ,CAAC;oBACD,IACE,WAAW,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,IAAI,6BAA6B;wBACpE,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,KAAK,aAAa,CAAC,iCAAiC;sBACxF,CAAC;wBACD,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;oBAC3D,CAAC;gBACH,CAAC;gBAED,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,IAAI,EAAE;oBACpC,IAAI,CAAC,WAAW;wBAAE,OAAO,IAAI,CAAC;oBAC9B,IAAI,WAAW,CAAC,KAAK;wBAAE,OAAO,IAAI,CAAC;oBACnC,MAAM,QAAQ,GACZ,WAAW,CAAC,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC;oBAElE,yEAAyE;oBACzE,IACE,QAAQ,EAAE,IAAI,KAAK,aAAa;wBAChC,QAAQ,EAAE,UAAU,EAAE,SAAS,KAAK,SAAS,EAC7C,CAAC;wBACD,OAAO,gBAAgB,CAAC,MAAM,CAAC,CAAC;oBAClC,CAAC;oBACD,MAAM,UAAU,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAC;oBAChD,MAAM,aAAa,GAAG,UAAU,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBACxD,IAAI,CAAC,aAAa;wBAAE,OAAO,IAAI,CAAC;oBAChC,OAAO,MAAM,eAAe,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;gBACzD,CAAC,CAAC,EAAE,CAAC;gBAEL,4FAA4F;gBAC5F,MAAM,SAAS,GACb,0BAA0B,GAAG,CAAC;oBAC5B,CAAC,CAAC,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,0BAA0B,CAAC,CAAC;oBAC/D,CAAC,CAAC,EAAE,CAAC;gBAET,OAAO;oBACL,SAAS,EAAE;wBACT,OAAO,EAAE,IAAI;wBACb,IAAI,EAAE;4BACJ,IAAI;4BACJ,OAAO,EAAE,WAAW;4BACpB,GAAG,CAAC,0BAA0B,GAAG,CAAC,CAAC,CAAC,CAAC,EAAC,SAAS,EAAC,CAAC,CAAC,CAAC,EAAE,CAAC;yBACvD;qBACF;oBACD,cAAc,EAAE;wBACd,KAAK,EAAE,cAAc;wBACrB,QAAQ;qBACT;iBACF,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,SAAS,EAAE;wBACT,OAAO,EAAE,KAAK;wBACd,OAAO,EAAE,yBAAyB;wBAClC,YAAY,EACV,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;qBAC3D;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;QACD,SAAS,EAAE,eAAe;KAC3B,CAAC,CAAC;AACL,CAAC;AAED;;;;;GAKG;AACH,KAAK,UAAU,eAAe,CAAC,SAA0B,EAAE,QAAgB;IACzE,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACxD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,CAAC;QACH,MAAM,aAAa,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC;QACxC,QAAQ;MACV,CAAC,CAAC;QACJ,OAAO,gBAAgB,CAAC,aAAa,CAAC,CAAC;IACzC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,IAAI,CAAC,yCAAyC,EAAE,KAAK,CAAC,CAAC;QAC/D,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC","sourcesContent":["import {extendedTool} from '@openassistant/utils';\nimport {AiSliceState, AiSliceTool} from '@sqlrooms/ai-core';\nimport {\n  arrowTableToJson,\n  DuckDbConnector,\n  DuckDbSliceState,\n  splitSqlStatements,\n} from '@sqlrooms/duckdb';\nimport type {StoreApi} from '@sqlrooms/room-shell';\nimport {z} from 'zod';\nimport {QueryToolResult} from './QueryToolResult';\n\nexport const QueryToolParameters = z.object({\n  type: z.literal('query'),\n  sqlQuery: z.string(),\n  reasoning: z.string(),\n});\nexport type QueryToolParameters = z.infer<typeof QueryToolParameters>;\n\nexport type QueryToolOptions = {\n  readOnly?: boolean;\n  autoSummary?: boolean;\n  numberOfRowsToShareWithLLM?: number;\n};\n\nexport function createQueryTool(\n  store: StoreApi<AiSliceState & DuckDbSliceState>,\n  options?: QueryToolOptions,\n): AiSliceTool {\n  const {\n    readOnly = true,\n    autoSummary = false,\n    numberOfRowsToShareWithLLM = 0,\n  } = options || {};\n  return extendedTool({\n    description: `A tool for running SQL queries on the tables in the database.\nPlease only run one query at a time.\nIf a query fails, please don't try to run it again with the same syntax.`,\n    parameters: QueryToolParameters,\n    execute: async ({type, sqlQuery}) => {\n      try {\n        const connector = await store.getState().db.getConnector();\n        // TODO use options.abortSignal: maybe call db.cancelPendingQuery\n        const result = await connector.query(sqlQuery);\n\n        const parsedQuery = await store.getState().db.sqlSelectToJson(sqlQuery);\n\n        if (\n          parsedQuery.error &&\n          // Only SELECT statements can be serialized to json, so we ignore not implemented errors\n          parsedQuery.error_type !== 'not implemented'\n        ) {\n          throw new Error(parsedQuery.error_message);\n        }\n\n        if (readOnly) {\n          if (parsedQuery.error) {\n            throw new Error(\n              `Query is not a valid SELECT statement: ${parsedQuery.error_message}`,\n            );\n          }\n          if (\n            parsedQuery.statements.length !== 1 || // only one statement allowed\n            parsedQuery.statements[0]?.node.type !== 'SELECT_NODE' // only SELECT statements allowed\n          ) {\n            throw new Error('Query is not a valid SELECT statement');\n          }\n        }\n\n        const summaryData = await (async () => {\n          if (!autoSummary) return null;\n          if (parsedQuery.error) return null;\n          const lastNode =\n            parsedQuery.statements[parsedQuery.statements.length - 1]?.node;\n\n          // Only get summary if the last statement isn't already a SUMMARIZE query\n          if (\n            lastNode?.type === 'SELECT_NODE' &&\n            lastNode?.from_table?.show_type === 'SUMMARY'\n          ) {\n            return arrowTableToJson(result);\n          }\n          const statements = splitSqlStatements(sqlQuery);\n          const lastStatement = statements[statements.length - 1];\n          if (!lastStatement) return null;\n          return await getQuerySummary(connector, lastStatement);\n        })();\n\n        // Conditionally get rows of the result as a json object based on numberOfRowsToShareWithLLM\n        const firstRows =\n          numberOfRowsToShareWithLLM > 0\n            ? arrowTableToJson(result.slice(0, numberOfRowsToShareWithLLM))\n            : [];\n\n        return {\n          llmResult: {\n            success: true,\n            data: {\n              type,\n              summary: summaryData,\n              ...(numberOfRowsToShareWithLLM > 0 ? {firstRows} : {}),\n            },\n          },\n          additionalData: {\n            title: 'Query Result',\n            sqlQuery,\n          },\n        };\n      } catch (error) {\n        return {\n          llmResult: {\n            success: false,\n            details: 'Query execution failed.',\n            errorMessage:\n              error instanceof Error ? error.message : 'Unknown error',\n          },\n        };\n      }\n    },\n    component: QueryToolResult,\n  });\n}\n\n/**\n * Generates summary statistics for a SQL query result\n * @param connector - DuckDB connection instance\n * @param sqlQuery - SQL SELECT query to analyze\n * @returns Summary statistics as JSON object, or null if the query is not a SELECT statement or if summary generation fails\n */\nasync function getQuerySummary(connector: DuckDbConnector, sqlQuery: string) {\n  if (!sqlQuery.toLowerCase().trim().startsWith('select')) {\n    return null;\n  }\n\n  try {\n    const summaryResult = await connector.query(`SUMMARIZE (\n      ${sqlQuery}\n    )`);\n    return arrowTableToJson(summaryResult);\n  } catch (error) {\n    console.warn('Failed to get summary for query. Error:', error);\n    return null;\n  }\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sqlrooms/ai",
-  "version": "0.25.0-rc.0",
+  "version": "0.26.0-rc.1",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "module": "dist/index.js",
@@ -18,26 +18,21 @@
     "access": "public"
   },
   "dependencies": {
-    "@ai-sdk/provider": "^1.1.3",
-    "@openassistant/core": "0.5.17",
-    "@openassistant/utils": "0.5.17",
-    "@paralleldrive/cuid2": "^2.2.2",
-    "@sqlrooms/data-table": "0.25.0-rc.0",
-    "@sqlrooms/duckdb": "0.25.0-rc.0",
-    "@sqlrooms/monaco-editor": "0.25.0-rc.0",
-    "@sqlrooms/recharts": "0.25.0-rc.0",
-    "@sqlrooms/room-config": "0.25.0-rc.0",
-    "@sqlrooms/room-shell": "0.25.0-rc.0",
-    "@sqlrooms/ui": "0.25.0-rc.0",
-    "@sqlrooms/utils": "0.25.0-rc.0",
+    "@sqlrooms/ai-config": "0.26.0-rc.1",
+    "@sqlrooms/ai-core": "0.26.0-rc.1",
+    "@sqlrooms/ai-settings": "0.26.0-rc.1",
+    "@sqlrooms/duckdb": "0.26.0-rc.1",
+    "@sqlrooms/room-shell": "0.26.0-rc.1",
+    "@sqlrooms/ui": "0.26.0-rc.1",
+    "@sqlrooms/utils": "0.26.0-rc.1",
     "ai": "^4.3.19",
-    "immer": "^10.1.1",
-    "lucide-react": "^0.475.0",
+    "immer": "^10.1.3",
+    "lucide-react": "^0.544.0",
     "react-markdown": "^10.1.0",
     "recharts": "^2.12.7",
     "rehype-raw": "^7.0.0",
     "remark-gfm": "^4.0.0",
-    "zod": "^3.25.73"
+    "zod": "^3.25.76"
   },
   "peerDependencies": {
     "apache-arrow": "17.0.0",
@@ -51,5 +46,5 @@
     "typecheck": "tsc --noEmit",
     "typedoc": "typedoc"
   },
-  "gitHead": "063d7608fe608031c690ea3b2e67bb1f4e25a8dd"
-}
+  "gitHead": "803cf5fa875f7099746a48db15a3111aee5c0359"
+}