improve ai chat

Author: alexphelps

app/api/chat/route.ts

@@ -3,6 +3,7 @@ import { convertToModelMessages, stepCountIs, streamText, tool, type UIMessage }
 import { z } from 'zod';
 import { source } from '@/lib/source';
 import { Document, type DocumentData } from 'flexsearch';
+import apiEndpoints from '@/lib/generated/api-endpoints.json';
 
 interface CustomDocument extends DocumentData {
   url: string;
@@ -39,6 +40,10 @@ async function createSearchServer() {
     if (doc) search.add(doc);
   }
 
+  for (const ep of apiEndpoints) {
+    search.add(ep as CustomDocument);
+  }
+
   return search;
 }
 
@@ -67,30 +72,22 @@ const systemPrompt = [
 ].join('\n');
 
 export async function POST(req: Request) {
-  const reqJson: { messages?: UIMessage[]; pageUrl?: string } = await req.json();
-
-  let contextPrompt = systemPrompt;
-  if (reqJson.pageUrl) {
-    const page = source.getPages().find((p) => p.url === reqJson.pageUrl);
-    if (page) {
-      contextPrompt += `\n\nThe user is currently viewing the page "${page.data.title}" (${reqJson.pageUrl}). Use this to prioritize relevant search results.`;
-    }
-  }
+  const reqJson: { messages?: UIMessage[] } = await req.json();
 
   const result = streamText({
     model: openrouter.chat(process.env.OPENROUTER_MODEL ?? 'anthropic/claude-3.5-sonnet'),
+    maxTokens: 4096,
     stopWhen: stepCountIs(5),
     tools: {
       search: searchTool,
     },
     messages: [
-      { role: 'system', content: contextPrompt },
+      { role: 'system', content: systemPrompt },
       ...(await convertToModelMessages(reqJson.messages ?? [])),
     ],
-    prepareStep: ({ stepNumber }) =>
-      stepNumber === 0 && reqJson.messages?.length === 1
-        ? { toolChoice: 'required' }
-        : { toolChoice: 'auto' },
+    prepareStep: ({ stepNumber }) => ({
+      toolChoice: stepNumber === 0 ? 'required' : 'auto',
+    }),
   });
 
   return result.toUIMessageStreamResponse();
@@ -100,11 +97,21 @@ const searchTool = tool({
   description: 'Search the docs content and return raw JSON results.',
   inputSchema: z.object({
     query: z.string(),
-    limit: z.number().int().min(1).max(100).default(10),
+    limit: z.number().int().min(1).max(20).default(5),
   }),
   async execute({ query, limit }) {
     const search = await searchServer;
-    return await search.searchAsync(query, { limit, merge: true, enrich: true });
+    const results = await search.searchAsync(query, { limit, merge: true, enrich: true });
+    // Truncate content to avoid flooding the context window
+    return results.map((r: any) => ({
+      ...r,
+      result: r.result?.map((doc: any) => ({
+        ...doc,
+        doc: doc.doc
+          ? { ...doc.doc, content: doc.doc.content?.slice(0, 1500) }
+          : doc.doc,
+      })),
+    }));
   },
 });
 

content/docs/campaigns/api/carts/cartsCalculate.mdx

@@ -0,0 +1,16 @@
+---
+title: Carts Calculate
+description: Calculate cart summary.
+full: true
+_openapi:
+  method: POST
+  toc: []
+  structuredData:
+    headings: []
+    contents:
+      - content: Calculate cart summary.
+---
+
+{/* This file was generated by Fumadocs. Do not edit this file directly. Any changes should be made by running the generation command again. */}
+
+<APIPage document={"public/api/campaigns/v1.yaml"} operations={[{"path":"/api/v1/carts/calculate/","method":"post"}]} />

lib/generated/api-endpoints.json

@@ -527,6 +527,7 @@
   "/docs/campaigns/api/campaigns/campaign-retrieve": "GET",
   "/docs/campaigns/api/campaigns/campaignRetrieve": "GET",
   "/docs/campaigns/api/carts/carts-create": "POST",
+  "/docs/campaigns/api/carts/cartsCalculate": "POST",
   "/docs/campaigns/api/carts/cartsCreate": "POST",
   "/docs/campaigns/api/cartsCreate": "POST",
   "/docs/campaigns/api/orderRetrieve": "GET",

lib/generated/api-methods.json

@@ -375,6 +375,29 @@ components:
       - lines
       - user
       type: object
+    CartCalculateSummary:
+      properties:
+        currency:
+          description: ISO 4217 currency code (e.g., USD, EUR, GBP), defaults to the
+            campaign currency if not provided
+          nullable: true
+          type: string
+        lines:
+          description: Array of package IDs
+          items:
+            $ref: '#/components/schemas/LineWithUpsell'
+          type: array
+        shipping_method:
+          description: Selected shipping method ID
+          type: integer
+        vouchers:
+          description: Campaign offer codes or store coupon code
+          items:
+            type: string
+          type: array
+      required:
+      - lines
+      type: object
     CartLine:
       properties:
         id:
@@ -427,6 +450,48 @@ components:
       - id
       - quantity
       type: object
+    CartSummary:
+      properties:
+        currency:
+          description: Cart currency in ISO 4217 format (e.g., USD, EUR, GBP)
+          maxLength: 3
+          type: string
+        lines:
+          items:
+            $ref: '#/components/schemas/SummaryLine'
+          type: array
+        offer_discounts:
+          description: Array of offer discounts applied to the cart
+          items:
+            $ref: '#/components/schemas/Discount'
+          type: array
+        shipping_method:
+          allOf:
+          - $ref: '#/components/schemas/SummaryShippingMethod'
+          description: Selected shipping method details
+        subtotal:
+          description: Cart subtotal before discounts
+          format: decimal
+          pattern: ^-?\d{0,10}(?:\.\d{0,2})?$
+          type: string
+        total:
+          description: Cart total after discounts
+          format: decimal
+          pattern: ^-?\d{0,10}(?:\.\d{0,2})?$
+          type: string
+        total_discount:
+          description: Total discount applied to the cart
+          format: decimal
+          pattern: ^-?\d{0,10}(?:\.\d{0,2})?$
+          type: string
+        voucher_discounts:
+          description: Array of voucher discounts applied to the cart
+          items:
+            $ref: '#/components/schemas/Discount'
+          type: array
+      required:
+      - currency
+      type: object
     Country:
       properties:
         code:
@@ -511,15 +576,24 @@ components:
     Discount:
       properties:
         amount:
+          description: Discount amount
           format: decimal
           pattern: ^-?\d{0,10}(?:\.\d{0,2})?$
           type: string
         description:
-          maxLength: 255
+          description: Description of the discount
           type: string
         name:
-          maxLength: 255
+          description: Name of the discount
           type: string
+        offer_id:
+          description: ID of the applied offer
+          type: integer
+      required:
+      - amount
+      - description
+      - name
+      - offer_id
       type: object
     IntervalEnum:
       enum:
@@ -1402,6 +1476,98 @@ components:
       required:
       - price
       type: object
+    SummaryLine:
+      properties:
+        discounts:
+          description: Array of discounts applied to the line
+          items:
+            $ref: '#/components/schemas/Discount'
+          type: array
+        original_package_price:
+          description: Package price before discounts
+          format: decimal
+          pattern: ^-?\d{0,10}(?:\.\d{0,2})?$
+          type: string
+        original_unit_price:
+          description: Unit price before discounts
+          format: decimal
+          pattern: ^-?\d{0,10}(?:\.\d{0,2})?$
+          type: string
+        package_id:
+          description: Package ID of products to add to the order
+          type: integer
+        package_price:
+          description: Package price after discounts
+          format: decimal
+          pattern: ^-?\d{0,10}(?:\.\d{0,2})?$
+          type: string
+        quantity:
+          description: Package quantity
+          type: integer
+        subtotal:
+          description: Line total before discounts
+          format: decimal
+          pattern: ^-?\d{0,10}(?:\.\d{0,2})?$
+          type: string
+        total:
+          description: Line total after discounts
+          format: decimal
+          pattern: ^-?\d{0,10}(?:\.\d{0,2})?$
+          type: string
+        total_discount:
+          description: Total discount applied to the line
+          format: decimal
+          pattern: ^-?\d{0,10}(?:\.\d{0,2})?$
+          type: string
+        unit_price:
+          description: Unit price after discounts
+          format: decimal
+          pattern: ^-?\d{0,10}(?:\.\d{0,2})?$
+          type: string
+      required:
+      - original_package_price
+      - original_unit_price
+      - package_id
+      - package_price
+      - quantity
+      - subtotal
+      - total
+      - total_discount
+      - unit_price
+      type: object
+    SummaryShippingMethod:
+      properties:
+        code:
+          description: Shipping method code
+          type: string
+        discounts:
+          description: Array of discounts applied to the shipping method
+          items:
+            $ref: '#/components/schemas/Discount'
+          type: array
+        id:
+          description: Shipping method ID
+          type: integer
+        name:
+          description: Shipping method name
+          type: string
+        original_price:
+          description: Shipping price before discounts
+          format: decimal
+          pattern: ^-?\d{0,10}(?:\.\d{0,2})?$
+          type: string
+        price:
+          description: Shipping price after discounts
+          format: decimal
+          pattern: ^-?\d{0,10}(?:\.\d{0,2})?$
+          type: string
+      required:
+      - code
+      - id
+      - name
+      - original_price
+      - price
+      type: object
     User:
       properties:
         accepts_marketing:
@@ -1653,21 +1819,12 @@ paths:
     get:
       description: Retrieve campaign details.
       operationId: campaignRetrieve
-      parameters:
-      - description: ISO 4217 currency code (e.g., USD, EUR, GBP) used to filter available
-          packages and shipping methods in the campaign
-        in: query
-        name: currency
-        schema:
-          type: string
       responses:
         '200':
           content:
             application/json:
               schema:
-                items:
-                  $ref: '#/components/schemas/Campaign'
-                type: array
+                $ref: '#/components/schemas/Campaign'
           description: ''
       security:
       - API Authentication: []
@@ -1700,6 +1857,33 @@ paths:
       - API Authentication: []
       tags:
       - carts
+  /api/v1/carts/calculate/:
+    post:
+      description: Calculate cart summary.
+      operationId: cartsCalculate
+      requestBody:
+        content:
+          application/json:
+            schema:
+              $ref: '#/components/schemas/CartCalculateSummary'
+          application/x-www-form-urlencoded:
+            schema:
+              $ref: '#/components/schemas/CartCalculateSummary'
+          multipart/form-data:
+            schema:
+              $ref: '#/components/schemas/CartCalculateSummary'
+        required: true
+      responses:
+        '200':
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/CartSummary'
+          description: ''
+      security:
+      - API Authentication: []
+      tags:
+      - carts
   /api/v1/orders/:
     post:
       description: Create a new order.