@sqlrooms/ai 0.1.0

package/dist/ToolCall.d.ts

@@ -0,0 +1,7 @@
+import { ToolCallSchema } from './schemas';
+interface ToolCallProps {
+    toolCall: ToolCallSchema;
+}
+export declare function ToolCall({ toolCall }: ToolCallProps): import("react/jsx-runtime").JSX.Element;
+export {};
+//# sourceMappingURL=ToolCall.d.ts.map

package/dist/ToolCall.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolCall.d.ts","sourceRoot":"","sources":["../src/ToolCall.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAC,cAAc,EAAC,MAAM,WAAW,CAAC;AAezC,UAAU,aAAa;IACrB,QAAQ,EAAE,cAAc,CAAC;CAC1B;AAED,wBAAgB,QAAQ,CAAC,EAAC,QAAQ,EAAC,EAAE,aAAa,2CA6EjD"}

package/dist/ToolCall.js

@@ -0,0 +1,13 @@
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { Badge, Button, Popover, PopoverContent, PopoverTrigger, Spinner, cn, } from '@sqlrooms/ui';
+import { CodeIcon } from 'lucide-react';
+import { Suspense } from 'react';
+import Markdown from 'react-markdown';
+import { VegaLiteChart } from '@sqlrooms/vega';
+export function ToolCall({ toolCall }) {
+    const { args, toolName, toolCallId } = toolCall;
+    const { type } = args;
+    return (_jsxs("div", { className: cn('border-2 relative bg-gray-900 px-5 py-6 rounded-md text-gray-300 text-xs', {
+            ' border-blue-500': toolName === 'answer',
+        }), children: [_jsx(Badge, { variant: "secondary", className: cn('text-xs absolute top-[-12px] left-2 text-gray-100', toolName === 'answer' && 'bg-blue-500'), children: toolName }), _jsx("div", { className: "absolute top-2 right-2", children: _jsxs(Popover, { children: [_jsx(PopoverTrigger, { asChild: true, children: _jsx(Button, { variant: "outline", size: "icon", className: "w-6 h-6", children: _jsx(CodeIcon, { className: "w-4 h-4" }) }) }), _jsx(PopoverContent, { className: "w-[400px] max-h-[400px] overflow-auto", align: "start", side: "right", children: _jsx("pre", { className: "text-xs", children: JSON.stringify(toolCall, null, 2) }) })] }) }), type === 'query' ? (_jsxs("div", { className: "flex flex-col gap-5", children: [_jsx("div", { className: "text-xs text-gray-400", children: args.reasoning }), _jsx("div", { className: "font-mono", children: args.sqlQuery })] })) : type === 'answer' ? (_jsxs("div", { className: "flex flex-col gap-5", children: [_jsx("div", { className: "text-sm", children: _jsx(Markdown, { children: args.answer }) }), args.chart && (_jsxs("div", { className: "flex flex-col gap-2", children: [_jsx("div", { className: "text-xs text-muted-foreground font-mono", children: args.chart.sqlQuery }), _jsx(Suspense, { fallback: _jsx("div", { className: "w-full h-full flex items-center justify-center", children: _jsx(Spinner, { className: "w-4 h-4" }) }), children: _jsx(VegaLiteChart, { width: 400, height: 250, sqlQuery: args.chart.sqlQuery, spec: args.chart.vegaLiteSpec }) })] }))] })) : (_jsx("pre", { children: JSON.stringify(args, null, 2) }))] }, toolCallId));
+}

package/dist/ToolResult.d.ts

@@ -0,0 +1,7 @@
+import { ToolResultSchema } from './schemas';
+interface ToolResultProps {
+    toolResult: ToolResultSchema;
+}
+export declare const ToolResult: React.FC<ToolResultProps>;
+export {};
+//# sourceMappingURL=ToolResult.d.ts.map

package/dist/ToolResult.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolResult.d.ts","sourceRoot":"","sources":["../src/ToolResult.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAC,gBAAgB,EAAC,MAAM,WAAW,CAAC;AAW3C,UAAU,eAAe;IACvB,UAAU,EAAE,gBAAgB,CAAC;CAC9B;AAED,eAAO,MAAM,UAAU,EAAE,KAAK,CAAC,EAAE,CAAC,eAAe,CA2DhD,CAAC"}

package/dist/ToolResult.js

@@ -0,0 +1,8 @@
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { Badge, Button, Popover, PopoverContent, PopoverTrigger, cn, } from '@sqlrooms/ui';
+import { CheckCircle2Icon, CodeIcon, XCircleIcon } from 'lucide-react';
+export const ToolResult = ({ toolResult }) => {
+    const { toolName, args, result } = toolResult;
+    const isSuccess = result.success;
+    return (_jsxs("div", { className: cn('border-2 relative bg-gray-900 px-5 py-6 rounded-md text-gray-300 text-xs', isSuccess ? 'border-green-500' : 'border-red-500'), children: [_jsxs(Badge, { variant: "secondary", className: cn('text-xs absolute top-[-12px] left-2 text-gray-100 flex items-center gap-1 border', isSuccess ? 'border-green-500' : 'border-red-500'), children: [isSuccess ? (_jsx(CheckCircle2Icon, { className: "w-3 h-3 text-green-500" })) : (_jsx(XCircleIcon, { className: "w-3 h-3 text-red-500" })), toolName] }), _jsx("div", { className: "absolute top-2 right-2", children: _jsxs(Popover, { children: [_jsx(PopoverTrigger, { asChild: true, children: _jsx(Button, { variant: "outline", size: "icon", className: "w-6 h-6", children: _jsx(CodeIcon, { className: "w-4 h-4" }) }) }), _jsx(PopoverContent, { className: "w-[400px] max-h-[300px] overflow-auto p-4", side: "right", align: "start", children: _jsx("pre", { className: "whitespace-no-wrap text-xs", children: JSON.stringify(toolResult, null, 2) }) })] }) }), _jsxs("div", { className: "flex flex-col gap-5", children: [args.reasoning && (_jsx("div", { className: "text-xs text-gray-400", children: args.reasoning })), args.sqlQuery && _jsx("div", { className: "font-mono", children: args.sqlQuery }), !result.success && (_jsxs("div", { className: "text-red-500 gap-2 flex flex-col", children: [_jsx("p", { className: "text-sm font-bold", children: "Oops! Something went wrong..." }), _jsx("p", { className: "text-xs", children: result.error })] }))] })] }));
+};

package/dist/analysis.d.ts

@@ -0,0 +1,143 @@
+import type { LanguageModelV1 } from '@ai-sdk/provider';
+import { StepResult, ToolExecutionOptions } from 'ai';
+/**
+ * Run analysis on the project data
+ * @param prompt - The prompt for the analysis
+ * @param abortSignal - An optional abort signal to cancel the analysis
+ * @returns The tool calls and the final answer
+ */
+export declare function runAnalysis({ model, abortSignal, onStepFinish, maxSteps, messages, }: {
+    abortSignal?: AbortSignal;
+    onStepFinish?: (event: StepResult<typeof TOOLS>) => Promise<void> | void;
+    apiKey: string;
+    model: LanguageModelV1;
+    maxSteps?: number;
+    messages?: any[];
+}): Promise<import("ai").GenerateTextResult<{
+    query: import("ai").Tool<import("zod").ZodObject<{
+        type: import("zod").ZodLiteral<"query">;
+        sqlQuery: import("zod").ZodString;
+        reasoning: import("zod").ZodString;
+    }, "strip", import("zod").ZodTypeAny, {
+        type: "query";
+        sqlQuery: string;
+        reasoning: string;
+    }, {
+        type: "query";
+        sqlQuery: string;
+        reasoning: string;
+    }>, {
+        success: false;
+        error: string;
+    } | {
+        success: true;
+        data: Record<string, any>;
+    }> & {
+        execute: (args: {
+            type: "query";
+            sqlQuery: string;
+            reasoning: string;
+        }, options: ToolExecutionOptions) => PromiseLike<{
+            success: false;
+            error: string;
+        } | {
+            success: true;
+            data: Record<string, any>;
+        }>;
+    };
+    answer: import("ai").Tool<import("zod").ZodObject<{
+        type: import("zod").ZodLiteral<"answer">;
+        answer: import("zod").ZodString;
+        chart: import("zod").ZodUnion<[import("zod").ZodObject<{
+            sqlQuery: import("zod").ZodString;
+            vegaLiteSpec: import("zod").ZodString;
+        }, "strip", import("zod").ZodTypeAny, {
+            sqlQuery: string;
+            vegaLiteSpec: string;
+        }, {
+            sqlQuery: string;
+            vegaLiteSpec: string;
+        }>, import("zod").ZodNull]>;
+    }, "strip", import("zod").ZodTypeAny, {
+        type: "answer";
+        answer: string;
+        chart: {
+            sqlQuery: string;
+            vegaLiteSpec: string;
+        } | null;
+    }, {
+        type: "answer";
+        answer: string;
+        chart: {
+            sqlQuery: string;
+            vegaLiteSpec: string;
+        } | null;
+    }>, unknown> & {
+        execute: undefined;
+    };
+}, never>>;
+declare const TOOLS: {
+    query: import("ai").Tool<import("zod").ZodObject<{
+        type: import("zod").ZodLiteral<"query">;
+        sqlQuery: import("zod").ZodString;
+        reasoning: import("zod").ZodString;
+    }, "strip", import("zod").ZodTypeAny, {
+        type: "query";
+        sqlQuery: string;
+        reasoning: string;
+    }, {
+        type: "query";
+        sqlQuery: string;
+        reasoning: string;
+    }>, {
+        success: false;
+        error: string;
+    } | {
+        success: true;
+        data: Record<string, any>;
+    }> & {
+        execute: (args: {
+            type: "query";
+            sqlQuery: string;
+            reasoning: string;
+        }, options: ToolExecutionOptions) => PromiseLike<{
+            success: false;
+            error: string;
+        } | {
+            success: true;
+            data: Record<string, any>;
+        }>;
+    };
+    answer: import("ai").Tool<import("zod").ZodObject<{
+        type: import("zod").ZodLiteral<"answer">;
+        answer: import("zod").ZodString;
+        chart: import("zod").ZodUnion<[import("zod").ZodObject<{
+            sqlQuery: import("zod").ZodString;
+            vegaLiteSpec: import("zod").ZodString;
+        }, "strip", import("zod").ZodTypeAny, {
+            sqlQuery: string;
+            vegaLiteSpec: string;
+        }, {
+            sqlQuery: string;
+            vegaLiteSpec: string;
+        }>, import("zod").ZodNull]>;
+    }, "strip", import("zod").ZodTypeAny, {
+        type: "answer";
+        answer: string;
+        chart: {
+            sqlQuery: string;
+            vegaLiteSpec: string;
+        } | null;
+    }, {
+        type: "answer";
+        answer: string;
+        chart: {
+            sqlQuery: string;
+            vegaLiteSpec: string;
+        } | null;
+    }>, unknown> & {
+        execute: undefined;
+    };
+};
+export {};
+//# sourceMappingURL=analysis.d.ts.map

package/dist/analysis.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"analysis.d.ts","sourceRoot":"","sources":["../src/analysis.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,eAAe,EAAC,MAAM,kBAAkB,CAAC;AAEtD,OAAO,EAAe,UAAU,EAAQ,oBAAoB,EAAC,MAAM,IAAI,CAAC;AAOxE;;;;;GAKG;AACH,wBAAsB,WAAW,CAAC,EAChC,KAAK,EAEL,WAAW,EACX,YAAY,EACZ,QAAc,EACd,QAAQ,GACT,EAAE;IAED,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,UAAU,CAAC,OAAO,KAAK,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACzE,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,eAAe,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,GAAG,EAAE,CAAC;CAClB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;WA+BA;AAED,QAAA,MAAM,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA0DV,CAAC"}

package/dist/analysis.js

@@ -0,0 +1,87 @@
+import { arrowTableToJson, getDuckDb } from '@sqlrooms/duckdb';
+import { generateText, tool } from 'ai';
+import { AnswerToolParameters, QueryToolParameters, } from './schemas';
+/**
+ * Run analysis on the project data
+ * @param prompt - The prompt for the analysis
+ * @param abortSignal - An optional abort signal to cancel the analysis
+ * @returns The tool calls and the final answer
+ */
+export async function runAnalysis({ model, 
+// prompt,
+abortSignal, onStepFinish, maxSteps = 100, messages, }) {
+    const result = await generateText({
+        model,
+        abortSignal,
+        // prompt,
+        messages,
+        tools: TOOLS,
+        toolChoice: 'required',
+        maxSteps,
+        maxRetries: 1,
+        system: 'You are analyzing tables in DuckDB database in the context of a project. ' +
+            'You can run SQL queries to perform analysis and answer questions. ' +
+            'Reason step by step. ' +
+            'When you give the final answer, provide an explanation for how you got it.',
+        onStepFinish,
+    });
+    // const answer = result.toolCalls.find((t) => t.toolName === 'answer');
+    // if (!answer) {
+    //   console.error('No answer tool call found', {result});
+    //   throw new Error('No answer tool call found');
+    // }
+    // return answer.args;
+    return result;
+}
+const TOOLS = {
+    query: tool({
+        description: 'A tool for executing SQL queries in DuckDB that is embedded in browser using duckdb-wasm. ' +
+            'You can obtain the structures of all tables and their column types by running `DESCRIBE`. ' +
+            'Query results are returned as a json object `{success: boolean, data: object[], error?: string}`. ' +
+            'You should only analyze tables which are in the main schema. ' +
+            'Avoid queries returning too much data to prevent the browser from crashing. ' +
+            'Include VegaLite charts in your response if the data is suitable for it. ' +
+            'Omit the data from the chart.vegaLiteSpec in the response, provide an sql query in chart.sqlQuery instead. ' +
+            'To obtain stats, use the `SUMMARIZE table_name` query. ' +
+            "Don't execute queries that modify data unless explicitly asked. ",
+        parameters: QueryToolParameters,
+        execute: async ({ sqlQuery }, options) => {
+            try {
+                const { conn } = await getDuckDb();
+                // TODO use options.abortSignal: maybe call db.cancelPendingQuery
+                const result = await conn.query(sqlQuery);
+                // if (options.abortSignal?.aborted) {
+                //   throw new Error('Query aborted');
+                // }
+                return {
+                    success: true,
+                    data: arrowTableToJson(result),
+                };
+            }
+            catch (error) {
+                console.error('SQL query error:', error);
+                const errorMessage = error instanceof Error
+                    ? error.cause instanceof Error
+                        ? error.cause.message
+                        : error.message
+                    : String(error);
+                return {
+                    success: false,
+                    error: errorMessage,
+                };
+            }
+        },
+        // TODO: consider experimental_toToolResultContent() for returning Arrow
+    }),
+    // answer tool: the LLM will provide a structured answer
+    answer: tool({
+        description: 'A tool for providing the final answer.',
+        parameters: AnswerToolParameters,
+        // execute: async ({answer}): Promise<ToolResultSchema['result']> => {
+        //   return {
+        //     success: true,
+        //     data: answer,
+        //   };
+        // },
+    }),
+};

package/dist/hooks/use-scroll-to-bottom.d.ts

@@ -0,0 +1,7 @@
+import { type RefObject } from 'react';
+export declare function useScrollToBottom<T extends HTMLElement>(options?: MutationObserverInit): [RefObject<T>, RefObject<T>];
+export declare function useScrollToBottomButton(containerRef: RefObject<HTMLElement>): {
+    showButton: boolean;
+    scrollToBottom: () => void;
+};
+//# sourceMappingURL=use-scroll-to-bottom.d.ts.map

package/dist/hooks/use-scroll-to-bottom.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-scroll-to-bottom.d.ts","sourceRoot":"","sources":["../../src/hooks/use-scroll-to-bottom.ts"],"names":[],"mappings":"AAAA,OAAO,EAAoB,KAAK,SAAS,EAAW,MAAM,OAAO,CAAC;AAElE,wBAAgB,iBAAiB,CAAC,CAAC,SAAS,WAAW,EACrD,OAAO,GAAE,oBAKR,GACA,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,CAoB9B;AAED,wBAAgB,uBAAuB,CAAC,YAAY,EAAE,SAAS,CAAC,WAAW,CAAC;;;EAiC3E"}

package/dist/hooks/use-scroll-to-bottom.js

@@ -0,0 +1,50 @@
+import { useEffect, useRef, useState } from 'react';
+export function useScrollToBottom(options = {
+    childList: true,
+    subtree: true,
+    characterData: true,
+    // attributes: true,
+}) {
+    const containerRef = useRef(null);
+    const endRef = useRef(null);
+    useEffect(() => {
+        const container = containerRef.current;
+        const end = endRef.current;
+        if (container && end) {
+            const observer = new MutationObserver(() => {
+                end.scrollIntoView({ behavior: 'instant', block: 'end' });
+            });
+            observer.observe(container, options);
+            return () => observer.disconnect();
+        }
+    }, [options]);
+    return [containerRef, endRef];
+}
+export function useScrollToBottomButton(containerRef) {
+    const [showButton, setShowButton] = useState(false);
+    useEffect(() => {
+        const container = containerRef.current;
+        if (!container)
+            return;
+        const handleScroll = () => {
+            const { scrollTop, scrollHeight, clientHeight } = container;
+            // Only show if we're scrolled up more than 200px from the bottom
+            const isNotAtBottom = scrollHeight - scrollTop - clientHeight > 200;
+            setShowButton(isNotAtBottom);
+        };
+        container.addEventListener('scroll', handleScroll);
+        // Initial check with a slight delay to ensure proper measurement
+        const timeoutId = setTimeout(handleScroll, 100);
+        return () => {
+            container.removeEventListener('scroll', handleScroll);
+            clearTimeout(timeoutId);
+        };
+    }, [containerRef]);
+    const scrollToBottom = () => {
+        containerRef.current?.scrollTo({
+            top: containerRef.current.scrollHeight,
+            behavior: 'smooth',
+        });
+    };
+    return { showButton, scrollToBottom };
+}

package/dist/index.d.ts

@@ -0,0 +1,7 @@
+export { AiSliceConfig, createAiSlice, useStoreWithAi as useAiStore, createDefaultAiConfig, } from './AiSlice';
+export type { AiSliceState } from './AiSlice';
+export { QueryControls } from './QueryControls';
+export { AnalysisResultsContainer } from './AnalysisResultsContainer';
+export { AnalysisResult } from './AnalysisResult';
+export { useScrollToBottom, useScrollToBottomButton, } from './hooks/use-scroll-to-bottom';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,aAAa,EACb,aAAa,EACb,cAAc,IAAI,UAAU,EAC5B,qBAAqB,GACtB,MAAM,WAAW,CAAC;AAEnB,YAAY,EAAC,YAAY,EAAC,MAAM,WAAW,CAAC;AAC5C,OAAO,EAAC,aAAa,EAAC,MAAM,iBAAiB,CAAC;AAC9C,OAAO,EAAC,wBAAwB,EAAC,MAAM,4BAA4B,CAAC;AACpE,OAAO,EAAC,cAAc,EAAC,MAAM,kBAAkB,CAAC;AAChD,OAAO,EACL,iBAAiB,EACjB,uBAAuB,GACxB,MAAM,8BAA8B,CAAC"}

package/dist/index.js

@@ -0,0 +1,5 @@
+export { AiSliceConfig, createAiSlice, useStoreWithAi as useAiStore, createDefaultAiConfig, } from './AiSlice';
+export { QueryControls } from './QueryControls';
+export { AnalysisResultsContainer } from './AnalysisResultsContainer';
+export { AnalysisResult } from './AnalysisResult';
+export { useScrollToBottom, useScrollToBottomButton, } from './hooks/use-scroll-to-bottom';