@contractspec/module.ai-chat 4.1.4 → 4.2.0

package/dist/node/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
 };