@falai/agent 0.9.0 → 1.0.0

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

@@ -31,15 +31,15 @@ interface BookingData {
 }
 
 // Booking confirmation tool
-const confirmBooking: Tool<unknown, BookingData, [], string> = {
+const confirmBooking: Tool<unknown, BookingData> = {
     id: "confirm_booking",
     description: "Confirm the hotel booking with all details",
     parameters: {
         type: "object",
         properties: {},
     },
-    handler: ({ data }) => {
-        const bookingData = data as Partial<BookingData>;
+    handler: (context, args) => {
+        const bookingData = context.data;
         const bookingId = `BK-${Date.now()}`;
 
         console.log(`📋 Confirming booking ${bookingId} for ${bookingData.customerName}`);
@@ -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));

package/examples/persistence/memory-sessions.ts

@@ -38,15 +38,15 @@ interface SupportContext {
 }
 
 // Support tools
-const createTicket: Tool<unknown, SupportTicketData, [], string> = {
+const createTicket: Tool<unknown, SupportTicketData> = {
   id: "create_support_ticket",
   description: "Create a new support ticket",
   parameters: {
     type: "object",
     properties: {},
   },
-  handler: ({ data }) => {
-    const ticketData = data as Partial<SupportTicketData>;
+  handler: (context, args) => {
+    const ticketData = context.data;
     const ticketId = `TICKET-${Date.now()}`;
     console.log(
       `Creating ticket ${ticketId} for ${ticketData?.category} issue`
@@ -110,7 +110,7 @@ const supportRoute = agent.createRoute({
       prompt:
         "Hi! I'm here to help with your support request. What's the issue you're experiencing?",
       collect: ["issue"],
-      skipIf: (data: Partial<SupportTicketData>) => !!data.issue,
+      skipIf: (ctx) => !!ctx.data?.issue,
     },
     {
       id: "ask_category",
@@ -119,7 +119,7 @@ const supportRoute = agent.createRoute({
         "What category does this issue fall under? (technical, billing, account, or general)",
       collect: ["category"],
       requires: ["issue"],
-      skipIf: (data: Partial<SupportTicketData>) => !!data.category,
+      skipIf: (ctx) => !!ctx.data?.category,
     },
     {
       id: "ask_priority",
@@ -128,7 +128,7 @@ const supportRoute = agent.createRoute({
         "How would you rate the priority of this issue? (low, medium, or high)",
       collect: ["priority"],
       requires: ["issue", "category"],
-      skipIf: (data: Partial<SupportTicketData>) => !!data.priority,
+      skipIf: (ctx) => !!ctx.data?.priority,
     },
     {
       id: "create_ticket",

package/examples/persistence/redis-persistence.ts

@@ -14,8 +14,6 @@ import {
   GeminiProvider,
   RedisAdapter,
   END_ROUTE,
-  MessageRole,
-  type HistoryItem,
 } from "../../src";
 // @ts-expect-error - Redis is not typed
 import Redis from "ioredis";
@@ -94,7 +92,7 @@ async function example() {
     schema: supportTicketSchema,
     // ✨ Redis adapter with custom options
     persistence: {
-      adapter: new RedisAdapter<ChatContext>({
+      adapter: new RedisAdapter({
         redis,
         keyPrefix: "support:", // Custom prefix
         sessionTTL: 24 * 60 * 60, // 24 hours
@@ -109,7 +107,7 @@ async function example() {
   const ticketRoute = agent.createRoute({
     title: "Create Support Ticket",
     description: "Help user create and track support tickets",
-    conditions: [
+    when: [
       "User needs help with an issue",
       "User wants to report a problem",
       "User mentions support, help, or issue",
@@ -125,18 +123,18 @@ async function example() {
     .nextStep({
       prompt: "Ask what the issue is",
       collect: ["issue", "category"],
-      skipIf: (data) => !!data.issue && !!data.category,
+      skipIf: (ctx) => !!ctx.data?.issue && !!ctx.data.category,
     })
     .nextStep({
       prompt: "Ask for priority",
       collect: ["priority"],
-      skipIf: (data) => !!data.priority,
+      skipIf: (ctx) => !!ctx.data?.priority,
       requires: ["issue", "category"],
     })
     .nextStep({
       prompt: "Ask for detailed description",
       collect: ["description"],
-      skipIf: (data) => !!data.description,
+      skipIf: (ctx) => !!ctx.data?.description,
       requires: ["issue", "category"],
     })
     .nextStep({
@@ -213,7 +211,7 @@ async function example() {
     // NEW: Agent-level schema (same as original agent)
     schema: supportTicketSchema,
     persistence: {
-      adapter: new RedisAdapter<ChatContext>({
+      adapter: new RedisAdapter({
         redis,
         keyPrefix: "support:",
         sessionTTL: 24 * 60 * 60,
@@ -228,7 +226,7 @@ async function example() {
   recoveredAgent.createRoute({
     title: "Create Support Ticket",
     description: "Help user create and track support tickets",
-    conditions: [
+    when: [
       "User needs help with an issue",
       "User wants to report a problem",
       "User mentions support, help, or issue",