Add ToolRenderResult interface for custom tool rendering

- Changed ToolRenderer return type from TemplateResult to ToolRenderResult
- ToolRenderResult = { content: TemplateResult, isCustom: boolean }
- isCustom: true = no card wrapper, false = wrap in card
- Updated all existing tool renderers to return new format
- Updated Messages.ts to handle custom rendering

This enables tools to render without default card chrome when needed.

packages/ai/src/agent/agent-loop.ts

@@ -2,13 +2,13 @@ import { streamSimple } from "../stream.js";
 import type { AssistantMessage, Context, Message, ToolResultMessage, UserMessage } from "../types.js";
 import { EventStream } from "../utils/event-stream.js";
 import { validateToolArguments } from "../utils/validation.js";
-import type { AgentContext, AgentEvent, AgentTool, AgentToolResult, PromptConfig } from "./types.js";
+import type { AgentContext, AgentEvent, AgentLoopConfig, AgentTool, AgentToolResult, QueuedMessage } from "./types.js";
 
 // Main prompt function - returns a stream of events
 export function agentLoop(
 	prompt: UserMessage,
 	context: AgentContext,
-	config: PromptConfig,
+	config: AgentLoopConfig,
 	signal?: AbortSignal,
 	streamFn?: typeof streamSimple,
 ): EventStream<AgentEvent, AgentContext["messages"]> {
@@ -36,15 +36,33 @@ export function agentLoop(
 			messages,
 		};
 
-		// Keep looping while we have tool calls
+		// Keep looping while we have tool calls or queued messages
 		let hasMoreToolCalls = true;
 		let firstTurn = true;
-		while (hasMoreToolCalls) {
+		let queuedMessages: QueuedMessage<any>[] = (await config.getQueuedMessages?.()) || [];
+
+		while (hasMoreToolCalls || queuedMessages.length > 0) {
 			if (!firstTurn) {
 				stream.push({ type: "turn_start" });
 			} else {
 				firstTurn = false;
 			}
+
+			// Process queued messages first (inject before next assistant response)
+			if (queuedMessages.length > 0) {
+				for (const { original, llm } of queuedMessages) {
+					stream.push({ type: "message_start", message: original });
+					stream.push({ type: "message_end", message: original });
+					if (llm) {
+						currentContext.messages.push(llm);
+						newMessages.push(llm);
+					}
+				}
+				queuedMessages = [];
+			}
+
+			console.log("agent-loop: ", [...currentContext.messages]);
+
 			// Stream assistant response
 			const message = await streamAssistantResponse(currentContext, config, signal, stream, streamFn);
 			newMessages.push(message);
@@ -69,6 +87,9 @@ export function agentLoop(
 				newMessages.push(...toolResults);
 			}
 			stream.push({ type: "turn_end", message, toolResults: toolResults });
+
+			// Get queued messages after turn completes
+			queuedMessages = (await config.getQueuedMessages?.()) || [];
 		}
 		stream.push({ type: "agent_end", messages: newMessages });
 		stream.end(newMessages);
@@ -80,7 +101,7 @@ export function agentLoop(
 // Helper functions
 async function streamAssistantResponse(
 	context: AgentContext,
-	config: PromptConfig,
+	config: AgentLoopConfig,
 	signal: AbortSignal | undefined,
 	stream: EventStream<AgentEvent, AgentContext["messages"]>,
 	streamFn?: typeof streamSimple,

packages/ai/src/agent/index.ts

@@ -1,3 +1,3 @@
 export { agentLoop } from "./agent-loop.js";
 export * from "./tools/index.js";
-export type { AgentContext, AgentEvent, AgentTool, PromptConfig } from "./types.js";
+export type { AgentContext, AgentEvent, AgentLoopConfig, AgentTool, QueuedMessage } from "./types.js";

packages/ai/src/agent/types.ts

@@ -62,8 +62,15 @@ export type AgentEvent =
 	// contained in messages, which can be appended to the context
 	| { type: "agent_end"; messages: AgentContext["messages"] };
 
-// Configuration for prompt execution
-export interface PromptConfig extends SimpleStreamOptions {
+// Queued message with optional LLM representation
+export interface QueuedMessage<TApp = Message> {
+	original: TApp; // Original message for UI events
+	llm?: Message; // Optional transformed message for loop context (undefined if filtered)
+}
+
+// Configuration for agent loop execution
+export interface AgentLoopConfig extends SimpleStreamOptions {
 	model: Model<any>;
 	preprocessor?: (messages: AgentContext["messages"], abortSignal?: AbortSignal) => Promise<AgentContext["messages"]>;
+	getQueuedMessages?: <T>() => Promise<QueuedMessage<T>[]>;
 }