@autobe/agent 0.17.1 → 0.19.1

package/src/AutoBeMockAgent.ts

@@ -8,6 +8,7 @@ import {
   IAutoBeAgent,
   IAutoBeCompiler,
   IAutoBeCompilerListener,
+  IAutoBePlaygroundReplay,
 } from "@autobe/interface";
 import { Singleton, randint, sleep_for } from "tstl";
 import { v4 } from "uuid";
@@ -88,7 +89,24 @@ export class AutoBeMockAgent extends AutoBeAgentBase implements IAutoBeAgent {
     const take = async (
       type: "analyze" | "prisma" | "interface" | "test" | "realize",
     ): Promise<void> => {
-      for (const s of this.getEventSnapshots(type)) {
+      const snapshots: AutoBeEventSnapshot[] | null =
+        this.getEventSnapshots(type);
+      if (snapshots === null) {
+        this.histories_.push(userMessage);
+        this.histories_.push({
+          id: v4(),
+          type: "assistantMessage",
+          text: [
+            "The histories are prepared until current state.",
+            "",
+            "Thanks for using AutoBE!",
+          ].join("\n"),
+          created_at: new Date().toISOString(),
+          completed_at: new Date().toISOString(),
+        });
+        return;
+      }
+      for (const s of snapshots) {
         const time: number = sleepMap[s.event.type] ?? 500;
         await sleep_for(randint(time * 0.2, time * 1.8));
         void this.dispatch(s.event).catch(() => {});
@@ -96,7 +114,7 @@ export class AutoBeMockAgent extends AutoBeAgentBase implements IAutoBeAgent {
       }
       this.histories_.push(userMessage);
       this.histories_.push(
-        this.props_.preset.histories.find((h) => h.type === type)!,
+        this.props_.replay.histories.find((h) => h.type === type)!,
       );
     };
     if (state.analyze === null) await take("analyze");
@@ -117,8 +135,8 @@ export class AutoBeMockAgent extends AutoBeAgentBase implements IAutoBeAgent {
 
   private getEventSnapshots(
     state: "analyze" | "prisma" | "interface" | "test" | "realize",
-  ): AutoBeEventSnapshot[] {
-    return this.props_.preset[state];
+  ): AutoBeEventSnapshot[] | null {
+    return this.props_.replay[state] ?? null;
   }
 }
 export namespace AutoBeMockAgent {
@@ -126,21 +144,21 @@ export namespace AutoBeMockAgent {
     compiler: (
       listener: IAutoBeCompilerListener,
     ) => IAutoBeCompiler | Promise<IAutoBeCompiler>;
-    preset: IPreset;
-  }
-  export interface IPreset {
-    histories: AutoBeHistory[];
-    analyze: AutoBeEventSnapshot[];
-    prisma: AutoBeEventSnapshot[];
-    interface: AutoBeEventSnapshot[];
-    test: AutoBeEventSnapshot[];
-    realize: AutoBeEventSnapshot[];
+    replay: IAutoBePlaygroundReplay;
   }
 }
 
-const sleepMap: Partial<Record<AutoBeEvent.Type, number>> = {
+const sleepMap: Record<AutoBeEvent.Type, number> = {
+  userMessage: 1_000,
+  assistantMessage: 1_000,
+  vendorRequest: 0,
+  vendorResponse: 0,
+  jsonParseError: 0,
+  jsonValidateError: 0,
+  consentFunctionCall: 0,
   // ANALYZE
   analyzeStart: 1_000,
+  analyzeScenario: 1_000,
   analyzeWrite: 500,
   analyzeReview: 300,
   analyzeComplete: 1_000,
@@ -148,20 +166,25 @@ const sleepMap: Partial<Record<AutoBeEvent.Type, number>> = {
   prismaStart: 1_000,
   prismaComponents: 1_000,
   prismaSchemas: 500,
+  prismaInsufficient: 1_000,
+  prismaReview: 500,
   prismaValidate: 2_000,
   prismaCorrect: 500,
-  prismaInsufficient: 1_000,
   prismaComplete: 1_000,
   // INTERFACE
   interfaceStart: 1_000,
+  interfaceGroups: 1_000,
   interfaceEndpoints: 1_000,
   interfaceOperations: 400,
+  interfaceOperationsReview: 400,
+  interfaceAuthorization: 400,
   interfaceSchemas: 400,
+  interfaceSchemasReview: 400,
   interfaceComplement: 2_000,
   interfaceComplete: 1_000,
   // TEST
   testStart: 1_000,
-  testScenario: 1_000,
+  testScenarios: 40,
   testWrite: 40,
   testValidate: 100,
   testCorrect: 100,
@@ -175,5 +198,10 @@ const sleepMap: Partial<Record<AutoBeEvent.Type, number>> = {
   realizeAuthorizationStart: 1_000,
   realizeAuthorizationWrite: 200,
   realizeAuthorizationValidate: 200,
+  realizeAuthorizationCorrect: 200,
   realizeAuthorizationComplete: 1_000,
+  realizeTestStart: 1_000,
+  realizeTestReset: 2_500,
+  realizeTestOperation: 400,
+  realizeTestComplete: 1_000,
 };