@sqlrooms/ai-core 0.26.0-rc.5 → 0.26.0

package/README.md

@@ -10,6 +10,7 @@ An AI integration package for SQLRooms that provides components and utilities fo
 - 🧩 **UI Components**: Ready-to-use components for AI interactions
 - 📝 **Query History**: Track and manage AI query history
 - 🎯 **Tool Integration**: Framework for AI tools and actions
+- 🤖 **Agent Framework**: Framework for building AI agents
 
 ## Installation
 
@@ -290,7 +291,6 @@ const weatherTool: OpenAssistantTool = {
 
 ### Transfer Additional Tool Output Data to Client
 
-
 #### The Problem
 
 When tools execute, they often generate additional data (like detailed search results, charts, metadata) that needs to be sent to the client for UI rendering, but should NOT be included in the conversation history sent back to the LLM.
@@ -322,7 +322,7 @@ writer.write({
 Backend (route.ts)
   │
   ├─> Tool executes
-  │   └─> writer.write({ 
+  │   └─> writer.write({
   │         type: 'data-tool-additional-output',
   │         transient: true,  // ✅ SDK handles exclusion
   │         data: {...}
@@ -366,6 +366,175 @@ const toolData = currentSession?.toolAdditionalData?.[toolCallId];
 
 AI SDK v5 supports message annotations, but these are still part of the message structure. The `transient` flag is specifically designed for data that should only be sent once and not persist in conversation history.
 
+## Agent Framework
+
+The agent framework enables building autonomous AI agents that can use tools and make multi-step decisions. Built on top of AI SDK v5's `Experimental_Agent` class, it provides utilities for integrating agents as tools within the main chat interface.
+
+### What are Agents?
+
+Agents differ from regular tools in that they:
+- Can autonomously decide which tools to call and when
+- Execute multiple steps to accomplish a goal
+- Maintain their own reasoning loop until completion
+- Can be embedded as tools within the main conversation
+
+### Creating an Agent Tool
+
+Agent tools follow the same OpenAssistantTool pattern but use the `processAgentStream` utility to handle streaming and progress tracking:
+
+```typescript
+import {Experimental_Agent as Agent, tool} from 'ai';
+import {processAgentStream} from '@sqlrooms/ai';
+import {z} from 'zod';
+
+export function weatherAgentTool(store: StoreApi<RoomState>) {
+  return {
+    name: 'agent-weather',
+    description: 'A specialized agent for weather-related queries',
+    parameters: z.object({
+      prompt: z.string().describe('The user\'s weather question')
+    }),
+    execute: async ({prompt}, options) => {
+      const state = store.getState();
+      const currentSession = state.ai.getCurrentSession();
+
+      // Create the agent with its own set of tools
+      const weatherAgent = new Agent({
+        model: getModel(state),
+        tools: {
+          weather: tool({
+            description: 'Get current weather in a location',
+            inputSchema: z.object({
+              location: z.string()
+            }),
+            execute: async ({location}) => ({
+              location,
+              temperature: 72 + Math.floor(Math.random() * 21) - 10
+            })
+          }),
+          convertTemperature: tool({
+            description: 'Convert temperature units',
+            inputSchema: z.object({
+              temperature: z.number(),
+              from: z.enum(['F', 'C']),
+              to: z.enum(['F', 'C'])
+            }),
+            execute: async ({temperature, from, to}) => {
+              // Conversion logic
+            }
+          })
+        },
+        stopWhen: stepCountIs(10) // Limit agent steps
+      });
+
+      // Stream the agent's execution
+      const agentResult = await weatherAgent.stream({prompt});
+
+      // Process the stream and track progress
+      const resultText = await processAgentStream(
+        agentResult,
+        store,
+        options.toolCallId
+      );
+
+      return {
+        llmResult: {
+          success: true,
+          details: resultText
+        }
+      };
+    }
+  };
+}
+```
+
+Use the agent as a tool in your main LLM:
+
+```typescript
+createAiSlice({
+  tools: {
+    query: QueryTool,
+    'agent-weather': weatherAgentTool(store)  // ⚡ Agent as tool
+  }
+})
+```
+
+### The `processAgentStream` Utility
+
+The `processAgentStream` function handles the complexity of integrating agent execution into the main conversation:
+
+```typescript
+await processAgentStream(agentResult, store, parentToolCallId)
+```
+
+**What it handles:**
+
+1. **Progress Tracking**: Monitors all tool calls made by the agent in real-time
+2. **State Updates**: Updates `toolAdditionalData` with agent progress for UI rendering
+3. **Error Handling**: Captures and reports tool execution errors
+4. **Result Aggregation**: Collects the final text output from the agent
+
+**Stored Data Structure:**
+
+Each agent tool call stores structured progress data:
+
+```typescript
+interface AgentToolCallAdditionalData {
+  agentToolCalls: Array<{
+    toolCallId: string;
+    toolName: string;
+    output?: unknown;
+    errorText?: string;
+    state: 'pending' | 'success' | 'error';
+  }>;
+  finalOutput?: string;
+  timestamp: string;
+}
+```
+
+### Rendering Agent Progress
+
+If you are using the `sqlrooms/ai-core` package, the rendering of agent progress is handled automatically.
+
+However, you can also use the `useRoomStore` hook to access the agent progress data in your components to show real-time execution:
+
+```tsx
+const currentSession = useRoomStore((state) => state.ai.getCurrentSession());
+const agentData = currentSession?.toolAdditionalData?.[toolCallId] as AgentToolCallAdditionalData;
+
+return (
+  <div>
+    <h3>Agent Progress:</h3>
+    {agentData?.agentToolCalls.map((call) => (
+      <div key={call.toolCallId}>
+        {call.toolName}: {call.state}
+        {call.output && <pre>{JSON.stringify(call.output, null, 2)}</pre>}
+        {call.errorText && <span className="error">{call.errorText}</span>}
+      </div>
+    ))}
+    {agentData?.finalOutput && (
+      <div className="final-result">{agentData.finalOutput}</div>
+    )}
+  </div>
+);
+```
+
+### Best Practices
+
+1. **Limit Steps**: Always use `stopWhen` to prevent infinite loops
+2. **Specific Tools**: Give agents focused, domain-specific tools
+3. **Clear Descriptions**: Write precise tool descriptions for better agent reasoning
+4. **Error Handling**: Handle agent errors gracefully with try-catch in execute
+5. **Progress UI**: Use `toolAdditionalData` to show agent progress to users
+
+### Use Cases
+
+- **Multi-step Analysis**: Agents that need to gather data from multiple sources
+- **Decision Trees**: Complex workflows requiring conditional logic
+- **Specialized Domains**: Weather, finance, or data analysis agents with domain-specific tools
+- **Autonomous Tasks**: Tasks that require planning and execution without human intervention
+
+
 ## Advanced Features
 
 - **Custom AI Tools**: Define custom tools for AI to use with the tool() function

package/dist/AiSlice.d.ts

@@ -1,21 +1,51 @@
 import { AiSliceConfig, AnalysisResultSchema, AnalysisSessionSchema } from '@sqlrooms/ai-config';
-import { BaseRoomConfig, type StateCreator } from '@sqlrooms/room-store';
-import { DefaultChatTransport, LanguageModel, UIMessage } from 'ai';
+import { type StateCreator } from '@sqlrooms/room-store';
+import { UIMessage, DefaultChatTransport, LanguageModel, UITools, ChatOnDataCallback, UIDataTypes } from 'ai';
+import { ToolCall } from './chatTransport';
 import { OpenAssistantToolSet } from '@openassistant/utils';
+import { AddToolResult } from './hooks/useAiChat';
 type ExtendedChatOnToolCallCallback = (args: {
-    toolCall: any;
-    addToolResult?: any;
-}) => Promise<any> | any;
+    toolCall: ToolCall;
+    addToolResult?: AddToolResult;
+}) => Promise<void> | void;
 export type AiSliceState = {
     ai: {
         config: AiSliceConfig;
         analysisPrompt: string;
         isRunningAnalysis: boolean;
+        promptSuggestionsVisible: boolean;
         tools: OpenAssistantToolSet;
         analysisAbortController?: AbortController;
         setConfig: (config: AiSliceConfig) => void;
+        setPromptSuggestionsVisible: (visible: boolean) => void;
+        /** Latest stop function from useChat to immediately halt local streaming */
+        chatStop?: () => void;
+        /** Register/replace the current chat stop function */
+        setChatStop: (stop: (() => void) | undefined) => void;
+        /** Latest sendMessage function from useChat to send messages */
+        chatSendMessage?: (message: {
+            text: string;
+        }) => void;
+        /** Register/replace the current chat sendMessage function */
+        setChatSendMessage: (sendMessage: ((message: {
+            text: string;
+        }) => void) | undefined) => void;
+        /** Latest addToolResult function from useChat to add tool results */
+        addToolResult?: AddToolResult;
+        /** Register/replace the current addToolResult function */
+        setAddToolResult: (addToolResult: AddToolResult | undefined) => void;
+        /** Wait for a tool result to be added by UI component */
+        waitForToolResult: (toolCallId: string, abortSignal?: AbortSignal) => Promise<void>;
         setAnalysisPrompt: (prompt: string) => void;
         addAnalysisResult: (message: UIMessage) => void;
+        sendPrompt: (prompt: string, options?: {
+            systemInstructions?: string;
+            modelProvider?: string;
+            modelName?: string;
+            baseUrl?: string;
+            abortSignal?: AbortSignal;
+            useTools?: boolean;
+        }) => Promise<string>;
         startAnalysis: (sendMessage: (message: {
             text: string;
         }) => void) => Promise<void>;
@@ -42,7 +72,7 @@ export type AiSliceState = {
         chatHeaders: Record<string, string>;
         getRemoteChatTransport: (endpoint: string, headers?: Record<string, string>) => DefaultChatTransport<UIMessage>;
         onChatToolCall: ExtendedChatOnToolCallCallback;
-        onChatData: (dataPart: any) => void;
+        onChatData: ChatOnDataCallback<UIMessage<unknown, UIDataTypes, UITools>>;
         onChatFinish: (args: {
             message: UIMessage;
             messages: UIMessage[];
@@ -77,7 +107,7 @@ export interface AiSliceOptions {
     /** Optional headers to send with remote endpoint */
     chatHeaders?: Record<string, string>;
 }
-export declare function createAiSlice<PC extends BaseRoomConfig>(params: AiSliceOptions): StateCreator<AiSliceState>;
+export declare function createAiSlice(params: AiSliceOptions): StateCreator<AiSliceState>;
 export declare function useStoreWithAi<T>(selector: (state: AiSliceState) => T): T;
 export {};
 //# sourceMappingURL=AiSlice.d.ts.map

package/dist/AiSlice.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AiSlice.d.ts","sourceRoot":"","sources":["../src/AiSlice.ts"],"names":[],"mappings":"AACA,OAAO,EACL,aAAa,EACb,oBAAoB,EACpB,qBAAqB,EAEtB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,cAAc,EAGd,KAAK,YAAY,EAClB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAC,oBAAoB,EAAE,aAAa,EAAE,SAAS,EAAC,MAAM,IAAI,CAAC;AAGlE,OAAO,EAAC,oBAAoB,EAAC,MAAM,sBAAsB,CAAC;AAS1D,KAAK,8BAA8B,GAAG,CAAC,IAAI,EAAE;IAC3C,QAAQ,EAAE,GAAG,CAAC;IACd,aAAa,CAAC,EAAE,GAAG,CAAC;CACrB,KAAK,OAAO,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC;AAEzB,MAAM,MAAM,YAAY,GAAG;IACzB,EAAE,EAAE;QACF,MAAM,EAAE,aAAa,CAAC;QACtB,cAAc,EAAE,MAAM,CAAC;QACvB,iBAAiB,EAAE,OAAO,CAAC;QAC3B,KAAK,EAAE,oBAAoB,CAAC;QAC5B,uBAAuB,CAAC,EAAE,eAAe,CAAC;QAC1C,SAAS,EAAE,CAAC,MAAM,EAAE,aAAa,KAAK,IAAI,CAAC;QAC3C,iBAAiB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;QAC5C,iBAAiB,EAAE,CAAC,OAAO,EAAE,SAAS,KAAK,IAAI,CAAC;QAChD,aAAa,EAAE,CACb,WAAW,EAAE,CAAC,OAAO,EAAE;YAAC,IAAI,EAAE,MAAM,CAAA;SAAC,KAAK,IAAI,KAC3C,OAAO,CAAC,IAAI,CAAC,CAAC;QACnB,cAAc,EAAE,MAAM,IAAI,CAAC;QAC3B,UAAU,EAAE,CAAC,aAAa,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;QAC3D,aAAa,EAAE,CACb,IAAI,CAAC,EAAE,MAAM,EACb,aAAa,CAAC,EAAE,MAAM,EACtB,KAAK,CAAC,EAAE,MAAM,KACX,IAAI,CAAC;QACV,aAAa,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;QAC3C,aAAa,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;QACzD,aAAa,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;QAC3C,iBAAiB,EAAE,MAAM,qBAAqB,GAAG,SAAS,CAAC;QAC3D,oBAAoB,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,KAAK,IAAI,CAAC;QAC3E,4BAA4B,EAAE,CAC5B,SAAS,EAAE,MAAM,EACjB,UAAU,EAAE,MAAM,EAClB,cAAc,EAAE,OAAO,KACpB,IAAI,CAAC;QACV,kBAAkB,EAAE,MAAM,oBAAoB,EAAE,CAAC;QACjD,oBAAoB,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAC;QACpE,wBAAwB,EAAE,CAAC,gBAAgB,EAAE,MAAM,KAAK,SAAS,CAAC,OAAO,CAAC,CAAC;QAC3E,iBAAiB,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,KAAK,CAAC,aAAa,GAAG,SAAS,CAAC;QACzE,qBAAqB,EAAE,MAAM,MAAM,CAAC;QACpC,sBAAsB,EAAE,MAAM,MAAM,GAAG,SAAS,CAAC;QACjD,uBAAuB,EAAE,MAAM,MAAM,CAAC;QACtC,mBAAmB,EAAE,MAAM,MAAM,CAAC;QAElC,qBAAqB,EAAE,MAAM,oBAAoB,CAAC,SAAS,CAAC,CAAC;QAC7D,kFAAkF;QAClF,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACpC,sBAAsB,EAAE,CACtB,QAAQ,EAAE,MAAM,EAChB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAC7B,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACrC,cAAc,EAAE,8BAA8B,CAAC;QAC/C,UAAU,EAAE,CAAC,QAAQ,EAAE,GAAG,KAAK,IAAI,CAAC;QACpC,YAAY,EAAE,CAAC,IAAI,EAAE;YACnB,OAAO,EAAE,SAAS,CAAC;YACnB,QAAQ,EAAE,SAAS,EAAE,CAAC;YACtB,OAAO,CAAC,EAAE,OAAO,CAAC;SACnB,KAAK,IAAI,CAAC;QACX,WAAW,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;KACvC,CAAC;CACH,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,MAAM,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;IAChC,sDAAsD;IACtD,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,uCAAuC;IACvC,KAAK,EAAE,oBAAoB,CAAC;IAE5B;;;OAGG;IACH,eAAe,EAAE,MAAM,MAAM,CAAC;IAC9B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,0EAA0E;IAC1E,cAAc,CAAC,EAAE,MAAM,aAAa,GAAG,SAAS,CAAC;IACjD,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,CAAC,aAAa,EAAE,MAAM,KAAK,MAAM,CAAC;IAC9C,UAAU,CAAC,EAAE,MAAM,MAAM,CAAC;IAC1B,kFAAkF;IAClF,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,oDAAoD;IACpD,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACtC;AAED,wBAAgB,aAAa,CAAC,EAAE,SAAS,cAAc,EACrD,MAAM,EAAE,cAAc,GACrB,YAAY,CAAC,YAAY,CAAC,CAwiB5B;AAYD,wBAAgB,cAAc,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,KAAK,EAAE,YAAY,KAAK,CAAC,GAAG,CAAC,CAEzE"}
+{"version":3,"file":"AiSlice.d.ts","sourceRoot":"","sources":["../src/AiSlice.ts"],"names":[],"mappings":"AACA,OAAO,EACL,aAAa,EACb,oBAAoB,EACpB,qBAAqB,EAEtB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAGL,KAAK,YAAY,EAClB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EACL,SAAS,EACT,oBAAoB,EACpB,aAAa,EACb,OAAO,EACP,kBAAkB,EAClB,WAAW,EAEZ,MAAM,IAAI,CAAC;AACZ,OAAO,EAIL,QAAQ,EAGT,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EAAC,oBAAoB,EAAC,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAC,aAAa,EAAC,MAAM,mBAAmB,CAAC;AAKhD,KAAK,8BAA8B,GAAG,CAAC,IAAI,EAAE;IAC3C,QAAQ,EAAE,QAAQ,CAAC;IACnB,aAAa,CAAC,EAAE,aAAa,CAAC;CAC/B,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;AAE3B,MAAM,MAAM,YAAY,GAAG;IACzB,EAAE,EAAE;QACF,MAAM,EAAE,aAAa,CAAC;QACtB,cAAc,EAAE,MAAM,CAAC;QACvB,iBAAiB,EAAE,OAAO,CAAC;QAC3B,wBAAwB,EAAE,OAAO,CAAC;QAClC,KAAK,EAAE,oBAAoB,CAAC;QAC5B,uBAAuB,CAAC,EAAE,eAAe,CAAC;QAC1C,SAAS,EAAE,CAAC,MAAM,EAAE,aAAa,KAAK,IAAI,CAAC;QAC3C,2BAA2B,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;QACxD,4EAA4E;QAC5E,QAAQ,CAAC,EAAE,MAAM,IAAI,CAAC;QACtB,sDAAsD;QACtD,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC,MAAM,IAAI,CAAC,GAAG,SAAS,KAAK,IAAI,CAAC;QACtD,gEAAgE;QAChE,eAAe,CAAC,EAAE,CAAC,OAAO,EAAE;YAAC,IAAI,EAAE,MAAM,CAAA;SAAC,KAAK,IAAI,CAAC;QACpD,6DAA6D;QAC7D,kBAAkB,EAAE,CAClB,WAAW,EAAE,CAAC,CAAC,OAAO,EAAE;YAAC,IAAI,EAAE,MAAM,CAAA;SAAC,KAAK,IAAI,CAAC,GAAG,SAAS,KACzD,IAAI,CAAC;QACV,qEAAqE;QACrE,aAAa,CAAC,EAAE,aAAa,CAAC;QAC9B,0DAA0D;QAC1D,gBAAgB,EAAE,CAAC,aAAa,EAAE,aAAa,GAAG,SAAS,KAAK,IAAI,CAAC;QACrE,yDAAyD;QACzD,iBAAiB,EAAE,CACjB,UAAU,EAAE,MAAM,EAClB,WAAW,CAAC,EAAE,WAAW,KACtB,OAAO,CAAC,IAAI,CAAC,CAAC;QACnB,iBAAiB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;QAC5C,iBAAiB,EAAE,CAAC,OAAO,EAAE,SAAS,KAAK,IAAI,CAAC;QAChD,UAAU,EAAE,CACV,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE;YACR,kBAAkB,CAAC,EAAE,MAAM,CAAC;YAC5B,aAAa,CAAC,EAAE,MAAM,CAAC;YACvB,SAAS,CAAC,EAAE,MAAM,CAAC;YACnB,OAAO,CAAC,EAAE,MAAM,CAAC;YACjB,WAAW,CAAC,EAAE,WAAW,CAAC;YAC1B,QAAQ,CAAC,EAAE,OAAO,CAAC;SACpB,KACE,OAAO,CAAC,MAAM,CAAC,CAAC;QACrB,aAAa,EAAE,CACb,WAAW,EAAE,CAAC,OAAO,EAAE;YAAC,IAAI,EAAE,MAAM,CAAA;SAAC,KAAK,IAAI,KAC3C,OAAO,CAAC,IAAI,CAAC,CAAC;QACnB,cAAc,EAAE,MAAM,IAAI,CAAC;QAC3B,UAAU,EAAE,CAAC,aAAa,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;QAC3D,aAAa,EAAE,CACb,IAAI,CAAC,EAAE,MAAM,EACb,aAAa,CAAC,EAAE,MAAM,EACtB,KAAK,CAAC,EAAE,MAAM,KACX,IAAI,CAAC;QACV,aAAa,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;QAC3C,aAAa,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;QACzD,aAAa,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;QAC3C,iBAAiB,EAAE,MAAM,qBAAqB,GAAG,SAAS,CAAC;QAC3D,oBAAoB,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,KAAK,IAAI,CAAC;QAC3E,4BAA4B,EAAE,CAC5B,SAAS,EAAE,MAAM,EACjB,UAAU,EAAE,MAAM,EAClB,cAAc,EAAE,OAAO,KACpB,IAAI,CAAC;QACV,kBAAkB,EAAE,MAAM,oBAAoB,EAAE,CAAC;QACjD,oBAAoB,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAC;QACpE,wBAAwB,EAAE,CAAC,gBAAgB,EAAE,MAAM,KAAK,SAAS,CAAC,OAAO,CAAC,CAAC;QAC3E,iBAAiB,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,KAAK,CAAC,aAAa,GAAG,SAAS,CAAC;QACzE,qBAAqB,EAAE,MAAM,MAAM,CAAC;QACpC,sBAAsB,EAAE,MAAM,MAAM,GAAG,SAAS,CAAC;QACjD,uBAAuB,EAAE,MAAM,MAAM,CAAC;QACtC,mBAAmB,EAAE,MAAM,MAAM,CAAC;QAElC,qBAAqB,EAAE,MAAM,oBAAoB,CAAC,SAAS,CAAC,CAAC;QAC7D,kFAAkF;QAClF,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACpC,sBAAsB,EAAE,CACtB,QAAQ,EAAE,MAAM,EAChB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAC7B,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACrC,cAAc,EAAE,8BAA8B,CAAC;QAC/C,UAAU,EAAE,kBAAkB,CAAC,SAAS,CAAC,OAAO,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC;QACzE,YAAY,EAAE,CAAC,IAAI,EAAE;YACnB,OAAO,EAAE,SAAS,CAAC;YACnB,QAAQ,EAAE,SAAS,EAAE,CAAC;YACtB,OAAO,CAAC,EAAE,OAAO,CAAC;SACnB,KAAK,IAAI,CAAC;QACX,WAAW,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;KACvC,CAAC;CACH,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,MAAM,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;IAChC,sDAAsD;IACtD,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,uCAAuC;IACvC,KAAK,EAAE,oBAAoB,CAAC;IAE5B;;;OAGG;IACH,eAAe,EAAE,MAAM,MAAM,CAAC;IAC9B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,0EAA0E;IAC1E,cAAc,CAAC,EAAE,MAAM,aAAa,GAAG,SAAS,CAAC;IACjD,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,CAAC,aAAa,EAAE,MAAM,KAAK,MAAM,CAAC;IAC9C,UAAU,CAAC,EAAE,MAAM,MAAM,CAAC;IAC1B,kFAAkF;IAClF,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,oDAAoD;IACpD,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACtC;AAED,wBAAgB,aAAa,CAC3B,MAAM,EAAE,cAAc,GACrB,YAAY,CAAC,YAAY,CAAC,CA+tB5B;AAYD,wBAAgB,cAAc,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,KAAK,EAAE,YAAY,KAAK,CAAC,GAAG,CAAC,CAEzE"}

package/dist/AiSlice.js

@@ -2,22 +2,112 @@ import { createId } from '@paralleldrive/cuid2';
 import { createDefaultAiConfig, } from '@sqlrooms/ai-config';
 import { createSlice, useBaseRoomStore, } from '@sqlrooms/room-store';
 import { produce } from 'immer';
-import { createChatHandlers, createLocalChatTransportFactory, createRemoteChatTransportFactory, } from './chatTransport';
+import { generateText, } from 'ai';
+import { createChatHandlers, createLocalChatTransportFactory, createRemoteChatTransportFactory, convertToAiSDKTools, completeIncompleteToolCalls, } from './chatTransport';
 import { hasAiSettingsConfig } from './hasAiSettingsConfig';
+import { cleanupPendingAnalysisResults } from './utils';
+import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
 export function createAiSlice(params) {
     const { initialAnalysisPrompt = '', tools, getApiKey, getBaseUrl, maxSteps = 50, getInstructions, defaultProvider = 'openai', defaultModel = 'gpt-4.1', getCustomModel, chatEndPoint = '', chatHeaders = {}, } = params;
     return createSlice((set, get, store) => {
+        // Clean up pending analysis results from persisted config
+        const cleanedConfig = params.config?.sessions
+            ? {
+                ...params.config,
+                sessions: params.config.sessions.map((session) => {
+                    const cleaned = cleanupPendingAnalysisResults(session);
+                    const completedUiMessages = Array.isArray(cleaned.uiMessages)
+                        ? completeIncompleteToolCalls(cleaned.uiMessages || [])
+                        : [];
+                    return {
+                        ...cleaned,
+                        uiMessages: completedUiMessages,
+                    };
+                }),
+            }
+            : params.config;
+        // Create a persistent Map for pending tool call resolvers (outside of immer draft)
+        const pendingToolCallResolvers = new Map();
         return {
             ai: {
-                config: createDefaultAiConfig(params.config),
+                config: createDefaultAiConfig(cleanedConfig),
                 analysisPrompt: initialAnalysisPrompt,
                 isRunningAnalysis: false,
+                promptSuggestionsVisible: true,
                 tools,
+                waitForToolResult: (toolCallId, abortSignal) => {
+                    return new Promise((resolve, reject) => {
+                        // Set up abort handler
+                        const abortHandler = () => {
+                            const resolver = pendingToolCallResolvers.get(toolCallId);
+                            if (resolver) {
+                                pendingToolCallResolvers.delete(toolCallId);
+                                resolver.reject(new Error('Tool call cancelled by user'));
+                            }
+                        };
+                        if (abortSignal) {
+                            if (abortSignal.aborted) {
+                                reject(new Error('Tool call cancelled by user'));
+                                return;
+                            }
+                            abortSignal.addEventListener('abort', abortHandler, { once: true });
+                        }
+                        // Store resolver (overwrites any existing one, which is fine for our use case)
+                        pendingToolCallResolvers.set(toolCallId, {
+                            resolve: () => {
+                                if (abortSignal) {
+                                    abortSignal.removeEventListener('abort', abortHandler);
+                                }
+                                pendingToolCallResolvers.delete(toolCallId);
+                                resolve();
+                            },
+                            reject: (error) => {
+                                if (abortSignal) {
+                                    abortSignal.removeEventListener('abort', abortHandler);
+                                }
+                                pendingToolCallResolvers.delete(toolCallId);
+                                reject(error);
+                            },
+                        });
+                    });
+                },
+                setChatStop: (stopFn) => {
+                    set((state) => produce(state, (draft) => {
+                        draft.ai.chatStop = stopFn;
+                    }));
+                },
                 setConfig: (config) => {
                     set((state) => produce(state, (draft) => {
                         draft.ai.config = config;
                     }));
                 },
+                setPromptSuggestionsVisible: (visible) => {
+                    set((state) => produce(state, (draft) => {
+                        draft.ai.promptSuggestionsVisible = visible;
+                    }));
+                },
+                setChatSendMessage: (sendMessageFn) => {
+                    set((state) => produce(state, (draft) => {
+                        draft.ai.chatSendMessage = sendMessageFn;
+                    }));
+                },
+                setAddToolResult: (addToolResultFn) => {
+                    // Wrap addToolResult to intercept calls and resolve pending promises
+                    const wrappedAddToolResult = addToolResultFn
+                        ? (options) => {
+                            // Call the original addToolResult
+                            addToolResultFn(options);
+                            // Resolve the promise if there's a pending waiter for this toolCallId
+                            const resolver = pendingToolCallResolvers.get(options.toolCallId);
+                            if (resolver) {
+                                resolver.resolve();
+                            }
+                        }
+                        : undefined;
+                    set((state) => produce(state, (draft) => {
+                        draft.ai.addToolResult = wrappedAddToolResult;
+                    }));
+                },
                 setAnalysisPrompt: (prompt) => {
                     set((state) => produce(state, (draft) => {
                         draft.ai.analysisPrompt = prompt;
@@ -228,6 +318,41 @@ export function createAiSlice(params) {
                     }
                     return instructions;
                 },
+                sendPrompt: async (prompt, options = {}) => {
+                    // One-shot generateText path with explicit abort lifecycle management
+                    const state = get();
+                    const currentSession = state.ai.getCurrentSession();
+                    const { systemInstructions, modelProvider, modelName, baseUrl, abortSignal, useTools = false, } = options;
+                    const provider = modelProvider || currentSession?.modelProvider || defaultProvider;
+                    const modelId = modelName || currentSession?.model || defaultModel;
+                    const baseURL = baseUrl ||
+                        state.ai.getBaseUrlFromSettings() ||
+                        'https://api.openai.com/v1';
+                    const tools = state.ai.tools;
+                    // remove execute from tools
+                    const toolsWithoutExecute = Object.fromEntries(Object.entries(tools).filter(([, tool]) => !tool.execute));
+                    const model = createOpenAICompatible({
+                        apiKey: state.ai.getApiKeyFromSettings(),
+                        name: provider,
+                        baseURL,
+                    }).chatModel(modelId);
+                    try {
+                        const response = await generateText({
+                            model,
+                            messages: [{ role: 'user', content: prompt }],
+                            system: systemInstructions || state.ai.getFullInstructions(),
+                            abortSignal: abortSignal,
+                            ...(useTools
+                                ? { tools: convertToAiSDKTools(toolsWithoutExecute) }
+                                : {}),
+                        });
+                        return response.text;
+                    }
+                    catch (error) {
+                        console.error('Error generating text:', error);
+                        return 'error: can not generate response';
+                    }
+                },
                 /**
                  * Start the analysis
                  * TODO: how to pass the history analysisResults?
@@ -243,9 +368,13 @@ export function createAiSlice(params) {
                     set((state) => produce(state, (draft) => {
                         draft.ai.analysisAbortController = abortController;
                         draft.ai.isRunningAnalysis = true;
+                        draft.ai.analysisPrompt = '';
+                        draft.ai.promptSuggestionsVisible = false;
                         // Add incomplete analysis result to session immediately for instant UI rendering
                         const session = draft.ai.config.sessions.find((s) => s.id === currentSession.id);
                         if (session) {
+                            // Remove any existing pending results (safety check for page refresh scenarios)
+                            session.analysisResults = session.analysisResults.filter((result) => result.id !== '__pending__');
                             // Add incomplete analysis result with a temporary ID
                             // This will be updated in onChatFinish with the actual user message ID
                             session.analysisResults.push({
@@ -259,21 +388,26 @@ export function createAiSlice(params) {
                     sendMessage({ text: promptText });
                 },
                 cancelAnalysis: () => {
-                    const currentSession = get().ai.getCurrentSession();
+                    const abortController = get().ai.analysisAbortController;
+                    // Stop local chat streaming immediately if available
+                    try {
+                        get().ai.chatStop?.();
+                    }
+                    catch {
+                        // no-op
+                    }
+                    // Call abort to signal cancellation
+                    // Keep the abort controller in state so that async handlers (onChatToolCall, onChatFinish)
+                    // can check if it was aborted. The onChatFinish handler will clean it up.
+                    abortController?.abort('Analysis cancelled');
                     set((state) => produce(state, (draft) => {
+                        // Set isRunningAnalysis to false to update UI
                         draft.ai.isRunningAnalysis = false;
-                        // Remove pending analysis result if it exists
-                        if (currentSession) {
-                            const session = draft.ai.config.sessions.find((s) => s.id === currentSession.id);
-                            if (session) {
-                                const pendingIndex = session.analysisResults.findIndex((result) => result.id === '__pending__');
-                                if (pendingIndex !== -1) {
-                                    session.analysisResults.splice(pendingIndex, 1);
-                                }
-                            }
-                        }
+                        // Keep analysisAbortController so handlers can check signal.aborted
+                        // It will be cleared by onChatFinish
+                        // Intentionally preserve any pending analysis result so the
+                        // conversation row remains visible until onChatFinish runs.
                     }));
-                    get().ai.analysisAbortController?.abort('Analysis cancelled');
                 },
                 /**
                  * Get the assistant message parts for a given analysis result ID