@falai/agent 0.9.0 → 1.0.0

package/examples/core-concepts/session-management.ts

@@ -10,6 +10,11 @@
  * - Data persistence across turns
  * - Session restoration
  * - Route transitions with session state
+ * - NEW: Flexible ConditionTemplate patterns for session-aware routing:
+ *   - String-only conditions: "user wants to order" (AI context only)
+ *   - Function-only conditions: (ctx) => !!ctx.data?.orderId (session state checks)
+ *   - Mixed arrays: ["payment needed", (ctx) => !ctx.data?.confirmed] (hybrid)
+ *   - Session-aware skipIf conditions for flow control
  */
 
 import {
@@ -17,6 +22,7 @@ import {
   GeminiProvider,
   MemoryAdapter,
   type Tool,
+  ValidationError,
 } from "../../src/index";
 
 // Define data types for our multi-step process
@@ -36,23 +42,23 @@ interface OrderContext {
 }
 
 interface PaymentData {
-  orderId: string;
+  orderId?: string; // Make this optional to match OrderData
   paymentMethod: "credit_card" | "paypal" | "bank_transfer";
   amount: number;
   confirmed: boolean;
 }
 
-// Tools for order processing
-const createOrder: Tool<unknown, UnifiedOrderData, [], string> = {
+// Tools for order processing - using unified Tool interface
+const createOrderTool: Tool<unknown, UnifiedOrderData> = {
   id: "create_order",
   description: "Create a new order from the collected order data",
   parameters: {
     type: "object",
     properties: {},
   },
-  handler: ({ data }) => {
+  handler: (context, args) => {
     const orderId = `ORD-${Date.now()}`;
-    console.log(`Creating order ${orderId} for ${data?.customerName}`);
+    console.log(`Creating order ${orderId} for ${context.data?.customerName}`);
 
     return {
       data: `Order ${orderId} created successfully!`,
@@ -63,22 +69,22 @@ const createOrder: Tool<unknown, UnifiedOrderData, [], string> = {
   },
 };
 
-const processPayment: Tool<unknown, UnifiedOrderData, [], string> = {
+const processPaymentTool: Tool<unknown, UnifiedOrderData> = {
   id: "process_payment",
   description: "Process payment for an order",
   parameters: {
     type: "object",
     properties: {},
   },
-  handler: ({ data }) => {
-    console.log(`Processing payment for order ${data?.orderId}`);
+  handler: (context, args) => {
+    console.log(`Processing payment for order ${context.data?.orderId}`);
 
     // Simulate payment processing
     const success = Math.random() > 0.1; // 90% success rate
 
     if (success) {
       return {
-        data: `Payment processed successfully! Order ${data?.orderId} is now confirmed.`,
+        data: `Payment processed successfully! Order ${context.data?.orderId} is now confirmed.`,
         dataUpdate: {
           confirmed: true,
         },
@@ -95,7 +101,7 @@ const processPayment: Tool<unknown, UnifiedOrderData, [], string> = {
 };
 
 // Define unified data schema for all order-related interactions
-interface UnifiedOrderData extends OrderData, PaymentData {}
+interface UnifiedOrderData extends OrderData, PaymentData { }
 
 const orderSchema = {
   type: "object",
@@ -133,10 +139,75 @@ const agent = new Agent<unknown, UnifiedOrderData>({
   },
 });
 
+// Demonstrate different tool registration patterns
+
+// Method 1: Register tools for ID-based reference in steps
+agent.tool.register(createOrderTool);
+agent.tool.register(processPaymentTool);
+
+// Method 2: Create specialized validation tool
+const orderValidationTool = agent.tool.createValidation({
+  id: "validate_order",
+  fields: ["customerName", "productType", "budget"] as const,
+  validator: async (context, data) => {
+    const errors: ValidationError[] = [];
+    if (!data.customerName || data.customerName.length < 2) {
+      errors.push({
+        field: "customerName",
+        value: data.customerName,
+        message: "Customer name must be at least 2 characters",
+        schemaPath: "customerName"
+      });
+    }
+    if (!data.budget || data.budget < 100) {
+      errors.push({
+        field: "budget",
+        value: data.budget,
+        message: "Budget must be at least $100",
+        schemaPath: "budget"
+      });
+    }
+    return {
+      valid: errors.length === 0,
+      errors,
+      warnings: [],
+    };
+  },
+});
+
+// Method 3: Create data enrichment tool
+const orderEnrichmentTool = agent.tool.createDataEnrichment({
+  id: "enrich_order",
+  fields: ["productType", "budget"] as const,
+  enricher: async (context, data) => {
+    // Enrich with fields that exist in UnifiedOrderData
+    const urgentDelivery = data.budget > 1000; // Premium orders get urgent delivery
+    const preferredColor = data.productType === "laptop" ? "silver" : "black";
+
+    return {
+      urgentDelivery,
+      preferredColor,
+    };
+  },
+});
+
 // Order collection route with sequential steps
 agent.createRoute({
   title: "Product Order",
   description: "Collect order details and create an order",
+  // Mixed condition: AI context + programmatic validation
+  when: [
+    "user wants to place an order or buy something",
+    (ctx) => {
+      const message = ctx.helpers.getLastUserMessage()?.toLowerCase() || '';
+      return message.includes('order') || message.includes('buy') || message.includes('purchase');
+    }
+  ],
+  // Skip if user already has a pending order
+  skipIf: [
+    "user already has an active order",
+    (ctx) => !!ctx.data?.orderId && !ctx.data?.confirmed
+  ],
   // NEW: Required fields for route completion
   requiredFields: ["customerName", "productType", "budget"],
   // NEW: Optional fields that enhance the experience
@@ -148,7 +219,8 @@ agent.createRoute({
       description: "Ask for customer name",
       prompt: "Hi! I'd like to help you place an order. What's your name?",
       collect: ["customerName"],
-      skipIf: (data: Partial<UnifiedOrderData>) => !!data.customerName,
+      // String-only skipIf for AI context
+      skipIf: "customer name already provided",
     },
     {
       id: "ask_product",
@@ -156,7 +228,8 @@ agent.createRoute({
       prompt: "What would you like to order? (laptop, phone, or tablet)",
       collect: ["productType"],
       requires: ["customerName"],
-      skipIf: (data: Partial<UnifiedOrderData>) => !!data.productType,
+      // Function-only skipIf for programmatic check
+      skipIf: (ctx) => !!ctx.data?.productType,
     },
     {
       id: "ask_budget",
@@ -164,7 +237,11 @@ agent.createRoute({
       prompt: "What's your budget for this purchase?",
       collect: ["budget"],
       requires: ["customerName", "productType"],
-      skipIf: (data: Partial<UnifiedOrderData>) => data.budget !== undefined,
+      // Mixed skipIf: AI context + programmatic logic
+      skipIf: [
+        "budget already specified",
+        (ctx) => ctx.data?.budget !== undefined
+      ],
     },
     {
       id: "ask_color",
@@ -172,7 +249,8 @@ agent.createRoute({
       prompt: "Do you have a preferred color?",
       collect: ["preferredColor"],
       requires: ["customerName", "productType", "budget"],
-      skipIf: (data: Partial<UnifiedOrderData>) => !!data.preferredColor,
+      // Function-only skipIf
+      skipIf: (ctx) => !!ctx.data?.preferredColor,
     },
     {
       id: "ask_urgent",
@@ -180,13 +258,14 @@ agent.createRoute({
       prompt: "Do you need urgent delivery?",
       collect: ["urgentDelivery"],
       requires: ["customerName", "productType", "budget"],
-      skipIf: (data: Partial<UnifiedOrderData>) => data.urgentDelivery !== undefined,
+      // String-only skipIf for simple cases
+      skipIf: "delivery preference already collected",
     },
     {
       id: "create_order",
       description: "Create the order",
       prompt: "Great! Let me create your order.",
-      tools: [createOrder],
+      tools: ["create_order"], // Reference by ID
       requires: ["customerName", "productType", "budget"],
     },
   ],
@@ -200,6 +279,10 @@ agent.createRoute({
 agent.createRoute({
   title: "Payment Processing",
   description: "Process payment for an order",
+  // Function-only condition for programmatic logic
+  when: (ctx) => !!ctx.data?.orderId && !ctx.data?.confirmed,
+  // Skip if payment already processed
+  skipIf: (ctx) => !!ctx.data?.confirmed,
   // NEW: Required fields for route completion
   requiredFields: ["orderId", "paymentMethod", "amount"],
   // NEW: Optional fields
@@ -212,7 +295,11 @@ agent.createRoute({
       prompt:
         "Now let's process payment for your order. What payment method would you prefer?",
       collect: ["paymentMethod"],
-      skipIf: (data: Partial<UnifiedOrderData>) => !!data.paymentMethod,
+      // Mixed skipIf: AI context + programmatic check
+      skipIf: [
+        "payment method already selected",
+        (ctx) => !!ctx.data?.paymentMethod
+      ],
     },
     {
       id: "ask_amount",
@@ -220,13 +307,14 @@ agent.createRoute({
       prompt: "What's the payment amount?",
       collect: ["amount"],
       requires: ["paymentMethod"],
-      skipIf: (data: Partial<UnifiedOrderData>) => data.amount !== undefined,
+      // Function-only skipIf
+      skipIf: (ctx) => ctx.data?.amount !== undefined,
     },
     {
       id: "process_payment",
       description: "Process the payment",
       prompt: "Processing your payment...",
-      tools: [processPayment],
+      tools: ["process_payment"], // Reference by ID
       requires: ["orderId", "paymentMethod", "amount"],
     },
   ],
@@ -255,7 +343,7 @@ async function demonstrateSessionManagement() {
 
   // Copy routes to the session agent
   agent.getRoutes().forEach(route => {
-    sessionAgent.createRoute(route);
+    sessionAgent.createRoute(route.toOptions());
   });
 
   // Turn 1: Start order process - simple message API
@@ -266,7 +354,7 @@ async function demonstrateSessionManagement() {
 
   console.log("Bot:", response1.message);
   console.log("Session ID:", sessionAgent.session.id);
-  console.log("Session data:", JSON.stringify(sessionAgent.session.getData<UnifiedOrderData>(), null, 2));
+  console.log("Session data:", JSON.stringify(sessionAgent.session.getData(), null, 2));
   console.log("History length:", sessionAgent.session.getHistory().length);
   console.log();
 
@@ -277,7 +365,7 @@ async function demonstrateSessionManagement() {
   const response2 = await sessionAgent.chat("My budget is $1500");
 
   console.log("Bot:", response2.message);
-  console.log("Session data:", JSON.stringify(sessionAgent.session.getData<UnifiedOrderData>(), null, 2));
+  console.log("Session data:", JSON.stringify(sessionAgent.session.getData(), null, 2));
   console.log("History length:", sessionAgent.session.getHistory().length);
   console.log();
 
@@ -288,7 +376,7 @@ async function demonstrateSessionManagement() {
   const response3 = await sessionAgent.chat("I want black color and urgent delivery please");
 
   console.log("Bot:", response3.message);
-  console.log("Session data:", JSON.stringify(sessionAgent.session.getData<UnifiedOrderData>(), null, 2));
+  console.log("Session data:", JSON.stringify(sessionAgent.session.getData(), null, 2));
   console.log("Route complete:", response3.isRouteComplete);
   console.log("History length:", sessionAgent.session.getHistory().length);
   console.log();
@@ -300,7 +388,7 @@ async function demonstrateSessionManagement() {
   const response4 = await sessionAgent.chat("I'll pay with credit card, amount is $1599");
 
   console.log("Bot:", response4.message);
-  console.log("Session data:", JSON.stringify(sessionAgent.session.getData<UnifiedOrderData>(), null, 2));
+  console.log("Session data:", JSON.stringify(sessionAgent.session.getData(), null, 2));
   console.log("Route complete:", response4.isRouteComplete);
   console.log("Final history length:", sessionAgent.session.getHistory().length);
 }
@@ -334,7 +422,7 @@ async function demonstrateSessionPersistence() {
 
   // Copy routes to the persistent agent
   agent.getRoutes().forEach(route => {
-    persistentAgent.createRoute(route);
+    persistentAgent.createRoute(route.toOptions());
   });
 
   console.log("Session ready:", persistentAgent.session.id);
@@ -344,7 +432,7 @@ async function demonstrateSessionPersistence() {
 
   console.log("After first interaction:");
   console.log("🤖 Agent:", response1.message);
-  console.log("Session data:", JSON.stringify(persistentAgent.session.getData<UnifiedOrderData>(), null, 2));
+  console.log("Session data:", JSON.stringify(persistentAgent.session.getData(), null, 2));
   console.log("History length:", persistentAgent.session.getHistory().length);
   console.log("Session automatically saved to persistence ✓");
 
@@ -373,20 +461,20 @@ async function demonstrateSessionPersistence() {
 
   // Copy routes to the restored agent
   agent.getRoutes().forEach(route => {
-    restoredAgent.createRoute(route);
+    restoredAgent.createRoute(route.toOptions());
   });
 
   console.log("Session automatically restored:");
   console.log("- Session ID:", restoredAgent.session.id);
   console.log("- History length:", restoredAgent.session.getHistory().length);
-  console.log("- Restored data:", JSON.stringify(restoredAgent.session.getData<UnifiedOrderData>(), null, 2));
+  console.log("- Restored data:", JSON.stringify(restoredAgent.session.getData(), null, 2));
 
   // Continue conversation seamlessly
   const response2 = await restoredAgent.chat("Actually, make it urgent delivery");
 
   console.log("\nAfter continuing with restored session:");
   console.log("🤖 Agent:", response2.message);
-  console.log("Session data:", JSON.stringify(restoredAgent.session.getData<UnifiedOrderData>(), null, 2));
+  console.log("Session data:", JSON.stringify(restoredAgent.session.getData(), null, 2));
   console.log("History length:", restoredAgent.session.getHistory().length);
   console.log("Session automatically saved again ✓");
 }

package/examples/integrations/database-integration.ts

@@ -155,7 +155,7 @@ async function example() {
   const onboardingRoute = agent.createRoute({
     title: "Customer Onboarding",
     description: "Collect customer information",
-    conditions: [
+    when: [
       "User is a new customer",
       "User needs to set up their account",
     ],
@@ -171,19 +171,19 @@ async function example() {
       id: "ask_name",
       prompt: "Ask for full name",
       collect: ["fullName"],
-      skipIf: (data: Partial<OnboardingData>) => !!data.fullName,
+      skipIf: (ctx) => !!ctx.data?.fullName,
     })
     .nextStep({
       id: "ask_email",
       prompt: "Ask for email address",
       collect: ["email"],
-      skipIf: (data: Partial<OnboardingData>) => !!data.email,
+      skipIf: (ctx) => !!ctx.data?.email,
     })
     .nextStep({
       id: "ask_company",
       prompt: "Ask for company name",
       collect: ["companyName"],
-      skipIf: (data: Partial<OnboardingData>) => !!data.companyName,
+      skipIf: (ctx) => !!ctx.data?.companyName,
     })
     .nextStep({
       id: "ask_phone",
@@ -414,7 +414,7 @@ async function syncSessionToDatabase(
 /**
  * Mock function to simulate processing the completed onboarding data.
  */
-async function processOnboarding(data: Partial<OnboardingData> | undefined) {
+async function processOnboarding(data) {
   console.log("\n🚀 Processing onboarding data...");
   console.log(
     `   - Creating account for ${data?.fullName} at ${data?.companyName}`
@@ -453,7 +453,7 @@ async function advancedExample() {
     schema: smartOnboardingSchema,
     hooks: {
       // Validate and enrich collected data
-      onDataUpdate: (data: Partial<OnboardingData>, _previous: Partial<OnboardingData>) => {
+      onDataUpdate: (data, _previous: Partial<OnboardingData>) => {
         console.log("🔄 Data collected, validating...");
 
         // Normalize email

package/examples/integrations/healthcare-integration.ts

@@ -39,7 +39,7 @@ interface SatisfactionData {
 
 // Tools
 
-const getUpcomingSlots: Tool<HealthcareContext, HealthcareData, unknown[], unknown> = {
+const getUpcomingSlots: Tool<HealthcareContext, HealthcareData> = {
   id: "get_upcoming_slots",
   name: "Available Appointment Slots",
   description: "Get upcoming appointment slots",
@@ -47,12 +47,12 @@ const getUpcomingSlots: Tool<HealthcareContext, HealthcareData, unknown[], unkno
     type: "object",
     properties: {},
   },
-  handler: () => {
-    return { data: ["Monday 10 AM", "Tuesday 2 PM", "Wednesday 1 PM"] };
+  handler: (context, args) => {
+    return { data: "Available slots: Monday 10 AM, Tuesday 2 PM, Wednesday 1 PM" };
   },
 };
 
-const getLaterSlots: Tool<HealthcareContext, HealthcareData, unknown[], unknown> = {
+const getLaterSlots: Tool<HealthcareContext, HealthcareData> = {
   id: "get_later_slots",
   name: "Extended Appointment Slots",
   description: "Get later appointment slots",
@@ -60,17 +60,12 @@ const getLaterSlots: Tool<HealthcareContext, HealthcareData, unknown[], unknown>
     type: "object",
     properties: {},
   },
-  handler: () => {
-    return { data: ["November 3, 11:30 AM", "November 12, 3 PM"] };
+  handler: (context, args) => {
+    return { data: "Later slots: November 3, 11:30 AM, November 12, 3 PM" };
   },
 };
 
-const scheduleAppointment: Tool<
-  HealthcareContext,
-  HealthcareData,
-  unknown[],
-  unknown
-> = {
+const scheduleAppointment: Tool<HealthcareContext, HealthcareData> = {
   id: "schedule_appointment",
   name: "Appointment Scheduler",
   description: "Schedule an appointment with a healthcare provider",
@@ -81,26 +76,21 @@ const scheduleAppointment: Tool<
     },
     required: ["datetime"],
   },
-  handler: ({ data }) => {
-    if (!data?.preferredDate || !data?.preferredTime) {
+  handler: (context, args) => {
+    if (!context.data?.preferredDate || !context.data?.preferredTime) {
       return {
         data: "Please specify preferred date and time for the appointment",
       };
     }
     return {
-      data: `Appointment scheduled for ${data.preferredDate} at ${
-        data.preferredTime
-      } for ${data.appointmentReason || "consultation"}`,
+      data: `Appointment scheduled for ${context.data.preferredDate} at ${
+        context.data.preferredTime
+      } for ${context.data.appointmentReason || "consultation"}`,
     };
   },
 };
 
-const getLabResults: Tool<
-  HealthcareContext,
-  HealthcareData,
-  unknown[],
-  unknown
-> = {
+const getLabResults: Tool<HealthcareContext, HealthcareData> = {
   id: "get_lab_results",
   name: "Lab Results Retriever",
   description: "Get lab test results",
@@ -108,13 +98,9 @@ const getLabResults: Tool<
     type: "object",
     properties: {},
   },
-  handler: ({ context, data }) => {
+  handler: (context, args) => {
     return {
-      data: {
-        report: `Lab results for ${data?.testType || "general"} tests`,
-        prognosis: "All tests are within the valid range",
-        patientName: context.patientName,
-      },
+      data: `Lab results for ${context.data?.testType || "general"} tests: All tests are within the valid range for ${context.context.patientName}`,
     };
   },
 };
@@ -262,7 +248,7 @@ function createHealthcareAgent() {
   const schedulingRoute = agent.createRoute({
     title: "Schedule an Appointment",
     description: "Helps the patient find a time for their appointment.",
-    conditions: ["The patient wants to schedule an appointment"],
+    when: ["The patient wants to schedule an appointment"],
     // Route-level identity for healthcare scheduling
     identity:
       "You are a compassionate healthcare scheduling assistant who helps patients book appointments. Be empathetic, prioritize urgent cases, and ensure patients feel supported throughout the process.",
@@ -295,7 +281,7 @@ function createHealthcareAgent() {
   const collectReason = schedulingRoute.initialStep.nextStep({
     prompt: "Ask what the patient needs an appointment for",
     collect: ["appointmentReason"],
-    skipIf: (data) => !!data.appointmentReason,
+    skipIf: (ctx) => !!ctx.data?.appointmentReason,
     when: "Patient hasn't specified reason for appointment yet",
   });
 
@@ -303,7 +289,7 @@ function createHealthcareAgent() {
   const checkUrgency = collectReason.nextStep({
     prompt: "Check if this is urgent and show available slots",
     collect: ["urgency"],
-    skipIf: (data) => !!data.urgency,
+    skipIf: (ctx) => !!ctx.data?.urgency,
     requires: ["appointmentReason"],
     when: "Reason provided, now assess urgency level",
   });
@@ -321,14 +307,14 @@ function createHealthcareAgent() {
   const collectPreferences = presentTimes.nextStep({
     prompt: "Collect preferred time and date",
     collect: ["preferredTime", "preferredDate"],
-    skipIf: (data) => !!data.preferredTime && !!data.preferredDate,
+    skipIf: (ctx) => !!ctx.data?.preferredTime && !!ctx.data?.preferredDate,
   });
 
   // Step 5: Confirm details and schedule
   const confirmDetails = collectPreferences.nextStep({
     prompt: "Confirm the details with the patient before scheduling",
     collect: ["appointmentType"],
-    skipIf: (data) => !!data.appointmentType,
+    skipIf: (ctx) => !!ctx.data?.appointmentType,
     requires: ["appointmentReason", "preferredTime", "preferredDate"],
   });
 
@@ -372,7 +358,7 @@ function createHealthcareAgent() {
   const labResultsRoute = agent.createRoute({
     title: "Lab Results",
     description: "Retrieves the patient's lab results and explains them.",
-    conditions: ["The patient wants to see their lab results"],
+    when: ["The patient wants to see their lab results"],
     // NEW: Required fields for route completion
     requiredFields: ["testType"],
     // NEW: Optional fields
@@ -383,14 +369,14 @@ function createHealthcareAgent() {
   const collectTestInfo = labResultsRoute.initialStep.nextStep({
     prompt: "Ask what type of test results they want to see",
     collect: ["testType"],
-    skipIf: (data) => !!data.testType,
+    skipIf: (ctx) => !!ctx.data?.testType,
   });
 
   // Step 2: Optional: collect test date
   const collectTestDate = collectTestInfo.nextStep({
     prompt: "Ask for the test date if they remember it",
     collect: ["testDate"],
-    skipIf: (data) => !!data.testDate,
+    skipIf: (ctx) => !!ctx.data?.testDate,
     requires: ["testType"],
   });
 
@@ -418,7 +404,7 @@ function createHealthcareAgent() {
   const satisfactionRoute = agent.createRoute({
     title: "Satisfaction Survey",
     description: "Quick satisfaction survey after scheduling",
-    conditions: ["Collect patient satisfaction feedback"],
+    when: ["Collect patient satisfaction feedback"],
 
     // Route-specific knowledge base for patient feedback
     knowledgeBase: {
@@ -456,7 +442,7 @@ function createHealthcareAgent() {
     prompt:
       "Ask for overall satisfaction rating from 1 to 5 with the scheduling experience",
     collect: ["rating"],
-    skipIf: (data) => !!data.rating,
+    skipIf: (ctx) => !!ctx.data?.rating,
   });
 
   const askComments = askRating.nextStep({

package/examples/integrations/search-integration.ts

@@ -76,7 +76,7 @@ async function example() {
   const userId = "user_123";
 
   // Create adapter
-  const adapter = new OpenSearchAdapter<ConversationContext>(client, {
+  const adapter = new OpenSearchAdapter(client, {
     indices: {
       sessions: "agent_sessions",
       messages: "agent_messages",
@@ -145,7 +145,7 @@ async function example() {
   const complaintRoute = agent.createRoute({
     title: "Handle Customer Complaint",
     description: "Process and resolve customer complaints",
-    conditions: [
+    when: [
       "User has a complaint",
       "User reports an issue or problem",
       "User is dissatisfied",
@@ -161,18 +161,18 @@ async function example() {
     .nextStep({
       prompt: "Understand the complaint",
       collect: ["category", "severity", "description"],
-      skipIf: (data: Partial<ComplaintData>) => !!data.description,
+      skipIf: (ctx) => !!ctx.data?.description,
     })
     .nextStep({
       prompt: "Identify affected service",
       collect: ["affectedService"],
-      skipIf: (data: Partial<ComplaintData>) => !!data.affectedService,
+      skipIf: (ctx) => !!ctx.data?.affectedService,
       requires: ["description"],
     })
     .nextStep({
       prompt: "Ask for desired resolution",
       collect: ["requestedResolution"],
-      skipIf: (data: Partial<ComplaintData>) => !!data.requestedResolution,
+      skipIf: (ctx) => !!ctx.data?.requestedResolution,
       requires: ["category", "description"],
     })
     .nextStep({
@@ -254,7 +254,7 @@ async function example() {
   recoveredAgent.createRoute({
     title: "Handle Customer Complaint",
     description: "Process and resolve customer complaints",
-    conditions: [
+    when: [
       "User has a complaint",
       "User reports an issue or problem",
       "User is dissatisfied",
@@ -324,7 +324,7 @@ async function analyticsExample() {
     auth: { username: "admin", password: "admin" },
   });
 
-  const adapter = new OpenSearchAdapter<ConversationContext>(client, {
+  const adapter = new OpenSearchAdapter(client, {
     indices: {
       sessions: "support_sessions",
       messages: "support_messages",
@@ -396,7 +396,7 @@ async function analyticsExample() {
       // NEW: Agent-level schema
       schema: ticketSchema,
       persistence: {
-        adapter: new OpenSearchAdapter<AnalyticsContext>(client),
+        adapter: new OpenSearchAdapter(client),
         autoSave: true,
       },
     });