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

package/dist/AiSlice.d.ts

@@ -1,453 +0,0 @@
-import { ExtendedTool } from '@openassistant/utils';
-import { DataTable } from '@sqlrooms/duckdb';
-import { BaseRoomConfig, RoomShellSliceState, type StateCreator } from '@sqlrooms/room-shell';
-import { z } from 'zod';
-import { DefaultToolsOptions } from './analysis';
-import { AnalysisSessionSchema } from './schemas';
-export declare const AiSliceConfig: z.ZodObject<{
-    ai: z.ZodObject<{
-        sessions: z.ZodArray<z.ZodEffects<z.ZodObject<{
-            id: z.ZodString;
-            name: z.ZodString;
-            modelProvider: z.ZodString;
-            model: z.ZodString;
-            customModelName: z.ZodOptional<z.ZodString>;
-            baseUrl: z.ZodOptional<z.ZodString>;
-            analysisResults: z.ZodArray<z.ZodObject<{
-                id: z.ZodString;
-                prompt: z.ZodString;
-                streamMessage: z.ZodEffects<z.ZodObject<{
-                    parts: z.ZodOptional<z.ZodArray<z.ZodUnion<[z.ZodObject<{
-                        type: z.ZodLiteral<"text">;
-                        text: z.ZodString;
-                        additionalData: z.ZodOptional<z.ZodAny>;
-                        isCompleted: z.ZodOptional<z.ZodBoolean>;
-                    }, "strip", z.ZodTypeAny, {
-                        type: "text";
-                        text: string;
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    }, {
-                        type: "text";
-                        text: string;
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    }>, z.ZodObject<{
-                        type: z.ZodLiteral<"tool-invocation">;
-                        toolInvocation: z.ZodObject<{
-                            toolCallId: z.ZodString;
-                            toolName: z.ZodString;
-                            args: z.ZodAny;
-                            state: z.ZodString;
-                            result: z.ZodOptional<z.ZodAny>;
-                        }, "strip", z.ZodTypeAny, {
-                            toolName: string;
-                            toolCallId: string;
-                            state: string;
-                            args?: any;
-                            result?: any;
-                        }, {
-                            toolName: string;
-                            toolCallId: string;
-                            state: string;
-                            args?: any;
-                            result?: any;
-                        }>;
-                        additionalData: z.ZodOptional<z.ZodAny>;
-                        isCompleted: z.ZodOptional<z.ZodBoolean>;
-                    }, "strip", z.ZodTypeAny, {
-                        type: "tool-invocation";
-                        toolInvocation: {
-                            toolName: string;
-                            toolCallId: string;
-                            state: string;
-                            args?: any;
-                            result?: any;
-                        };
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    }, {
-                        type: "tool-invocation";
-                        toolInvocation: {
-                            toolName: string;
-                            toolCallId: string;
-                            state: string;
-                            args?: any;
-                            result?: any;
-                        };
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    }>, z.ZodObject<{
-                        type: z.ZodString;
-                        additionalData: z.ZodOptional<z.ZodAny>;
-                        isCompleted: z.ZodOptional<z.ZodBoolean>;
-                    }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
-                        type: z.ZodString;
-                        additionalData: z.ZodOptional<z.ZodAny>;
-                        isCompleted: z.ZodOptional<z.ZodBoolean>;
-                    }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
-                        type: z.ZodString;
-                        additionalData: z.ZodOptional<z.ZodAny>;
-                        isCompleted: z.ZodOptional<z.ZodBoolean>;
-                    }, z.ZodTypeAny, "passthrough">>]>, "many">>;
-                }, "strip", z.ZodTypeAny, {
-                    parts?: ({
-                        type: "text";
-                        text: string;
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | {
-                        type: "tool-invocation";
-                        toolInvocation: {
-                            toolName: string;
-                            toolCallId: string;
-                            state: string;
-                            args?: any;
-                            result?: any;
-                        };
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | z.objectOutputType<{
-                        type: z.ZodString;
-                        additionalData: z.ZodOptional<z.ZodAny>;
-                        isCompleted: z.ZodOptional<z.ZodBoolean>;
-                    }, z.ZodTypeAny, "passthrough">)[] | undefined;
-                }, {
-                    parts?: ({
-                        type: "text";
-                        text: string;
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | {
-                        type: "tool-invocation";
-                        toolInvocation: {
-                            toolName: string;
-                            toolCallId: string;
-                            state: string;
-                            args?: any;
-                            result?: any;
-                        };
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | z.objectInputType<{
-                        type: z.ZodString;
-                        additionalData: z.ZodOptional<z.ZodAny>;
-                        isCompleted: z.ZodOptional<z.ZodBoolean>;
-                    }, z.ZodTypeAny, "passthrough">)[] | undefined;
-                }>, {
-                    parts?: ({
-                        type: "text";
-                        text: string;
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | {
-                        type: "tool-invocation";
-                        toolInvocation: {
-                            toolName: string;
-                            toolCallId: string;
-                            state: string;
-                            args?: any;
-                            result?: any;
-                        };
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | z.objectOutputType<{
-                        type: z.ZodString;
-                        additionalData: z.ZodOptional<z.ZodAny>;
-                        isCompleted: z.ZodOptional<z.ZodBoolean>;
-                    }, z.ZodTypeAny, "passthrough">)[] | undefined;
-                }, unknown>;
-                errorMessage: z.ZodOptional<z.ZodObject<{
-                    error: z.ZodString;
-                }, "strip", z.ZodTypeAny, {
-                    error: string;
-                }, {
-                    error: string;
-                }>>;
-                isCompleted: z.ZodBoolean;
-            }, "strip", z.ZodTypeAny, {
-                id: string;
-                isCompleted: boolean;
-                prompt: string;
-                streamMessage: {
-                    parts?: ({
-                        type: "text";
-                        text: string;
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | {
-                        type: "tool-invocation";
-                        toolInvocation: {
-                            toolName: string;
-                            toolCallId: string;
-                            state: string;
-                            args?: any;
-                            result?: any;
-                        };
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | z.objectOutputType<{
-                        type: z.ZodString;
-                        additionalData: z.ZodOptional<z.ZodAny>;
-                        isCompleted: z.ZodOptional<z.ZodBoolean>;
-                    }, z.ZodTypeAny, "passthrough">)[] | undefined;
-                };
-                errorMessage?: {
-                    error: string;
-                } | undefined;
-            }, {
-                id: string;
-                isCompleted: boolean;
-                prompt: string;
-                streamMessage?: unknown;
-                errorMessage?: {
-                    error: string;
-                } | undefined;
-            }>, "many">;
-            createdAt: z.ZodOptional<z.ZodDate>;
-        }, "strip", z.ZodTypeAny, {
-            id: string;
-            name: string;
-            modelProvider: string;
-            model: string;
-            analysisResults: {
-                id: string;
-                isCompleted: boolean;
-                prompt: string;
-                streamMessage: {
-                    parts?: ({
-                        type: "text";
-                        text: string;
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | {
-                        type: "tool-invocation";
-                        toolInvocation: {
-                            toolName: string;
-                            toolCallId: string;
-                            state: string;
-                            args?: any;
-                            result?: any;
-                        };
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | z.objectOutputType<{
-                        type: z.ZodString;
-                        additionalData: z.ZodOptional<z.ZodAny>;
-                        isCompleted: z.ZodOptional<z.ZodBoolean>;
-                    }, z.ZodTypeAny, "passthrough">)[] | undefined;
-                };
-                errorMessage?: {
-                    error: string;
-                } | undefined;
-            }[];
-            baseUrl?: string | undefined;
-            customModelName?: string | undefined;
-            createdAt?: Date | undefined;
-        }, {
-            id: string;
-            name: string;
-            modelProvider: string;
-            model: string;
-            analysisResults: {
-                id: string;
-                isCompleted: boolean;
-                prompt: string;
-                streamMessage?: unknown;
-                errorMessage?: {
-                    error: string;
-                } | undefined;
-            }[];
-            baseUrl?: string | undefined;
-            customModelName?: string | undefined;
-            createdAt?: Date | undefined;
-        }>, {
-            id: string;
-            name: string;
-            modelProvider: string;
-            model: string;
-            analysisResults: {
-                id: string;
-                isCompleted: boolean;
-                prompt: string;
-                streamMessage: {
-                    parts?: ({
-                        type: "text";
-                        text: string;
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | {
-                        type: "tool-invocation";
-                        toolInvocation: {
-                            toolName: string;
-                            toolCallId: string;
-                            state: string;
-                            args?: any;
-                            result?: any;
-                        };
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | z.objectOutputType<{
-                        type: z.ZodString;
-                        additionalData: z.ZodOptional<z.ZodAny>;
-                        isCompleted: z.ZodOptional<z.ZodBoolean>;
-                    }, z.ZodTypeAny, "passthrough">)[] | undefined;
-                };
-                errorMessage?: {
-                    error: string;
-                } | undefined;
-            }[];
-            baseUrl?: string | undefined;
-            customModelName?: string | undefined;
-            createdAt?: Date | undefined;
-        }, unknown>, "many">;
-        currentSessionId: z.ZodOptional<z.ZodString>;
-    }, "strip", z.ZodTypeAny, {
-        sessions: {
-            id: string;
-            name: string;
-            modelProvider: string;
-            model: string;
-            analysisResults: {
-                id: string;
-                isCompleted: boolean;
-                prompt: string;
-                streamMessage: {
-                    parts?: ({
-                        type: "text";
-                        text: string;
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | {
-                        type: "tool-invocation";
-                        toolInvocation: {
-                            toolName: string;
-                            toolCallId: string;
-                            state: string;
-                            args?: any;
-                            result?: any;
-                        };
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | z.objectOutputType<{
-                        type: z.ZodString;
-                        additionalData: z.ZodOptional<z.ZodAny>;
-                        isCompleted: z.ZodOptional<z.ZodBoolean>;
-                    }, z.ZodTypeAny, "passthrough">)[] | undefined;
-                };
-                errorMessage?: {
-                    error: string;
-                } | undefined;
-            }[];
-            baseUrl?: string | undefined;
-            customModelName?: string | undefined;
-            createdAt?: Date | undefined;
-        }[];
-        currentSessionId?: string | undefined;
-    }, {
-        sessions: unknown[];
-        currentSessionId?: string | undefined;
-    }>;
-}, "strip", z.ZodTypeAny, {
-    ai: {
-        sessions: {
-            id: string;
-            name: string;
-            modelProvider: string;
-            model: string;
-            analysisResults: {
-                id: string;
-                isCompleted: boolean;
-                prompt: string;
-                streamMessage: {
-                    parts?: ({
-                        type: "text";
-                        text: string;
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | {
-                        type: "tool-invocation";
-                        toolInvocation: {
-                            toolName: string;
-                            toolCallId: string;
-                            state: string;
-                            args?: any;
-                            result?: any;
-                        };
-                        isCompleted?: boolean | undefined;
-                        additionalData?: any;
-                    } | z.objectOutputType<{
-                        type: z.ZodString;
-                        additionalData: z.ZodOptional<z.ZodAny>;
-                        isCompleted: z.ZodOptional<z.ZodBoolean>;
-                    }, z.ZodTypeAny, "passthrough">)[] | undefined;
-                };
-                errorMessage?: {
-                    error: string;
-                } | undefined;
-            }[];
-            baseUrl?: string | undefined;
-            customModelName?: string | undefined;
-            createdAt?: Date | undefined;
-        }[];
-        currentSessionId?: string | undefined;
-    };
-}, {
-    ai: {
-        sessions: unknown[];
-        currentSessionId?: string | undefined;
-    };
-}>;
-export type AiSliceConfig = z.infer<typeof AiSliceConfig>;
-export declare function createDefaultAiConfig(props: Partial<AiSliceConfig['ai']>): AiSliceConfig;
-export type AiSliceTool = ExtendedTool<z.ZodTypeAny, unknown, unknown, unknown>;
-export type AiSliceState = {
-    ai: {
-        analysisPrompt: string;
-        isRunningAnalysis: boolean;
-        tools: Record<string, AiSliceTool>;
-        analysisAbortController?: AbortController;
-        setAnalysisPrompt: (prompt: string) => void;
-        startAnalysis: () => Promise<void>;
-        cancelAnalysis: () => 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;
-        getApiKeyFromSettings: () => string;
-        getBaseUrlFromSettings: () => string | undefined;
-        getMaxStepsFromSettings: () => number;
-        getInstructionsFromSettings: () => string;
-    };
-};
-/**
- * Configuration options for creating an AI slice
- */
-export interface AiSliceOptions {
-    /** Initial prompt to display in the analysis input */
-    initialAnalysisPrompt?: string;
-    /** Custom tools to add to the AI assistant */
-    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;
-    toolsOptions?: DefaultToolsOptions;
-    defaultProvider?: string;
-    defaultModel?: string;
-    maxSteps?: number;
-    getApiKey?: (modelProvider: string) => string;
-    getBaseUrl?: () => string;
-}
-export declare function createAiSlice<PC extends BaseRoomConfig & AiSliceConfig>(params: AiSliceOptions): StateCreator<AiSliceState>;
-type RoomConfigWithAi = BaseRoomConfig & AiSliceConfig;
-type RoomShellSliceStateWithAi = RoomShellSliceState<RoomConfigWithAi> & AiSliceState;
-export declare function useStoreWithAi<T>(selector: (state: RoomShellSliceStateWithAi) => T): T;
-export {};
-//# sourceMappingURL=AiSlice.d.ts.map

package/dist/AiSlice.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"AiSlice.d.ts","sourceRoot":"","sources":["../src/AiSlice.ts"],"names":[],"mappings":"AACA,OAAO,EAAC,YAAY,EAAC,MAAM,sBAAsB,CAAC;AAElD,OAAO,EAAC,SAAS,EAAC,MAAM,kBAAkB,CAAC;AAC3C,OAAO,EACL,cAAc,EAEd,mBAAmB,EAEnB,KAAK,YAAY,EAClB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EACL,mBAAmB,EAIpB,MAAM,YAAY,CAAC;AACpB,OAAO,EAEL,qBAAqB,EAEtB,MAAM,WAAW,CAAC;AAGnB,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;gCA6Gd,CAAC;kCAAqB,CAAC;;;;;gCAGT,CAAC;kCACf,CAAX;;;;;;;;;;gCAU0B,CAAA;kCACb,CAAL;;;;;;;;;;gCAKG,CAAV;kCACQ,CAAF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;gCAPmB,CAAA;kCACb,CAAL;;;;;;;;;;;;;;;;;;;;;gCAKG,CAAV;kCACQ,CAAF;;;;;;;;;;;;;;;;;;;;;gCAPmB,CAAA;kCACb,CAAL;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;gCADkB,CAAA;kCACb,CAAL;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;gCADkB,CAAA;kCACb,CAAL;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;gCADkB,CAAA;kCACb,CAAL;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;gCADkB,CAAA;kCACb,CAAL;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;gCADkB,CAAA;kCACb,CAAL;;;;;;;;;;;;;;;;;;;;;;;;;EAvHP,CAAC;AACH,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC;AAE1D,wBAAgB,qBAAqB,CACnC,KAAK,EAAE,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,GAClC,aAAa,CAkBf;AAGD,MAAM,MAAM,WAAW,GAAG,YAAY,CAAC,CAAC,CAAC,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;AAEhF,MAAM,MAAM,YAAY,GAAG;IACzB,EAAE,EAAE;QACF,cAAc,EAAE,MAAM,CAAC;QACvB,iBAAiB,EAAE,OAAO,CAAC;QAC3B,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QACnC,uBAAuB,CAAC,EAAE,eAAe,CAAC;QAC1C,iBAAiB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;QAC5C,aAAa,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;QACnC,cAAc,EAAE,MAAM,IAAI,CAAC;QAC3B,UAAU,EAAE,CAAC,aAAa,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;QAC3D,aAAa,EAAE,CACb,IAAI,CAAC,EAAE,MAAM,EACb,aAAa,CAAC,EAAE,MAAM,EACtB,KAAK,CAAC,EAAE,MAAM,KACX,IAAI,CAAC;QACV,aAAa,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;QAC3C,aAAa,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;QACzD,aAAa,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;QAC3C,iBAAiB,EAAE,MAAM,qBAAqB,GAAG,SAAS,CAAC;QAC3D,oBAAoB,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAC;QACpE,iBAAiB,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,KAAK,CAAC,aAAa,GAAG,SAAS,CAAC;QACzE,qBAAqB,EAAE,MAAM,MAAM,CAAC;QACpC,sBAAsB,EAAE,MAAM,MAAM,GAAG,SAAS,CAAC;QACjD,uBAAuB,EAAE,MAAM,MAAM,CAAC;QACtC,2BAA2B,EAAE,MAAM,MAAM,CAAC;KAC3C,CAAC;CACH,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,sDAAsD;IACtD,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,8CAA8C;IAC9C,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IAC1C;;;;OAIG;IACH,eAAe,CAAC,EAAE,CAAC,YAAY,EAAE,SAAS,EAAE,KAAK,MAAM,CAAC;IACxD,YAAY,CAAC,EAAE,mBAAmB,CAAC;IACnC,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,CAAC,aAAa,EAAE,MAAM,KAAK,MAAM,CAAC;IAC9C,UAAU,CAAC,EAAE,MAAM,MAAM,CAAC;CAC3B;AAED,wBAAgB,aAAa,CAAC,EAAE,SAAS,cAAc,GAAG,aAAa,EACrE,MAAM,EAAE,cAAc,GACrB,YAAY,CAAC,YAAY,CAAC,CA4X5B;AAkHD,KAAK,gBAAgB,GAAG,cAAc,GAAG,aAAa,CAAC;AACvD,KAAK,yBAAyB,GAAG,mBAAmB,CAAC,gBAAgB,CAAC,GACpE,YAAY,CAAC;AAEf,wBAAgB,cAAc,CAAC,CAAC,EAC9B,QAAQ,EAAE,CAAC,KAAK,EAAE,yBAAyB,KAAK,CAAC,GAChD,CAAC,CAMH"}