@autobe/rpc 0.22.0 → 0.23.0

package/lib/AutoBeRpcService.d.ts

@@ -1,4 +1,4 @@
-import { AutoBeHistory, AutoBeUserMessageContent, IAutoBeAgent, IAutoBeGetFilesOptions, IAutoBeRpcListener, IAutoBeRpcService, IAutoBeTokenUsageJson } from "@autobe/interface";
+import { AutoBeHistory, AutoBePhase, AutoBeUserMessageContent, IAutoBeAgent, IAutoBeGetFilesOptions, IAutoBeRpcListener, IAutoBeRpcService, IAutoBeTokenUsageJson } from "@autobe/interface";
 /**
  * WebSocket RPC service implementation that wraps AutoBeAgent for remote
  * access.
@@ -45,6 +45,7 @@ export declare class AutoBeRpcService implements IAutoBeRpcService {
     getFiles(options?: Partial<IAutoBeGetFilesOptions>): Promise<Record<string, string>>;
     getHistories(): Promise<AutoBeHistory[]>;
     getTokenUsage(): Promise<IAutoBeTokenUsageJson>;
+    getPhase(): Promise<AutoBePhase | null>;
 }
 export declare namespace AgenticaRpcService {
     /**
@@ -78,5 +79,7 @@ export declare namespace AgenticaRpcService {
          * the remote client.
          */
         listener: IAutoBeRpcListener;
+        onStart?: (content: string | AutoBeUserMessageContent | AutoBeUserMessageContent[]) => void;
+        onComplete?: (result: AutoBeHistory[]) => void;
     }
 }

package/lib/AutoBeRpcService.js

@@ -103,14 +103,31 @@ class AutoBeRpcService {
             "realizeValidate",
             "testComplete",
             "testStart",
-            "testValidate"
-        ])
+            "testValidate",
+            "enable"
+        ]) {
+            if (key === "enable")
+                continue;
             agent.on(key, (event) => {
                 listener[key](event).catch(() => { });
             });
+        }
     }
     conversate(content) {
-        return this.props.agent.conversate(content);
+        return __awaiter(this, void 0, void 0, function* () {
+            if (this.props.onStart)
+                this.props.onStart(content);
+            this.props.listener.enable(false).catch(() => { });
+            try {
+                const result = yield this.props.agent.conversate(content);
+                if (this.props.onComplete)
+                    this.props.onComplete(result);
+                return result;
+            }
+            finally {
+                this.props.listener.enable(true).catch(() => { });
+            }
+        });
     }
     getFiles(options) {
         return __awaiter(this, void 0, void 0, function* () {
@@ -127,6 +144,11 @@ class AutoBeRpcService {
             return this.props.agent.getTokenUsage();
         });
     }
+    getPhase() {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this.props.agent.getPhase();
+        });
+    }
 }
 exports.AutoBeRpcService = AutoBeRpcService;
 //# sourceMappingURL=AutoBeRpcService.js.map

package/lib/AutoBeRpcService.js.map

@@ -1 +1 @@
-{"version":3,"file":"AutoBeRpcService.js","sourceRoot":"","sources":["../src/AutoBeRpcService.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AASA,kDAA0B;AAE1B;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAa,gBAAgB;IAC3B;;;;;;;;;;;;;;;OAeG;IACH,YAAoC,KAAgC;QAAhC,UAAK,GAAL,KAAK,CAA2B;QAClE,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC;QACvC,KAAK,MAAM,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;YACZ,KAAK,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,EAAE,EAAE;gBACtB,QAAQ,CAAC,GAAG,CAAE,CAAC,KAAY,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;IACP,CAAC;IAEM,UAAU,CACf,OAAuE;QAEvE,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAC9C,CAAC;IAEY,QAAQ,CACnB,OAAyC;;YAEzC,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC5C,CAAC;KAAA;IAEY,YAAY;;YACvB,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;QACzC,CAAC;KAAA;IAEY,aAAa;;YACxB,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,EAAE,CAAC;QAC1C,CAAC;KAAA;CACF;AA5CD,4CA4CC"}
+{"version":3,"file":"AutoBeRpcService.js","sourceRoot":"","sources":["../src/AutoBeRpcService.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAUA,kDAA0B;AAE1B;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAa,gBAAgB;IAC3B;;;;;;;;;;;;;;;OAeG;IACH,YAAoC,KAAgC;QAAhC,UAAK,GAAL,KAAK,CAA2B;QAClE,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC;QACvC,KAAK,MAAM,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;WAAqD,CAAC;YAClE,IAAI,GAAG,KAAK,QAAQ;gBAAE,SAAS;YAC/B,KAAK,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,EAAE,EAAE;gBACtB,QAAQ,CAAC,GAAG,CAAE,CAAC,KAAY,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEY,UAAU,CACrB,OAAuE;;YAEvE,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO;gBAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAEpD,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAClD,IAAI,CAAC;gBACH,MAAM,MAAM,GACV,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;gBAC7C,IAAI,IAAI,CAAC,KAAK,CAAC,UAAU;oBAAE,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;gBACzD,OAAO,MAAM,CAAC;YAChB,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACnD,CAAC;QACH,CAAC;KAAA;IAEY,QAAQ,CACnB,OAAyC;;YAEzC,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC5C,CAAC;KAAA;IAEY,YAAY;;YACvB,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;QACzC,CAAC;KAAA;IAEY,aAAa;;YACxB,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,EAAE,CAAC;QAC1C,CAAC;KAAA;IAEY,QAAQ;;YACnB,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QACrC,CAAC;KAAA;CACF;AA5DD,4CA4DC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@autobe/rpc",
-  "version": "0.22.0",
+  "version": "0.23.0",
   "description": "AI backend server code generator",
   "main": "lib/index.js",
   "keywords": [],
@@ -26,7 +26,7 @@
   "dependencies": {
     "@samchon/openapi": "^4.7.1",
     "typia": "^9.7.2",
-    "@autobe/interface": "^0.22.0"
+    "@autobe/interface": "^0.23.0"
   },
   "devDependencies": {
     "rimraf": "^6.0.1",

package/src/AutoBeRpcService.ts

@@ -1,5 +1,6 @@
 import {
   AutoBeHistory,
+  AutoBePhase,
   AutoBeUserMessageContent,
   IAutoBeAgent,
   IAutoBeGetFilesOptions,
@@ -51,16 +52,28 @@ export class AutoBeRpcService implements IAutoBeRpcService {
    */
   public constructor(private readonly props: AgenticaRpcService.IProps) {
     const { agent, listener } = this.props;
-    for (const key of typia.misc.literals<keyof IAutoBeRpcListener>())
+    for (const key of typia.misc.literals<keyof IAutoBeRpcListener>()) {
+      if (key === "enable") continue;
       agent.on(key, (event) => {
         listener[key]!(event as any).catch(() => {});
       });
+    }
   }
 
-  public conversate(
+  public async conversate(
     content: string | AutoBeUserMessageContent | AutoBeUserMessageContent[],
   ): Promise<AutoBeHistory[]> {
-    return this.props.agent.conversate(content);
+    if (this.props.onStart) this.props.onStart(content);
+
+    this.props.listener.enable(false).catch(() => {});
+    try {
+      const result: AutoBeHistory[] =
+        await this.props.agent.conversate(content);
+      if (this.props.onComplete) this.props.onComplete(result);
+      return result;
+    } finally {
+      this.props.listener.enable(true).catch(() => {});
+    }
   }
 
   public async getFiles(
@@ -76,6 +89,10 @@ export class AutoBeRpcService implements IAutoBeRpcService {
   public async getTokenUsage(): Promise<IAutoBeTokenUsageJson> {
     return this.props.agent.getTokenUsage();
   }
+
+  public async getPhase(): Promise<AutoBePhase | null> {
+    return this.props.agent.getPhase();
+  }
 }
 
 export namespace AgenticaRpcService {
@@ -111,5 +128,11 @@ export namespace AgenticaRpcService {
      * the remote client.
      */
     listener: IAutoBeRpcListener;
+
+    onStart?: (
+      content: string | AutoBeUserMessageContent | AutoBeUserMessageContent[],
+    ) => void;
+
+    onComplete?: (result: AutoBeHistory[]) => void;
   }
 }