@sqlrooms/ai 0.5.1 → 0.6.0

package/dist/AiSlice.d.ts

@@ -1,12 +1,13 @@
 import { ProjectState, type StateCreator } from '@sqlrooms/project-builder';
 import { BaseProjectConfig } from '@sqlrooms/project-config';
-import { CoreAssistantMessage, CoreToolMessage, CoreUserMessage, LanguageModelV1 } from 'ai';
+import { CoreAssistantMessage, CoreToolMessage, CoreUserMessage } from 'ai';
 import { z } from 'zod';
 type AiMessage = (CoreToolMessage | CoreAssistantMessage | CoreUserMessage) & {
     id: string;
 };
 export declare const AiSliceConfig: z.ZodObject<{
     ai: z.ZodObject<{
+        modelProvider: z.ZodString;
         model: z.ZodString;
         analysisResults: z.ZodArray<z.ZodObject<{
             id: z.ZodString;
@@ -73,32 +74,17 @@ export declare const AiSliceConfig: z.ZodObject<{
                     sqlQuery: string;
                     reasoning: string;
                 }>, z.ZodObject<{
-                    type: z.ZodLiteral<"answer">;
-                    answer: z.ZodString;
-                    chart: z.ZodUnion<[z.ZodObject<{
-                        sqlQuery: z.ZodString;
-                        vegaLiteSpec: z.ZodString;
-                    }, "strip", z.ZodTypeAny, {
-                        sqlQuery: string;
-                        vegaLiteSpec: string;
-                    }, {
-                        sqlQuery: string;
-                        vegaLiteSpec: string;
-                    }>, z.ZodNull]>;
+                    sqlQuery: z.ZodString;
+                    vegaLiteSpec: z.ZodString;
+                    reasoning: z.ZodString;
                 }, "strip", z.ZodTypeAny, {
-                    type: "answer";
-                    answer: string;
-                    chart: {
-                        sqlQuery: string;
-                        vegaLiteSpec: string;
-                    } | null;
+                    sqlQuery: string;
+                    reasoning: string;
+                    vegaLiteSpec: string;
                 }, {
-                    type: "answer";
-                    answer: string;
-                    chart: {
-                        sqlQuery: string;
-                        vegaLiteSpec: string;
-                    } | null;
+                    sqlQuery: string;
+                    reasoning: string;
+                    vegaLiteSpec: string;
                 }>]>;
             }, "strip", z.ZodTypeAny, {
                 toolName: string;
@@ -108,12 +94,9 @@ export declare const AiSliceConfig: z.ZodObject<{
                     sqlQuery: string;
                     reasoning: string;
                 } | {
-                    type: "answer";
-                    answer: string;
-                    chart: {
-                        sqlQuery: string;
-                        vegaLiteSpec: string;
-                    } | null;
+                    sqlQuery: string;
+                    reasoning: string;
+                    vegaLiteSpec: string;
                 };
             }, {
                 toolName: string;
@@ -123,14 +106,23 @@ export declare const AiSliceConfig: z.ZodObject<{
                     sqlQuery: string;
                     reasoning: string;
                 } | {
-                    type: "answer";
-                    answer: string;
-                    chart: {
-                        sqlQuery: string;
-                        vegaLiteSpec: string;
-                    } | null;
+                    sqlQuery: string;
+                    reasoning: string;
+                    vegaLiteSpec: string;
                 };
             }>, "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;
         }, "strip", z.ZodTypeAny, {
             id: string;
             prompt: string;
@@ -154,14 +146,17 @@ export declare const AiSliceConfig: z.ZodObject<{
                     sqlQuery: string;
                     reasoning: string;
                 } | {
-                    type: "answer";
-                    answer: string;
-                    chart: {
-                        sqlQuery: string;
-                        vegaLiteSpec: string;
-                    } | null;
+                    sqlQuery: string;
+                    reasoning: string;
+                    vegaLiteSpec: string;
                 };
             }[];
+            toolCallMessages: {
+                toolCallId: string;
+                element?: any;
+            }[];
+            analysis: string;
+            isCompleted: boolean;
         }, {
             id: string;
             prompt: string;
@@ -185,16 +180,20 @@ export declare const AiSliceConfig: z.ZodObject<{
                     sqlQuery: string;
                     reasoning: string;
                 } | {
-                    type: "answer";
-                    answer: string;
-                    chart: {
-                        sqlQuery: string;
-                        vegaLiteSpec: string;
-                    } | null;
+                    sqlQuery: string;
+                    reasoning: string;
+                    vegaLiteSpec: string;
                 };
             }[];
+            toolCallMessages: {
+                toolCallId: string;
+                element?: any;
+            }[];
+            analysis: string;
+            isCompleted: boolean;
         }>, "many">;
     }, "strip", z.ZodTypeAny, {
+        modelProvider: string;
         model: string;
         analysisResults: {
             id: string;
@@ -219,16 +218,20 @@ export declare const AiSliceConfig: z.ZodObject<{
                     sqlQuery: string;
                     reasoning: string;
                 } | {
-                    type: "answer";
-                    answer: string;
-                    chart: {
-                        sqlQuery: string;
-                        vegaLiteSpec: string;
-                    } | null;
+                    sqlQuery: string;
+                    reasoning: string;
+                    vegaLiteSpec: string;
                 };
             }[];
+            toolCallMessages: {
+                toolCallId: string;
+                element?: any;
+            }[];
+            analysis: string;
+            isCompleted: boolean;
         }[];
     }, {
+        modelProvider: string;
         model: string;
         analysisResults: {
             id: string;
@@ -253,18 +256,22 @@ export declare const AiSliceConfig: z.ZodObject<{
                     sqlQuery: string;
                     reasoning: string;
                 } | {
-                    type: "answer";
-                    answer: string;
-                    chart: {
-                        sqlQuery: string;
-                        vegaLiteSpec: string;
-                    } | null;
+                    sqlQuery: string;
+                    reasoning: string;
+                    vegaLiteSpec: string;
                 };
             }[];
+            toolCallMessages: {
+                toolCallId: string;
+                element?: any;
+            }[];
+            analysis: string;
+            isCompleted: boolean;
         }[];
     }>;
 }, "strip", z.ZodTypeAny, {
     ai: {
+        modelProvider: string;
         model: string;
         analysisResults: {
             id: string;
@@ -289,18 +296,22 @@ export declare const AiSliceConfig: z.ZodObject<{
                     sqlQuery: string;
                     reasoning: string;
                 } | {
-                    type: "answer";
-                    answer: string;
-                    chart: {
-                        sqlQuery: string;
-                        vegaLiteSpec: string;
-                    } | null;
+                    sqlQuery: string;
+                    reasoning: string;
+                    vegaLiteSpec: string;
                 };
             }[];
+            toolCallMessages: {
+                toolCallId: string;
+                element?: any;
+            }[];
+            analysis: string;
+            isCompleted: boolean;
         }[];
     };
 }, {
     ai: {
+        modelProvider: string;
         model: string;
         analysisResults: {
             id: string;
@@ -325,14 +336,17 @@ export declare const AiSliceConfig: z.ZodObject<{
                     sqlQuery: string;
                     reasoning: string;
                 } | {
-                    type: "answer";
-                    answer: string;
-                    chart: {
-                        sqlQuery: string;
-                        vegaLiteSpec: string;
-                    } | null;
+                    sqlQuery: string;
+                    reasoning: string;
+                    vegaLiteSpec: string;
                 };
             }[];
+            toolCallMessages: {
+                toolCallId: string;
+                element?: any;
+            }[];
+            analysis: string;
+            isCompleted: boolean;
         }[];
     };
 }>;
@@ -349,10 +363,11 @@ export type AiSliceState = {
         messagesById: Map<string, AiMessage>;
         addMessages: (messages: AiMessage[]) => void;
         getMessages: () => AiMessage[];
+        setAiModel: (model: string) => void;
     };
 };
-export declare function createAiSlice<PC extends BaseProjectConfig & AiSliceConfig>({ createModel, initialAnalysisPrompt, }: {
-    createModel: (model: string) => LanguageModelV1;
+export declare function createAiSlice<PC extends BaseProjectConfig & AiSliceConfig>({ getApiKey, initialAnalysisPrompt, }: {
+    getApiKey: () => string;
     initialAnalysisPrompt?: string;
 }): StateCreator<AiSliceState>;
 type ProjectConfigWithAi = BaseProjectConfig & AiSliceConfig;

package/dist/AiSlice.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AiSlice.d.ts","sourceRoot":"","sources":["../src/AiSlice.ts"],"names":[],"mappings":"AACA,OAAO,EAEL,YAAY,EAEZ,KAAK,YAAY,EAClB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAC,iBAAiB,EAAC,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EACL,oBAAoB,EACpB,eAAe,EACf,eAAe,EACf,eAAe,EAChB,MAAM,IAAI,CAAC;AAEZ,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAQtB,KAAK,SAAS,GAAG,CAAC,eAAe,GAAG,oBAAoB,GAAG,eAAe,CAAC,GAAG;IAC5E,EAAE,EAAE,MAAM,CAAC;CACZ,CAAC;AAEF,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAKxB,CAAC;AACH,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC;AAE1D,wBAAgB,qBAAqB,IAAI,aAAa,CAOrD;AAED,MAAM,MAAM,YAAY,GAAG;IACzB,EAAE,EAAE;QACF,cAAc,EAAE,MAAM,CAAC;QACvB,iBAAiB,EAAE,OAAO,CAAC;QAC3B,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,YAAY,EAAE,GAAG,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QACrC,WAAW,EAAE,CAAC,QAAQ,EAAE,SAAS,EAAE,KAAK,IAAI,CAAC;QAC7C,WAAW,EAAE,MAAM,SAAS,EAAE,CAAC;KAChC,CAAC;CACH,CAAC;AAEF,wBAAgB,aAAa,CAAC,EAAE,SAAS,iBAAiB,GAAG,aAAa,EAAE,EAC1E,WAAW,EACX,qBAAgI,GACjI,EAAE;IACD,WAAW,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,eAAe,CAAC;IAChD,qBAAqB,CAAC,EAAE,MAAM,CAAC;CAChC,GAAG,YAAY,CAAC,YAAY,CAAC,CAuJ7B;AAsCD,KAAK,mBAAmB,GAAG,iBAAiB,GAAG,aAAa,CAAC;AAC7D,KAAK,kBAAkB,GAAG,YAAY,CAAC,mBAAmB,CAAC,GAAG,YAAY,CAAC;AAE3E,wBAAgB,cAAc,CAAC,CAAC,EAC9B,QAAQ,EAAE,CAAC,KAAK,EAAE,kBAAkB,KAAK,CAAC,GACzC,CAAC,CAMH"}
+{"version":3,"file":"AiSlice.d.ts","sourceRoot":"","sources":["../src/AiSlice.ts"],"names":[],"mappings":"AACA,OAAO,EAEL,YAAY,EAEZ,KAAK,YAAY,EAClB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAC,iBAAiB,EAAC,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EACL,oBAAoB,EACpB,eAAe,EACf,eAAe,EAGhB,MAAM,IAAI,CAAC;AAEZ,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAQtB,KAAK,SAAS,GAAG,CAAC,eAAe,GAAG,oBAAoB,GAAG,eAAe,CAAC,GAAG;IAC5E,EAAE,EAAE,MAAM,CAAC;CACZ,CAAC;AAEF,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAMxB,CAAC;AACH,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC;AAE1D,wBAAgB,qBAAqB,IAAI,aAAa,CAQrD;AAED,MAAM,MAAM,YAAY,GAAG;IACzB,EAAE,EAAE;QACF,cAAc,EAAE,MAAM,CAAC;QACvB,iBAAiB,EAAE,OAAO,CAAC;QAC3B,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,YAAY,EAAE,GAAG,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QACrC,WAAW,EAAE,CAAC,QAAQ,EAAE,SAAS,EAAE,KAAK,IAAI,CAAC;QAC7C,WAAW,EAAE,MAAM,SAAS,EAAE,CAAC;QAC/B,UAAU,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;KACrC,CAAC;CACH,CAAC;AAwFF,wBAAgB,aAAa,CAAC,EAAE,SAAS,iBAAiB,GAAG,aAAa,EAAE,EAC1E,SAAS,EACT,qBAAgI,GACjI,EAAE;IACD,SAAS,EAAE,MAAM,MAAM,CAAC;IACxB,qBAAqB,CAAC,EAAE,MAAM,CAAC;CAChC,GAAG,YAAY,CAAC,YAAY,CAAC,CAiH7B;AA+DD,KAAK,mBAAmB,GAAG,iBAAiB,GAAG,aAAa,CAAC;AAC7D,KAAK,kBAAkB,GAAG,YAAY,CAAC,mBAAmB,CAAC,GAAG,YAAY,CAAC;AAE3E,wBAAgB,cAAc,CAAC,CAAC,EAC9B,QAAQ,EAAE,CAAC,KAAK,EAAE,kBAAkB,KAAK,CAAC,GACzC,CAAC,CAMH"}

package/dist/AiSlice.js

@@ -6,6 +6,7 @@ import { runAnalysis } from './analysis';
 import { AnalysisResultSchema, } from './schemas';
 export const AiSliceConfig = z.object({
     ai: z.object({
+        modelProvider: z.string(),
         model: z.string(),
         analysisResults: z.array(AnalysisResultSchema),
     }),
@@ -13,23 +14,86 @@ export const AiSliceConfig = z.object({
 export function createDefaultAiConfig() {
     return {
         ai: {
+            modelProvider: 'openai',
             model: 'gpt-4o-mini',
             analysisResults: [],
         },
     };
 }
-export function createAiSlice({ createModel, initialAnalysisPrompt = 'Describe the data in the tables and make a chart providing an overview of the most important features.', }) {
+/**
+ * Execute the analysis. It will be used by the action `startAnalysis`.
+ *
+ * Each analysis contains an array of toolCalls and the results of the tool calls (toolResults).
+ * After all the tool calls have been executed, the LLM will stream the results as text stored in `analysis`.
+ *
+ * @param resultId - The result id
+ * @param prompt - The prompt
+ * @param model - The model
+ * @param apiKey - The api key
+ * @param abortController - The abort controller
+ * @param addMessages - The add messages function
+ * @param set - The set function
+ */
+async function executeAnalysis({ resultId, prompt, modelProvider, model, apiKey, abortController, addMessages, set, }) {
+    try {
+        await runAnalysis({
+            modelProvider,
+            model,
+            apiKey,
+            prompt,
+            abortController,
+            onStepFinish: (event, toolCallMessages) => {
+                addMessages(event.response.messages);
+                set(makeResultsAppender({
+                    resultId,
+                    toolResults: event.toolResults,
+                    toolCalls: event.toolCalls,
+                    toolCallMessages,
+                }));
+            },
+            onStreamResult: (message, isCompleted) => {
+                set(makeResultsAppender({
+                    resultId,
+                    analysis: message,
+                    isCompleted,
+                }));
+            },
+        });
+    }
+    catch (err) {
+        set(makeResultsAppender({
+            resultId,
+            isCompleted: true,
+            toolResults: [
+                {
+                    toolName: 'error',
+                    toolCallId: createId(),
+                    args: {},
+                    result: {
+                        success: false,
+                        error: err instanceof Error ? err.message : String(err),
+                    },
+                },
+            ],
+            toolCalls: [],
+        }));
+    }
+}
+export function createAiSlice({ getApiKey, initialAnalysisPrompt = 'Describe the data in the tables and make a chart providing an overview of the most important features.', }) {
     return createSlice((set, get) => ({
         ai: {
             analysisPrompt: initialAnalysisPrompt,
             isRunningAnalysis: false,
             messagesById: new Map(),
-            apiKey: null,
             setAnalysisPrompt: (prompt) => {
                 set((state) => produce(state, (draft) => {
                     draft.ai.analysisPrompt = prompt;
                 }));
             },
+            /**
+             * Set the AI model
+             * @param model - The model to set
+             */
             setAiModel: (model) => {
                 set((state) => produce(state, (draft) => {
                     draft.project.config.ai.model = model;
@@ -49,7 +113,6 @@ export function createAiSlice({ createModel, initialAnalysisPrompt = 'Describe t
                         }
                         newMessagesById.set(m.id, m);
                     }
-                    console.log('newMessagesById', Array.from(newMessagesById.values()));
                     return {
                         ai: {
                             ...state.ai,
@@ -72,6 +135,9 @@ export function createAiSlice({ createModel, initialAnalysisPrompt = 'Describe t
                         prompt: get().ai.analysisPrompt,
                         toolResults: [],
                         toolCalls: [],
+                        toolCallMessages: [],
+                        analysis: '',
+                        isCompleted: false,
                     });
                 }));
                 get().ai.addMessages([
@@ -81,63 +147,22 @@ export function createAiSlice({ createModel, initialAnalysisPrompt = 'Describe t
                         content: get().ai.analysisPrompt,
                     },
                 ]);
-                set((state) => produce(state, (draft) => {
-                    draft.ai.analysisPrompt = '';
-                }));
                 try {
-                    const { toolResults, toolCalls, ...rest } = await runAnalysis({
-                        model: createModel(get().project.config.ai.model),
-                        // prompt: get().analysisPrompt,
-                        messages: get().ai.getMessages(),
-                        onStepFinish: (event) => {
-                            console.log('onStepFinish', event);
-                            get().ai.addMessages(event.response.messages);
-                            set(makeResultsAppender({
-                                resultId,
-                                toolResults: event.toolResults,
-                                toolCalls: event.toolCalls,
-                            }));
-                        },
-                        abortSignal: abortController.signal,
-                    });
-                    console.log('final result', { toolResults, toolCalls, ...rest });
-                    // get().ai.addMessages([
-                    //   {
-                    //     id: createId(),
-                    //     role: 'tool',
-                    //     content: [],
-                    //     // @ts-ignore
-                    //     tool_call_id: toolCalls[toolCalls.length - 1].toolCallId,
-                    //   } satisfies AiMessage,
-                    // ]);
-                    // set(
-                    //   makeResultsAppender({
-                    //     resultId,
-                    //     toolResults,
-                    //     toolCalls: rest.toolCalls,
-                    //   }),
-                    // );
-                }
-                catch (err) {
-                    set(makeResultsAppender({
+                    await executeAnalysis({
                         resultId,
-                        toolResults: [
-                            {
-                                toolName: 'answer',
-                                toolCallId: createId(),
-                                args: {},
-                                result: {
-                                    success: false,
-                                    error: err instanceof Error ? err.message : String(err),
-                                },
-                            },
-                        ],
-                        toolCalls: [],
-                    }));
+                        prompt: get().ai.analysisPrompt,
+                        modelProvider: get().project.config.ai.modelProvider,
+                        model: get().project.config.ai.model,
+                        apiKey: getApiKey(),
+                        abortController,
+                        addMessages: get().ai.addMessages,
+                        set,
+                    });
                 }
                 finally {
                     set((state) => produce(state, (draft) => {
                         draft.ai.isRunningAnalysis = false;
+                        draft.ai.analysisPrompt = '';
                     }));
                 }
             },
@@ -154,19 +179,38 @@ function findResultById(analysisResults, id) {
     return analysisResults.find((r) => r.id === id);
 }
 /**
- * Returns a function that will update the state by appending new results
- * to the analysis results.
+ * Returns a function that will update the state by appending new results to the analysis results.
+ *
  * @param resultId - The result id
- * @param toolResults - The new tool results
- * @param toolCalls - The new tool calls
+ * @param toolCalls - The tool calls that were executed by the LLM, e.g. "query" or "chart" ("map" will be added soon). See {@link ToolCallSchema} for more details.
+ * @param toolResults - The results of the tool calls that were executed by the LLM. See {@link ToolResultSchema} for more details.
+ * @param toolCallMessages - The tool call messages that were created by some of our defined TOOLS, e.g. the table with query result. It's an array of React/JSX elements. It is linked to the tool call by the toolCallId.
+ * @param analysis - The analysis is the content generated after all the tool calls have been executed
+ * @param isCompleted - Whether the analysis is completed
  * @returns The new state
  */
-function makeResultsAppender({ resultId, toolResults, toolCalls, }) {
+function makeResultsAppender({ resultId, toolResults, toolCalls, analysis, isCompleted, toolCallMessages, }) {
     return (state) => produce(state, (draft) => {
         const result = findResultById(draft.project.config.ai.analysisResults, resultId);
         if (result) {
-            result.toolResults = [...result.toolResults, ...toolResults];
-            result.toolCalls = [...result.toolCalls, ...toolCalls];
+            if (toolResults) {
+                result.toolResults = [...result.toolResults, ...toolResults];
+            }
+            if (toolCalls) {
+                result.toolCalls = [...result.toolCalls, ...toolCalls];
+            }
+            if (toolCallMessages) {
+                result.toolCallMessages = [
+                    ...result.toolCallMessages,
+                    ...toolCallMessages,
+                ];
+            }
+            if (analysis) {
+                result.analysis = analysis;
+            }
+            if (isCompleted) {
+                result.isCompleted = isCompleted;
+            }
         }
         else {
             console.error('Result not found', resultId);

package/dist/AiSlice.js.map

@@ -1 +1 @@
-{"version":3,"file":"AiSlice.js","sourceRoot":"","sources":["../src/AiSlice.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,QAAQ,EAAC,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EACL,WAAW,EAEX,mBAAmB,GAEpB,MAAM,2BAA2B,CAAC;AAQnC,OAAO,EAAC,OAAO,EAAC,MAAM,OAAO,CAAC;AAC9B,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EAAC,WAAW,EAAC,MAAM,YAAY,CAAC;AACvC,OAAO,EACL,oBAAoB,GAGrB,MAAM,WAAW,CAAC;AAMnB,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC;QACX,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;QACjB,eAAe,EAAE,CAAC,CAAC,KAAK,CAAC,oBAAoB,CAAC;KAC/C,CAAC;CACH,CAAC,CAAC;AAGH,MAAM,UAAU,qBAAqB;IACnC,OAAO;QACL,EAAE,EAAE;YACF,KAAK,EAAE,aAAa;YACpB,eAAe,EAAE,EAAE;SACpB;KACF,CAAC;AACJ,CAAC;AAgBD,MAAM,UAAU,aAAa,CAA+C,EAC1E,WAAW,EACX,qBAAqB,GAAG,wGAAwG,GAIjI;IACC,OAAO,WAAW,CAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QAClD,EAAE,EAAE;YACF,cAAc,EAAE,qBAAqB;YACrC,iBAAiB,EAAE,KAAK;YACxB,YAAY,EAAE,IAAI,GAAG,EAAE;YACvB,MAAM,EAAE,IAAI;YAEZ,iBAAiB,EAAE,CAAC,MAAc,EAAE,EAAE;gBACpC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,EAAE,CAAC,cAAc,GAAG,MAAM,CAAC;gBACnC,CAAC,CAAC,CACH,CAAC;YACJ,CAAC;YAED,UAAU,EAAE,CAAC,KAAa,EAAE,EAAE;gBAC5B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,GAAG,KAAK,CAAC;gBACxC,CAAC,CAAC,CACH,CAAC;YACJ,CAAC;YAED;;;eAGG;YACH,WAAW,EAAE,CAAC,QAAqB,EAAE,EAAE;gBACrC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;oBACZ,MAAM,WAAW,GAAG,QAAQ,CAAC,MAAM,CACjC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CACxC,CAAC;oBACF,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC;oBACvD,KAAK,MAAM,CAAC,IAAI,WAAW,EAAE,CAAC;wBAC5B,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;4BACV,OAAO,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC,CAAC,CAAC;wBACvC,CAAC;wBACD,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;oBAC/B,CAAC;oBACD,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;oBACrE,OAAO;wBACL,EAAE,EAAE;4BACF,GAAG,KAAK,CAAC,EAAE;4BACX,YAAY,EAAE,eAAe;yBAC9B;qBACF,CAAC;gBACJ,CAAC,CAAC,CAAC;YACL,CAAC;YAED,WAAW,EAAE,GAAG,EAAE;gBAChB,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;YACpD,CAAC;YACD,aAAa,EAAE,KAAK,IAAI,EAAE;gBACxB,MAAM,QAAQ,GAAG,QAAQ,EAAE,CAAC;gBAC5B,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;gBAC9C,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,EAAE,CAAC,uBAAuB,GAAG,eAAe,CAAC;oBACnD,KAAK,CAAC,EAAE,CAAC,iBAAiB,GAAG,IAAI,CAAC;oBAClC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;wBAC3C,EAAE,EAAE,QAAQ;wBACZ,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,cAAc;wBAC/B,WAAW,EAAE,EAAE;wBACf,SAAS,EAAE,EAAE;qBACd,CAAC,CAAC;gBACL,CAAC,CAAC,CACH,CAAC;gBACF,GAAG,EAAE,CAAC,EAAE,CAAC,WAAW,CAAC;oBACnB;wBACE,EAAE,EAAE,QAAQ,EAAE;wBACd,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,cAAc;qBACjC;iBACF,CAAC,CAAC;gBACH,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,EAAE,CAAC,cAAc,GAAG,EAAE,CAAC;gBAC/B,CAAC,CAAC,CACH,CAAC;gBACF,IAAI,CAAC;oBACH,MAAM,EAAC,WAAW,EAAE,SAAS,EAAE,GAAG,IAAI,EAAC,GAAG,MAAM,WAAW,CAAC;wBAC1D,KAAK,EAAE,WAAW,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC;wBACjD,gCAAgC;wBAChC,QAAQ,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,WAAW,EAAE;wBAChC,YAAY,EAAE,CAAC,KAAK,EAAE,EAAE;4BACtB,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;4BACnC,GAAG,EAAE,CAAC,EAAE,CAAC,WAAW,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;4BAC9C,GAAG,CACD,mBAAmB,CAAC;gCAClB,QAAQ;gCACR,WAAW,EAAE,KAAK,CAAC,WAAW;gCAC9B,SAAS,EAAE,KAAK,CAAC,SAAS;6BAC3B,CAAC,CACH,CAAC;wBACJ,CAAC;wBACD,WAAW,EAAE,eAAe,CAAC,MAAM;qBACpC,CAAC,CAAC;oBACH,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,EAAC,WAAW,EAAE,SAAS,EAAE,GAAG,IAAI,EAAC,CAAC,CAAC;oBAC/D,yBAAyB;oBACzB,MAAM;oBACN,sBAAsB;oBACtB,oBAAoB;oBACpB,mBAAmB;oBACnB,oBAAoB;oBACpB,gEAAgE;oBAChE,2BAA2B;oBAC3B,MAAM;oBACN,OAAO;oBACP,0BAA0B;oBAC1B,gBAAgB;oBAChB,mBAAmB;oBACnB,iCAAiC;oBACjC,QAAQ;oBACR,KAAK;gBACP,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,GAAG,CACD,mBAAmB,CAAC;wBAClB,QAAQ;wBACR,WAAW,EAAE;4BACX;gCACE,QAAQ,EAAE,QAAQ;gCAClB,UAAU,EAAE,QAAQ,EAAE;gCACtB,IAAI,EAAE,EAAE;gCACR,MAAM,EAAE;oCACN,OAAO,EAAE,KAAK;oCACd,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;iCACxD;6BACF;yBACF;wBACD,SAAS,EAAE,EAAE;qBACd,CAAC,CACH,CAAC;gBACJ,CAAC;wBAAS,CAAC;oBACT,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,EAAE,CAAC,iBAAiB,GAAG,KAAK,CAAC;oBACrC,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,cAAc,EAAE,GAAG,EAAE;gBACnB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,EAAE,CAAC,iBAAiB,GAAG,KAAK,CAAC;gBACrC,CAAC,CAAC,CACH,CAAC;gBACF,GAAG,EAAE,CAAC,EAAE,CAAC,uBAAuB,EAAE,KAAK,CAAC,oBAAoB,CAAC,CAAC;YAChE,CAAC;SACF;KACF,CAAC,CAAC,CAAC;AACN,CAAC;AAED,SAAS,cAAc,CAAC,eAAuC,EAAE,EAAU;IACzE,OAAO,eAAe,CAAC,IAAI,CAAC,CAAC,CAAuB,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;AACxE,CAAC;AAED;;;;;;;GAOG;AACH,SAAS,mBAAmB,CAA+C,EACzE,QAAQ,EACR,WAAW,EACX,SAAS,GAKV;IACC,OAAO,CAAC,KAAuB,EAAE,EAAE,CACjC,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,cAAc,CAC3B,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,eAAe,EACvC,QAAQ,CACT,CAAC;QACF,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,WAAW,GAAG,CAAC,GAAG,MAAM,CAAC,WAAW,EAAE,GAAG,WAAW,CAAC,CAAC;YAC7D,MAAM,CAAC,SAAS,GAAG,CAAC,GAAG,MAAM,CAAC,SAAS,EAAE,GAAG,SAAS,CAAC,CAAC;QACzD,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,KAAK,CAAC,kBAAkB,EAAE,QAAQ,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC;AAKD,MAAM,UAAU,cAAc,CAC5B,QAA0C;IAE1C,OAAO,mBAAmB,CAIxB,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,KAAsC,CAAC,CAAC,CAAC;AACjE,CAAC","sourcesContent":["import {createId} from '@paralleldrive/cuid2';\nimport {\n  createSlice,\n  ProjectState,\n  useBaseProjectStore,\n  type StateCreator,\n} from '@sqlrooms/project-builder';\nimport {BaseProjectConfig} from '@sqlrooms/project-config';\nimport {\n  CoreAssistantMessage,\n  CoreToolMessage,\n  CoreUserMessage,\n  LanguageModelV1,\n} from 'ai';\nimport {produce} from 'immer';\nimport {z} from 'zod';\nimport {runAnalysis} from './analysis';\nimport {\n  AnalysisResultSchema,\n  ToolCallSchema,\n  ToolResultSchema,\n} from './schemas';\n\ntype AiMessage = (CoreToolMessage | CoreAssistantMessage | CoreUserMessage) & {\n  id: string;\n};\n\nexport const AiSliceConfig = z.object({\n  ai: z.object({\n    model: z.string(),\n    analysisResults: z.array(AnalysisResultSchema),\n  }),\n});\nexport type AiSliceConfig = z.infer<typeof AiSliceConfig>;\n\nexport function createDefaultAiConfig(): AiSliceConfig {\n  return {\n    ai: {\n      model: 'gpt-4o-mini',\n      analysisResults: [],\n    },\n  };\n}\n\nexport type AiSliceState = {\n  ai: {\n    analysisPrompt: string;\n    isRunningAnalysis: boolean;\n    analysisAbortController?: AbortController;\n    setAnalysisPrompt: (prompt: string) => void;\n    startAnalysis: () => Promise<void>;\n    cancelAnalysis: () => void;\n    messagesById: Map<string, AiMessage>;\n    addMessages: (messages: AiMessage[]) => void;\n    getMessages: () => AiMessage[];\n  };\n};\n\nexport function createAiSlice<PC extends BaseProjectConfig & AiSliceConfig>({\n  createModel,\n  initialAnalysisPrompt = 'Describe the data in the tables and make a chart providing an overview of the most important features.',\n}: {\n  createModel: (model: string) => LanguageModelV1;\n  initialAnalysisPrompt?: string;\n}): StateCreator<AiSliceState> {\n  return createSlice<PC, AiSliceState>((set, get) => ({\n    ai: {\n      analysisPrompt: initialAnalysisPrompt,\n      isRunningAnalysis: false,\n      messagesById: new Map(),\n      apiKey: null,\n\n      setAnalysisPrompt: (prompt: string) => {\n        set((state) =>\n          produce(state, (draft) => {\n            draft.ai.analysisPrompt = prompt;\n          }),\n        );\n      },\n\n      setAiModel: (model: string) => {\n        set((state) =>\n          produce(state, (draft) => {\n            draft.project.config.ai.model = model;\n          }),\n        );\n      },\n\n      /**\n       * Add messages to the project store uniquely by id\n       * @param messages - The messages to add.\n       */\n      addMessages: (messages: AiMessage[]) => {\n        set((state) => {\n          const newMessages = messages.filter(\n            (m) => !state.ai.messagesById.has(m.id),\n          );\n          const newMessagesById = new Map(state.ai.messagesById);\n          for (const m of newMessages) {\n            if (!m.id) {\n              console.warn('Message has no id', m);\n            }\n            newMessagesById.set(m.id, m);\n          }\n          console.log('newMessagesById', Array.from(newMessagesById.values()));\n          return {\n            ai: {\n              ...state.ai,\n              messagesById: newMessagesById,\n            },\n          };\n        });\n      },\n\n      getMessages: () => {\n        return Array.from(get().ai.messagesById.values());\n      },\n      startAnalysis: async () => {\n        const resultId = createId();\n        const abortController = new AbortController();\n        set((state) =>\n          produce(state, (draft) => {\n            draft.ai.analysisAbortController = abortController;\n            draft.ai.isRunningAnalysis = true;\n            draft.project.config.ai.analysisResults.push({\n              id: resultId,\n              prompt: get().ai.analysisPrompt,\n              toolResults: [],\n              toolCalls: [],\n            });\n          }),\n        );\n        get().ai.addMessages([\n          {\n            id: createId(),\n            role: 'user',\n            content: get().ai.analysisPrompt,\n          },\n        ]);\n        set((state) =>\n          produce(state, (draft) => {\n            draft.ai.analysisPrompt = '';\n          }),\n        );\n        try {\n          const {toolResults, toolCalls, ...rest} = await runAnalysis({\n            model: createModel(get().project.config.ai.model),\n            // prompt: get().analysisPrompt,\n            messages: get().ai.getMessages(),\n            onStepFinish: (event) => {\n              console.log('onStepFinish', event);\n              get().ai.addMessages(event.response.messages);\n              set(\n                makeResultsAppender({\n                  resultId,\n                  toolResults: event.toolResults,\n                  toolCalls: event.toolCalls,\n                }),\n              );\n            },\n            abortSignal: abortController.signal,\n          });\n          console.log('final result', {toolResults, toolCalls, ...rest});\n          // get().ai.addMessages([\n          //   {\n          //     id: createId(),\n          //     role: 'tool',\n          //     content: [],\n          //     // @ts-ignore\n          //     tool_call_id: toolCalls[toolCalls.length - 1].toolCallId,\n          //   } satisfies AiMessage,\n          // ]);\n          // set(\n          //   makeResultsAppender({\n          //     resultId,\n          //     toolResults,\n          //     toolCalls: rest.toolCalls,\n          //   }),\n          // );\n        } catch (err) {\n          set(\n            makeResultsAppender({\n              resultId,\n              toolResults: [\n                {\n                  toolName: 'answer',\n                  toolCallId: createId(),\n                  args: {},\n                  result: {\n                    success: false,\n                    error: err instanceof Error ? err.message : String(err),\n                  },\n                },\n              ],\n              toolCalls: [],\n            }),\n          );\n        } finally {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.ai.isRunningAnalysis = false;\n            }),\n          );\n        }\n      },\n      cancelAnalysis: () => {\n        set((state) =>\n          produce(state, (draft) => {\n            draft.ai.isRunningAnalysis = false;\n          }),\n        );\n        get().ai.analysisAbortController?.abort('Analysis cancelled');\n      },\n    },\n  }));\n}\n\nfunction findResultById(analysisResults: AnalysisResultSchema[], id: string) {\n  return analysisResults.find((r: AnalysisResultSchema) => r.id === id);\n}\n\n/**\n * Returns a function that will update the state by appending new results\n * to the analysis results.\n * @param resultId - The result id\n * @param toolResults - The new tool results\n * @param toolCalls - The new tool calls\n * @returns The new state\n */\nfunction makeResultsAppender<PC extends BaseProjectConfig & AiSliceConfig>({\n  resultId,\n  toolResults,\n  toolCalls,\n}: {\n  resultId: string;\n  toolResults: ToolResultSchema[];\n  toolCalls: ToolCallSchema[];\n}) {\n  return (state: ProjectState<PC>) =>\n    produce(state, (draft) => {\n      const result = findResultById(\n        draft.project.config.ai.analysisResults,\n        resultId,\n      );\n      if (result) {\n        result.toolResults = [...result.toolResults, ...toolResults];\n        result.toolCalls = [...result.toolCalls, ...toolCalls];\n      } else {\n        console.error('Result not found', resultId);\n      }\n    });\n}\n\ntype ProjectConfigWithAi = BaseProjectConfig & AiSliceConfig;\ntype ProjectStateWithAi = ProjectState<ProjectConfigWithAi> & AiSliceState;\n\nexport function useStoreWithAi<T>(\n  selector: (state: ProjectStateWithAi) => T,\n): T {\n  return useBaseProjectStore<\n    BaseProjectConfig & AiSliceConfig,\n    ProjectState<ProjectConfigWithAi>,\n    T\n  >((state) => selector(state as unknown as ProjectStateWithAi));\n}\n"]}
+{"version":3,"file":"AiSlice.js","sourceRoot":"","sources":["../src/AiSlice.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,QAAQ,EAAC,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EACL,WAAW,EAEX,mBAAmB,GAEpB,MAAM,2BAA2B,CAAC;AASnC,OAAO,EAAC,OAAO,EAAC,MAAM,OAAO,CAAC;AAC9B,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EAAC,WAAW,EAAC,MAAM,YAAY,CAAC;AACvC,OAAO,EACL,oBAAoB,GAGrB,MAAM,WAAW,CAAC;AAMnB,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC;QACX,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;QACzB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;QACjB,eAAe,EAAE,CAAC,CAAC,KAAK,CAAC,oBAAoB,CAAC;KAC/C,CAAC;CACH,CAAC,CAAC;AAGH,MAAM,UAAU,qBAAqB;IACnC,OAAO;QACL,EAAE,EAAE;YACF,aAAa,EAAE,QAAQ;YACvB,KAAK,EAAE,aAAa;YACpB,eAAe,EAAE,EAAE;SACpB;KACF,CAAC;AACJ,CAAC;AAiBD;;;;;;;;;;;;;GAaG;AACH,KAAK,UAAU,eAAe,CAAC,EAC7B,QAAQ,EACR,MAAM,EACN,aAAa,EACb,KAAK,EACL,MAAM,EACN,eAAe,EACf,WAAW,EACX,GAAG,GAUJ;IACC,IAAI,CAAC;QACH,MAAM,WAAW,CAAC;YAChB,aAAa;YACb,KAAK;YACL,MAAM;YACN,MAAM;YACN,eAAe;YACf,YAAY,EAAE,CACZ,KAA0B,EAC1B,gBAAmC,EACnC,EAAE;gBACF,WAAW,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;gBACrC,GAAG,CACD,mBAAmB,CAAC;oBAClB,QAAQ;oBACR,WAAW,EAAE,KAAK,CAAC,WAAW;oBAC9B,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,gBAAgB;iBACjB,CAAC,CACH,CAAC;YACJ,CAAC;YACD,cAAc,EAAE,CAAC,OAAO,EAAE,WAAW,EAAE,EAAE;gBACvC,GAAG,CACD,mBAAmB,CAAC;oBAClB,QAAQ;oBACR,QAAQ,EAAE,OAAO;oBACjB,WAAW;iBACZ,CAAC,CACH,CAAC;YACJ,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,GAAG,CACD,mBAAmB,CAAC;YAClB,QAAQ;YACR,WAAW,EAAE,IAAI;YACjB,WAAW,EAAE;gBACX;oBACE,QAAQ,EAAE,OAAO;oBACjB,UAAU,EAAE,QAAQ,EAAE;oBACtB,IAAI,EAAE,EAAE;oBACR,MAAM,EAAE;wBACN,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;qBACxD;iBACF;aACF;YACD,SAAS,EAAE,EAAE;SACd,CAAC,CACH,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAA+C,EAC1E,SAAS,EACT,qBAAqB,GAAG,wGAAwG,GAIjI;IACC,OAAO,WAAW,CAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QAClD,EAAE,EAAE;YACF,cAAc,EAAE,qBAAqB;YACrC,iBAAiB,EAAE,KAAK;YACxB,YAAY,EAAE,IAAI,GAAG,EAAE;YAEvB,iBAAiB,EAAE,CAAC,MAAc,EAAE,EAAE;gBACpC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,EAAE,CAAC,cAAc,GAAG,MAAM,CAAC;gBACnC,CAAC,CAAC,CACH,CAAC;YACJ,CAAC;YAED;;;eAGG;YACH,UAAU,EAAE,CAAC,KAAa,EAAE,EAAE;gBAC5B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,GAAG,KAAK,CAAC;gBACxC,CAAC,CAAC,CACH,CAAC;YACJ,CAAC;YAED;;;eAGG;YACH,WAAW,EAAE,CAAC,QAAqB,EAAE,EAAE;gBACrC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;oBACZ,MAAM,WAAW,GAAG,QAAQ,CAAC,MAAM,CACjC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CACxC,CAAC;oBACF,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC;oBACvD,KAAK,MAAM,CAAC,IAAI,WAAW,EAAE,CAAC;wBAC5B,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;4BACV,OAAO,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC,CAAC,CAAC;wBACvC,CAAC;wBACD,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;oBAC/B,CAAC;oBACD,OAAO;wBACL,EAAE,EAAE;4BACF,GAAG,KAAK,CAAC,EAAE;4BACX,YAAY,EAAE,eAAe;yBAC9B;qBACF,CAAC;gBACJ,CAAC,CAAC,CAAC;YACL,CAAC;YAED,WAAW,EAAE,GAAG,EAAE;gBAChB,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;YACpD,CAAC;YAED,aAAa,EAAE,KAAK,IAAI,EAAE;gBACxB,MAAM,QAAQ,GAAG,QAAQ,EAAE,CAAC;gBAC5B,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;gBAC9C,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,EAAE,CAAC,uBAAuB,GAAG,eAAe,CAAC;oBACnD,KAAK,CAAC,EAAE,CAAC,iBAAiB,GAAG,IAAI,CAAC;oBAClC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;wBAC3C,EAAE,EAAE,QAAQ;wBACZ,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,cAAc;wBAC/B,WAAW,EAAE,EAAE;wBACf,SAAS,EAAE,EAAE;wBACb,gBAAgB,EAAE,EAAE;wBACpB,QAAQ,EAAE,EAAE;wBACZ,WAAW,EAAE,KAAK;qBACnB,CAAC,CAAC;gBACL,CAAC,CAAC,CACH,CAAC;gBAEF,GAAG,EAAE,CAAC,EAAE,CAAC,WAAW,CAAC;oBACnB;wBACE,EAAE,EAAE,QAAQ,EAAE;wBACd,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,cAAc;qBACjC;iBACF,CAAC,CAAC;gBAEH,IAAI,CAAC;oBACH,MAAM,eAAe,CAAC;wBACpB,QAAQ;wBACR,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,cAAc;wBAC/B,aAAa,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,aAAa;wBACpD,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK;wBACpC,MAAM,EAAE,SAAS,EAAE;wBACnB,eAAe;wBACf,WAAW,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,WAAW;wBACjC,GAAG;qBACJ,CAAC,CAAC;gBACL,CAAC;wBAAS,CAAC;oBACT,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;wBACvB,KAAK,CAAC,EAAE,CAAC,iBAAiB,GAAG,KAAK,CAAC;wBACnC,KAAK,CAAC,EAAE,CAAC,cAAc,GAAG,EAAE,CAAC;oBAC/B,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,cAAc,EAAE,GAAG,EAAE;gBACnB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACvB,KAAK,CAAC,EAAE,CAAC,iBAAiB,GAAG,KAAK,CAAC;gBACrC,CAAC,CAAC,CACH,CAAC;gBACF,GAAG,EAAE,CAAC,EAAE,CAAC,uBAAuB,EAAE,KAAK,CAAC,oBAAoB,CAAC,CAAC;YAChE,CAAC;SACF;KACF,CAAC,CAAC,CAAC;AACN,CAAC;AAED,SAAS,cAAc,CAAC,eAAuC,EAAE,EAAU;IACzE,OAAO,eAAe,CAAC,IAAI,CAAC,CAAC,CAAuB,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;AACxE,CAAC;AAED;;;;;;;;;;GAUG;AACH,SAAS,mBAAmB,CAA+C,EACzE,QAAQ,EACR,WAAW,EACX,SAAS,EACT,QAAQ,EACR,WAAW,EACX,gBAAgB,GAQjB;IACC,OAAO,CAAC,KAAuB,EAAE,EAAE,CACjC,OAAO,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,cAAc,CAC3B,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,eAAe,EACvC,QAAQ,CACT,CAAC;QACF,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,WAAW,EAAE,CAAC;gBAChB,MAAM,CAAC,WAAW,GAAG,CAAC,GAAG,MAAM,CAAC,WAAW,EAAE,GAAG,WAAW,CAAC,CAAC;YAC/D,CAAC;YACD,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,CAAC,SAAS,GAAG,CAAC,GAAG,MAAM,CAAC,SAAS,EAAE,GAAG,SAAS,CAAC,CAAC;YACzD,CAAC;YACD,IAAI,gBAAgB,EAAE,CAAC;gBACrB,MAAM,CAAC,gBAAgB,GAAG;oBACxB,GAAG,MAAM,CAAC,gBAAgB;oBAC1B,GAAG,gBAAgB;iBACpB,CAAC;YACJ,CAAC;YACD,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,CAAC,QAAQ,GAAG,QAAQ,CAAC;YAC7B,CAAC;YACD,IAAI,WAAW,EAAE,CAAC;gBAChB,MAAM,CAAC,WAAW,GAAG,WAAW,CAAC;YACnC,CAAC;QACH,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,KAAK,CAAC,kBAAkB,EAAE,QAAQ,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC;AAKD,MAAM,UAAU,cAAc,CAC5B,QAA0C;IAE1C,OAAO,mBAAmB,CAIxB,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,KAAsC,CAAC,CAAC,CAAC;AACjE,CAAC","sourcesContent":["import {createId} from '@paralleldrive/cuid2';\nimport {\n  createSlice,\n  ProjectState,\n  useBaseProjectStore,\n  type StateCreator,\n} from '@sqlrooms/project-builder';\nimport {BaseProjectConfig} from '@sqlrooms/project-config';\nimport {\n  CoreAssistantMessage,\n  CoreToolMessage,\n  CoreUserMessage,\n  StepResult,\n  ToolSet,\n} from 'ai';\nimport {produce} from 'immer';\nimport {z} from 'zod';\nimport {runAnalysis} from './analysis';\nimport {\n  AnalysisResultSchema,\n  ToolCallSchema,\n  ToolResultSchema,\n} from './schemas';\nimport {ToolCallMessage} from '@openassistant/core';\ntype AiMessage = (CoreToolMessage | CoreAssistantMessage | CoreUserMessage) & {\n  id: string;\n};\n\nexport const AiSliceConfig = z.object({\n  ai: z.object({\n    modelProvider: z.string(),\n    model: z.string(),\n    analysisResults: z.array(AnalysisResultSchema),\n  }),\n});\nexport type AiSliceConfig = z.infer<typeof AiSliceConfig>;\n\nexport function createDefaultAiConfig(): AiSliceConfig {\n  return {\n    ai: {\n      modelProvider: 'openai',\n      model: 'gpt-4o-mini',\n      analysisResults: [],\n    },\n  };\n}\n\nexport type AiSliceState = {\n  ai: {\n    analysisPrompt: string;\n    isRunningAnalysis: boolean;\n    analysisAbortController?: AbortController;\n    setAnalysisPrompt: (prompt: string) => void;\n    startAnalysis: () => Promise<void>;\n    cancelAnalysis: () => void;\n    messagesById: Map<string, AiMessage>;\n    addMessages: (messages: AiMessage[]) => void;\n    getMessages: () => AiMessage[];\n    setAiModel: (model: string) => void;\n  };\n};\n\n/**\n * Execute the analysis. It will be used by the action `startAnalysis`.\n *\n * Each analysis contains an array of toolCalls and the results of the tool calls (toolResults).\n * After all the tool calls have been executed, the LLM will stream the results as text stored in `analysis`.\n *\n * @param resultId - The result id\n * @param prompt - The prompt\n * @param model - The model\n * @param apiKey - The api key\n * @param abortController - The abort controller\n * @param addMessages - The add messages function\n * @param set - The set function\n */\nasync function executeAnalysis({\n  resultId,\n  prompt,\n  modelProvider,\n  model,\n  apiKey,\n  abortController,\n  addMessages,\n  set,\n}: {\n  resultId: string;\n  prompt: string;\n  modelProvider: string;\n  model: string;\n  apiKey: string;\n  abortController: AbortController;\n  addMessages: (messages: AiMessage[]) => void;\n  set: <T>(fn: (state: T) => T) => void;\n}) {\n  try {\n    await runAnalysis({\n      modelProvider,\n      model,\n      apiKey,\n      prompt,\n      abortController,\n      onStepFinish: (\n        event: StepResult<ToolSet>,\n        toolCallMessages: ToolCallMessage[],\n      ) => {\n        addMessages(event.response.messages);\n        set(\n          makeResultsAppender({\n            resultId,\n            toolResults: event.toolResults,\n            toolCalls: event.toolCalls,\n            toolCallMessages,\n          }),\n        );\n      },\n      onStreamResult: (message, isCompleted) => {\n        set(\n          makeResultsAppender({\n            resultId,\n            analysis: message,\n            isCompleted,\n          }),\n        );\n      },\n    });\n  } catch (err) {\n    set(\n      makeResultsAppender({\n        resultId,\n        isCompleted: true,\n        toolResults: [\n          {\n            toolName: 'error',\n            toolCallId: createId(),\n            args: {},\n            result: {\n              success: false,\n              error: err instanceof Error ? err.message : String(err),\n            },\n          },\n        ],\n        toolCalls: [],\n      }),\n    );\n  }\n}\n\nexport function createAiSlice<PC extends BaseProjectConfig & AiSliceConfig>({\n  getApiKey,\n  initialAnalysisPrompt = 'Describe the data in the tables and make a chart providing an overview of the most important features.',\n}: {\n  getApiKey: () => string;\n  initialAnalysisPrompt?: string;\n}): StateCreator<AiSliceState> {\n  return createSlice<PC, AiSliceState>((set, get) => ({\n    ai: {\n      analysisPrompt: initialAnalysisPrompt,\n      isRunningAnalysis: false,\n      messagesById: new Map(),\n\n      setAnalysisPrompt: (prompt: string) => {\n        set((state) =>\n          produce(state, (draft) => {\n            draft.ai.analysisPrompt = prompt;\n          }),\n        );\n      },\n\n      /**\n       * Set the AI model\n       * @param model - The model to set\n       */\n      setAiModel: (model: string) => {\n        set((state) =>\n          produce(state, (draft) => {\n            draft.project.config.ai.model = model;\n          }),\n        );\n      },\n\n      /**\n       * Add messages to the project store uniquely by id\n       * @param messages - The messages to add.\n       */\n      addMessages: (messages: AiMessage[]) => {\n        set((state) => {\n          const newMessages = messages.filter(\n            (m) => !state.ai.messagesById.has(m.id),\n          );\n          const newMessagesById = new Map(state.ai.messagesById);\n          for (const m of newMessages) {\n            if (!m.id) {\n              console.warn('Message has no id', m);\n            }\n            newMessagesById.set(m.id, m);\n          }\n          return {\n            ai: {\n              ...state.ai,\n              messagesById: newMessagesById,\n            },\n          };\n        });\n      },\n\n      getMessages: () => {\n        return Array.from(get().ai.messagesById.values());\n      },\n\n      startAnalysis: async () => {\n        const resultId = createId();\n        const abortController = new AbortController();\n        set((state) =>\n          produce(state, (draft) => {\n            draft.ai.analysisAbortController = abortController;\n            draft.ai.isRunningAnalysis = true;\n            draft.project.config.ai.analysisResults.push({\n              id: resultId,\n              prompt: get().ai.analysisPrompt,\n              toolResults: [],\n              toolCalls: [],\n              toolCallMessages: [],\n              analysis: '',\n              isCompleted: false,\n            });\n          }),\n        );\n\n        get().ai.addMessages([\n          {\n            id: createId(),\n            role: 'user',\n            content: get().ai.analysisPrompt,\n          },\n        ]);\n\n        try {\n          await executeAnalysis({\n            resultId,\n            prompt: get().ai.analysisPrompt,\n            modelProvider: get().project.config.ai.modelProvider,\n            model: get().project.config.ai.model,\n            apiKey: getApiKey(),\n            abortController,\n            addMessages: get().ai.addMessages,\n            set,\n          });\n        } finally {\n          set((state) =>\n            produce(state, (draft) => {\n              draft.ai.isRunningAnalysis = false;\n              draft.ai.analysisPrompt = '';\n            }),\n          );\n        }\n      },\n      cancelAnalysis: () => {\n        set((state) =>\n          produce(state, (draft) => {\n            draft.ai.isRunningAnalysis = false;\n          }),\n        );\n        get().ai.analysisAbortController?.abort('Analysis cancelled');\n      },\n    },\n  }));\n}\n\nfunction findResultById(analysisResults: AnalysisResultSchema[], id: string) {\n  return analysisResults.find((r: AnalysisResultSchema) => r.id === id);\n}\n\n/**\n * Returns a function that will update the state by appending new results to the analysis results.\n *\n * @param resultId - The result id\n * @param toolCalls - The tool calls that were executed by the LLM, e.g. \"query\" or \"chart\" (\"map\" will be added soon). See {@link ToolCallSchema} for more details.\n * @param toolResults - The results of the tool calls that were executed by the LLM. See {@link ToolResultSchema} for more details.\n * @param toolCallMessages - The tool call messages that were created by some of our defined TOOLS, e.g. the table with query result. It's an array of React/JSX elements. It is linked to the tool call by the toolCallId.\n * @param analysis - The analysis is the content generated after all the tool calls have been executed\n * @param isCompleted - Whether the analysis is completed\n * @returns The new state\n */\nfunction makeResultsAppender<PC extends BaseProjectConfig & AiSliceConfig>({\n  resultId,\n  toolResults,\n  toolCalls,\n  analysis,\n  isCompleted,\n  toolCallMessages,\n}: {\n  resultId: string;\n  toolResults?: ToolResultSchema[];\n  toolCalls?: ToolCallSchema[];\n  analysis?: string;\n  isCompleted?: boolean;\n  toolCallMessages?: ToolCallMessage[];\n}) {\n  return (state: ProjectState<PC>) =>\n    produce(state, (draft) => {\n      const result = findResultById(\n        draft.project.config.ai.analysisResults,\n        resultId,\n      );\n      if (result) {\n        if (toolResults) {\n          result.toolResults = [...result.toolResults, ...toolResults];\n        }\n        if (toolCalls) {\n          result.toolCalls = [...result.toolCalls, ...toolCalls];\n        }\n        if (toolCallMessages) {\n          result.toolCallMessages = [\n            ...result.toolCallMessages,\n            ...toolCallMessages,\n          ];\n        }\n        if (analysis) {\n          result.analysis = analysis;\n        }\n        if (isCompleted) {\n          result.isCompleted = isCompleted;\n        }\n      } else {\n        console.error('Result not found', resultId);\n      }\n    });\n}\n\ntype ProjectConfigWithAi = BaseProjectConfig & AiSliceConfig;\ntype ProjectStateWithAi = ProjectState<ProjectConfigWithAi> & AiSliceState;\n\nexport function useStoreWithAi<T>(\n  selector: (state: ProjectStateWithAi) => T,\n): T {\n  return useBaseProjectStore<\n    BaseProjectConfig & AiSliceConfig,\n    ProjectState<ProjectConfigWithAi>,\n    T\n  >((state) => selector(state as unknown as ProjectStateWithAi));\n}\n"]}

package/dist/AnalysisResult.d.ts

@@ -1,7 +1,21 @@
 import { AnalysisResultSchema } from './schemas';
-interface AnalysisResultProps {
+/**
+ * Props for the AnalysisResult component
+ * @property {AnalysisResultSchema} result - The result of the analysis containing prompt, tool calls, and analysis data
+ */
+type AnalysisResultProps = {
     result: AnalysisResultSchema;
-}
+};
+/**
+ * Component that displays the results of an AI analysis.
+ * Shows the original prompt, intermediate tool calls, final analysis text,
+ * and any tool results.
+ *
+ * @component
+ * @param props - Component props
+ * @param props.result - The analysis result data to display
+ * @returns A React component displaying the analysis results
+ */
 export declare const AnalysisResult: React.FC<AnalysisResultProps>;
 export {};
 //# sourceMappingURL=AnalysisResult.d.ts.map

package/dist/AnalysisResult.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AnalysisResult.d.ts","sourceRoot":"","sources":["../src/AnalysisResult.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAC,oBAAoB,EAAC,MAAM,WAAW,CAAC;AAM/C,UAAU,mBAAmB;IAC3B,MAAM,EAAE,oBAAoB,CAAC;CAC9B;AAED,eAAO,MAAM,cAAc,EAAE,KAAK,CAAC,EAAE,CAAC,mBAAmB,CAwCxD,CAAC"}
+{"version":3,"file":"AnalysisResult.d.ts","sourceRoot":"","sources":["../src/AnalysisResult.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAC,oBAAoB,EAAC,MAAM,WAAW,CAAC;AAK/C;;;GAGG;AACH,KAAK,mBAAmB,GAAG;IACzB,MAAM,EAAE,oBAAoB,CAAC;CAC9B,CAAC;AA8BF;;;;;;;;;GASG;AACH,eAAO,MAAM,cAAc,EAAE,KAAK,CAAC,EAAE,CAAC,mBAAmB,CAsDxD,CAAC"}