ai 2.2.29 → 2.2.31

package/react/dist/index.js

@@ -137,6 +137,23 @@ var dataMessageStreamPart = {
     };
   }
 };
+var toolCallStreamPart = {
+  code: "7",
+  name: "tool_calls",
+  parse: (value) => {
+    if (value == null || typeof value !== "object" || !("tool_calls" in value) || typeof value.tool_calls !== "object" || value.tool_calls == null || !Array.isArray(value.tool_calls) || value.tool_calls.some((tc) => {
+      tc == null || typeof tc !== "object" || !("id" in tc) || typeof tc.id !== "string" || !("type" in tc) || typeof tc.type !== "string" || !("function" in tc) || tc.function == null || typeof tc.function !== "object" || !("arguments" in tc.function) || typeof tc.function.name !== "string" || typeof tc.function.arguments !== "string";
+    })) {
+      throw new Error(
+        '"tool_calls" parts expect an object with a ToolCallPayload.'
+      );
+    }
+    return {
+      type: "tool_calls",
+      value
+    };
+  }
+};
 var streamParts = [
   textStreamPart,
   functionCallStreamPart,
@@ -144,7 +161,8 @@ var streamParts = [
   errorStreamPart,
   assistantMessageStreamPart,
   assistantControlDataStreamPart,
-  dataMessageStreamPart
+  dataMessageStreamPart,
+  toolCallStreamPart
 ];
 var streamPartsByCode = {
   [textStreamPart.code]: textStreamPart,
@@ -153,7 +171,8 @@ var streamPartsByCode = {
   [errorStreamPart.code]: errorStreamPart,
   [assistantMessageStreamPart.code]: assistantMessageStreamPart,
   [assistantControlDataStreamPart.code]: assistantControlDataStreamPart,
-  [dataMessageStreamPart.code]: dataMessageStreamPart
+  [dataMessageStreamPart.code]: dataMessageStreamPart,
+  [toolCallStreamPart.code]: toolCallStreamPart
 };
 var StreamStringPrefixes = {
   [textStreamPart.name]: textStreamPart.code,
@@ -162,7 +181,8 @@ var StreamStringPrefixes = {
   [errorStreamPart.name]: errorStreamPart.code,
   [assistantMessageStreamPart.name]: assistantMessageStreamPart.code,
   [assistantControlDataStreamPart.name]: assistantControlDataStreamPart.code,
-  [dataMessageStreamPart.name]: dataMessageStreamPart.code
+  [dataMessageStreamPart.name]: dataMessageStreamPart.code,
+  [toolCallStreamPart.name]: toolCallStreamPart.code
 };
 var validCodes = streamParts.map((part) => part.code);
 var parseStreamPart = (line) => {
@@ -288,20 +308,35 @@ async function parseComplexResponse({
       };
       functionCallMessage = prefixMap["function_call"];
     }
+    let toolCallMessage = null;
+    if (type === "tool_calls") {
+      prefixMap["tool_calls"] = {
+        id: generateId(),
+        role: "assistant",
+        content: "",
+        tool_calls: value.tool_calls,
+        createdAt
+      };
+      toolCallMessage = prefixMap["tool_calls"];
+    }
     if (type === "data") {
       prefixMap["data"].push(...value);
     }
     const responseMessage = prefixMap["text"];
-    const merged = [functionCallMessage, responseMessage].filter(
-      Boolean
-    );
+    const merged = [
+      functionCallMessage,
+      toolCallMessage,
+      responseMessage
+    ].filter(Boolean);
     update(merged, [...prefixMap["data"]]);
   }
   onFinish == null ? void 0 : onFinish(prefixMap);
   return {
-    messages: [prefixMap.text, prefixMap.function_call].filter(
-      Boolean
-    ),
+    messages: [
+      prefixMap.text,
+      prefixMap.function_call,
+      prefixMap.tool_calls
+    ].filter(Boolean),
     data: prefixMap.data
   };
 }
@@ -387,6 +422,8 @@ async function callChatApi({
       streamedResponse += decode(value);
       if (streamedResponse.startsWith('{"function_call":')) {
         responseMessage["function_call"] = streamedResponse;
+      } else if (streamedResponse.startsWith('{"tool_calls":')) {
+        responseMessage["tool_calls"] = streamedResponse;
       } else {
         responseMessage["content"] = streamedResponse;
       }
@@ -401,6 +438,11 @@ async function callChatApi({
       responseMessage["function_call"] = parsedFunctionCall;
       appendMessage({ ...responseMessage });
     }
+    if (streamedResponse.startsWith('{"tool_calls":')) {
+      const parsedToolCalls = JSON.parse(streamedResponse).tool_calls;
+      responseMessage["tool_calls"] = parsedToolCalls;
+      appendMessage({ ...responseMessage });
+    }
     if (onFinish) {
       onFinish(responseMessage);
     }
@@ -412,6 +454,7 @@ async function callChatApi({
 async function processChatStream({
   getStreamedResponse: getStreamedResponse2,
   experimental_onFunctionCall,
+  experimental_onToolCall,
   updateChatRequest,
   getCurrentMessages
 }) {
@@ -420,12 +463,18 @@ async function processChatStream({
     if ("messages" in messagesAndDataOrJustMessage) {
       let hasFollowingResponse = false;
       for (const message of messagesAndDataOrJustMessage.messages) {
-        if (message.function_call === void 0 || typeof message.function_call === "string") {
+        if ((message.function_call === void 0 || typeof message.function_call === "string") && (message.tool_calls === void 0 || typeof message.tool_calls === "string")) {
           continue;
         }
         hasFollowingResponse = true;
         if (experimental_onFunctionCall) {
           const functionCall = message.function_call;
+          if (typeof functionCall !== "object") {
+            console.warn(
+              "experimental_onFunctionCall should not be defined when using tools"
+            );
+            continue;
+          }
           const functionCallResponse = await experimental_onFunctionCall(
             getCurrentMessages(),
             functionCall
@@ -436,22 +485,83 @@ async function processChatStream({
           }
           updateChatRequest(functionCallResponse);
         }
+        if (experimental_onToolCall) {
+          const toolCalls = message.tool_calls;
+          if (!Array.isArray(toolCalls) || toolCalls.some((toolCall) => typeof toolCall !== "object")) {
+            console.warn(
+              "experimental_onToolCall should not be defined when using tools"
+            );
+            continue;
+          }
+          const toolCallResponse = await experimental_onToolCall(getCurrentMessages(), toolCalls);
+          if (toolCallResponse === void 0) {
+            hasFollowingResponse = false;
+            break;
+          }
+          updateChatRequest(toolCallResponse);
+        }
       }
       if (!hasFollowingResponse) {
         break;
       }
     } else {
+      let fixFunctionCallArguments2 = function(response) {
+        for (const message of response.messages) {
+          if (message.tool_calls !== void 0) {
+            for (const toolCall of message.tool_calls) {
+              if (typeof toolCall === "object") {
+                if (toolCall.function.arguments && typeof toolCall.function.arguments !== "string") {
+                  toolCall.function.arguments = JSON.stringify(
+                    toolCall.function.arguments
+                  );
+                }
+              }
+            }
+          }
+          if (message.function_call !== void 0) {
+            if (typeof message.function_call === "object") {
+              if (message.function_call.arguments && typeof message.function_call.arguments !== "string") {
+                message.function_call.arguments = JSON.stringify(
+                  message.function_call.arguments
+                );
+              }
+            }
+          }
+        }
+      };
+      var fixFunctionCallArguments = fixFunctionCallArguments2;
       const streamedResponseMessage = messagesAndDataOrJustMessage;
-      if (streamedResponseMessage.function_call === void 0 || typeof streamedResponseMessage.function_call === "string") {
+      if ((streamedResponseMessage.function_call === void 0 || typeof streamedResponseMessage.function_call === "string") && (streamedResponseMessage.tool_calls === void 0 || typeof streamedResponseMessage.tool_calls === "string")) {
         break;
       }
       if (experimental_onFunctionCall) {
         const functionCall = streamedResponseMessage.function_call;
+        if (!(typeof functionCall === "object")) {
+          console.warn(
+            "experimental_onFunctionCall should not be defined when using tools"
+          );
+          continue;
+        }
         const functionCallResponse = await experimental_onFunctionCall(getCurrentMessages(), functionCall);
         if (functionCallResponse === void 0)
           break;
+        fixFunctionCallArguments2(functionCallResponse);
         updateChatRequest(functionCallResponse);
       }
+      if (experimental_onToolCall) {
+        const toolCalls = streamedResponseMessage.tool_calls;
+        if (!(typeof toolCalls === "object")) {
+          console.warn(
+            "experimental_onToolCall should not be defined when using functions"
+          );
+          continue;
+        }
+        const toolCallResponse = await experimental_onToolCall(getCurrentMessages(), toolCalls);
+        if (toolCallResponse === void 0)
+          break;
+        fixFunctionCallArguments2(toolCallResponse);
+        updateChatRequest(toolCallResponse);
+      }
     }
   }
 }
@@ -461,14 +571,20 @@ var getStreamedResponse = async (api, chatRequest, mutate, mutateStreamData, exi
   var _a, _b;
   const previousMessages = messagesRef.current;
   mutate(chatRequest.messages, false);
-  const constructedMessagesPayload = sendExtraMessageFields ? chatRequest.messages : chatRequest.messages.map(({ role, content, name, function_call }) => ({
-    role,
-    content,
-    ...name !== void 0 && { name },
-    ...function_call !== void 0 && {
-      function_call
-    }
-  }));
+  const constructedMessagesPayload = sendExtraMessageFields ? chatRequest.messages : chatRequest.messages.map(
+    ({ role, content, name, function_call, tool_calls, tool_call_id }) => ({
+      role,
+      content,
+      tool_call_id,
+      ...name !== void 0 && { name },
+      ...function_call !== void 0 && {
+        function_call
+      },
+      ...tool_calls !== void 0 && {
+        tool_calls
+      }
+    })
+  );
   if (typeof api !== "string") {
     const replyId = generateId();
     const createdAt = /* @__PURE__ */ new Date();
@@ -514,6 +630,12 @@ var getStreamedResponse = async (api, chatRequest, mutate, mutateStreamData, exi
       },
       ...chatRequest.function_call !== void 0 && {
         function_call: chatRequest.function_call
+      },
+      ...chatRequest.tools !== void 0 && {
+        tools: chatRequest.tools
+      },
+      ...chatRequest.tool_choice !== void 0 && {
+        tool_choice: chatRequest.tool_choice
       }
     },
     credentials: extraMetadataRef.current.credentials,
@@ -544,6 +666,7 @@ function useChat({
   initialInput = "",
   sendExtraMessageFields,
   experimental_onFunctionCall,
+  experimental_onToolCall,
   onResponse,
   onFinish,
   onError,
@@ -607,6 +730,7 @@ function useChat({
             sendExtraMessageFields
           ),
           experimental_onFunctionCall,
+          experimental_onToolCall,
           updateChatRequest: (chatRequestParam) => {
             chatRequest = chatRequestParam;
           },
@@ -639,13 +763,21 @@ function useChat({
       streamData,
       sendExtraMessageFields,
       experimental_onFunctionCall,
+      experimental_onToolCall,
       messagesRef,
       abortControllerRef,
       generateId
     ]
   );
   const append = (0, import_react.useCallback)(
-    async (message, { options, functions, function_call, data } = {}) => {
+    async (message, {
+      options,
+      functions,
+      function_call,
+      tools,
+      tool_choice,
+      data
+    } = {}) => {
       if (!message.id) {
         message.id = generateId();
       }
@@ -654,14 +786,22 @@ function useChat({
         options,
         data,
         ...functions !== void 0 && { functions },
-        ...function_call !== void 0 && { function_call }
+        ...function_call !== void 0 && { function_call },
+        ...tools !== void 0 && { tools },
+        ...tool_choice !== void 0 && { tool_choice }
       };
       return triggerRequest(chatRequest);
     },
     [triggerRequest, generateId]
   );
   const reload = (0, import_react.useCallback)(
-    async ({ options, functions, function_call } = {}) => {
+    async ({
+      options,
+      functions,
+      function_call,
+      tools,
+      tool_choice
+    } = {}) => {
       if (messagesRef.current.length === 0)
         return null;
       const lastMessage = messagesRef.current[messagesRef.current.length - 1];
@@ -670,7 +810,9 @@ function useChat({
           messages: messagesRef.current.slice(0, -1),
           options,
           ...functions !== void 0 && { functions },
-          ...function_call !== void 0 && { function_call }
+          ...function_call !== void 0 && { function_call },
+          ...tools !== void 0 && { tools },
+          ...tool_choice !== void 0 && { tool_choice }
         };
         return triggerRequest(chatRequest2);
       }
@@ -678,7 +820,9 @@ function useChat({
         messages: messagesRef.current,
         options,
         ...functions !== void 0 && { functions },
-        ...function_call !== void 0 && { function_call }
+        ...function_call !== void 0 && { function_call },
+        ...tools !== void 0 && { tools },
+        ...tool_choice !== void 0 && { tool_choice }
       };
       return triggerRequest(chatRequest);
     },
@@ -1070,6 +1214,7 @@ function experimental_useAssistant({
     messages,
     threadId,
     input,
+    setInput,
     handleInputChange,
     submitMessage,
     status,