agent-swarm-kit 1.1.73 → 1.1.74

package/build/index.cjs

@@ -24072,6 +24072,62 @@ const RETRY_DELAY = 5000;
  */
 class AdapterUtils {
     constructor() {
+        /**
+         * Creates a function to interact with Grok's chat completions API.
+         * @param {any} grok - The Grok client instance.
+         * @param {string} [model="grok-3-mini"] - The model to use for completions (defaults to "grok-3-mini").
+         * @returns {TCompleteFn} A function that processes completion arguments and returns a response from Grok.
+         */
+        this.fromGrok = (grok, model = "grok-3-mini") => 
+        /**
+         * Handles a completion request to Grok, transforming messages and tools into the required format.
+         * Executes requests in a pool to limit concurrency with retry logic for reliability.
+         * @param {ICompletionArgs} args - The arguments for the completion request.
+         * @param {string} args.agentName - The name of the agent making the request.
+         * @param {IModelMessage[]} args.messages - The array of messages to send to Grok.
+         * @param {string} args.mode - The mode of the completion (e.g., "user" or "tool").
+         * @param {any[]} args.tools - The tools available for the completion, if any.
+         * @returns {Promise<IModelMessage>} The response from Grok in `agent-swarm-kit` format.
+         */
+        functoolsKit.execpool(functoolsKit.retry(async ({ agentName, messages: rawMessages, mode, tools, clientId, }) => {
+            LoggerAdapter.logClient(clientId, "AdapterUtils fromGrok completion", JSON.stringify(rawMessages));
+            const messages = rawMessages.map(({ role, tool_call_id, tool_calls, content }) => ({
+                role,
+                tool_call_id,
+                content,
+                tool_calls: tool_calls?.map(({ function: f, ...rest }) => ({
+                    ...rest,
+                    function: {
+                        name: f.name,
+                        arguments: JSON.stringify(f.arguments),
+                    },
+                })),
+            }));
+            const { choices: [{ message: { content, role, tool_calls }, },], } = await grok.chat.completions.create({
+                model,
+                messages: messages,
+                tools: tools,
+                response_format: {
+                    type: "text",
+                },
+            });
+            return {
+                content: content,
+                mode,
+                agentName,
+                role,
+                tool_calls: tool_calls?.map(({ function: f, ...rest }) => ({
+                    ...rest,
+                    function: {
+                        name: f.name,
+                        arguments: JSON.parse(f.arguments),
+                    },
+                })),
+            };
+        }, RETRY_COUNT, RETRY_DELAY), {
+            maxExec: EXECPOOL_SIZE,
+            delay: EXECPOOL_WAIT,
+        });
         /**
          * Creates a function to interact with CohereClientV2 chat completions API.
          * @param {any} openai - The CohereClientV2 client instance.

package/build/index.mjs

@@ -24070,6 +24070,62 @@ const RETRY_DELAY = 5000;
  */
 class AdapterUtils {
     constructor() {
+        /**
+         * Creates a function to interact with Grok's chat completions API.
+         * @param {any} grok - The Grok client instance.
+         * @param {string} [model="grok-3-mini"] - The model to use for completions (defaults to "grok-3-mini").
+         * @returns {TCompleteFn} A function that processes completion arguments and returns a response from Grok.
+         */
+        this.fromGrok = (grok, model = "grok-3-mini") => 
+        /**
+         * Handles a completion request to Grok, transforming messages and tools into the required format.
+         * Executes requests in a pool to limit concurrency with retry logic for reliability.
+         * @param {ICompletionArgs} args - The arguments for the completion request.
+         * @param {string} args.agentName - The name of the agent making the request.
+         * @param {IModelMessage[]} args.messages - The array of messages to send to Grok.
+         * @param {string} args.mode - The mode of the completion (e.g., "user" or "tool").
+         * @param {any[]} args.tools - The tools available for the completion, if any.
+         * @returns {Promise<IModelMessage>} The response from Grok in `agent-swarm-kit` format.
+         */
+        execpool(retry(async ({ agentName, messages: rawMessages, mode, tools, clientId, }) => {
+            LoggerAdapter.logClient(clientId, "AdapterUtils fromGrok completion", JSON.stringify(rawMessages));
+            const messages = rawMessages.map(({ role, tool_call_id, tool_calls, content }) => ({
+                role,
+                tool_call_id,
+                content,
+                tool_calls: tool_calls?.map(({ function: f, ...rest }) => ({
+                    ...rest,
+                    function: {
+                        name: f.name,
+                        arguments: JSON.stringify(f.arguments),
+                    },
+                })),
+            }));
+            const { choices: [{ message: { content, role, tool_calls }, },], } = await grok.chat.completions.create({
+                model,
+                messages: messages,
+                tools: tools,
+                response_format: {
+                    type: "text",
+                },
+            });
+            return {
+                content: content,
+                mode,
+                agentName,
+                role,
+                tool_calls: tool_calls?.map(({ function: f, ...rest }) => ({
+                    ...rest,
+                    function: {
+                        name: f.name,
+                        arguments: JSON.parse(f.arguments),
+                    },
+                })),
+            };
+        }, RETRY_COUNT, RETRY_DELAY), {
+            maxExec: EXECPOOL_SIZE,
+            delay: EXECPOOL_WAIT,
+        });
         /**
          * Creates a function to interact with CohereClientV2 chat completions API.
          * @param {any} openai - The CohereClientV2 client instance.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-swarm-kit",
-  "version": "1.1.73",
+  "version": "1.1.74",
   "description": "A TypeScript library for building orchestrated framework-agnostic multi-agent AI systems",
   "author": {
     "name": "Petr Tripolsky",

package/types.d.ts

@@ -14751,6 +14751,13 @@ type TCompleteFn = (args: ICompletionArgs) => Promise<IModelMessage>;
  * Utility class providing adapter functions for interacting with various AI completion providers.
  */
 declare class AdapterUtils {
+    /**
+     * Creates a function to interact with Grok's chat completions API.
+     * @param {any} grok - The Grok client instance.
+     * @param {string} [model="grok-3-mini"] - The model to use for completions (defaults to "grok-3-mini").
+     * @returns {TCompleteFn} A function that processes completion arguments and returns a response from Grok.
+     */
+    fromGrok: (grok: any, model?: string) => TCompleteFn;
     /**
      * Creates a function to interact with CohereClientV2 chat completions API.
      * @param {any} openai - The CohereClientV2 client instance.