@falai/agent 0.9.2 → 1.0.0

package/examples/core-concepts/schema-driven-extraction.ts

@@ -8,7 +8,11 @@
  * - Schema-first data definition
  * - Type-safe data extraction
  * - Automatic validation
- * - SkipIf conditions for smart flow control
+ * - NEW: Flexible ConditionTemplate patterns for routing and step control:
+ *   - String-only conditions: "user wants to register" (AI context only)
+ *   - Function-only conditions: (ctx) => ctx.data?.complete (programmatic only)
+ *   - Mixed arrays: ["user provided info", (ctx) => !!ctx.data?.name] (hybrid)
+ *   - Enhanced skipIf conditions for smart flow control
  */
 
 import { Agent, GeminiProvider, type Tool } from "../../src/index";
@@ -106,6 +110,16 @@ agent.addTool(saveUserProfileTool);
 agent.createRoute({
   title: "User Profile Collection",
   description: "Collect comprehensive user profile information",
+  // Mixed condition: AI context + programmatic validation
+  when: [
+    "user wants to create a profile or register",
+    (ctx) => {
+      const message = ctx.helpers.getLastUserMessage()?.toLowerCase() || '';
+      return message.includes('profile') || message.includes('register') || message.includes('sign up');
+    }
+  ],
+  // Skip if user already has a complete profile
+  skipIf: (ctx) => Boolean(ctx.data?.name && ctx.data?.email && ctx.data?.age),
   // NEW: Required fields for route completion (instead of schema)
   requiredFields: ["name", "email"],
   // NEW: Optional fields that enhance the experience
@@ -117,7 +131,8 @@ agent.createRoute({
       description: "Ask for user's name",
       prompt: "Hi! I'd like to create a profile for you. What's your name?",
       collect: ["name"],
-      skipIf: (data: Partial<UserProfileData>) => !!data.name,
+      // String-only skipIf for AI context
+      skipIf: "user already provided their name",
     },
     {
       id: "ask_email",
@@ -125,7 +140,8 @@ agent.createRoute({
       prompt: "What's your email address?",
       collect: ["email"],
       requires: ["name"],
-      skipIf: (data: Partial<UserProfileData>) => !!data.email,
+      // Function-only skipIf for programmatic logic
+      skipIf: (ctx) => !!ctx.data?.email,
     },
     {
       id: "ask_age",
@@ -133,7 +149,11 @@ agent.createRoute({
       prompt: "How old are you? (optional)",
       collect: ["age"],
       requires: ["name", "email"],
-      skipIf: (data: Partial<UserProfileData>) => data.age !== undefined,
+      // Mixed skipIf: AI context + programmatic check
+      skipIf: [
+        "age already collected",
+        (ctx) => ctx.data?.age !== undefined
+      ],
     },
     {
       id: "ask_interests",
@@ -141,8 +161,11 @@ agent.createRoute({
       prompt: "What are your interests or hobbies? (optional)",
       collect: ["interests"],
       requires: ["name", "email"],
-      skipIf: (data: Partial<UserProfileData>) =>
-        !!(data.interests && data.interests.length > 0),
+      // Mixed skipIf with complex logic
+      skipIf: [
+        "interests already provided",
+        (ctx) => !!(ctx.data?.interests && ctx.data.interests.length > 0)
+      ],
     },
     {
       id: "ask_contact_preference",
@@ -150,7 +173,8 @@ agent.createRoute({
       prompt: "What's your preferred contact method?",
       collect: ["preferredContact"],
       requires: ["name", "email"],
-      skipIf: (data: Partial<UserProfileData>) => !!data.preferredContact,
+      // Function-only skipIf
+      skipIf: (ctx) => !!ctx.data?.preferredContact,
     },
     {
       id: "ask_newsletter",
@@ -158,8 +182,8 @@ agent.createRoute({
       prompt: "Would you like to subscribe to our newsletter?",
       collect: ["newsletterOptIn"],
       requires: ["name", "email"],
-      skipIf: (data: Partial<UserProfileData>) =>
-        data.newsletterOptIn !== undefined,
+      // String-only skipIf for simple cases
+      skipIf: "newsletter preference already collected",
     },
     {
       id: "save_profile",

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 {
@@ -96,7 +101,7 @@ const processPaymentTool: Tool<unknown, UnifiedOrderData> = {
 };
 
 // Define unified data schema for all order-related interactions
-interface UnifiedOrderData extends OrderData, PaymentData {}
+interface UnifiedOrderData extends OrderData, PaymentData { }
 
 const orderSchema = {
   type: "object",
@@ -147,16 +152,16 @@ const orderValidationTool = agent.tool.createValidation({
   validator: async (context, data) => {
     const errors: ValidationError[] = [];
     if (!data.customerName || data.customerName.length < 2) {
-      errors.push({ 
-        field: "customerName", 
+      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", 
+      errors.push({
+        field: "budget",
         value: data.budget,
         message: "Budget must be at least $100",
         schemaPath: "budget"
@@ -178,7 +183,7 @@ const orderEnrichmentTool = agent.tool.createDataEnrichment({
     // 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,
@@ -190,6 +195,19 @@ const orderEnrichmentTool = agent.tool.createDataEnrichment({
 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
@@ -201,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",
@@ -209,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",
@@ -217,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",
@@ -225,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",
@@ -233,7 +258,8 @@ 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",
@@ -253,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
@@ -265,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",
@@ -273,7 +307,8 @@ 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",
@@ -308,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
@@ -387,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);
@@ -426,7 +461,7 @@ 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:");

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

@@ -248,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.",
@@ -281,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",
   });
 
@@ -289,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",
   });
@@ -307,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"],
   });
 
@@ -358,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
@@ -369,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"],
   });
 
@@ -404,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: {
@@ -442,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,
       },
     });

package/examples/integrations/server-session-management.ts

@@ -92,7 +92,7 @@ function createBookingAgent(sessionId?: string) {
             {
                 title: "Hotel Booking",
                 description: "Collect booking details and confirm reservation",
-                conditions: ["User wants to book a hotel room"],
+                when: ["User wants to book a hotel room"],
                 // NEW: Required fields for route completion
                 requiredFields: ["customerName", "hotelName", "checkInDate", "checkOutDate", "guests"],
                 // NEW: Optional fields that enhance the experience
@@ -103,7 +103,7 @@ function createBookingAgent(sessionId?: string) {
                         description: "Ask for customer name",
                         prompt: "Hi! I'd be happy to help you book a hotel room. What's your name?",
                         collect: ["customerName"],
-                        skipIf: (data: Partial<BookingData>) => !!data.customerName,
+                        skipIf: (ctx) => !!ctx.data?.customerName,
                     },
                     {
                         id: "ask_hotel",
@@ -111,7 +111,7 @@ function createBookingAgent(sessionId?: string) {
                         prompt: "Which hotel would you like to book?",
                         collect: ["hotelName"],
                         requires: ["customerName"],
-                        skipIf: (data: Partial<BookingData>) => !!data.hotelName,
+                        skipIf: (ctx) => !!ctx.data?.hotelName,
                     },
                     {
                         id: "ask_dates",
@@ -119,7 +119,7 @@ function createBookingAgent(sessionId?: string) {
                         prompt: "What are your check-in and check-out dates?",
                         collect: ["checkInDate", "checkOutDate"],
                         requires: ["customerName", "hotelName"],
-                        skipIf: (data: Partial<BookingData>) => !!data.checkInDate && !!data.checkOutDate,
+                        skipIf: (ctx) => !!ctx.data?.checkInDate && !!ctx.data.checkOutDate,
                     },
                     {
                         id: "ask_guests",
@@ -127,7 +127,7 @@ function createBookingAgent(sessionId?: string) {
                         prompt: "How many guests will be staying?",
                         collect: ["guests"],
                         requires: ["customerName", "hotelName", "checkInDate", "checkOutDate"],
-                        skipIf: (data: Partial<BookingData>) => data.guests !== undefined,
+                        skipIf: (ctx) => ctx.data?.guests !== undefined,
                     },
                     {
                         id: "ask_room_type",
@@ -135,7 +135,7 @@ function createBookingAgent(sessionId?: string) {
                         prompt: "What type of room would you prefer? (standard, deluxe, or suite)",
                         collect: ["roomType"],
                         requires: ["customerName", "hotelName", "checkInDate", "checkOutDate", "guests"],
-                        skipIf: (data: Partial<BookingData>) => !!data.roomType,
+                        skipIf: (ctx) => !!ctx.data?.roomType,
                     },
                     {
                         id: "ask_special_requests",
@@ -143,7 +143,7 @@ function createBookingAgent(sessionId?: string) {
                         prompt: "Do you have any special requests for your stay?",
                         collect: ["specialRequests"],
                         requires: ["customerName", "hotelName", "checkInDate", "checkOutDate", "guests"],
-                        skipIf: (data: Partial<BookingData>) => data.specialRequests !== undefined,
+                        skipIf: (ctx) => ctx.data?.specialRequests !== undefined,
                     },
                     {
                         id: "confirm_booking",
@@ -168,7 +168,7 @@ interface ChatResponse {
     message: string;
     sessionId: string;
     isComplete: boolean;
-    data?: Partial<BookingData>;
+    data?;
 }
 
 // Simulate POST /chat endpoint

package/examples/persistence/database-persistence.ts

@@ -97,7 +97,7 @@ async function example() {
       userName: "Alice",
     },
     persistence: {
-      adapter: new PrismaAdapter<ConversationContext>({ prisma }),
+      adapter: new PrismaAdapter({ prisma }),
       autoSave: true, // Auto-saves session after each response
     },
     schema: {
@@ -139,7 +139,7 @@ async function example() {
   const flightRoute = agent.createRoute({
     title: "Book a Flight",
     description: "Help user book a flight ticket",
-    conditions: [
+    when: [
       "User wants to book a flight",
       "User mentions travel, flying, or booking tickets",
     ],
@@ -150,14 +150,14 @@ async function example() {
     id: "ask_destination", // Custom step ID for easier tracking
     prompt: "Ask where they want to fly",
     collect: ["destination"],
-    skipIf: (data: Partial<FlightBookingData>) => !!data.destination,
+    skipIf: (ctx) => !!ctx.data?.destination,
   });
 
   const askDates = askDestination.nextStep({
     id: "ask_dates", // Custom step ID
     prompt: "Ask about travel dates",
     collect: ["departureDate", "returnDate"],
-    skipIf: (data: Partial<FlightBookingData>) => !!data.departureDate,
+    skipIf: (ctx) => !!ctx.data?.departureDate,
     requires: ["destination"],
   });
 
@@ -165,7 +165,7 @@ async function example() {
     id: "ask_passengers", // Custom step ID
     prompt: "Ask how many passengers",
     collect: ["passengers"],
-    skipIf: (data: Partial<FlightBookingData>) => !!data.passengers,
+    skipIf: (ctx) => !!ctx.data?.passengers,
     requires: ["destination", "departureDate"],
   });
 
@@ -173,7 +173,7 @@ async function example() {
     id: "ask_cabin_class", // Custom step ID
     prompt: "Ask about cabin class preference",
     collect: ["cabinClass"],
-    skipIf: (data: Partial<FlightBookingData>) => !!data.cabinClass,
+    skipIf: (ctx) => !!ctx.data?.cabinClass,
     requires: ["destination", "departureDate", "passengers"],
   });
 
@@ -275,7 +275,7 @@ async function example() {
       required: ["destination", "departureDate", "passengers", "cabinClass"],
     },
     persistence: {
-      adapter: new PrismaAdapter<ConversationContext>({ prisma }),
+      adapter: new PrismaAdapter({ prisma }),
     },
     sessionId, // Same sessionId - will load existing session
   });
@@ -338,8 +338,8 @@ async function advancedExample() {
     // Lifecycle hooks for data enrichment
     hooks: {
       onDataUpdate: async (
-        data: Partial<OnboardingData>,
-        previous: Partial<OnboardingData>
+        data,
+        previous
       ) => {
         console.log("🔄 Data updated:", { data, previous });
 
@@ -365,7 +365,7 @@ async function advancedExample() {
       },
     },
     persistence: {
-      adapter: new PrismaAdapter<UserContext>({ prisma }),
+      adapter: new PrismaAdapter({ prisma }),
       autoSave: true,
     },
     sessionId,
@@ -391,12 +391,12 @@ async function advancedExample() {
     .nextStep({
       prompt: "Welcome and ask for name",
       collect: ["fullName"],
-      skipIf: (data: Partial<OnboardingData>) => !!data.fullName,
+      skipIf: (ctx) => !!ctx.data?.fullName,
     })
     .nextStep({
       prompt: "Ask for email",
       collect: ["email"],
-      skipIf: (data: Partial<OnboardingData>) => !!data.email,
+      skipIf: (ctx) => !!ctx.data?.email,
     })
     .nextStep({
       prompt: "Ask for phone number (optional)",
@@ -405,7 +405,7 @@ async function advancedExample() {
     .nextStep({
       prompt: "Ask for country",
       collect: ["country"],
-      skipIf: (data: Partial<OnboardingData>) => !!data.country,
+      skipIf: (ctx) => !!ctx.data?.country,
     })
     .nextStep({
       prompt: "Confirm and complete onboarding",
@@ -536,7 +536,7 @@ async function serverEndpointExample() {
  * Mock function to send a flight confirmation email.
  */
 async function sendFlightConfirmation(
-  data: Partial<FlightBookingData> | undefined
+  data
 ) {
   console.log("\n" + "=".repeat(60));
   console.log("🚀 Sending Flight Confirmation...");
@@ -552,7 +552,7 @@ async function sendFlightConfirmation(
 /**
  * Mock function to send an onboarding email.
  */
-async function sendOnboardingEmail(data: Partial<OnboardingData> | undefined) {
+async function sendOnboardingEmail(data) {
   console.log("\n" + "=".repeat(60));
   console.log("🚀 Sending Onboarding Email...");
   console.log("=".repeat(60));