@autobe/agent 0.25.3 → 0.25.4

package/lib/orchestrate/prisma/orchestratePrismaCorrect.js

@@ -62,7 +62,7 @@ function orchestratePrismaCorrect(ctx, application) {
             return true;
         });
     application.files = application.files.filter((f) => f.models.length !== 0);
-    return iterate(ctx, application, Math.max(ctx.retry, 5));
+    return iterate(ctx, application, Math.max(ctx.retry, 8));
 }
 function iterate(ctx, application, life) {
     return __awaiter(this, void 0, void 0, function* () {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@autobe/agent",
-  "version": "0.25.3",
+  "version": "0.25.4",
   "description": "AI backend server code generator",
   "main": "lib/index.js",
   "author": "Wrtn Technologies",
@@ -27,8 +27,8 @@
     "tstl": "^3.0.0",
     "typia": "^9.7.2",
     "uuid": "^11.1.0",
-    "@autobe/utils": "^0.25.3",
-    "@autobe/interface": "^0.25.3"
+    "@autobe/interface": "^0.25.4",
+    "@autobe/utils": "^0.25.4"
   },
   "devDependencies": {
     "@rollup/plugin-json": "^6.1.0",
@@ -40,8 +40,8 @@
     "ts-node": "^10.9.2",
     "ts-patch": "^3.3.0",
     "typescript": "~5.9.2",
-    "@autobe/compiler": "^0.25.3",
-    "@autobe/filesystem": "^0.25.3"
+    "@autobe/compiler": "^0.25.4",
+    "@autobe/filesystem": "^0.25.4"
   },
   "keywords": [
     "ai",

package/src/AutoBeAgent.ts

@@ -27,6 +27,7 @@ import { createAgenticaHistory } from "./factory/createAgenticaHistory";
 import { createAutoBeController } from "./factory/createAutoBeApplication";
 import { createAutoBeContext } from "./factory/createAutoBeContext";
 import { createAutoBeState } from "./factory/createAutoBeState";
+import { supportMistral } from "./factory/supportMistral";
 import { transformFacadeStateMessage } from "./orchestrate/facade/transformFacadeStateMessage";
 import { IAutoBeProps } from "./structures/IAutoBeProps";
 import { randomBackoffStrategy } from "./utils/backoffRetry";
@@ -162,6 +163,7 @@ export class AutoBeAgent<Model extends ILlmSchema.Model>
         }),
       ],
     });
+    supportMistral(this.agentica_, props.vendor);
 
     // HISTORIES MANIPULATION
     this.agentica_.getHistories().push(

package/src/factory/consentFunctionCall.ts

@@ -16,6 +16,7 @@ import { AutoBeConfigConstant } from "../constants/AutoBeConfigConstant";
 import { AutoBeSystemPromptConstant } from "../constants/AutoBeSystemPromptConstant";
 import { IAutoBeConfig } from "../structures/IAutoBeConfig";
 import { IAutoBeVendor } from "../structures/IAutoBeVendor";
+import { supportMistral } from "./supportMistral";
 
 export const consentFunctionCall = async (props: {
   dispatch: (event: AutoBeEvent) => void;
@@ -67,6 +68,8 @@ export const consentFunctionCall = async (props: {
       } satisfies IConsentApplication),
     ],
   });
+  supportMistral(agent, props.vendor);
+
   const histories: MicroAgenticaHistory<"chatgpt">[] = await agent.conversate(
     "Analyze and judge this assistant message please.",
   );

package/src/factory/createAutoBeContext.ts

@@ -41,6 +41,7 @@ import { AutoBeTimeoutError } from "../utils/AutoBeTimeoutError";
 import { TimedConversation } from "../utils/TimedConversation";
 import { consentFunctionCall } from "./consentFunctionCall";
 import { getCriticalCompiler } from "./getCriticalCompiler";
+import { supportMistral } from "./supportMistral";
 
 export const createAutoBeContext = <Model extends ILlmSchema.Model>(props: {
   model: Model;
@@ -104,6 +105,7 @@ export const createAutoBeContext = <Model extends ILlmSchema.Model>(props: {
           histories: next.histories,
           controllers: [next.controller],
         });
+        supportMistral(agent, props.vendor);
 
         // ADD EVENT LISTENERS
         agent.on("request", async (event) => {

package/src/factory/supportMistral.ts

@@ -0,0 +1,122 @@
+import { MicroAgentica } from "@agentica/core";
+import { ILlmSchema } from "@samchon/openapi";
+import OpenAI from "openai";
+
+import { IAutoBeVendor } from "../structures/IAutoBeVendor";
+
+export const supportMistral = <Model extends ILlmSchema.Model>(
+  agent: MicroAgentica<Model>,
+  vendor: IAutoBeVendor,
+): void => {
+  if (
+    vendor.model.includes("mistral") ||
+    vendor.model.includes("devstral") ||
+    vendor.model.includes("codestral")
+  ) {
+    agent.on("request", async (e) => {
+      const newMessages: OpenAI.ChatCompletionMessageParam[] = [];
+      for (const m of e.body.messages) {
+        newMessages.push(m);
+        if (m.role === "tool") {
+          m.tool_call_id = uuidToShortId(m.tool_call_id);
+          newMessages.push({
+            role: "assistant",
+            content: "A tool has been called.",
+          });
+        } else if (m.role === "assistant") {
+          for (const call of m.tool_calls ?? [])
+            call.id = uuidToShortId(call.id);
+        }
+      }
+      e.body.messages = newMessages;
+    });
+
+    // agent.on("request", (e) => {
+    //   const toolCalls: OpenAI.ChatCompletionMessageFunctionToolCall[] =
+    //     e.body.messages
+    //       .filter((m) => m.role === "assistant")
+    //       .filter((m) => !!m.tool_calls?.length)
+    //       .map((m) => m.tool_calls ?? [])
+    //       .flat()
+    //       .filter((c) => c.type === "function");
+    //   e.body.messages.forEach((m, i, array) => {
+    //     if (m.role !== "tool") return;
+    //     const call: OpenAI.ChatCompletionMessageFunctionToolCall | undefined =
+    //       toolCalls.find((c) => c.id === m.tool_call_id);
+    //     const content: string = getFunctionCallMessage(m, call);
+    //     array[i] = {
+    //       role: "assistant",
+    //       content,
+    //     };
+    //   });
+    //   e.body.messages = e.body.messages.filter(
+    //     (m) => m.role !== "assistant" || !m.tool_calls?.length,
+    //   );
+    // });
+  }
+};
+
+// const getFunctionCallMessage = (
+//   param: OpenAI.ChatCompletionToolMessageParam,
+//   call: OpenAI.ChatCompletionMessageFunctionToolCall | undefined,
+// ): string => {
+//   if (call === undefined) {
+//     // unreachable
+//     return StringUtil.trim`
+//       ## Function Call
+
+//       A function has been called, but could not find its arguments.
+
+//       - id: ${param.tool_call_id}
+//       - content: ${param.content}
+//     `;
+//   }
+//   return StringUtil.trim`
+//     ## Function Call
+
+//     - id: ${call.id}
+//     - function name: ${call.function.name}
+//     - arguments: ${JSON.stringify(call.function.arguments)}
+//     - content: ${param.content}
+//   `;
+// };
+
+const BASE62_CHARS =
+  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
+
+function simpleHash(str: string): number {
+  let h1 = 0xdeadbeef;
+  let h2 = 0x41c6ce57;
+
+  for (let i = 0; i < str.length; i++) {
+    const ch = str.charCodeAt(i);
+    h1 = Math.imul(h1 ^ ch, 2654435761);
+    h2 = Math.imul(h2 ^ ch, 1597334677);
+  }
+
+  h1 =
+    Math.imul(h1 ^ (h1 >>> 16), 2246822507) ^
+    Math.imul(h2 ^ (h2 >>> 13), 3266489909);
+  h2 =
+    Math.imul(h2 ^ (h2 >>> 16), 2246822507) ^
+    Math.imul(h1 ^ (h1 >>> 13), 3266489909);
+
+  return 4294967296 * (2097151 & h2) + (h1 >>> 0);
+}
+
+function toBase62(num: number, length: number): string {
+  let result = "";
+  let n = num;
+
+  while (n > 0 && result.length < length) {
+    result = BASE62_CHARS[n % 62] + result;
+    n = Math.floor(n / 62);
+  }
+
+  return result.padStart(length, "0");
+}
+
+function uuidToShortId(uuid: string): string {
+  const hash = simpleHash(uuid);
+  return toBase62(hash, 9);
+}

package/src/orchestrate/prisma/orchestratePrismaCorrect.ts

@@ -27,7 +27,7 @@ export function orchestratePrismaCorrect<Model extends ILlmSchema.Model>(
       return true;
     });
   application.files = application.files.filter((f) => f.models.length !== 0);
-  return iterate(ctx, application, Math.max(ctx.retry, 5));
+  return iterate(ctx, application, Math.max(ctx.retry, 8));
 }
 
 async function iterate<Model extends ILlmSchema.Model>(