@absolutejs/ai 0.0.18 → 0.0.20

package/dist/ai/index.js

@@ -2977,6 +2977,69 @@ var generateAI = async (options) => {
   }
   return { text, toolCalls, usage };
 };
+var DEFAULT_TOOL_MAX_TURNS = 6;
+var mergeUsage = (left, right) => {
+  if (!left)
+    return right;
+  if (!right)
+    return left;
+  const add = (a, b) => a === undefined && b === undefined ? undefined : (a ?? 0) + (b ?? 0);
+  return {
+    cacheReadInputTokens: add(left.cacheReadInputTokens, right.cacheReadInputTokens),
+    cacheWriteInputTokens: add(left.cacheWriteInputTokens, right.cacheWriteInputTokens),
+    inputTokens: (left.inputTokens ?? 0) + (right.inputTokens ?? 0),
+    outputTokens: (left.outputTokens ?? 0) + (right.outputTokens ?? 0)
+  };
+};
+var toProviderTools = (tools) => Object.entries(tools).map(([name, definition]) => ({
+  description: definition.description,
+  input_schema: definition.input,
+  name
+}));
+var generateAIWithTools = async (options) => {
+  const { maxTurns = DEFAULT_TOOL_MAX_TURNS, onToolUse, tools, ...base } = options;
+  const providerTools = toProviderTools(tools);
+  const toolCalls = [];
+  let usage;
+  const runTurn = async (messages, turnsLeft) => {
+    const result = await generateAI({
+      ...base,
+      messages,
+      toolChoice: "auto",
+      tools: providerTools
+    });
+    usage = mergeUsage(usage, result.usage);
+    if (result.toolCalls.length === 0 || turnsLeft <= 1) {
+      return { messages, text: result.text, toolCalls, usage };
+    }
+    toolCalls.push(...result.toolCalls);
+    const assistantBlocks = [
+      ...result.text ? [{ content: result.text, type: "text" }] : [],
+      ...result.toolCalls.map((call) => ({
+        id: call.id,
+        input: call.input,
+        name: call.name,
+        type: "tool_use"
+      }))
+    ];
+    const resultBlocks = await Promise.all(result.toolCalls.map(async (call) => {
+      const definition = tools[call.name];
+      const output = definition ? await Promise.resolve(definition.handler(call.input)).catch((err) => `Error: ${err instanceof Error ? err.message : String(err)}`) : `Error: unknown tool "${call.name}"`;
+      onToolUse?.(call.name, call.input, output);
+      return {
+        content: output,
+        tool_use_id: call.id,
+        type: "tool_result"
+      };
+    }));
+    return runTurn([
+      ...messages,
+      { content: assistantBlocks, role: "assistant" },
+      { content: resultBlocks, role: "user" }
+    ], turnsLeft - 1);
+  };
+  return runTurn(options.messages, maxTurns);
+};
 var generateObjectAI = async (options) => {
   const toolName = options.toolName ?? DEFAULT_OBJECT_TOOL_NAME;
   const tool = {
@@ -3843,6 +3906,7 @@ export {
   google,
   generateObjectAI,
   generateId,
+  generateAIWithTools,
   generateAI,
   gemini,
   deepseek,
@@ -3857,5 +3921,5 @@ export {
   aiChat
 };
 
-//# debugId=F448AA17322C183164756E2164756E21
+//# debugId=3AB7E01A7CCE98C464756E2164756E21
 //# sourceMappingURL=index.js.map