@contractspec/module.ai-chat 4.1.4 → 4.2.0

package/dist/core/create-chat-route.d.ts

@@ -6,16 +6,19 @@
  * @example
  * ```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({ provider: 'openai', apiKey: process.env.OPENAI_API_KEY });
  * export const POST = createChatRoute({ provider });
+ * export const maxDuration = CHAT_ROUTE_MAX_DURATION;
  * ```
  */
-import { type ToolSet } from 'ai';
+import { type LanguageModel, type ToolSet } from 'ai';
 import type { Provider as ChatProvider } from '@contractspec/lib.ai-providers';
 import type { ThinkingLevel } from './thinking-levels';
+/** Recommended maxDuration for Next.js route handlers (Vercel/serverless). */
+export declare const CHAT_ROUTE_MAX_DURATION = 30;
 export interface CreateChatRouteOptions {
     /** LLM provider (from createProvider) */
     provider: ChatProvider;
@@ -25,6 +28,14 @@ export interface CreateChatRouteOptions {
     tools?: ToolSet;
     /** Default thinking level (can be overridden by request body) */
     thinkingLevel?: ThinkingLevel;
+    /** Send source citations to client (e.g. Perplexity). Default: true when thinkingLevel is set. */
+    sendSources?: boolean;
+    /** Send reasoning parts to client (extended thinking). Default: true when thinkingLevel is set and not instant. */
+    sendReasoning?: boolean;
+    /** Resolve model from request body (for model picker). If not provided, uses provider.getModel(). */
+    getModel?: (body: {
+        model?: string;
+    }) => LanguageModel;
 }
 /**
  * Create a route handler that streams chat using streamText + toUIMessageStreamResponse.
@@ -32,7 +43,7 @@ export interface CreateChatRouteOptions {
  * Compatible with @ai-sdk/react useChat when used as the API endpoint.
  * Supports tool approval when tools have needsApproval.
  *
- * @param options - Provider, system prompt, and optional tools
+ * @param options - Provider, system prompt, optional tools, sendSources, sendReasoning
  * @returns A function (req: Request) => Promise<Response> suitable for Next.js route handlers
  */
 export declare function createChatRoute(options: CreateChatRouteOptions): (req: Request) => Promise<Response>;

package/dist/core/index.js

@@ -1084,39 +1084,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
@@ -1609,5 +1644,6 @@ export {
   THINKING_LEVEL_DESCRIPTIONS,
   LocalStorageConversationStore,
   InMemoryConversationStore,
-  ChatService
+  ChatService,
+  CHAT_ROUTE_MAX_DURATION
 };

package/dist/core/message-types.d.ts

@@ -42,6 +42,10 @@ export interface ChatToolCall {
     result?: unknown;
     status: 'pending' | 'running' | 'completed' | 'error';
     error?: string;
+    /** When true, result is streaming (preliminary) */
+    preliminary?: boolean;
+    /** Nested UIMessage parts (subagent output) */
+    nestedParts?: unknown[];
 }
 /**
  * Source/citation in a message