@sqlrooms/ai-core 0.26.0-rc.2 → 0.26.0-rc.3

package/dist/hooks/useAiChat.js

@@ -0,0 +1,92 @@
+import { useEffect, useMemo } from 'react';
+import { useChat } from '@ai-sdk/react';
+import { lastAssistantMessageIsCompleteWithToolCalls, } from 'ai';
+import { useStoreWithAi } from '../AiSlice';
+/**
+ * Custom hook that provides AI chat functionality with automatic transport setup,
+ * message syncing, and tool call handling.
+ *
+ * This hook encapsulates all the logic needed to integrate the AI SDK's useChat
+ * with the AI slice state management.
+ *
+ * @returns An object containing messages and sendMessage from useChat
+ *
+ * @example
+ * ```tsx
+ * function MyComponent() {
+ *   const {messages, sendMessage} = useAiChat();
+ *
+ *   const handleSubmit = () => {
+ *     sendMessage({text: 'Hello!'});
+ *   };
+ *
+ *   return (
+ *     <div>
+ *       {messages.map(msg => <div key={msg.id}>{msg.content}</div>)}
+ *     </div>
+ *   );
+ * }
+ * ```
+ */
+export function useAiChat() {
+    // Get current session and configuration
+    const currentSession = useStoreWithAi((s) => s.ai.getCurrentSession());
+    const sessionId = currentSession?.id;
+    const model = currentSession?.model;
+    // Use messagesRevision to force reset only when messages are explicitly deleted
+    const messagesRevision = currentSession?.messagesRevision ?? 0;
+    // Get chat transport configuration
+    const getLocalChatTransport = useStoreWithAi((s) => s.ai.getLocalChatTransport);
+    const getRemoteChatTransport = useStoreWithAi((s) => s.ai.getRemoteChatTransport);
+    const endPoint = useStoreWithAi((s) => s.ai.chatEndPoint);
+    const headers = useStoreWithAi((s) => s.ai.chatHeaders);
+    // Get chat handlers
+    const onChatToolCall = useStoreWithAi((s) => s.ai.onChatToolCall);
+    const onChatFinish = useStoreWithAi((s) => s.ai.onChatFinish);
+    const onChatData = useStoreWithAi((s) => s.ai.onChatData);
+    const onChatError = useStoreWithAi((s) => s.ai.onChatError);
+    const setSessionUiMessages = useStoreWithAi((s) => s.ai.setSessionUiMessages);
+    // Create transport (recreate when model changes)
+    const transport = useMemo(() => {
+        // Recreate transport when the model changes
+        void model;
+        const trimmed = (endPoint || '').trim();
+        if (trimmed.length > 0) {
+            return getRemoteChatTransport(trimmed, headers);
+        }
+        return getLocalChatTransport();
+    }, [getLocalChatTransport, getRemoteChatTransport, headers, endPoint, model]);
+    // Setup useChat with all configuration
+    // Include messagesRevision in the id to force reset only when messages are explicitly deleted
+    // Store addToolResult in a variable that can be captured by the onToolCall closure
+    let capturedAddToolResult;
+    const { messages, sendMessage, addToolResult } = useChat({
+        id: `${sessionId}-${messagesRevision}`,
+        transport,
+        messages: currentSession?.uiMessages ?? [],
+        onToolCall: async ({ toolCall }) => {
+            // Wrap the store's onChatToolCall to provide addToolResult
+            // Use the captured addToolResult from the outer scope
+            return onChatToolCall?.({ toolCall, addToolResult: capturedAddToolResult });
+        },
+        onFinish: onChatFinish,
+        onError: onChatError,
+        onData: onChatData,
+        // Automatically submit when all tool results are available
+        // NOTE: When using sendAutomaticallyWhen, don't use await with addToolResult inside onChatToolCall as it can cause deadlocks.
+        sendAutomaticallyWhen: lastAssistantMessageIsCompleteWithToolCalls,
+    });
+    // Capture addToolResult for use in onToolCall
+    capturedAddToolResult = addToolResult;
+    // Sync streaming updates into the store so UiMessages renders incrementally
+    useEffect(() => {
+        if (!sessionId)
+            return;
+        setSessionUiMessages(sessionId, messages);
+    }, [messages, sessionId, setSessionUiMessages]);
+    return {
+        messages,
+        sendMessage,
+    };
+}
+//# sourceMappingURL=useAiChat.js.map

package/dist/hooks/useAiChat.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useAiChat.js","sourceRoot":"","sources":["../../src/hooks/useAiChat.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,SAAS,EAAE,OAAO,EAAC,MAAM,OAAO,CAAC;AACzC,OAAO,EAAC,OAAO,EAAC,MAAM,eAAe,CAAC;AACtC,OAAO,EAEL,2CAA2C,GAC5C,MAAM,IAAI,CAAC;AAEZ,OAAO,EAAC,cAAc,EAAC,MAAM,YAAY,CAAC;AAa1C;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,MAAM,UAAU,SAAS;IACvB,wCAAwC;IACxC,MAAM,cAAc,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC;IACvE,MAAM,SAAS,GAAG,cAAc,EAAE,EAAE,CAAC;IACrC,MAAM,KAAK,GAAG,cAAc,EAAE,KAAK,CAAC;IACpC,gFAAgF;IAChF,MAAM,gBAAgB,GAAG,cAAc,EAAE,gBAAgB,IAAI,CAAC,CAAC;IAE/D,mCAAmC;IACnC,MAAM,qBAAqB,GAAG,cAAc,CAC1C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,qBAAqB,CAClC,CAAC;IACF,MAAM,sBAAsB,GAAG,cAAc,CAC3C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,sBAAsB,CACnC,CAAC;IACF,MAAM,QAAQ,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC;IAC1D,MAAM,OAAO,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC;IAExD,oBAAoB;IACpB,MAAM,cAAc,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,cAAc,CAAC,CAAC;IAClE,MAAM,YAAY,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC;IAC9D,MAAM,UAAU,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC;IAC1D,MAAM,WAAW,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC;IAC5D,MAAM,oBAAoB,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,oBAAoB,CAAC,CAAC;IAE9E,iDAAiD;IACjD,MAAM,SAAS,GAAoC,OAAO,CAAC,GAAG,EAAE;QAC9D,4CAA4C;QAC5C,KAAK,KAAK,CAAC;QACX,MAAM,OAAO,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACxC,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,OAAO,sBAAsB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAClD,CAAC;QACD,OAAO,qBAAqB,EAAE,CAAC;IACjC,CAAC,EAAE,CAAC,qBAAqB,EAAE,sBAAsB,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC;IAE9E,uCAAuC;IACvC,8FAA8F;IAC9F,mFAAmF;IACnF,IAAI,qBAAoC,CAAC;IAEzC,MAAM,EAAC,QAAQ,EAAE,WAAW,EAAE,aAAa,EAAC,GAAG,OAAO,CAAC;QACrD,EAAE,EAAE,GAAG,SAAS,IAAI,gBAAgB,EAAE;QACtC,SAAS;QACT,QAAQ,EAAG,cAAc,EAAE,UAAqC,IAAI,EAAE;QACtE,UAAU,EAAE,KAAK,EAAE,EAAC,QAAQ,EAAkB,EAAE,EAAE;YAChD,2DAA2D;YAC3D,sDAAsD;YACtD,OAAO,cAAc,EAAE,CAAC,EAAC,QAAQ,EAAE,aAAa,EAAE,qBAAqB,EAAC,CAAC,CAAC;QAC5E,CAAC;QACD,QAAQ,EAAE,YAAY;QACtB,OAAO,EAAE,WAAW;QACpB,MAAM,EAAE,UAAU;QAClB,2DAA2D;QAC3D,8HAA8H;QAC9H,qBAAqB,EAAE,2CAA2C;KACnE,CAAC,CAAC;IAEH,8CAA8C;IAC9C,qBAAqB,GAAG,aAAa,CAAC;IAEtC,4EAA4E;IAC5E,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,SAAS;YAAE,OAAO;QACvB,oBAAoB,CAAC,SAAS,EAAE,QAAuB,CAAC,CAAC;IAC3D,CAAC,EAAE,CAAC,QAAQ,EAAE,SAAS,EAAE,oBAAoB,CAAC,CAAC,CAAC;IAEhD,OAAO;QACL,QAAQ;QACR,WAAW;KACZ,CAAC;AACJ,CAAC","sourcesContent":["import {useEffect, useMemo} from 'react';\nimport {useChat} from '@ai-sdk/react';\nimport {\n  DefaultChatTransport,\n  lastAssistantMessageIsCompleteWithToolCalls,\n} from 'ai';\nimport type {UIMessage} from 'ai';\nimport {useStoreWithAi} from '../AiSlice';\n\nexport type AddToolResult = (\n  options:\n    | {tool: string; toolCallId: string; output: unknown}\n    | {\n        tool: string;\n        toolCallId: string;\n        state: 'output-error';\n        errorText: string;\n      },\n) => void;\n\n/**\n * Custom hook that provides AI chat functionality with automatic transport setup,\n * message syncing, and tool call handling.\n *\n * This hook encapsulates all the logic needed to integrate the AI SDK's useChat\n * with the AI slice state management.\n *\n * @returns An object containing messages and sendMessage from useChat\n *\n * @example\n * ```tsx\n * function MyComponent() {\n *   const {messages, sendMessage} = useAiChat();\n *\n *   const handleSubmit = () => {\n *     sendMessage({text: 'Hello!'});\n *   };\n *\n *   return (\n *     <div>\n *       {messages.map(msg => <div key={msg.id}>{msg.content}</div>)}\n *     </div>\n *   );\n * }\n * ```\n */\nexport function useAiChat() {\n  // Get current session and configuration\n  const currentSession = useStoreWithAi((s) => s.ai.getCurrentSession());\n  const sessionId = currentSession?.id;\n  const model = currentSession?.model;\n  // Use messagesRevision to force reset only when messages are explicitly deleted\n  const messagesRevision = currentSession?.messagesRevision ?? 0;\n\n  // Get chat transport configuration\n  const getLocalChatTransport = useStoreWithAi(\n    (s) => s.ai.getLocalChatTransport,\n  );\n  const getRemoteChatTransport = useStoreWithAi(\n    (s) => s.ai.getRemoteChatTransport,\n  );\n  const endPoint = useStoreWithAi((s) => s.ai.chatEndPoint);\n  const headers = useStoreWithAi((s) => s.ai.chatHeaders);\n\n  // Get chat handlers\n  const onChatToolCall = useStoreWithAi((s) => s.ai.onChatToolCall);\n  const onChatFinish = useStoreWithAi((s) => s.ai.onChatFinish);\n  const onChatData = useStoreWithAi((s) => s.ai.onChatData);\n  const onChatError = useStoreWithAi((s) => s.ai.onChatError);\n  const setSessionUiMessages = useStoreWithAi((s) => s.ai.setSessionUiMessages);\n\n  // Create transport (recreate when model changes)\n  const transport: DefaultChatTransport<UIMessage> = useMemo(() => {\n    // Recreate transport when the model changes\n    void model;\n    const trimmed = (endPoint || '').trim();\n    if (trimmed.length > 0) {\n      return getRemoteChatTransport(trimmed, headers);\n    }\n    return getLocalChatTransport();\n  }, [getLocalChatTransport, getRemoteChatTransport, headers, endPoint, model]);\n\n  // Setup useChat with all configuration\n  // Include messagesRevision in the id to force reset only when messages are explicitly deleted\n  // Store addToolResult in a variable that can be captured by the onToolCall closure\n  let capturedAddToolResult: AddToolResult;\n\n  const {messages, sendMessage, addToolResult} = useChat({\n    id: `${sessionId}-${messagesRevision}`,\n    transport,\n    messages: (currentSession?.uiMessages as unknown as UIMessage[]) ?? [],\n    onToolCall: async ({toolCall}: {toolCall: any}) => {\n      // Wrap the store's onChatToolCall to provide addToolResult\n      // Use the captured addToolResult from the outer scope\n      return onChatToolCall?.({toolCall, addToolResult: capturedAddToolResult});\n    },\n    onFinish: onChatFinish,\n    onError: onChatError,\n    onData: onChatData,\n    // Automatically submit when all tool results are available\n    // NOTE: When using sendAutomaticallyWhen, don't use await with addToolResult inside onChatToolCall as it can cause deadlocks.\n    sendAutomaticallyWhen: lastAssistantMessageIsCompleteWithToolCalls,\n  });\n\n  // Capture addToolResult for use in onToolCall\n  capturedAddToolResult = addToolResult;\n\n  // Sync streaming updates into the store so UiMessages renders incrementally\n  useEffect(() => {\n    if (!sessionId) return;\n    setSessionUiMessages(sessionId, messages as UIMessage[]);\n  }, [messages, sessionId, setSessionUiMessages]);\n\n  return {\n    messages,\n    sendMessage,\n  };\n}\n"]}

package/dist/index.d.ts

@@ -7,7 +7,7 @@ export type { AiSliceState } from './AiSlice';
 export { AnalysisResultsContainer } from './components/AnalysisResultsContainer';
 export { AnalysisResult } from './components/AnalysisResult';
 export { useScrollToBottom } from './hooks/useScrollToBottom';
-export type { AiSliceTool } from './AiSlice';
+export { useAiChat } from './hooks/useAiChat';
 export { ModelSelector } from './components/ModelSelector';
 export { SessionControls } from './components/SessionControls';
 export { QueryControls } from './components/QueryControls';
@@ -17,5 +17,7 @@ export { SessionDropdown } from './components/session/SessionDropdown';
 export { SessionTitle } from './components/session/SessionTitle';
 export type { SessionType } from './components/session/SessionType';
 export { ToolErrorMessage } from './components/tools/ToolErrorMessage';
+export { ToolCallInfo } from './components/ToolCallInfo';
 export { AiSliceConfig, createDefaultAiConfig } from '@sqlrooms/ai-config';
+export { AiThinkingDots } from './components/AiThinkingDots';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAC,aAAa,EAAE,cAAc,EAAC,MAAM,WAAW,CAAC;AAExD,YAAY,EAAC,YAAY,EAAC,MAAM,WAAW,CAAC;AAC5C,OAAO,EAAC,wBAAwB,EAAC,MAAM,uCAAuC,CAAC;AAC/E,OAAO,EAAC,cAAc,EAAC,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAC,iBAAiB,EAAC,MAAM,2BAA2B,CAAC;AAC5D,YAAY,EAAC,WAAW,EAAC,MAAM,WAAW,CAAC;AAE3C,OAAO,EAAC,aAAa,EAAC,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAC,eAAe,EAAC,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAC,aAAa,EAAC,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAC,mBAAmB,EAAC,MAAM,0CAA0C,CAAC;AAC7E,OAAO,EAAC,cAAc,EAAC,MAAM,qCAAqC,CAAC;AACnE,OAAO,EAAC,eAAe,EAAC,MAAM,sCAAsC,CAAC;AACrE,OAAO,EAAC,YAAY,EAAC,MAAM,mCAAmC,CAAC;AAC/D,YAAY,EAAC,WAAW,EAAC,MAAM,kCAAkC,CAAC;AAClE,OAAO,EAAC,gBAAgB,EAAC,MAAM,qCAAqC,CAAC;AAErE,OAAO,EAAC,aAAa,EAAE,qBAAqB,EAAC,MAAM,qBAAqB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAC,aAAa,EAAE,cAAc,EAAC,MAAM,WAAW,CAAC;AAExD,YAAY,EAAC,YAAY,EAAC,MAAM,WAAW,CAAC;AAC5C,OAAO,EAAC,wBAAwB,EAAC,MAAM,uCAAuC,CAAC;AAC/E,OAAO,EAAC,cAAc,EAAC,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAC,iBAAiB,EAAC,MAAM,2BAA2B,CAAC;AAC5D,OAAO,EAAC,SAAS,EAAC,MAAM,mBAAmB,CAAC;AAE5C,OAAO,EAAC,aAAa,EAAC,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAC,eAAe,EAAC,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAC,aAAa,EAAC,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAC,mBAAmB,EAAC,MAAM,0CAA0C,CAAC;AAC7E,OAAO,EAAC,cAAc,EAAC,MAAM,qCAAqC,CAAC;AACnE,OAAO,EAAC,eAAe,EAAC,MAAM,sCAAsC,CAAC;AACrE,OAAO,EAAC,YAAY,EAAC,MAAM,mCAAmC,CAAC;AAC/D,YAAY,EAAC,WAAW,EAAC,MAAM,kCAAkC,CAAC;AAClE,OAAO,EAAC,gBAAgB,EAAC,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAC,YAAY,EAAC,MAAM,2BAA2B,CAAC;AAEvD,OAAO,EAAC,aAAa,EAAE,qBAAqB,EAAC,MAAM,qBAAqB,CAAC;AACzE,OAAO,EAAC,cAAc,EAAC,MAAM,6BAA6B,CAAC"}

package/dist/index.js

@@ -6,6 +6,7 @@ export { createAiSlice, useStoreWithAi } from './AiSlice';
 export { AnalysisResultsContainer } from './components/AnalysisResultsContainer';
 export { AnalysisResult } from './components/AnalysisResult';
 export { useScrollToBottom } from './hooks/useScrollToBottom';
+export { useAiChat } from './hooks/useAiChat';
 export { ModelSelector } from './components/ModelSelector';
 export { SessionControls } from './components/SessionControls';
 export { QueryControls } from './components/QueryControls';
@@ -14,5 +15,7 @@ export { SessionActions } from './components/session/SessionActions';
 export { SessionDropdown } from './components/session/SessionDropdown';
 export { SessionTitle } from './components/session/SessionTitle';
 export { ToolErrorMessage } from './components/tools/ToolErrorMessage';
+export { ToolCallInfo } from './components/ToolCallInfo';
 export { AiSliceConfig, createDefaultAiConfig } from '@sqlrooms/ai-config';
+export { AiThinkingDots } from './components/AiThinkingDots';
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAC,aAAa,EAAE,cAAc,EAAC,MAAM,WAAW,CAAC;AAGxD,OAAO,EAAC,wBAAwB,EAAC,MAAM,uCAAuC,CAAC;AAC/E,OAAO,EAAC,cAAc,EAAC,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAC,iBAAiB,EAAC,MAAM,2BAA2B,CAAC;AAG5D,OAAO,EAAC,aAAa,EAAC,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAC,eAAe,EAAC,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAC,aAAa,EAAC,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAC,mBAAmB,EAAC,MAAM,0CAA0C,CAAC;AAC7E,OAAO,EAAC,cAAc,EAAC,MAAM,qCAAqC,CAAC;AACnE,OAAO,EAAC,eAAe,EAAC,MAAM,sCAAsC,CAAC;AACrE,OAAO,EAAC,YAAY,EAAC,MAAM,mCAAmC,CAAC;AAE/D,OAAO,EAAC,gBAAgB,EAAC,MAAM,qCAAqC,CAAC;AAErE,OAAO,EAAC,aAAa,EAAE,qBAAqB,EAAC,MAAM,qBAAqB,CAAC","sourcesContent":["/**\n * {@include ../README.md}\n * @packageDocumentation\n */\n\nexport {createAiSlice, useStoreWithAi} from './AiSlice';\n\nexport type {AiSliceState} from './AiSlice';\nexport {AnalysisResultsContainer} from './components/AnalysisResultsContainer';\nexport {AnalysisResult} from './components/AnalysisResult';\nexport {useScrollToBottom} from './hooks/useScrollToBottom';\nexport type {AiSliceTool} from './AiSlice';\n\nexport {ModelSelector} from './components/ModelSelector';\nexport {SessionControls} from './components/SessionControls';\nexport {QueryControls} from './components/QueryControls';\nexport {DeleteSessionDialog} from './components/session/DeleteSessionDialog';\nexport {SessionActions} from './components/session/SessionActions';\nexport {SessionDropdown} from './components/session/SessionDropdown';\nexport {SessionTitle} from './components/session/SessionTitle';\nexport type {SessionType} from './components/session/SessionType';\nexport {ToolErrorMessage} from './components/tools/ToolErrorMessage';\n\nexport {AiSliceConfig, createDefaultAiConfig} from '@sqlrooms/ai-config';\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAC,aAAa,EAAE,cAAc,EAAC,MAAM,WAAW,CAAC;AAGxD,OAAO,EAAC,wBAAwB,EAAC,MAAM,uCAAuC,CAAC;AAC/E,OAAO,EAAC,cAAc,EAAC,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAC,iBAAiB,EAAC,MAAM,2BAA2B,CAAC;AAC5D,OAAO,EAAC,SAAS,EAAC,MAAM,mBAAmB,CAAC;AAE5C,OAAO,EAAC,aAAa,EAAC,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAC,eAAe,EAAC,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAC,aAAa,EAAC,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAC,mBAAmB,EAAC,MAAM,0CAA0C,CAAC;AAC7E,OAAO,EAAC,cAAc,EAAC,MAAM,qCAAqC,CAAC;AACnE,OAAO,EAAC,eAAe,EAAC,MAAM,sCAAsC,CAAC;AACrE,OAAO,EAAC,YAAY,EAAC,MAAM,mCAAmC,CAAC;AAE/D,OAAO,EAAC,gBAAgB,EAAC,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAC,YAAY,EAAC,MAAM,2BAA2B,CAAC;AAEvD,OAAO,EAAC,aAAa,EAAE,qBAAqB,EAAC,MAAM,qBAAqB,CAAC;AACzE,OAAO,EAAC,cAAc,EAAC,MAAM,6BAA6B,CAAC","sourcesContent":["/**\n * {@include ../README.md}\n * @packageDocumentation\n */\n\nexport {createAiSlice, useStoreWithAi} from './AiSlice';\n\nexport type {AiSliceState} from './AiSlice';\nexport {AnalysisResultsContainer} from './components/AnalysisResultsContainer';\nexport {AnalysisResult} from './components/AnalysisResult';\nexport {useScrollToBottom} from './hooks/useScrollToBottom';\nexport {useAiChat} from './hooks/useAiChat';\n\nexport {ModelSelector} from './components/ModelSelector';\nexport {SessionControls} from './components/SessionControls';\nexport {QueryControls} from './components/QueryControls';\nexport {DeleteSessionDialog} from './components/session/DeleteSessionDialog';\nexport {SessionActions} from './components/session/SessionActions';\nexport {SessionDropdown} from './components/session/SessionDropdown';\nexport {SessionTitle} from './components/session/SessionTitle';\nexport type {SessionType} from './components/session/SessionType';\nexport {ToolErrorMessage} from './components/tools/ToolErrorMessage';\nexport {ToolCallInfo} from './components/ToolCallInfo';\n\nexport {AiSliceConfig, createDefaultAiConfig} from '@sqlrooms/ai-config';\nexport {AiThinkingDots} from './components/AiThinkingDots';\n"]}

package/dist/utils.d.ts

@@ -1,7 +1,8 @@
 /**
  * Utility functions for AI Chat UI configuration
  */
-import { AiSettingsSliceConfig } from '@sqlrooms/ai-config';
+import { AiSettingsSliceConfig, UIMessagePart } from '@sqlrooms/ai-config';
+import { TextUIPart, ToolUIPart } from 'ai';
 /**
  * Extract models from aiSettings in the format expected by ModelSelector
  * @param config - The AI model configuration
@@ -12,4 +13,25 @@ export declare function extractModelsFromSettings(config: AiSettingsSliceConfig)
     label: string;
     value: string;
 }>;
+/**
+ * Type guard to check if a UIMessagePart is a text part
+ * @param part - The message part to check
+ * @returns True if the part is a text part
+ */
+export declare function isTextPart(part: UIMessagePart): part is TextUIPart;
+/**
+ * Type guard to check if a UIMessagePart is a reasoning part
+ * @param part - The message part to check
+ * @returns True if the part is a reasoning part
+ */
+export declare function isReasoningPart(part: UIMessagePart): part is Extract<UIMessagePart, {
+    type: 'reasoning';
+    text: string;
+}>;
+/**
+ * Type guard to check if a UIMessagePart is a tool part (type starts with 'tool-')
+ * @param part - The message part to check
+ * @returns True if the part is a tool part
+ */
+export declare function isToolPart(part: UIMessagePart): part is ToolUIPart;
 //# sourceMappingURL=utils.d.ts.map

package/dist/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAC,qBAAqB,EAAC,MAAM,qBAAqB,CAAC;AAE1D;;;;GAIG;AACH,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,qBAAqB,GAC5B,KAAK,CAAC;IACP,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;CACf,CAAC,CA4BD"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACL,qBAAqB,EAErB,aAAa,EACd,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAY,UAAU,EAAE,UAAU,EAAC,MAAM,IAAI,CAAC;AAErD;;;;GAIG;AACH,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,qBAAqB,GAC5B,KAAK,CAAC;IACP,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;CACf,CAAC,CA4BD;AAED;;;;GAIG;AACH,wBAAgB,UAAU,CAAC,IAAI,EAAE,aAAa,GAAG,IAAI,IAAI,UAAU,CAElE;AAED;;;;GAIG;AACH,wBAAgB,eAAe,CAC7B,IAAI,EAAE,aAAa,GAClB,IAAI,IAAI,OAAO,CAAC,aAAa,EAAE;IAAC,IAAI,EAAE,WAAW,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAC,CAAC,CAEnE;AAED;;;;GAIG;AACH,wBAAgB,UAAU,CAAC,IAAI,EAAE,aAAa,GAAG,IAAI,IAAI,UAAU,CAElE"}

package/dist/utils.js

@@ -28,4 +28,28 @@ export function extractModelsFromSettings(config) {
     });
     return models;
 }
+/**
+ * Type guard to check if a UIMessagePart is a text part
+ * @param part - The message part to check
+ * @returns True if the part is a text part
+ */
+export function isTextPart(part) {
+    return part.type === 'text';
+}
+/**
+ * Type guard to check if a UIMessagePart is a reasoning part
+ * @param part - The message part to check
+ * @returns True if the part is a reasoning part
+ */
+export function isReasoningPart(part) {
+    return part.type === 'reasoning';
+}
+/**
+ * Type guard to check if a UIMessagePart is a tool part (type starts with 'tool-')
+ * @param part - The message part to check
+ * @returns True if the part is a tool part
+ */
+export function isToolPart(part) {
+    return typeof part.type === 'string' && part.type.startsWith('tool-');
+}
 //# sourceMappingURL=utils.js.map

package/dist/utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"utils.js","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA;;GAEG;AAIH;;;;GAIG;AACH,MAAM,UAAU,yBAAyB,CACvC,MAA6B;IAM7B,MAAM,MAAM,GAIP,EAAE,CAAC;IAER,gCAAgC;IAChC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,EAAE,QAAQ,CAAC,EAAE,EAAE;QACnE,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;YAChC,MAAM,CAAC,IAAI,CAAC;gBACV,QAAQ,EAAE,WAAW;gBACrB,KAAK,EAAE,KAAK,CAAC,SAAS;gBACtB,KAAK,EAAE,KAAK,CAAC,SAAS;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,oBAAoB;IACpB,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE;QAC1C,MAAM,CAAC,IAAI,CAAC;YACV,QAAQ,EAAE,QAAQ;YAClB,KAAK,EAAE,WAAW,CAAC,SAAS;YAC5B,KAAK,EAAE,WAAW,CAAC,SAAS;SAC7B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC","sourcesContent":["/**\n * Utility functions for AI Chat UI configuration\n */\n\nimport {AiSettingsSliceConfig} from '@sqlrooms/ai-config';\n\n/**\n * Extract models from aiSettings in the format expected by ModelSelector\n * @param config - The AI model configuration\n * @returns Array of models with provider, label, and value properties\n */\nexport function extractModelsFromSettings(\n  config: AiSettingsSliceConfig,\n): Array<{\n  provider: string;\n  label: string;\n  value: string;\n}> {\n  const models: Array<{\n    provider: string;\n    label: string;\n    value: string;\n  }> = [];\n\n  // Extract models from providers\n  Object.entries(config.providers).forEach(([providerKey, provider]) => {\n    provider.models.forEach((model) => {\n      models.push({\n        provider: providerKey,\n        label: model.modelName,\n        value: model.modelName,\n      });\n    });\n  });\n\n  // Add custom models\n  config.customModels.forEach((customModel) => {\n    models.push({\n      provider: 'custom',\n      label: customModel.modelName,\n      value: customModel.modelName,\n    });\n  });\n\n  return models;\n}\n"]}
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA;;GAEG;AASH;;;;GAIG;AACH,MAAM,UAAU,yBAAyB,CACvC,MAA6B;IAM7B,MAAM,MAAM,GAIP,EAAE,CAAC;IAER,gCAAgC;IAChC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,EAAE,QAAQ,CAAC,EAAE,EAAE;QACnE,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;YAChC,MAAM,CAAC,IAAI,CAAC;gBACV,QAAQ,EAAE,WAAW;gBACrB,KAAK,EAAE,KAAK,CAAC,SAAS;gBACtB,KAAK,EAAE,KAAK,CAAC,SAAS;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,oBAAoB;IACpB,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE;QAC1C,MAAM,CAAC,IAAI,CAAC;YACV,QAAQ,EAAE,QAAQ;YAClB,KAAK,EAAE,WAAW,CAAC,SAAS;YAC5B,KAAK,EAAE,WAAW,CAAC,SAAS;SAC7B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,UAAU,CAAC,IAAmB;IAC5C,OAAO,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;AAC9B,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,eAAe,CAC7B,IAAmB;IAEnB,OAAO,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC;AACnC,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,UAAU,CAAC,IAAmB;IAC5C,OAAO,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;AACxE,CAAC","sourcesContent":["/**\n * Utility functions for AI Chat UI configuration\n */\n\nimport {\n  AiSettingsSliceConfig,\n  AnalysisResultSchema,\n  UIMessagePart,\n} from '@sqlrooms/ai-config';\nimport {UIMessage, TextUIPart, ToolUIPart} from 'ai';\n\n/**\n * Extract models from aiSettings in the format expected by ModelSelector\n * @param config - The AI model configuration\n * @returns Array of models with provider, label, and value properties\n */\nexport function extractModelsFromSettings(\n  config: AiSettingsSliceConfig,\n): Array<{\n  provider: string;\n  label: string;\n  value: string;\n}> {\n  const models: Array<{\n    provider: string;\n    label: string;\n    value: string;\n  }> = [];\n\n  // Extract models from providers\n  Object.entries(config.providers).forEach(([providerKey, provider]) => {\n    provider.models.forEach((model) => {\n      models.push({\n        provider: providerKey,\n        label: model.modelName,\n        value: model.modelName,\n      });\n    });\n  });\n\n  // Add custom models\n  config.customModels.forEach((customModel) => {\n    models.push({\n      provider: 'custom',\n      label: customModel.modelName,\n      value: customModel.modelName,\n    });\n  });\n\n  return models;\n}\n\n/**\n * Type guard to check if a UIMessagePart is a text part\n * @param part - The message part to check\n * @returns True if the part is a text part\n */\nexport function isTextPart(part: UIMessagePart): part is TextUIPart {\n  return part.type === 'text';\n}\n\n/**\n * Type guard to check if a UIMessagePart is a reasoning part\n * @param part - The message part to check\n * @returns True if the part is a reasoning part\n */\nexport function isReasoningPart(\n  part: UIMessagePart,\n): part is Extract<UIMessagePart, {type: 'reasoning'; text: string}> {\n  return part.type === 'reasoning';\n}\n\n/**\n * Type guard to check if a UIMessagePart is a tool part (type starts with 'tool-')\n * @param part - The message part to check\n * @returns True if the part is a tool part\n */\nexport function isToolPart(part: UIMessagePart): part is ToolUIPart {\n  return typeof part.type === 'string' && part.type.startsWith('tool-');\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sqlrooms/ai-core",
-  "version": "0.26.0-rc.2",
+  "version": "0.26.0-rc.3",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "module": "dist/index.js",
@@ -18,23 +18,24 @@
     "access": "public"
   },
   "dependencies": {
-    "@ai-sdk/provider": "^1.1.3",
-    "@openassistant/core": "0.5.17",
-    "@openassistant/utils": "0.5.17",
+    "@ai-sdk/openai": "^2.0.32",
+    "@ai-sdk/openai-compatible": "^1.0.18",
+    "@ai-sdk/react": "^2.0.44",
+    "@openassistant/utils": "1.0.0-alpha.0",
     "@paralleldrive/cuid2": "^2.2.2",
-    "@sqlrooms/ai-config": "0.26.0-rc.2",
-    "@sqlrooms/monaco-editor": "0.26.0-rc.2",
-    "@sqlrooms/room-config": "0.26.0-rc.2",
-    "@sqlrooms/room-store": "0.26.0-rc.2",
-    "@sqlrooms/ui": "0.26.0-rc.2",
-    "@sqlrooms/utils": "0.26.0-rc.2",
-    "ai": "^4.3.19",
+    "@sqlrooms/ai-config": "0.26.0-rc.3",
+    "@sqlrooms/monaco-editor": "0.26.0-rc.3",
+    "@sqlrooms/room-config": "0.26.0-rc.3",
+    "@sqlrooms/room-store": "0.26.0-rc.3",
+    "@sqlrooms/ui": "0.26.0-rc.3",
+    "@sqlrooms/utils": "0.26.0-rc.3",
+    "ai": "^5.0.44",
     "immer": "^10.1.1",
     "lucide-react": "^0.475.0",
     "react-markdown": "^10.1.0",
     "rehype-raw": "^7.0.0",
     "remark-gfm": "^4.0.0",
-    "zod": "^3.25.76"
+    "zod": "^4.1.8"
   },
   "peerDependencies": {
     "react": ">=18",
@@ -47,5 +48,5 @@
     "typecheck": "tsc --noEmit",
     "typedoc": "typedoc"
   },
-  "gitHead": "86e1f2915278944e6bec6b19ab8ac16dc094cbb9"
-}
+  "gitHead": "536764b2aa924e5bb6650fe0bc674113179ff444"
+}

package/dist/analysis.d.ts

@@ -1,51 +0,0 @@
-import { StreamMessage } from '@openassistant/core';
-import { AnalysisResultSchema } from '@sqlrooms/ai-config';
-import { AiSliceTool } from './AiSlice';
-/**
- * Configuration options for running an AI analysis session
- */
-type AnalysisParameters = {
-    /** Assistant instance identifier (default: 'sqlrooms-ai') */
-    name?: string;
-    /** AI model provider (e.g., 'openai', 'anthropic') */
-    modelProvider: string;
-    /** Model identifier (e.g., 'gpt-4', 'claude-3') */
-    model: string;
-    /** Authentication key for the model provider's API */
-    apiKey: string;
-    /** Analysis prompt or question to be processed */
-    prompt: string;
-    /** Optional controller for canceling the analysis operation */
-    abortController?: AbortController;
-    /** Maximum number of analysis steps allowed (default: 100) */
-    maxSteps?: number;
-    /** The history of analysis results (e.g. saved in localStorage) */
-    historyAnalysis?: AnalysisResultSchema[];
-    /** Tools to use in the analysis */
-    tools?: Record<string, AiSliceTool>;
-    /** Base URL for Ollama provider (required when modelProvider is 'ollama') */
-    baseUrl?: string;
-    /**
-     * Function to get custom instructions for the AI assistant
-     * @returns The instructions string to use
-     */
-    getInstructions: () => string;
-    /**
-     * Callback for handling streaming results
-     * @param isCompleted - Indicates if this is the final message in the stream
-     * @param streamMessage - Current message content being streamed
-     */
-    onStreamResult: (isCompleted: boolean, streamMessage?: StreamMessage) => void;
-};
-/**
- * Executes an AI analysis session on the room data
- *
- * @param config - Analysis configuration options. See {@link AnalysisParameters} for more details.
- * @returns Object containing tool calls executed and the final analysis result
- */
-export declare function runAnalysis({ name, modelProvider, model, apiKey, prompt, abortController, historyAnalysis, onStreamResult, maxSteps, tools, getInstructions, baseUrl, }: AnalysisParameters): Promise<{
-    streamMessage: StreamMessage;
-    messages: import("ai").CoreMessage[];
-}>;
-export {};
-//# sourceMappingURL=analysis.d.ts.map

package/dist/analysis.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"analysis.d.ts","sourceRoot":"","sources":["../src/analysis.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,aAAa,EACd,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAC,oBAAoB,EAAC,MAAM,qBAAqB,CAAC;AAEzD,OAAO,EAAC,WAAW,EAAC,MAAM,WAAW,CAAC;AAEtC;;GAEG;AACH,KAAK,kBAAkB,GAAG;IACxB,6DAA6D;IAC7D,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd,sDAAsD;IACtD,aAAa,EAAE,MAAM,CAAC;IAEtB,mDAAmD;IACnD,KAAK,EAAE,MAAM,CAAC;IAEd,sDAAsD;IACtD,MAAM,EAAE,MAAM,CAAC;IAEf,kDAAkD;IAClD,MAAM,EAAE,MAAM,CAAC;IAEf,+DAA+D;IAC/D,eAAe,CAAC,EAAE,eAAe,CAAC;IAElC,8DAA8D;IAC9D,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB,mEAAmE;IACnE,eAAe,CAAC,EAAE,oBAAoB,EAAE,CAAC;IAEzC,mCAAmC;IACnC,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IAEpC,6EAA6E;IAC7E,OAAO,CAAC,EAAE,MAAM,CAAC;IAEjB;;;OAGG;IACH,eAAe,EAAE,MAAM,MAAM,CAAC;IAE9B;;;;OAIG;IACH,cAAc,EAAE,CAAC,WAAW,EAAE,OAAO,EAAE,aAAa,CAAC,EAAE,aAAa,KAAK,IAAI,CAAC;CAC/E,CAAC;AAEF;;;;;GAKG;AACH,wBAAsB,WAAW,CAAC,EAChC,IAAoB,EACpB,aAAa,EACb,KAAK,EACL,MAAM,EACN,MAAM,EACN,eAAe,EACf,eAAe,EACf,cAAc,EACd,QAAY,EACZ,KAAU,EACV,eAAe,EACf,OAAO,GACR,EAAE,kBAAkB;;;GA0CpB"}

package/dist/analysis.js

@@ -1,43 +0,0 @@
-import { createAssistant, rebuildMessages, } from '@openassistant/core';
-import { convertToCoreMessages } from 'ai';
-/**
- * Executes an AI analysis session on the room data
- *
- * @param config - Analysis configuration options. See {@link AnalysisParameters} for more details.
- * @returns Object containing tool calls executed and the final analysis result
- */
-export async function runAnalysis({ name = 'sqlrooms-ai', modelProvider, model, apiKey, prompt, abortController, historyAnalysis, onStreamResult, maxSteps = 5, tools = {}, getInstructions, baseUrl, }) {
-    // get the singleton assistant instance
-    const assistant = await createAssistant({
-        name,
-        modelProvider,
-        model,
-        apiKey,
-        version: 'v1',
-        instructions: getInstructions(),
-        tools: tools,
-        temperature: 0,
-        toolChoice: 'auto', // this will enable streaming
-        maxSteps,
-        ...(abortController ? { abortController } : {}),
-        baseUrl, // ollama base url or LLM proxy server url
-    });
-    // restore ai messages from historyAnalysis?
-    if (historyAnalysis) {
-        const historyMessages = historyAnalysis.map((analysis) => ({
-            prompt: analysis.prompt,
-            response: analysis.streamMessage,
-        }));
-        const initialMessages = rebuildMessages(historyMessages);
-        assistant.setMessages(convertToCoreMessages(initialMessages));
-    }
-    // process the prompt
-    const newMessages = await assistant.processTextMessage({
-        textMessage: prompt,
-        streamMessageCallback: ({ isCompleted, message, }) => {
-            onStreamResult(isCompleted ?? false, message);
-        },
-    });
-    return newMessages;
-}
-//# sourceMappingURL=analysis.js.map

package/dist/analysis.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"analysis.js","sourceRoot":"","sources":["../src/analysis.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,eAAe,EACf,eAAe,GAEhB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAC,qBAAqB,EAAC,MAAM,IAAI,CAAC;AAmDzC;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,EAChC,IAAI,GAAG,aAAa,EACpB,aAAa,EACb,KAAK,EACL,MAAM,EACN,MAAM,EACN,eAAe,EACf,eAAe,EACf,cAAc,EACd,QAAQ,GAAG,CAAC,EACZ,KAAK,GAAG,EAAE,EACV,eAAe,EACf,OAAO,GACY;IACnB,uCAAuC;IACvC,MAAM,SAAS,GAAG,MAAM,eAAe,CAAC;QACtC,IAAI;QACJ,aAAa;QACb,KAAK;QACL,MAAM;QACN,OAAO,EAAE,IAAI;QACb,YAAY,EAAE,eAAe,EAAE;QAC/B,KAAK,EAAE,KAAK;QACZ,WAAW,EAAE,CAAC;QACd,UAAU,EAAE,MAAM,EAAE,6BAA6B;QACjD,QAAQ;QACR,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,EAAC,eAAe,EAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7C,OAAO,EAAE,0CAA0C;KACpD,CAAC,CAAC;IAEH,4CAA4C;IAC5C,IAAI,eAAe,EAAE,CAAC;QACpB,MAAM,eAAe,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YACzD,MAAM,EAAE,QAAQ,CAAC,MAAM;YACvB,QAAQ,EAAE,QAAQ,CAAC,aAA8B;SAClD,CAAC,CAAC,CAAC;QACJ,MAAM,eAAe,GAAG,eAAe,CAAC,eAAe,CAAC,CAAC;QACzD,SAAS,CAAC,WAAW,CAAC,qBAAqB,CAAC,eAAe,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,qBAAqB;IACrB,MAAM,WAAW,GAAG,MAAM,SAAS,CAAC,kBAAkB,CAAC;QACrD,WAAW,EAAE,MAAM;QACnB,qBAAqB,EAAE,CAAC,EACtB,WAAW,EACX,OAAO,GAIR,EAAE,EAAE;YACH,cAAc,CAAC,WAAW,IAAI,KAAK,EAAE,OAAO,CAAC,CAAC;QAChD,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,WAAW,CAAC;AACrB,CAAC","sourcesContent":["import {\n  createAssistant,\n  rebuildMessages,\n  StreamMessage,\n} from '@openassistant/core';\nimport {AnalysisResultSchema} from '@sqlrooms/ai-config';\nimport {convertToCoreMessages} from 'ai';\nimport {AiSliceTool} from './AiSlice';\n\n/**\n * Configuration options for running an AI analysis session\n */\ntype AnalysisParameters = {\n  /** Assistant instance identifier (default: 'sqlrooms-ai') */\n  name?: string;\n\n  /** AI model provider (e.g., 'openai', 'anthropic') */\n  modelProvider: string;\n\n  /** Model identifier (e.g., 'gpt-4', 'claude-3') */\n  model: string;\n\n  /** Authentication key for the model provider's API */\n  apiKey: string;\n\n  /** Analysis prompt or question to be processed */\n  prompt: string;\n\n  /** Optional controller for canceling the analysis operation */\n  abortController?: AbortController;\n\n  /** Maximum number of analysis steps allowed (default: 100) */\n  maxSteps?: number;\n\n  /** The history of analysis results (e.g. saved in localStorage) */\n  historyAnalysis?: AnalysisResultSchema[];\n\n  /** Tools to use in the analysis */\n  tools?: Record<string, AiSliceTool>;\n\n  /** Base URL for Ollama provider (required when modelProvider is 'ollama') */\n  baseUrl?: string;\n\n  /**\n   * Function to get custom instructions for the AI assistant\n   * @returns The instructions string to use\n   */\n  getInstructions: () => string;\n\n  /**\n   * Callback for handling streaming results\n   * @param isCompleted - Indicates if this is the final message in the stream\n   * @param streamMessage - Current message content being streamed\n   */\n  onStreamResult: (isCompleted: boolean, streamMessage?: StreamMessage) => void;\n};\n\n/**\n * Executes an AI analysis session on the room data\n *\n * @param config - Analysis configuration options. See {@link AnalysisParameters} for more details.\n * @returns Object containing tool calls executed and the final analysis result\n */\nexport async function runAnalysis({\n  name = 'sqlrooms-ai',\n  modelProvider,\n  model,\n  apiKey,\n  prompt,\n  abortController,\n  historyAnalysis,\n  onStreamResult,\n  maxSteps = 5,\n  tools = {},\n  getInstructions,\n  baseUrl,\n}: AnalysisParameters) {\n  // get the singleton assistant instance\n  const assistant = await createAssistant({\n    name,\n    modelProvider,\n    model,\n    apiKey,\n    version: 'v1',\n    instructions: getInstructions(),\n    tools: tools,\n    temperature: 0,\n    toolChoice: 'auto', // this will enable streaming\n    maxSteps,\n    ...(abortController ? {abortController} : {}),\n    baseUrl, // ollama base url or LLM proxy server url\n  });\n\n  // restore ai messages from historyAnalysis?\n  if (historyAnalysis) {\n    const historyMessages = historyAnalysis.map((analysis) => ({\n      prompt: analysis.prompt,\n      response: analysis.streamMessage as StreamMessage,\n    }));\n    const initialMessages = rebuildMessages(historyMessages);\n    assistant.setMessages(convertToCoreMessages(initialMessages));\n  }\n\n  // process the prompt\n  const newMessages = await assistant.processTextMessage({\n    textMessage: prompt,\n    streamMessageCallback: ({\n      isCompleted,\n      message,\n    }: {\n      isCompleted?: boolean;\n      message?: StreamMessage;\n    }) => {\n      onStreamResult(isCompleted ?? false, message);\n    },\n  });\n\n  return newMessages;\n}\n"]}