@contractspec/module.ai-chat 4.1.5 → 4.3.0

package/README.md

@@ -162,7 +162,10 @@ For full AI SDK compatibility including tool approval, use `createChatRoute` wit
 
 ```ts
 // app/api/chat/route.ts (Next.js App Router)
-import { createChatRoute } from '@contractspec/module.ai-chat/core';
+import {
+  createChatRoute,
+  CHAT_ROUTE_MAX_DURATION,
+} from '@contractspec/module.ai-chat/core';
 import { createProvider } from '@contractspec/lib.ai-providers';
 
 const provider = createProvider({
@@ -172,8 +175,11 @@ const provider = createProvider({
 });
 
 export const POST = createChatRoute({ provider });
+export const maxDuration = CHAT_ROUTE_MAX_DURATION;
 ```
 
+`maxDuration` is required for long-running streaming on Vercel/serverless. The route sends sources and reasoning by default (AI Elements parity). Override with `sendSources` and `sendReasoning` in options. Use `getModel` to support a model picker from the request body.
+
 ```tsx
 // Client: use @ai-sdk/react useChat with DefaultChatTransport
 import { useChat } from '@ai-sdk/react';
@@ -229,9 +235,29 @@ const chatService = new ChatService({
 // based on thinking level when modelSelector is set
 ```
 
-### Optional AI Elements
+### AI Elements-Style Components
+
+The module includes native implementations of [Reasoning](https://elements.ai-sdk.dev/components/reasoning), [Sources](https://elements.ai-sdk.dev/components/sources), [Suggestion](https://elements.ai-sdk.dev/components/suggestion), and [Chain of Thought](https://elements.ai-sdk.dev/components/chain-of-thought) patterns. Pass `components` to `ChatMessage` or `ChatWithExport` to override with ai-elements when installed:
+
+```tsx
+import { ChatWithSidebar, type ChatMessageComponents } from '@contractspec/module.ai-chat';
+// When using ai-elements: import { Reasoning, Sources, ... } from '@/components/ai-elements/...';
+
+<ChatWithSidebar
+  components={{
+    Reasoning: AiElementsReasoning,
+    Sources: AiElementsSources,
+    SourcesTrigger: AiElementsSourcesTrigger,
+    Source: AiElementsSource,
+    ChainOfThought: AiElementsChainOfThought,
+    ChainOfThoughtStep: AiElementsChainOfThoughtStep,
+  }}
+  suggestions={['Explain how X works', 'What is Y?']}
+  showSuggestionsWhenEmpty
+/>
+```
 
-Apps can optionally use [AI Elements](https://elements.ai-sdk.dev) for UI. This module does not depend on ai-elements; provide an adapter from `ChatMessage` to `UIMessage` when integrating.
+Use `suggestions` and `onSuggestionClick` for clickable prompt chips. Apps can optionally use [AI Elements](https://elements.ai-sdk.dev) for enhanced UI; provide an adapter from `ChatMessage` to `UIMessage` when integrating.
 
 ### useCompletion (Non-Chat Completion)
 

package/dist/browser/core/index.js

@@ -1089,39 +1089,74 @@ import {
   convertToModelMessages,
   streamText as streamText2
 } from "ai";
+import { z as z4 } from "zod";
+var CHAT_ROUTE_MAX_DURATION = 30;
+var REQUEST_BODY_SCHEMA = z4.object({
+  messages: z4.array(z4.unknown()).min(1, "messages array required"),
+  thinkingLevel: z4.enum(["instant", "thinking", "extra_thinking", "max"]).optional(),
+  model: z4.string().optional()
+});
 var DEFAULT_SYSTEM_PROMPT2 = `You are a helpful AI assistant.`;
+function defaultSendReasoning(thinkingLevel) {
+  return Boolean(thinkingLevel && thinkingLevel !== "instant");
+}
+function defaultSendSources() {
+  return true;
+}
 function createChatRoute(options) {
   const {
     provider,
     systemPrompt = DEFAULT_SYSTEM_PROMPT2,
     tools,
-    thinkingLevel: defaultThinkingLevel
+    thinkingLevel: defaultThinkingLevel,
+    sendSources = defaultSendSources(),
+    sendReasoning = defaultSendReasoning(defaultThinkingLevel),
+    getModel
   } = options;
   return async (req) => {
     if (req.method !== "POST") {
       return new Response("Method not allowed", { status: 405 });
     }
-    let body;
+    let rawBody;
     try {
-      body = await req.json();
+      rawBody = await req.json();
     } catch {
-      return new Response("Invalid JSON body", { status: 400 });
+      return new Response(JSON.stringify({ error: "Invalid JSON body" }), {
+        status: 400,
+        headers: { "Content-Type": "application/json" }
+      });
     }
-    const messages = body.messages ?? [];
-    if (!Array.isArray(messages) || messages.length === 0) {
-      return new Response("messages array required", { status: 400 });
+    const parsed = REQUEST_BODY_SCHEMA.safeParse(rawBody);
+    if (!parsed.success) {
+      const message = parsed.error.issues[0]?.message ?? "Invalid request body";
+      return new Response(JSON.stringify({ error: message }), {
+        status: 400,
+        headers: { "Content-Type": "application/json" }
+      });
     }
+    const body = parsed.data;
     const thinkingLevel = body.thinkingLevel ?? defaultThinkingLevel;
     const providerOptions = getProviderOptions(thinkingLevel, provider.name);
-    const model = provider.getModel();
-    const result = streamText2({
-      model,
-      messages: await convertToModelMessages(messages),
-      system: systemPrompt,
-      tools,
-      providerOptions: Object.keys(providerOptions).length > 0 ? providerOptions : undefined
-    });
-    return result.toUIMessageStreamResponse();
+    const model = getModel ? getModel(body) : provider.getModel();
+    try {
+      const result = streamText2({
+        model,
+        messages: await convertToModelMessages(body.messages),
+        system: systemPrompt,
+        tools,
+        providerOptions: Object.keys(providerOptions).length > 0 ? providerOptions : undefined
+      });
+      return result.toUIMessageStreamResponse({
+        sendSources,
+        sendReasoning
+      });
+    } catch (error) {
+      const message = error instanceof Error ? error.message : "An error occurred";
+      return new Response(JSON.stringify({ error: message }), {
+        status: 500,
+        headers: { "Content-Type": "application/json" }
+      });
+    }
   };
 }
 // src/core/create-completion-route.ts
@@ -1614,5 +1649,6 @@ export {
   THINKING_LEVEL_DESCRIPTIONS,
   LocalStorageConversationStore,
   InMemoryConversationStore,
-  ChatService
+  ChatService,
+  CHAT_ROUTE_MAX_DURATION
 };