@autobe/agent 0.12.0 → 0.14.2

package/src/orchestrate/realize/writeCodeUntilCompilePassed.ts

@@ -1,5 +1,7 @@
 import {
   AutoBeOpenApi,
+  AutoBeRealizeAuthorization,
+  AutoBeRealizeFunction,
   IAutoBeTypeScriptCompileResult,
 } from "@autobe/interface";
 import { ILlmSchema } from "@samchon/openapi";
@@ -10,7 +12,6 @@ import { AutoBeContext } from "../../context/AutoBeContext";
 import { pipe } from "./RealizePipe";
 import { orchestrateRealizeCoder } from "./orchestrateRealizeCoder";
 import { orchestrateRealizePlanner } from "./orchestrateRealizePlanner";
-import { IAutoBeRealizeCoderApplication } from "./structures/IAutoBeRealizeCoderApplication";
 import { IAutoBeRealizeCompile } from "./structures/IAutoBeRealizeCompile";
 import { FAILED } from "./structures/IAutoBeRealizeFailedSymbol";
 
@@ -19,13 +20,17 @@ export async function writeCodeUntilCompilePassed<
 >(
   ctx: AutoBeContext<Model>,
   ops: AutoBeOpenApi.IOperation[],
-  retry: number = 3,
-): Promise<
-  Pick<
-    IAutoBeRealizeCoderApplication.RealizeCoderOutput,
-    "filename" | "implementationCode"
-  >[]
-> {
+  authorizations: AutoBeRealizeAuthorization[],
+  retry: number,
+): Promise<AutoBeRealizeFunction[]> {
+  const payloads = authorizations
+    .map((el) => {
+      return {
+        [el.payload.location]: el.payload.content,
+      };
+    })
+    .reduce<Record<string, string>>((acc, cur) => Object.assign(acc, cur), {});
+
   const files = Object.entries(await ctx.files({ dbms: "postgres" }))
     .filter(([key]) => {
       return key.startsWith("src");
@@ -35,7 +40,7 @@ export async function writeCodeUntilCompilePassed<
       {},
     );
 
-  const templateFiles = ["src/providers/jwtDecode.ts", "src/MyGlobal.ts"];
+  const templateFiles = ["src/MyGlobal.ts", "src/util/toISOStringSafe.ts"];
   const entireCodes: IAutoBeRealizeCompile.FileContentMap = {
     ...(await loadTemplateFiles(templateFiles)),
   };
@@ -45,7 +50,6 @@ export async function writeCodeUntilCompilePassed<
     total: [],
   };
 
-  ops = ops.filter((_el, i) => i < 10);
   for (let i = 0; i < retry; i++) {
     const targets = ops.filter((op) =>
       shouldProcessOperation(op, diagnostics.current),
@@ -57,7 +61,10 @@ export async function writeCodeUntilCompilePassed<
       | IAutoBeRealizeCompile.Fail
     )[] = await Promise.all(
       targets.map((op) => {
-        return process(ctx, metadata, op, diagnostics, entireCodes);
+        const role = op.authorizationRole;
+        const decorator = authorizations.find((el) => el.role === role);
+
+        return process(ctx, metadata, op, diagnostics, entireCodes, decorator);
       }),
     );
 
@@ -66,6 +73,12 @@ export async function writeCodeUntilCompilePassed<
         entireCodes[c.result.filename] = {
           content: c.result.implementationCode,
           result: "success",
+          endpoint: {
+            method: c.op.method,
+            path: c.op.path,
+          },
+          location: c.result.filename,
+          name: c.result.name,
         };
       }
     }
@@ -75,6 +88,7 @@ export async function writeCodeUntilCompilePassed<
     const compiler = await ctx.compiler();
     const compiled = await compiler.typescript.compile({
       files: {
+        ...payloads,
         ...files,
         ...nodeModules,
         ...Object.entries(entireCodes)
@@ -102,28 +116,34 @@ export async function writeCodeUntilCompilePassed<
       console.log(
         JSON.stringify(diagnostics.current, null, 2),
         `현재 에러의 수: ${diagnostics.current.length}\n`,
-        `현재 시도 수: ${i}`,
+        `현재 시도 수: ${i + 1}`,
       );
     }
   }
 
-  return Object.entries(entireCodes).map(([filename, { content }]) => {
-    return {
-      filename,
-      implementationCode: content,
-    };
-  });
+  return Object.entries(entireCodes)
+    .filter(([filename]) => filename.startsWith("src/providers")) // filter only provider files
+    .map(([filename, value]) => {
+      return {
+        filename,
+        content: value.content,
+        endpoint: value.endpoint!,
+        location: value.location!,
+        name: value.name!,
+        role: value.role!,
+      };
+    });
 }
 
 async function loadTemplateFiles(
   templateFiles: string[],
-): Promise<Record<string, { content: string; result: "success" }>> {
+): Promise<IAutoBeRealizeCompile.FileContentMap> {
   const templateBasePath = path.join(
     __dirname,
     "../../../../../internals/template/realize",
   );
 
-  const result: Record<string, { content: string; result: "success" }> = {};
+  const result: IAutoBeRealizeCompile.FileContentMap = {};
 
   for (const filePath of templateFiles) {
     result[filePath] = {
@@ -131,6 +151,8 @@ async function loadTemplateFiles(
         encoding: "utf-8",
       }),
       result: "success",
+      location: filePath,
+      role: null,
     };
   }
 
@@ -143,10 +165,11 @@ async function process<Model extends ILlmSchema.Model>(
   op: AutoBeOpenApi.IOperation,
   diagnostics: IAutoBeRealizeCompile.CompileDiagnostics,
   entireCodes: IAutoBeRealizeCompile.FileContentMap,
+  decorator?: AutoBeRealizeAuthorization,
 ) {
   const result = await pipe(
     op,
-    (op) => orchestrateRealizePlanner(ctx, op),
+    (op) => orchestrateRealizePlanner(ctx, op, decorator),
     async (p) => {
       const filename = `src/providers/${p.functionName}.ts` as const;
       const t = diagnostics.total.filter((el) => el.file === filename);
@@ -155,19 +178,21 @@ async function process<Model extends ILlmSchema.Model>(
       const c = entireCodes[filename]?.content ?? null;
 
       return orchestrateRealizeCoder(ctx, op, p, c, t, d).then((res) => {
+        ctx.dispatch({
+          type: "realizeProgress",
+          filename: filename,
+          content: res === FAILED ? "FAILED" : res.implementationCode,
+          completed: ++metadata.count,
+          created_at: new Date().toISOString(),
+          step: ctx.state().analyze?.step ?? 0,
+          total: metadata.total,
+        });
+
         if (res === FAILED) {
-        } else {
-          ctx.dispatch({
-            type: "realizeProgress",
-            filename: res.filename,
-            content: res.implementationCode,
-            completed: ++metadata.count,
-            created_at: new Date().toISOString(),
-            step: ctx.state().analyze?.step ?? 0,
-            total: metadata.total,
-          });
+          return res;
         }
-        return res;
+
+        return { ...res, name: p.functionName };
       });
     },
   );
@@ -176,7 +201,7 @@ async function process<Model extends ILlmSchema.Model>(
     return { type: "failed", op: op, result } as const;
   }
 
-  return { type: "success", op: op, result: result } as const;
+  return { type: "success", op: op, result } as const;
 }
 
 function shouldProcessOperation(

package/lib/orchestrate/realize/orchestrateRealizeDecorator.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"orchestrateRealizeDecorator.js","sourceRoot":"","sources":["../../../src/orchestrate/realize/orchestrateRealizeDecorator.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwBA,kEAgEC;;AAxFD,yCAAoE;AAMpE,2DAA6B;AAC7B,gDAAwB;AAExB,kDAA0B;AAG1B,uEAAoE;AACpE,iEAA8D;AAE9D,2EAAiF;AACjF,2GAAwG;AAExG;;;;;GAKG;AACH,SAAsB,2BAA2B,CAG/C,GAAyB;;;QAEzB,MAAM,KAAK,GACT,MAAA,MAAA,MAAA,GAAG;aACA,KAAK,EAAE;aACP,SAAS,0CAAE,QAAQ,CAAC,UAAU,CAAC,aAAa,0CAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,mCACzE,EAAE,CAAC;QAEL,MAAM,UAAU,GAAgD,EAAE,CAAC;QAEnE,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,MAAM,aAAa,GAAG;YACpB,iBAAiB,EAAE,MAAM,kBAAE,CAAC,QAAQ,CAClC,cAAI,CAAC,IAAI,CACP,SAAS,EACT,2DAA2D,CAC5D,EACD,OAAO,CACR;YACD,qCAAqC,EAAE,MAAM,kBAAE,CAAC,QAAQ,CACtD,cAAI,CAAC,IAAI,CACP,SAAS,EACT,yEAAyE,CAC1E,EACD,OAAO,CACR;SACF,CAAC;QAEF,MAAM,KAAK,qBACN,aAAa,CACjB,CAAC;QAEF,MAAM,OAAO,CAAC,GAAG,CACf,KAAK,CAAC,GAAG,CAAC,CAAO,IAAI,EAAE,EAAE;;YACvB,MAAM,SAAS,GACb,MAAM,OAAO,CAAC,GAAG,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC;YAE1C,KAAK,CAAC,kBAAkB,SAAS,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC;gBACpD,SAAS,CAAC,SAAS,CAAC,IAAI,CAAC;YAC3B,KAAK,CAAC,uBAAuB,SAAS,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC;gBACxD,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC;YAC1B,KAAK,CAAC,6BAA6B,SAAS,CAAC,aAAa,CAAC,IAAI,KAAK,CAAC;gBACnE,SAAS,CAAC,aAAa,CAAC,IAAI,CAAC;YAE/B,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAE3B,MAAM,MAAM,GAAgC;gBAC1C,IAAI,EAAE,kBAAkB;gBACxB,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACpC,KAAK;gBACL,SAAS,EAAE,EAAE,SAAS;gBACtB,KAAK,EAAE,KAAK,CAAC,MAAM;gBACnB,IAAI,EAAE,MAAA,MAAA,GAAG,CAAC,KAAK,EAAE,CAAC,IAAI,0CAAE,IAAI,mCAAI,CAAC;aAClC,CAAC;YAEF,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACvB,CAAC,CAAA,CAAC,CACH,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC;CAAA;AAED,SAAe,OAAO,CACpB,GAAyB,EACzB,IAAY,EACZ,aAAqC;;;QAErC,MAAM,OAAO,GAA+D;YAC1E,KAAK,EAAE,IAAI;SACZ,CAAC;QAEF,MAAM,QAAQ,GAAyB,IAAI,oBAAa,CAAC;YACvD,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,MAAM,kCACD,CAAC,MAAA,GAAG,CAAC,MAAM,mCAAI,EAAE,CAAC,KACrB,QAAQ,EAAE;oBACR,QAAQ,EAAE,IAAI;iBACf,GACF;YACD,SAAS,EAAE,IAAA,8DAAkC,EAAC,GAAG,EAAE,IAAI,CAAC;YACxD,WAAW,EAAE;gBACX,iBAAiB,CAAC;oBAChB,KAAK,EAAE,GAAG,CAAC,KAAK;oBAChB,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE;wBACd,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC;oBACvB,CAAC;iBACF,CAAC;aACH;SACF,CAAC,CAAC;QAEH,IAAA,iCAAe,EAAC,QAAQ,CAAC,CAAC;QAE1B,MAAM,QAAQ;aACX,UAAU,CAAC,8CAA8C,CAAC;aAC1D,OAAO,CAAC,GAAG,EAAE;YACZ,MAAM,UAAU,GAAG,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC5C,GAAG,CAAC,KAAK,EAAE,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEL,IAAI,OAAO,CAAC,KAAK,KAAK,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QAE3E,MAAM,QAAQ,GAAG,MAAA,GAAG,CAAC,KAAK,EAAE,CAAC,MAAM,0CAAE,QAAQ,CAAC;QAE9C,MAAM,aAAa,GACjB,CAAA,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,IAAI,MAAK,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;QAE3D,OAAO,MAAM,gBAAgB,CAC3B,GAAG,EACH,OAAO,CAAC,KAAK,EACb,aAAa,EACb,aAAa,CACd,CAAC;IACJ,CAAC;CAAA;AAED,SAAe,gBAAgB;yDAC7B,GAAyB,EACzB,MAAiD,EACjD,aAAqC,EACrC,aAAqC,EACrC,OAAe,CAAC;;QAEhB,gBAAgB;QAChB,MAAM,KAAK,iDACN,aAAa,GACb,aAAa,KAChB,CAAC,kBAAkB,MAAM,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,EAAE,MAAM,CAAC,SAAS,CAAC,IAAI,EACrE,CAAC,uBAAuB,MAAM,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,EACxE,CAAC,6BAA6B,MAAM,CAAC,aAAa,CAAC,IAAI,KAAK,CAAC,EAC3D,MAAM,CAAC,aAAa,CAAC,IAAI,GAC5B,CAAC;QAEF,MAAM,QAAQ,GAAoB,MAAM,GAAG,CAAC,QAAQ,EAAE,CAAC;QAEvD,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC;YACjD,KAAK;SACN,CAAC,CAAC;QAEH,GAAG,CAAC,QAAQ,CAAC;YACX,IAAI,EAAE,0BAA0B;YAChC,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACpC,MAAM,EAAE,QAAQ;YAChB,KAAK;YACL,IAAI,EAAE,MAAA,MAAA,GAAG,CAAC,KAAK,EAAE,CAAC,IAAI,0CAAE,IAAI,mCAAI,CAAC;SAClC,CAAC,CAAC;QAEH,IAAI,QAAQ,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAChC,OAAO,MAAM,CAAC;QAChB,CAAC;aAAM,IAAI,QAAQ,CAAC,IAAI,KAAK,WAAW,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;YACvD,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,MAAM,OAAO,GAA+D;YAC1E,KAAK,EAAE,IAAI;SACZ,CAAC;QAEF,MAAM,QAAQ,GAAyB,IAAI,oBAAa,CAAC;YACvD,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,MAAM,kCACD,CAAC,MAAA,GAAG,CAAC,MAAM,mCAAI,EAAE,CAAC,KACrB,QAAQ,EAAE;oBACR,QAAQ,EAAE,IAAI;iBACf,GACF;YACD,SAAS,EAAE,IAAA,qFAAyC,EAClD,GAAG,EACH,MAAM,EACN,aAAa,EACb,QAAQ,CAAC,WAAW,CACrB;YACD,WAAW,EAAE;gBACX,iBAAiB,CAAC;oBAChB,KAAK,EAAE,GAAG,CAAC,KAAK;oBAChB,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE;wBACd,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC;oBACvB,CAAC;iBACF,CAAC;aACH;SACF,CAAC,CAAC;QACH,IAAA,iCAAe,EAAC,QAAQ,CAAC,CAAC;QAE1B,MAAM,QAAQ;aACX,UAAU,CAAC,gDAAgD,CAAC;aAC5D,OAAO,CAAC,GAAG,EAAE;YACZ,MAAM,UAAU,GAAG,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC5C,GAAG,CAAC,KAAK,EAAE,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEL,IAAI,OAAO,CAAC,KAAK,KAAK,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAE5E,MAAM,cAAc,mCACf,KAAK,KACR,CAAC,kBAAkB,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,EACnD,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAC9B,CAAC,uBAAuB,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC,EACvD,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAC7B,CAAC,6BAA6B,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,KAAK,CAAC,EAClE,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,GACnC,CAAC;QAEF,GAAG,CAAC,QAAQ,CAAC;YACX,IAAI,EAAE,yBAAyB;YAC/B,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACpC,KAAK,EAAE,cAAc;YACrB,MAAM,EAAE,QAAQ;YAChB,IAAI,EAAE,MAAA,MAAA,GAAG,CAAC,KAAK,EAAE,CAAC,IAAI,0CAAE,IAAI,mCAAI,CAAC;SAClC,CAAC,CAAC;QAEH,OAAO,MAAM,gBAAgB,CAC3B,GAAG,EACH,OAAO,CAAC,KAAK,EACb,aAAa,EACb,aAAa,EACb,IAAI,GAAG,CAAC,CACT,CAAC;IACJ,CAAC;CAAA;AAED,SAAS,iBAAiB,CAAiC,KAG1D;IACC,IAAA,qCAAiB,EAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAE/B,MAAM,WAAW,GAA2B,UAAU,CACpD,KAAK,CAAC,KAAK,CACyB,CAAC;IAEvC,OAAO;QACL,QAAQ,EAAE,OAAO;QACjB,IAAI,EAAE,kBAAkB;QACxB,WAAW;QACX,OAAO,EAAE;YACP,eAAe,EAAE,CAAC,IAAI,EAAE,EAAE;gBACxB,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpB,CAAC;SAC2C;KAC/C,CAAC;AACJ,CAAC;AAED,MAAM,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAMT,CAAC;AACJ,MAAM,UAAU,GAAG;IACjB,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KAIJ;IACH,MAAM;IACN,KAAK,EAAE,MAAM;IACb,QAAQ,EAAE,MAAM;IAChB,KAAK,EAAE,MAAM;CACd,CAAC"}