@sqlrooms/ai 0.7.0 → 0.8.0

package/dist/AiSlice.d.ts

@@ -1,374 +1,556 @@
+import { ExtendedTool } from '@openassistant/core';
 import { ProjectState, type StateCreator } from '@sqlrooms/project-builder';
 import { BaseProjectConfig } from '@sqlrooms/project-config';
-import { CoreAssistantMessage, CoreToolMessage, CoreUserMessage } from 'ai';
 import { z } from 'zod';
-type AiMessage = (CoreToolMessage | CoreAssistantMessage | CoreUserMessage) & {
-    id: string;
-};
+import { AnalysisSessionSchema } from './schemas';
+import { DataTable } from '@sqlrooms/duckdb';
 export declare const AiSliceConfig: z.ZodObject<{
     ai: z.ZodObject<{
-        modelProvider: z.ZodString;
-        model: z.ZodString;
-        analysisResults: z.ZodArray<z.ZodObject<{
+        sessions: z.ZodArray<z.ZodObject<{
             id: z.ZodString;
-            prompt: z.ZodString;
-            toolResults: z.ZodArray<z.ZodObject<{
-                toolName: z.ZodString;
-                toolCallId: z.ZodString;
-                args: z.ZodRecord<z.ZodString, z.ZodAny>;
-                result: z.ZodUnion<[z.ZodObject<{
-                    success: z.ZodLiteral<false>;
+            name: z.ZodString;
+            modelProvider: z.ZodString;
+            model: z.ZodString;
+            analysisResults: z.ZodArray<z.ZodObject<{
+                id: z.ZodString;
+                prompt: z.ZodString;
+                streamMessage: z.ZodObject<{
+                    reasoning: z.ZodOptional<z.ZodString>;
+                    toolCallMessages: z.ZodOptional<z.ZodArray<z.ZodObject<{
+                        toolName: z.ZodString;
+                        toolCallId: z.ZodString;
+                        args: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+                        isCompleted: z.ZodBoolean;
+                        llmResult: z.ZodOptional<z.ZodUnknown>;
+                        additionalData: z.ZodOptional<z.ZodUnknown>;
+                        text: z.ZodOptional<z.ZodString>;
+                    }, "strip", z.ZodTypeAny, {
+                        toolName: string;
+                        toolCallId: string;
+                        args: Record<string, unknown>;
+                        isCompleted: boolean;
+                        text?: string | undefined;
+                        llmResult?: unknown;
+                        additionalData?: unknown;
+                    }, {
+                        toolName: string;
+                        toolCallId: string;
+                        args: Record<string, unknown>;
+                        isCompleted: boolean;
+                        text?: string | undefined;
+                        llmResult?: unknown;
+                        additionalData?: unknown;
+                    }>, "many">>;
+                    analysis: z.ZodOptional<z.ZodString>;
+                    text: z.ZodOptional<z.ZodString>;
+                    parts: z.ZodOptional<z.ZodArray<z.ZodUnion<[z.ZodObject<{
+                        type: z.ZodLiteral<"text">;
+                        text: z.ZodString;
+                    }, "strip", z.ZodTypeAny, {
+                        text: string;
+                        type: "text";
+                    }, {
+                        text: string;
+                        type: "text";
+                    }>, z.ZodObject<{
+                        type: z.ZodLiteral<"tool">;
+                        toolCallMessages: z.ZodArray<z.ZodObject<{
+                            toolName: z.ZodString;
+                            toolCallId: z.ZodString;
+                            args: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+                            isCompleted: z.ZodBoolean;
+                            llmResult: z.ZodOptional<z.ZodUnknown>;
+                            additionalData: z.ZodOptional<z.ZodUnknown>;
+                            text: z.ZodOptional<z.ZodString>;
+                        }, "strip", z.ZodTypeAny, {
+                            toolName: string;
+                            toolCallId: string;
+                            args: Record<string, unknown>;
+                            isCompleted: boolean;
+                            text?: string | undefined;
+                            llmResult?: unknown;
+                            additionalData?: unknown;
+                        }, {
+                            toolName: string;
+                            toolCallId: string;
+                            args: Record<string, unknown>;
+                            isCompleted: boolean;
+                            text?: string | undefined;
+                            llmResult?: unknown;
+                            additionalData?: unknown;
+                        }>, "many">;
+                    }, "strip", z.ZodTypeAny, {
+                        type: "tool";
+                        toolCallMessages: {
+                            toolName: string;
+                            toolCallId: string;
+                            args: Record<string, unknown>;
+                            isCompleted: boolean;
+                            text?: string | undefined;
+                            llmResult?: unknown;
+                            additionalData?: unknown;
+                        }[];
+                    }, {
+                        type: "tool";
+                        toolCallMessages: {
+                            toolName: string;
+                            toolCallId: string;
+                            args: Record<string, unknown>;
+                            isCompleted: boolean;
+                            text?: string | undefined;
+                            llmResult?: unknown;
+                            additionalData?: unknown;
+                        }[];
+                    }>]>, "many">>;
+                }, "strip", z.ZodTypeAny, {
+                    text?: string | undefined;
+                    toolCallMessages?: {
+                        toolName: string;
+                        toolCallId: string;
+                        args: Record<string, unknown>;
+                        isCompleted: boolean;
+                        text?: string | undefined;
+                        llmResult?: unknown;
+                        additionalData?: unknown;
+                    }[] | undefined;
+                    reasoning?: string | undefined;
+                    analysis?: string | undefined;
+                    parts?: ({
+                        text: string;
+                        type: "text";
+                    } | {
+                        type: "tool";
+                        toolCallMessages: {
+                            toolName: string;
+                            toolCallId: string;
+                            args: Record<string, unknown>;
+                            isCompleted: boolean;
+                            text?: string | undefined;
+                            llmResult?: unknown;
+                            additionalData?: unknown;
+                        }[];
+                    })[] | undefined;
+                }, {
+                    text?: string | undefined;
+                    toolCallMessages?: {
+                        toolName: string;
+                        toolCallId: string;
+                        args: Record<string, unknown>;
+                        isCompleted: boolean;
+                        text?: string | undefined;
+                        llmResult?: unknown;
+                        additionalData?: unknown;
+                    }[] | undefined;
+                    reasoning?: string | undefined;
+                    analysis?: string | undefined;
+                    parts?: ({
+                        text: string;
+                        type: "text";
+                    } | {
+                        type: "tool";
+                        toolCallMessages: {
+                            toolName: string;
+                            toolCallId: string;
+                            args: Record<string, unknown>;
+                            isCompleted: boolean;
+                            text?: string | undefined;
+                            llmResult?: unknown;
+                            additionalData?: unknown;
+                        }[];
+                    })[] | undefined;
+                }>;
+                errorMessage: z.ZodOptional<z.ZodObject<{
                     error: z.ZodString;
                 }, "strip", z.ZodTypeAny, {
-                    success: false;
                     error: string;
                 }, {
-                    success: false;
                     error: string;
-                }>, z.ZodObject<{
-                    success: z.ZodLiteral<true>;
-                    data: z.ZodRecord<z.ZodString, z.ZodAny>;
-                }, "strip", z.ZodTypeAny, {
-                    success: true;
-                    data: Record<string, any>;
-                }, {
-                    success: true;
-                    data: Record<string, any>;
-                }>]>;
+                }>>;
+                isCompleted: z.ZodBoolean;
             }, "strip", z.ZodTypeAny, {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, any>;
-                result: {
-                    success: false;
-                    error: string;
-                } | {
-                    success: true;
-                    data: Record<string, any>;
+                id: string;
+                prompt: string;
+                streamMessage: {
+                    text?: string | undefined;
+                    toolCallMessages?: {
+                        toolName: string;
+                        toolCallId: string;
+                        args: Record<string, unknown>;
+                        isCompleted: boolean;
+                        text?: string | undefined;
+                        llmResult?: unknown;
+                        additionalData?: unknown;
+                    }[] | undefined;
+                    reasoning?: string | undefined;
+                    analysis?: string | undefined;
+                    parts?: ({
+                        text: string;
+                        type: "text";
+                    } | {
+                        type: "tool";
+                        toolCallMessages: {
+                            toolName: string;
+                            toolCallId: string;
+                            args: Record<string, unknown>;
+                            isCompleted: boolean;
+                            text?: string | undefined;
+                            llmResult?: unknown;
+                            additionalData?: unknown;
+                        }[];
+                    })[] | undefined;
                 };
-            }, {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, any>;
-                result: {
-                    success: false;
+                isCompleted: boolean;
+                errorMessage?: {
                     error: string;
-                } | {
-                    success: true;
-                    data: Record<string, any>;
-                };
-            }>, "many">;
-            toolCalls: z.ZodArray<z.ZodObject<{
-                toolName: z.ZodString;
-                toolCallId: z.ZodString;
-                args: z.ZodUnion<[z.ZodObject<{
-                    type: z.ZodLiteral<"query">;
-                    sqlQuery: z.ZodString;
-                    reasoning: z.ZodString;
-                }, "strip", z.ZodTypeAny, {
-                    type: "query";
-                    sqlQuery: string;
-                    reasoning: string;
-                }, {
-                    type: "query";
-                    sqlQuery: string;
-                    reasoning: string;
-                }>, z.ZodObject<{
-                    sqlQuery: z.ZodString;
-                    vegaLiteSpec: z.ZodString;
-                    reasoning: z.ZodString;
-                }, "strip", z.ZodTypeAny, {
-                    sqlQuery: string;
-                    reasoning: string;
-                    vegaLiteSpec: string;
-                }, {
-                    sqlQuery: string;
-                    reasoning: string;
-                    vegaLiteSpec: string;
-                }>]>;
-            }, "strip", z.ZodTypeAny, {
-                toolName: string;
-                toolCallId: string;
-                args: {
-                    type: "query";
-                    sqlQuery: string;
-                    reasoning: string;
-                } | {
-                    sqlQuery: string;
-                    reasoning: string;
-                    vegaLiteSpec: string;
-                };
+                } | undefined;
             }, {
-                toolName: string;
-                toolCallId: string;
-                args: {
-                    type: "query";
-                    sqlQuery: string;
-                    reasoning: string;
-                } | {
-                    sqlQuery: string;
-                    reasoning: string;
-                    vegaLiteSpec: string;
+                id: string;
+                prompt: string;
+                streamMessage: {
+                    text?: string | undefined;
+                    toolCallMessages?: {
+                        toolName: string;
+                        toolCallId: string;
+                        args: Record<string, unknown>;
+                        isCompleted: boolean;
+                        text?: string | undefined;
+                        llmResult?: unknown;
+                        additionalData?: unknown;
+                    }[] | undefined;
+                    reasoning?: string | undefined;
+                    analysis?: string | undefined;
+                    parts?: ({
+                        text: string;
+                        type: "text";
+                    } | {
+                        type: "tool";
+                        toolCallMessages: {
+                            toolName: string;
+                            toolCallId: string;
+                            args: Record<string, unknown>;
+                            isCompleted: boolean;
+                            text?: string | undefined;
+                            llmResult?: unknown;
+                            additionalData?: unknown;
+                        }[];
+                    })[] | undefined;
                 };
+                isCompleted: boolean;
+                errorMessage?: {
+                    error: string;
+                } | undefined;
             }>, "many">;
-            toolCallMessages: z.ZodArray<z.ZodObject<{
-                toolCallId: z.ZodString;
-                element: z.ZodAny;
-            }, "strip", z.ZodTypeAny, {
-                toolCallId: string;
-                element?: any;
-            }, {
-                toolCallId: string;
-                element?: any;
-            }>, "many">;
-            analysis: z.ZodString;
-            isCompleted: z.ZodBoolean;
+            createdAt: z.ZodOptional<z.ZodDate>;
         }, "strip", z.ZodTypeAny, {
             id: string;
-            prompt: string;
-            toolResults: {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, any>;
-                result: {
-                    success: false;
-                    error: string;
-                } | {
-                    success: true;
-                    data: Record<string, any>;
-                };
-            }[];
-            toolCalls: {
-                toolName: string;
-                toolCallId: string;
-                args: {
-                    type: "query";
-                    sqlQuery: string;
-                    reasoning: string;
-                } | {
-                    sqlQuery: string;
-                    reasoning: string;
-                    vegaLiteSpec: string;
+            name: string;
+            modelProvider: string;
+            model: string;
+            analysisResults: {
+                id: string;
+                prompt: string;
+                streamMessage: {
+                    text?: string | undefined;
+                    toolCallMessages?: {
+                        toolName: string;
+                        toolCallId: string;
+                        args: Record<string, unknown>;
+                        isCompleted: boolean;
+                        text?: string | undefined;
+                        llmResult?: unknown;
+                        additionalData?: unknown;
+                    }[] | undefined;
+                    reasoning?: string | undefined;
+                    analysis?: string | undefined;
+                    parts?: ({
+                        text: string;
+                        type: "text";
+                    } | {
+                        type: "tool";
+                        toolCallMessages: {
+                            toolName: string;
+                            toolCallId: string;
+                            args: Record<string, unknown>;
+                            isCompleted: boolean;
+                            text?: string | undefined;
+                            llmResult?: unknown;
+                            additionalData?: unknown;
+                        }[];
+                    })[] | undefined;
                 };
+                isCompleted: boolean;
+                errorMessage?: {
+                    error: string;
+                } | undefined;
             }[];
-            toolCallMessages: {
-                toolCallId: string;
-                element?: any;
-            }[];
-            analysis: string;
-            isCompleted: boolean;
+            createdAt?: Date | undefined;
         }, {
             id: string;
-            prompt: string;
-            toolResults: {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, any>;
-                result: {
-                    success: false;
-                    error: string;
-                } | {
-                    success: true;
-                    data: Record<string, any>;
+            name: string;
+            modelProvider: string;
+            model: string;
+            analysisResults: {
+                id: string;
+                prompt: string;
+                streamMessage: {
+                    text?: string | undefined;
+                    toolCallMessages?: {
+                        toolName: string;
+                        toolCallId: string;
+                        args: Record<string, unknown>;
+                        isCompleted: boolean;
+                        text?: string | undefined;
+                        llmResult?: unknown;
+                        additionalData?: unknown;
+                    }[] | undefined;
+                    reasoning?: string | undefined;
+                    analysis?: string | undefined;
+                    parts?: ({
+                        text: string;
+                        type: "text";
+                    } | {
+                        type: "tool";
+                        toolCallMessages: {
+                            toolName: string;
+                            toolCallId: string;
+                            args: Record<string, unknown>;
+                            isCompleted: boolean;
+                            text?: string | undefined;
+                            llmResult?: unknown;
+                            additionalData?: unknown;
+                        }[];
+                    })[] | undefined;
                 };
+                isCompleted: boolean;
+                errorMessage?: {
+                    error: string;
+                } | undefined;
             }[];
-            toolCalls: {
-                toolName: string;
-                toolCallId: string;
-                args: {
-                    type: "query";
-                    sqlQuery: string;
-                    reasoning: string;
-                } | {
-                    sqlQuery: string;
-                    reasoning: string;
-                    vegaLiteSpec: string;
-                };
-            }[];
-            toolCallMessages: {
-                toolCallId: string;
-                element?: any;
-            }[];
-            analysis: string;
-            isCompleted: boolean;
+            createdAt?: Date | undefined;
         }>, "many">;
+        currentSessionId: z.ZodOptional<z.ZodString>;
     }, "strip", z.ZodTypeAny, {
-        modelProvider: string;
-        model: string;
-        analysisResults: {
+        sessions: {
             id: string;
-            prompt: string;
-            toolResults: {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, any>;
-                result: {
-                    success: false;
-                    error: string;
-                } | {
-                    success: true;
-                    data: Record<string, any>;
-                };
-            }[];
-            toolCalls: {
-                toolName: string;
-                toolCallId: string;
-                args: {
-                    type: "query";
-                    sqlQuery: string;
-                    reasoning: string;
-                } | {
-                    sqlQuery: string;
-                    reasoning: string;
-                    vegaLiteSpec: string;
+            name: string;
+            modelProvider: string;
+            model: string;
+            analysisResults: {
+                id: string;
+                prompt: string;
+                streamMessage: {
+                    text?: string | undefined;
+                    toolCallMessages?: {
+                        toolName: string;
+                        toolCallId: string;
+                        args: Record<string, unknown>;
+                        isCompleted: boolean;
+                        text?: string | undefined;
+                        llmResult?: unknown;
+                        additionalData?: unknown;
+                    }[] | undefined;
+                    reasoning?: string | undefined;
+                    analysis?: string | undefined;
+                    parts?: ({
+                        text: string;
+                        type: "text";
+                    } | {
+                        type: "tool";
+                        toolCallMessages: {
+                            toolName: string;
+                            toolCallId: string;
+                            args: Record<string, unknown>;
+                            isCompleted: boolean;
+                            text?: string | undefined;
+                            llmResult?: unknown;
+                            additionalData?: unknown;
+                        }[];
+                    })[] | undefined;
                 };
+                isCompleted: boolean;
+                errorMessage?: {
+                    error: string;
+                } | undefined;
             }[];
-            toolCallMessages: {
-                toolCallId: string;
-                element?: any;
-            }[];
-            analysis: string;
-            isCompleted: boolean;
+            createdAt?: Date | undefined;
         }[];
+        currentSessionId?: string | undefined;
     }, {
-        modelProvider: string;
-        model: string;
-        analysisResults: {
+        sessions: {
             id: string;
-            prompt: string;
-            toolResults: {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, any>;
-                result: {
-                    success: false;
-                    error: string;
-                } | {
-                    success: true;
-                    data: Record<string, any>;
-                };
-            }[];
-            toolCalls: {
-                toolName: string;
-                toolCallId: string;
-                args: {
-                    type: "query";
-                    sqlQuery: string;
-                    reasoning: string;
-                } | {
-                    sqlQuery: string;
-                    reasoning: string;
-                    vegaLiteSpec: string;
+            name: string;
+            modelProvider: string;
+            model: string;
+            analysisResults: {
+                id: string;
+                prompt: string;
+                streamMessage: {
+                    text?: string | undefined;
+                    toolCallMessages?: {
+                        toolName: string;
+                        toolCallId: string;
+                        args: Record<string, unknown>;
+                        isCompleted: boolean;
+                        text?: string | undefined;
+                        llmResult?: unknown;
+                        additionalData?: unknown;
+                    }[] | undefined;
+                    reasoning?: string | undefined;
+                    analysis?: string | undefined;
+                    parts?: ({
+                        text: string;
+                        type: "text";
+                    } | {
+                        type: "tool";
+                        toolCallMessages: {
+                            toolName: string;
+                            toolCallId: string;
+                            args: Record<string, unknown>;
+                            isCompleted: boolean;
+                            text?: string | undefined;
+                            llmResult?: unknown;
+                            additionalData?: unknown;
+                        }[];
+                    })[] | undefined;
                 };
+                isCompleted: boolean;
+                errorMessage?: {
+                    error: string;
+                } | undefined;
             }[];
-            toolCallMessages: {
-                toolCallId: string;
-                element?: any;
-            }[];
-            analysis: string;
-            isCompleted: boolean;
+            createdAt?: Date | undefined;
         }[];
+        currentSessionId?: string | undefined;
     }>;
 }, "strip", z.ZodTypeAny, {
     ai: {
-        modelProvider: string;
-        model: string;
-        analysisResults: {
+        sessions: {
             id: string;
-            prompt: string;
-            toolResults: {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, any>;
-                result: {
-                    success: false;
-                    error: string;
-                } | {
-                    success: true;
-                    data: Record<string, any>;
+            name: string;
+            modelProvider: string;
+            model: string;
+            analysisResults: {
+                id: string;
+                prompt: string;
+                streamMessage: {
+                    text?: string | undefined;
+                    toolCallMessages?: {
+                        toolName: string;
+                        toolCallId: string;
+                        args: Record<string, unknown>;
+                        isCompleted: boolean;
+                        text?: string | undefined;
+                        llmResult?: unknown;
+                        additionalData?: unknown;
+                    }[] | undefined;
+                    reasoning?: string | undefined;
+                    analysis?: string | undefined;
+                    parts?: ({
+                        text: string;
+                        type: "text";
+                    } | {
+                        type: "tool";
+                        toolCallMessages: {
+                            toolName: string;
+                            toolCallId: string;
+                            args: Record<string, unknown>;
+                            isCompleted: boolean;
+                            text?: string | undefined;
+                            llmResult?: unknown;
+                            additionalData?: unknown;
+                        }[];
+                    })[] | undefined;
                 };
+                isCompleted: boolean;
+                errorMessage?: {
+                    error: string;
+                } | undefined;
             }[];
-            toolCalls: {
-                toolName: string;
-                toolCallId: string;
-                args: {
-                    type: "query";
-                    sqlQuery: string;
-                    reasoning: string;
-                } | {
-                    sqlQuery: string;
-                    reasoning: string;
-                    vegaLiteSpec: string;
-                };
-            }[];
-            toolCallMessages: {
-                toolCallId: string;
-                element?: any;
-            }[];
-            analysis: string;
-            isCompleted: boolean;
+            createdAt?: Date | undefined;
         }[];
+        currentSessionId?: string | undefined;
     };
 }, {
     ai: {
-        modelProvider: string;
-        model: string;
-        analysisResults: {
+        sessions: {
             id: string;
-            prompt: string;
-            toolResults: {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, any>;
-                result: {
-                    success: false;
-                    error: string;
-                } | {
-                    success: true;
-                    data: Record<string, any>;
-                };
-            }[];
-            toolCalls: {
-                toolName: string;
-                toolCallId: string;
-                args: {
-                    type: "query";
-                    sqlQuery: string;
-                    reasoning: string;
-                } | {
-                    sqlQuery: string;
-                    reasoning: string;
-                    vegaLiteSpec: string;
+            name: string;
+            modelProvider: string;
+            model: string;
+            analysisResults: {
+                id: string;
+                prompt: string;
+                streamMessage: {
+                    text?: string | undefined;
+                    toolCallMessages?: {
+                        toolName: string;
+                        toolCallId: string;
+                        args: Record<string, unknown>;
+                        isCompleted: boolean;
+                        text?: string | undefined;
+                        llmResult?: unknown;
+                        additionalData?: unknown;
+                    }[] | undefined;
+                    reasoning?: string | undefined;
+                    analysis?: string | undefined;
+                    parts?: ({
+                        text: string;
+                        type: "text";
+                    } | {
+                        type: "tool";
+                        toolCallMessages: {
+                            toolName: string;
+                            toolCallId: string;
+                            args: Record<string, unknown>;
+                            isCompleted: boolean;
+                            text?: string | undefined;
+                            llmResult?: unknown;
+                            additionalData?: unknown;
+                        }[];
+                    })[] | undefined;
                 };
+                isCompleted: boolean;
+                errorMessage?: {
+                    error: string;
+                } | undefined;
             }[];
-            toolCallMessages: {
-                toolCallId: string;
-                element?: any;
-            }[];
-            analysis: string;
-            isCompleted: boolean;
+            createdAt?: Date | undefined;
         }[];
+        currentSessionId?: string | undefined;
     };
 }>;
 export type AiSliceConfig = z.infer<typeof AiSliceConfig>;
-export declare function createDefaultAiConfig(): AiSliceConfig;
+export declare function createDefaultAiConfig(props: Partial<AiSliceConfig['ai']>): AiSliceConfig;
+export type AiSliceTool = ExtendedTool<any>;
 export type AiSliceState = {
     ai: {
         analysisPrompt: string;
         isRunningAnalysis: boolean;
+        tools: Record<string, AiSliceTool>;
         analysisAbortController?: AbortController;
         setAnalysisPrompt: (prompt: string) => void;
         startAnalysis: () => Promise<void>;
         cancelAnalysis: () => void;
-        messagesById: Map<string, AiMessage>;
-        addMessages: (messages: AiMessage[]) => void;
-        getMessages: () => AiMessage[];
-        setAiModel: (model: string) => void;
+        setAiModel: (modelProvider: string, model: string) => void;
+        createSession: (name?: string, modelProvider?: string, model?: string) => void;
+        switchSession: (sessionId: string) => void;
+        renameSession: (sessionId: string, name: string) => void;
+        deleteSession: (sessionId: string) => void;
+        getCurrentSession: () => AnalysisSessionSchema | undefined;
+        deleteAnalysisResult: (sessionId: string, resultId: string) => void;
+        findToolComponent: (toolName: string) => React.ComponentType | undefined;
     };
 };
-export declare function createAiSlice<PC extends BaseProjectConfig & AiSliceConfig>({ getApiKey, initialAnalysisPrompt, }: {
-    getApiKey: () => string;
+export declare function createAiSlice<PC extends BaseProjectConfig & AiSliceConfig>({ getApiKey, initialAnalysisPrompt, customTools, getInstructions, }: {
+    getApiKey: (modelProvider: string) => string;
     initialAnalysisPrompt?: string;
+    customTools?: Record<string, AiSliceTool>;
+    /**
+     * Function to get custom instructions for the AI assistant
+     * @param tablesSchema - The schema of the tables in the database
+     * @returns The instructions string to use
+     */
+    getInstructions?: (tablesSchema: DataTable[]) => string;
 }): StateCreator<AiSliceState>;
 type ProjectConfigWithAi = BaseProjectConfig & AiSliceConfig;
 type ProjectStateWithAi = ProjectState<ProjectConfigWithAi> & AiSliceState;