ai 2.2.21 → 2.2.22

package/dist/index.d.ts

@@ -574,7 +574,9 @@ interface Prediction {
  * return new StreamingTextResponse(stream)
  *
  */
-declare function ReplicateStream(res: Prediction, cb?: AIStreamCallbacksAndOptions): Promise<ReadableStream>;
+declare function ReplicateStream(res: Prediction, cb?: AIStreamCallbacksAndOptions, options?: {
+    headers?: Record<string, string>;
+}): Promise<ReadableStream>;
 
 declare const nanoid: (size?: number | undefined) => string;
 declare function createChunkDecoder(): (chunk: Uint8Array | undefined) => string;

package/dist/index.js

@@ -691,7 +691,7 @@ function LangChainStream(callbacks) {
 }
 
 // streams/replicate-stream.ts
-async function ReplicateStream(res, cb) {
+async function ReplicateStream(res, cb, options) {
   var _a;
   const url = (_a = res.urls) == null ? void 0 : _a.stream;
   if (!url) {
@@ -703,7 +703,8 @@ async function ReplicateStream(res, cb) {
   const eventStream = await fetch(url, {
     method: "GET",
     headers: {
-      Accept: "text/event-stream"
+      Accept: "text/event-stream",
+      ...options == null ? void 0 : options.headers
     }
   });
   return AIStream(eventStream, void 0, cb).pipeThrough(

package/dist/index.mjs

@@ -645,7 +645,7 @@ function LangChainStream(callbacks) {
 }
 
 // streams/replicate-stream.ts
-async function ReplicateStream(res, cb) {
+async function ReplicateStream(res, cb, options) {
   var _a;
   const url = (_a = res.urls) == null ? void 0 : _a.stream;
   if (!url) {
@@ -657,7 +657,8 @@ async function ReplicateStream(res, cb) {
   const eventStream = await fetch(url, {
     method: "GET",
     headers: {
-      Accept: "text/event-stream"
+      Accept: "text/event-stream",
+      ...options == null ? void 0 : options.headers
     }
   });
   return AIStream(eventStream, void 0, cb).pipeThrough(

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai",
-  "version": "2.2.21",
+  "version": "2.2.22",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",

package/react/dist/index.d.ts

@@ -264,7 +264,7 @@ type UseChatHelpers = {
 type StreamingReactResponseAction = (payload: {
     messages: Message[];
 }) => Promise<experimental_StreamingReactResponse>;
-declare function useChat({ api, id, initialMessages: initialMessagesParam, initialInput, sendExtraMessageFields, experimental_onFunctionCall, onResponse, onFinish, onError, credentials, headers, body, }?: Omit<UseChatOptions, 'api'> & {
+declare function useChat({ api, id, initialMessages, initialInput, sendExtraMessageFields, experimental_onFunctionCall, onResponse, onFinish, onError, credentials, headers, body, }?: Omit<UseChatOptions, 'api'> & {
     api?: string | StreamingReactResponseAction;
 }): UseChatHelpers;
 

package/react/dist/index.js

@@ -332,7 +332,7 @@ var getStreamedResponse = async (api, chatRequest, mutate, mutateStreamData, exi
 function useChat({
   api = "/api/chat",
   id,
-  initialMessages: initialMessagesParam,
+  initialMessages,
   initialInput = "",
   sendExtraMessageFields,
   experimental_onFunctionCall,
@@ -345,9 +345,9 @@ function useChat({
 } = {}) {
   const hookId = (0, import_react.useId)();
   const chatId = id || hookId;
-  const [initialMessages] = (0, import_react.useState)(initialMessagesParam != null ? initialMessagesParam : []);
+  const [initialMessagesFallback] = (0, import_react.useState)([]);
   const { data: messages, mutate } = (0, import_swr.default)([api, chatId], null, {
-    fallbackData: initialMessages
+    fallbackData: initialMessages != null ? initialMessages : initialMessagesFallback
   });
   const { data: isLoading = false, mutate: mutateLoading } = (0, import_swr.default)(
     [chatId, "loading"],

package/react/dist/index.mjs

@@ -296,7 +296,7 @@ var getStreamedResponse = async (api, chatRequest, mutate, mutateStreamData, exi
 function useChat({
   api = "/api/chat",
   id,
-  initialMessages: initialMessagesParam,
+  initialMessages,
   initialInput = "",
   sendExtraMessageFields,
   experimental_onFunctionCall,
@@ -309,9 +309,9 @@ function useChat({
 } = {}) {
   const hookId = useId();
   const chatId = id || hookId;
-  const [initialMessages] = useState(initialMessagesParam != null ? initialMessagesParam : []);
+  const [initialMessagesFallback] = useState([]);
   const { data: messages, mutate } = useSWR([api, chatId], null, {
-    fallbackData: initialMessages
+    fallbackData: initialMessages != null ? initialMessages : initialMessagesFallback
   });
   const { data: isLoading = false, mutate: mutateLoading } = useSWR(
     [chatId, "loading"],