ai 3.1.15 → 3.1.17

package/rsc/dist/rsc-server.mjs

@@ -500,6 +500,33 @@ function convertZodToJSONSchema(zodSchema) {
   return zodToJsonSchema(zodSchema);
 }
 
+// core/util/is-non-empty-object.ts
+function isNonEmptyObject(object) {
+  return object != null && Object.keys(object).length > 0;
+}
+
+// core/prompt/prepare-tools-and-tool-choice.ts
+function prepareToolsAndToolChoice({
+  tools,
+  toolChoice
+}) {
+  if (!isNonEmptyObject(tools)) {
+    return {
+      tools: void 0,
+      toolChoice: void 0
+    };
+  }
+  return {
+    tools: Object.entries(tools).map(([name, tool]) => ({
+      type: "function",
+      name,
+      description: tool.description,
+      parameters: convertZodToJSONSchema(tool.parameters)
+    })),
+    toolChoice: toolChoice == null ? { type: "auto" } : typeof toolChoice === "string" ? { type: toolChoice } : { type: "tool", toolName: toolChoice.toolName }
+  };
+}
+
 // shared/stream-parts.ts
 var textStreamPart = {
   code: "0",
@@ -1641,6 +1668,7 @@ var defaultTextRenderer = ({ content }) => content;
 async function streamUI({
   model,
   tools,
+  toolChoice,
   system,
   prompt,
   messages,
@@ -1720,12 +1748,7 @@ async function streamUI({
     () => model.doStream({
       mode: {
         type: "regular",
-        tools: tools == null ? void 0 : Object.entries(tools).map(([name, tool]) => ({
-          type: "function",
-          name,
-          description: tool.description,
-          parameters: convertZodToJSONSchema(tool.parameters)
-        }))
+        ...prepareToolsAndToolChoice({ tools, toolChoice })
       },
       ...prepareCallSettings(settings),
       inputFormat: validatedPrompt.type,