@falai/agent 0.8.1 → 0.9.0-alpha-1

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

@@ -0,0 +1,301 @@
+/**
+ * Schema-Driven Data Extraction Example
+ *
+ * This example demonstrates how to use JSON Schema to define data contracts
+ * and extract structured data reliably from conversations.
+ *
+ * Key concepts:
+ * - Schema-first data definition
+ * - Type-safe data extraction
+ * - Automatic validation
+ * - SkipIf conditions for smart flow control
+ */
+
+import { Agent, GeminiProvider, type Tool } from "../../src/index";
+
+// Define the data structure we want to collect
+interface UserProfileData {
+  name: string;
+  email: string;
+  age: number;
+  interests: string[];
+  preferredContact: "email" | "phone" | "sms";
+  newsletterOptIn: boolean;
+}
+
+// Define a tool that uses the collected data
+const saveUserProfile: Tool<unknown, [], string, UserProfileData> = {
+  id: "save_user_profile",
+  description: "Save the collected user profile information",
+  parameters: {
+    type: "object",
+    properties: {},
+  },
+  handler: ({ data }) => {
+    console.log("Saving user profile:", data);
+
+    // Simulate saving to database
+    console.log("Profile data:", data);
+
+    const userData = data as Partial<UserProfileData>;
+    return {
+      data: `Profile saved successfully! Welcome ${userData?.name}!`,
+    };
+  },
+};
+
+// Create the agent
+const agent = new Agent({
+  name: "ProfileBot",
+  description:
+    "A bot that collects user profile information using schema-driven extraction",
+  provider: new GeminiProvider({
+    apiKey: process.env.GEMINI_API_KEY!,
+    model: "models/gemini-2.5-flash",
+  }),
+});
+
+// Define the schema for data validation and extraction
+const userProfileSchema = {
+  type: "object",
+  properties: {
+    name: {
+      type: "string",
+      description: "The user's full name",
+      minLength: 2,
+      maxLength: 100,
+    },
+    email: {
+      type: "string",
+      format: "email",
+      description: "The user's email address",
+    },
+    age: {
+      type: "number",
+      description: "The user's age in years",
+      minimum: 13,
+      maximum: 120,
+    },
+    interests: {
+      type: "array",
+      items: { type: "string" },
+      description: "User's interests and hobbies",
+      minItems: 1,
+      maxItems: 10,
+    },
+    preferredContact: {
+      type: "string",
+      enum: ["email", "phone", "sms"],
+      description: "Preferred contact method",
+      default: "email",
+    },
+    newsletterOptIn: {
+      type: "boolean",
+      description: "Whether user wants to receive newsletters",
+      default: false,
+    },
+  },
+  required: ["name", "email"],
+};
+
+// Create a route that collects profile information step by step
+agent.createRoute<UserProfileData>({
+  title: "User Profile Collection",
+  description: "Collect comprehensive user profile information",
+  schema: userProfileSchema,
+  // Use sequential steps for a linear flow with smart skipIf conditions
+  steps: [
+    {
+      id: "ask_name",
+      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,
+    },
+    {
+      id: "ask_email",
+      description: "Ask for user's email",
+      prompt: "What's your email address?",
+      collect: ["email"],
+      requires: ["name"],
+      skipIf: (data: Partial<UserProfileData>) => !!data.email,
+    },
+    {
+      id: "ask_age",
+      description: "Ask for user's age (optional)",
+      prompt: "How old are you? (optional)",
+      collect: ["age"],
+      requires: ["name", "email"],
+      skipIf: (data: Partial<UserProfileData>) => data.age !== undefined,
+    },
+    {
+      id: "ask_interests",
+      description: "Ask for user's interests (optional)",
+      prompt: "What are your interests or hobbies? (optional)",
+      collect: ["interests"],
+      requires: ["name", "email"],
+      skipIf: (data: Partial<UserProfileData>) =>
+        !!(data.interests && data.interests.length > 0),
+    },
+    {
+      id: "ask_contact_preference",
+      description: "Ask for preferred contact method",
+      prompt: "What's your preferred contact method?",
+      collect: ["preferredContact"],
+      requires: ["name", "email"],
+      skipIf: (data: Partial<UserProfileData>) => !!data.preferredContact,
+    },
+    {
+      id: "ask_newsletter",
+      description: "Ask about newsletter subscription",
+      prompt: "Would you like to subscribe to our newsletter?",
+      collect: ["newsletterOptIn"],
+      requires: ["name", "email"],
+      skipIf: (data: Partial<UserProfileData>) =>
+        data.newsletterOptIn !== undefined,
+    },
+    {
+      id: "save_profile",
+      description: "Save the collected profile",
+      prompt:
+        "Thanks for providing your information! Let me save your profile.",
+      tools: [saveUserProfile],
+      requires: ["name", "email"],
+    },
+  ],
+});
+
+// Example conversation demonstrating schema-driven extraction
+async function demonstrateSchemaExtraction() {
+  console.log("=== Schema-Driven Data Extraction Demo ===\n");
+
+  // First message - user provides name and email
+  console.log("User: Hi, I'm John Doe and my email is john@example.com");
+  const response1 = await agent.respond({
+    history: [
+      {
+        role: "user",
+        content: "Hi, I'm John Doe and my email is john@example.com",
+        name: "John",
+      },
+    ],
+  });
+
+  console.log("Bot:", response1.message);
+  console.log(
+    "Collected data:",
+    JSON.stringify(response1.session?.data as Partial<UserProfileData>, null, 2)
+  );
+
+  // Second message - user provides more information
+  console.log(
+    "\nUser: I'm 30 years old and I like hiking, reading, and photography"
+  );
+  const response2 = await agent.respond({
+    history: [
+      {
+        role: "user",
+        content: "Hi, I'm John Doe and my email is john@example.com",
+        name: "John",
+      },
+      {
+        role: "assistant",
+        content: response1.message,
+      },
+      {
+        role: "user",
+        content: "I'm 30 years old and I like hiking, reading, and photography",
+        name: "John",
+      },
+    ],
+    session: response1.session,
+  });
+
+  console.log("Bot:", response2.message);
+  console.log(
+    "Collected data:",
+    JSON.stringify(response2.session?.data as Partial<UserProfileData>, null, 2)
+  );
+
+  // Third message - user provides contact preference and completes profile
+  console.log(
+    "\nUser: Please contact me by email and yes, I'd like the newsletter"
+  );
+  const response3 = await agent.respond({
+    history: [
+      {
+        role: "user",
+        content: "Hi, I'm John Doe and my email is john@example.com",
+        name: "John",
+      },
+      {
+        role: "assistant",
+        content: response1.message,
+      },
+      {
+        role: "user",
+        content: "I'm 30 years old and I like hiking, reading, and photography",
+        name: "John",
+      },
+      {
+        role: "assistant",
+        content: response2.message,
+      },
+      {
+        role: "user",
+        content: "Please contact me by email and yes, I'd like the newsletter",
+        name: "John",
+      },
+    ],
+    session: response2.session,
+  });
+
+  console.log("Bot:", response3.message);
+  console.log(
+    "Final collected data:",
+    JSON.stringify(response3.session?.data as Partial<UserProfileData>, null, 2)
+  );
+  console.log("Route complete:", response3.isRouteComplete);
+}
+
+// Demonstrate with a single comprehensive message
+async function demonstrateSmartSkipping() {
+  console.log("\n=== Smart Skipping Demo ===\n");
+
+  console.log("User provides everything in one message:");
+  console.log(
+    "User: I'm Sarah Johnson, sarah@email.com, 28 years old. I enjoy cooking, travel, and music. Contact me by phone and yes to newsletter."
+  );
+
+  const response = await agent.respond({
+    history: [
+      {
+        role: "user",
+        content:
+          "I'm Sarah Johnson, sarah@email.com, 28 years old. I enjoy cooking, travel, and music. Contact me by phone and yes to newsletter.",
+        name: "Sarah",
+      },
+    ],
+  });
+
+  console.log("Bot:", response.message);
+  console.log(
+    "Collected data:",
+    JSON.stringify(response.session?.data as Partial<UserProfileData>, null, 2)
+  );
+  console.log("Route complete:", response.isRouteComplete);
+}
+
+// Run the demonstrations
+async function main() {
+  try {
+    await demonstrateSchemaExtraction();
+    await demonstrateSmartSkipping();
+  } catch (error) {
+    console.error("Error:", error);
+  }
+}
+
+if (import.meta.url === `file://${process.argv[1]}`) {
+  main().catch(console.error);
+}

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

@@ -0,0 +1,394 @@
+/**
+ * Session Management Example
+ *
+ * This example demonstrates how to manage conversation state across multiple turns
+ * using session management, persistence, and state restoration.
+ *
+ * Key concepts:
+ * - Session state management
+ * - Multi-turn conversations
+ * - Data persistence across turns
+ * - Session restoration
+ * - Route transitions with session state
+ */
+
+import {
+  Agent,
+  GeminiProvider,
+  MemoryAdapter,
+  type Tool,
+} from "../../src/index";
+
+// Define data types for our multi-step process
+interface OrderData {
+  customerName: string;
+  productType: "laptop" | "phone" | "tablet";
+  budget: number;
+  preferredColor?: string;
+  urgentDelivery: boolean;
+  orderId?: string;
+}
+
+interface OrderContext {
+  userId: string;
+  userName: string;
+  isVip: boolean;
+}
+
+interface PaymentData {
+  orderId: string;
+  paymentMethod: "credit_card" | "paypal" | "bank_transfer";
+  amount: number;
+  confirmed: boolean;
+}
+
+// Tools for order processing
+const createOrder: Tool<unknown, [], string, OrderData> = {
+  id: "create_order",
+  description: "Create a new order from the collected order data",
+  parameters: {
+    type: "object",
+    properties: {},
+  },
+  handler: ({ data }) => {
+    const orderId = `ORD-${Date.now()}`;
+    const orderData = data as Partial<OrderData>;
+    console.log(`Creating order ${orderId} for ${orderData?.customerName}`);
+
+    return {
+      data: `Order ${orderId} created successfully!`,
+      dataUpdate: {
+        orderId,
+      },
+    };
+  },
+};
+
+const processPayment: Tool<unknown, [], string, PaymentData> = {
+  id: "process_payment",
+  description: "Process payment for an order",
+  parameters: {
+    type: "object",
+    properties: {},
+  },
+  handler: ({ data }) => {
+    const paymentData = data as Partial<PaymentData>;
+    console.log(`Processing payment for order ${paymentData?.orderId}`);
+
+    // Simulate payment processing
+    const success = Math.random() > 0.1; // 90% success rate
+
+    if (success) {
+      return {
+        data: `Payment processed successfully! Order ${paymentData?.orderId} is now confirmed.`,
+        dataUpdate: {
+          confirmed: true,
+        },
+      };
+    } else {
+      return {
+        data: "Payment failed. Please try again or contact support.",
+        dataUpdate: {
+          confirmed: false,
+        },
+      };
+    }
+  },
+};
+
+// Create agent with persistence
+const agent = new Agent({
+  name: "OrderBot",
+  description:
+    "A bot that handles multi-step order processing with session management",
+  provider: new GeminiProvider({
+    apiKey: process.env.GEMINI_API_KEY!,
+    model: "models/gemini-2.5-flash",
+  }),
+  persistence: {
+    adapter: new MemoryAdapter(), // In production, use RedisAdapter, PrismaAdapter, etc.
+  },
+});
+
+// Order collection route with sequential steps
+agent.createRoute<OrderData>({
+  title: "Product Order",
+  description: "Collect order details and create an order",
+  schema: {
+    type: "object",
+    properties: {
+      customerName: { type: "string" },
+      productType: { type: "string", enum: ["laptop", "phone", "tablet"] },
+      budget: { type: "number", minimum: 100 },
+      preferredColor: { type: "string" },
+      urgentDelivery: { type: "boolean" },
+      orderId: { type: "string" },
+    },
+    required: ["customerName", "productType", "budget"],
+  },
+  // Sequential steps for order collection
+  steps: [
+    {
+      id: "ask_name",
+      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<OrderData>) => !!data.customerName,
+    },
+    {
+      id: "ask_product",
+      description: "Ask for product type",
+      prompt: "What would you like to order? (laptop, phone, or tablet)",
+      collect: ["productType"],
+      requires: ["customerName"],
+      skipIf: (data: Partial<OrderData>) => !!data.productType,
+    },
+    {
+      id: "ask_budget",
+      description: "Ask for budget",
+      prompt: "What's your budget for this purchase?",
+      collect: ["budget"],
+      requires: ["customerName", "productType"],
+      skipIf: (data: Partial<OrderData>) => data.budget !== undefined,
+    },
+    {
+      id: "ask_color",
+      description: "Ask for preferred color",
+      prompt: "Do you have a preferred color?",
+      collect: ["preferredColor"],
+      requires: ["customerName", "productType", "budget"],
+      skipIf: (data: Partial<OrderData>) => !!data.preferredColor,
+    },
+    {
+      id: "ask_urgent",
+      description: "Ask about urgent delivery",
+      prompt: "Do you need urgent delivery?",
+      collect: ["urgentDelivery"],
+      requires: ["customerName", "productType", "budget"],
+      skipIf: (data: Partial<OrderData>) => data.urgentDelivery !== undefined,
+    },
+    {
+      id: "create_order",
+      description: "Create the order",
+      prompt: "Great! Let me create your order.",
+      tools: [createOrder],
+      requires: ["customerName", "productType", "budget"],
+    },
+  ],
+  onComplete: {
+    nextStep: "Payment Processing",
+    condition: "Order has been created successfully",
+  },
+});
+
+// Payment route with sequential steps
+agent.createRoute<PaymentData>({
+  title: "Payment Processing",
+  description: "Process payment for an order",
+  schema: {
+    type: "object",
+    properties: {
+      orderId: { type: "string" },
+      paymentMethod: {
+        type: "string",
+        enum: ["credit_card", "paypal", "bank_transfer"],
+      },
+      amount: { type: "number" },
+      confirmed: { type: "boolean" },
+    },
+    required: ["orderId", "paymentMethod", "amount"],
+  },
+  // Sequential steps for payment processing
+  steps: [
+    {
+      id: "ask_payment_method",
+      description: "Ask for payment method",
+      prompt:
+        "Now let's process payment for your order. What payment method would you prefer?",
+      collect: ["paymentMethod"],
+      skipIf: (data: Partial<PaymentData>) => !!data.paymentMethod,
+    },
+    {
+      id: "ask_amount",
+      description: "Ask for payment amount",
+      prompt: "What's the payment amount?",
+      collect: ["amount"],
+      requires: ["paymentMethod"],
+      skipIf: (data: Partial<PaymentData>) => data.amount !== undefined,
+    },
+    {
+      id: "process_payment",
+      description: "Process the payment",
+      prompt: "Processing your payment...",
+      tools: [processPayment],
+      requires: ["orderId", "paymentMethod", "amount"],
+    },
+  ],
+  onComplete: "Order Complete", // End conversation when payment is complete
+});
+
+// Demonstration of automatic session management
+async function demonstrateSessionManagement() {
+  console.log("=== Automatic Session Management Demo ===\n");
+
+  // Create agent with automatic session management
+  const sessionAgent = new Agent({
+    name: "OrderBot",
+    description: "A bot that handles multi-step order processing",
+    provider: new GeminiProvider({
+      apiKey: process.env.GEMINI_API_KEY!,
+      model: "models/gemini-2.5-flash",
+    }),
+    persistence: {
+      adapter: new MemoryAdapter(),
+    },
+    sessionId: "user-alice-123", // Automatically creates/loads this session
+  });
+
+  // Copy routes to the session agent
+  agent.getRoutes().forEach(route => {
+    sessionAgent.createRoute(route);
+  });
+
+  // Turn 1: Start order process - simple message API
+  console.log("Turn 1: Starting order process");
+  console.log("User: Hi, I'm Alice and I want to buy a laptop");
+
+  const response1 = await sessionAgent.chat("Hi, I'm Alice and I want to buy a laptop");
+
+  console.log("Bot:", response1.message);
+  console.log("Session ID:", sessionAgent.session.id);
+  console.log("Session data:", JSON.stringify(sessionAgent.session.getData<OrderData>(), null, 2));
+  console.log("History length:", sessionAgent.session.getHistory().length);
+  console.log();
+
+  // Turn 2: Provide budget - session automatically maintained
+  console.log("Turn 2: Providing budget");
+  console.log("User: My budget is $1500");
+
+  const response2 = await sessionAgent.chat("My budget is $1500");
+
+  console.log("Bot:", response2.message);
+  console.log("Session data:", JSON.stringify(sessionAgent.session.getData<OrderData>(), null, 2));
+  console.log("History length:", sessionAgent.session.getHistory().length);
+  console.log();
+
+  // Turn 3: Complete order details - session automatically updated
+  console.log("Turn 3: Completing order and transitioning to payment");
+  console.log("User: I want black color and urgent delivery please");
+
+  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<OrderData>(), null, 2));
+  console.log("Route complete:", response3.isRouteComplete);
+  console.log("History length:", sessionAgent.session.getHistory().length);
+  console.log();
+
+  // Turn 4: Process payment - automatic route transition
+  console.log("Turn 4: Processing payment in transitioned route");
+  console.log("User: I'll pay with credit card, amount is $1599");
+
+  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<PaymentData>(), null, 2));
+  console.log("Route complete:", response4.isRouteComplete);
+  console.log("Final history length:", sessionAgent.session.getHistory().length);
+}
+
+// Demonstrate automatic session persistence and restoration
+async function demonstrateSessionPersistence() {
+  console.log("\n=== Automatic Session Persistence Demo ===\n");
+
+  // Create agent with specific sessionId for demonstration
+  const sessionId = "demo-session-456";
+  const persistentAgent = new Agent<OrderContext>({
+    name: "Order Assistant",
+    description: "Help customers place orders with automatic persistence",
+    provider: new GeminiProvider({
+      apiKey: process.env.GEMINI_API_KEY!,
+      model: "models/gemini-2.5-flash",
+    }),
+    context: {
+      userId: "user_alice",
+      userName: "Alice",
+      isVip: false,
+    },
+    persistence: {
+      adapter: new MemoryAdapter(),
+      autoSave: true, // Automatic persistence
+    },
+    sessionId, // Agent automatically loads or creates this session
+  });
+
+  // Copy routes to the persistent agent
+  agent.getRoutes().forEach(route => {
+    persistentAgent.createRoute(route);
+  });
+
+  console.log("Session ready:", persistentAgent.session.id);
+
+  // Start conversation - session automatically managed
+  const response1 = await persistentAgent.chat("I want to buy a phone for $800");
+
+  console.log("After first interaction:");
+  console.log("🤖 Agent:", response1.message);
+  console.log("Session data:", JSON.stringify(persistentAgent.session.getData<OrderData>(), null, 2));
+  console.log("History length:", persistentAgent.session.getHistory().length);
+  console.log("Session automatically saved to persistence ✓");
+
+  // Simulate server restart - create new agent instance with same sessionId
+  console.log("\n🔄 Simulating server restart - creating new agent instance...");
+
+  const restoredAgent = new Agent<OrderContext>({
+    name: "Order Assistant",
+    provider: new GeminiProvider({
+      apiKey: process.env.GEMINI_API_KEY!,
+      model: "models/gemini-2.5-flash",
+    }),
+    context: {
+      userId: "user_alice",
+      userName: "Alice",
+      isVip: false,
+    },
+    persistence: {
+      adapter: new MemoryAdapter(), // Same adapter instance for demo
+      autoSave: true,
+    },
+    sessionId, // Same sessionId - automatically loads existing session
+  });
+
+  // Copy routes to the restored agent
+  agent.getRoutes().forEach(route => {
+    restoredAgent.createRoute(route);
+  });
+
+  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<OrderData>(), 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<OrderData>(), null, 2));
+  console.log("History length:", restoredAgent.session.getHistory().length);
+  console.log("Session automatically saved again ✓");
+}
+
+// Run demonstrations
+async function main() {
+  try {
+    await demonstrateSessionManagement();
+    await demonstrateSessionPersistence();
+  } catch (error) {
+    console.error("Error:", error);
+  }
+}
+
+if (import.meta.url === `file://${process.argv[1]}`) {
+  main().catch(console.error);
+}