@falai/agent 0.9.0-alpha-1 → 0.9.0

package/src/core/PersistenceManager.ts

@@ -53,7 +53,6 @@ export class PersistenceManager<TData = Record<string, unknown>> {
       status: "active",
       collectedData: {
         data: options.initialData || {},
-        dataByRoute: {},
         routeHistory: [],
         metadata: {},
       },
@@ -235,32 +234,57 @@ export class PersistenceManager<TData = Record<string, unknown>> {
     sessionId: string,
     sessionStep: SessionState<TData>
   ): Promise<SessionData<TData> | null> {
-    const persistenceData = sessionStepToData(sessionStep);
-
-    // First try to find existing session
-    const existingSession = await this.sessionRepository.findById(sessionId);
-
-    if (existingSession) {
-      // Update existing session
-      return await this.sessionRepository.update(sessionId, {
-        currentRoute: persistenceData.currentRoute,
-        currentStep: persistenceData.currentStep,
-        collectedData: persistenceData.collectedData,
-        lastMessageAt: new Date(),
-      });
-    } else {
-      // Create new session if it doesn't exist
-      return await this.sessionRepository.create({
-        id: sessionId,
-        userId: persistenceData.collectedData.metadata?.userId
-          ? JSON.stringify(persistenceData.collectedData.metadata?.userId)
-          : this.config.userId,
-        status: "active",
-        currentRoute: persistenceData.currentRoute,
-        currentStep: persistenceData.currentStep,
-        collectedData: persistenceData.collectedData,
-        messageCount: 0,
-      });
+    // Validate input parameters
+    if (!sessionId || typeof sessionId !== 'string') {
+      throw new Error('Session ID must be a non-empty string');
+    }
+
+    if (!sessionStep || typeof sessionStep !== 'object') {
+      throw new Error('Session step must be a valid object');
+    }
+
+    // Validate session data structure
+    if (sessionStep.data && typeof sessionStep.data !== 'object') {
+      throw new Error('Session data must be an object');
+    }
+
+    let persistenceData;
+    try {
+      persistenceData = sessionStepToData(sessionStep);
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      throw new Error(`Failed to convert session step to persistence data: ${errorMessage}`);
+    }
+
+    try {
+      // First try to find existing session
+      const existingSession = await this.sessionRepository.findById(sessionId);
+
+      if (existingSession) {
+        // Update existing session
+        return await this.sessionRepository.update(sessionId, {
+          currentRoute: persistenceData.currentRoute,
+          currentStep: persistenceData.currentStep,
+          collectedData: persistenceData.collectedData,
+          lastMessageAt: new Date(),
+        });
+      } else {
+        // Create new session if it doesn't exist
+        return await this.sessionRepository.create({
+          id: sessionId,
+          userId: persistenceData.collectedData.metadata?.userId
+            ? JSON.stringify(persistenceData.collectedData.metadata?.userId)
+            : this.config.userId,
+          status: "active",
+          currentRoute: persistenceData.currentRoute,
+          currentStep: persistenceData.currentStep,
+          collectedData: persistenceData.collectedData,
+          messageCount: 0,
+        });
+      }
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      throw new Error(`Failed to save session state to persistence: ${errorMessage}`);
     }
   }
 

package/src/core/PromptComposer.ts

@@ -13,7 +13,7 @@ export class PromptComposer<TContext = unknown, TData = unknown> {
 
   // Specific, typed sections tailored to the framework
 
-  async addAgentMeta(agent: AgentOptions<TContext>): Promise<this> {
+  async addAgentMeta(agent: AgentOptions<TContext, TData>): Promise<this> {
     const lines: string[] = [];
     lines.push("## Agent");
     lines.push(`**Name:** ${agent.name}`);

package/src/core/ResponseEngine.ts

@@ -23,27 +23,30 @@ export interface BuildResponsePromptParams<
   directives: string[] | undefined;
   history: Event[];
   lastMessage: string;
-  agentOptions?: AgentOptions<TContext>;
+  agentOptions?: AgentOptions<TContext, TData>;
   // Combined properties from agent and route
-  combinedGuidelines?: Guideline<TContext>[];
-  combinedTerms?: Term<TContext>[];
+  combinedGuidelines?: Guideline<TContext, TData>[];
+  combinedTerms?: Term<TContext, TData>[];
   context?: TContext;
   session?: SessionState<TData>;
+  // NEW: Agent-level schema for data validation
+  agentSchema?: StructuredSchema;
 }
 
-export interface BuildFallbackPromptParams<TContext = unknown> {
+export interface BuildFallbackPromptParams<TContext = unknown, TData = unknown> {
   history: Event[];
-  agentOptions: AgentOptions<TContext>;
-  terms: Term<TContext>[];
-  guidelines: Guideline<TContext>[];
+  agentOptions: AgentOptions<TContext, TData>;
+  terms: Term<TContext, TData>[];
+  guidelines: Guideline<TContext, TData>[];
   context?: TContext;
-  session?: SessionState;
+  session?: SessionState<TData>;
 }
 
-export class ResponseEngine<TContext = unknown> {
-  responseSchemaForRoute<TData = unknown>(
+export class ResponseEngine<TContext = unknown, TData = unknown> {
+  responseSchemaForRoute(
     route: Route<TContext, TData>,
-    currentStep?: Step<TContext, TData>
+    currentStep?: Step<TContext, TData>,
+    agentSchema?: StructuredSchema
   ): StructuredSchema {
     const base: StructuredSchema = {
       type: "object",
@@ -59,12 +62,12 @@ export class ResponseEngine<TContext = unknown> {
       base.properties!.data = route.responseOutputSchema;
     }
 
-    // Add collect fields from current step
-    if (currentStep?.collect && route.schema?.properties) {
+    // Add collect fields from current step using agent-level schema
+    if (currentStep?.collect && agentSchema?.properties) {
       for (const field of currentStep.collect) {
-        const fieldSchema = route.schema.properties[field];
+        const fieldSchema = agentSchema.properties[field as string];
         if (fieldSchema) {
-          base.properties![field] = fieldSchema;
+          base.properties![field as string] = fieldSchema;
         }
       }
     }
@@ -73,7 +76,7 @@ export class ResponseEngine<TContext = unknown> {
   }
 
   async buildResponsePrompt(
-    params: BuildResponsePromptParams<TContext>
+    params: BuildResponsePromptParams<TContext, TData>
   ): Promise<string> {
     const {
       route,
@@ -107,8 +110,7 @@ export class ResponseEngine<TContext = unknown> {
       await pc.addAgentMeta(effectiveAgentOptions);
     }
     await pc.addInstruction(
-      `Route: ${route.title}${
-        route.description ? ` — ${route.description}` : ""
+      `Route: ${route.title}${route.description ? ` — ${route.description}` : ""
       }`
     );
     if (currentStep.prompt) {
@@ -148,7 +150,7 @@ export class ResponseEngine<TContext = unknown> {
   }
 
   async buildFallbackPrompt(
-    params: BuildFallbackPromptParams<TContext>
+    params: BuildFallbackPromptParams<TContext, TData>
   ): Promise<string> {
     const { history, agentOptions, terms, guidelines, context, session } =
       params;