ai 2.1.8 → 2.1.9

package/dist/index.d.ts

@@ -89,6 +89,8 @@ declare function streamToResponse(res: ReadableStream, response: ServerResponse,
 
 declare function HuggingFaceStream(res: AsyncGenerator<any>, callbacks?: AIStreamCallbacks): ReadableStream;
 
+declare function CohereStream(res: AsyncGenerator<any>, callbacks?: AIStreamCallbacks): ReadableStream;
+
 declare function AnthropicStream(res: Response, cb?: AIStreamCallbacks): ReadableStream;
 
 declare function LangChainStream(callbacks?: AIStreamCallbacks): {
@@ -103,23 +105,23 @@ declare function LangChainStream(callbacks?: AIStreamCallbacks): {
 /**
  * Shared types between the API and UI packages.
  */
-declare type Message = {
+type Message = {
     id: string;
     createdAt?: Date;
     content: string;
     role: 'system' | 'user' | 'assistant';
 };
-declare type CreateMessage = {
+type CreateMessage = {
     id?: string;
     createdAt?: Date;
     content: string;
     role: 'system' | 'user' | 'assistant';
 };
-declare type RequestOptions = {
+type RequestOptions = {
     headers?: Record<string, string> | Headers;
     body?: object;
 };
-declare type UseChatOptions = {
+type UseChatOptions = {
     /**
      * The API endpoint that accepts a `{ messages: Message[] }` object and returns
      * a stream of tokens of the AI chat response. Defaults to `/api/chat`.
@@ -175,7 +177,7 @@ declare type UseChatOptions = {
      */
     sendExtraMessageFields?: boolean;
 };
-declare type UseCompletionOptions = {
+type UseCompletionOptions = {
     /**
      * The API endpoint that accepts a `{ prompt: string }` object and returns
      * a stream of tokens of the AI completion response. Defaults to `/api/completion`.
@@ -226,4 +228,4 @@ declare type UseCompletionOptions = {
     body?: object;
 };
 
-export { AIStream, AIStreamCallbacks, AIStreamParser, AnthropicStream, CreateMessage, HuggingFaceStream, LangChainStream, Message, OpenAIStream, RequestOptions, StreamingTextResponse, UseChatOptions, UseCompletionOptions, createCallbacksTransformer, createEventStreamTransformer, streamToResponse, trimStartOfStreamHelper };
+export { AIStream, AIStreamCallbacks, AIStreamParser, AnthropicStream, CohereStream, CreateMessage, HuggingFaceStream, LangChainStream, Message, OpenAIStream, RequestOptions, StreamingTextResponse, UseChatOptions, UseCompletionOptions, createCallbacksTransformer, createEventStreamTransformer, streamToResponse, trimStartOfStreamHelper };

package/dist/index.js

@@ -59,6 +59,7 @@ var streams_exports = {};
 __export(streams_exports, {
   AIStream: () => AIStream,
   AnthropicStream: () => AnthropicStream,
+  CohereStream: () => CohereStream,
   HuggingFaceStream: () => HuggingFaceStream,
   LangChainStream: () => LangChainStream,
   OpenAIStream: () => OpenAIStream,
@@ -231,6 +232,30 @@ function HuggingFaceStream(res, callbacks) {
   return createParser2(res).pipeThrough(createCallbacksTransformer(callbacks));
 }
 
+// streams/cohere-stream.ts
+function createParser3(res) {
+  return new ReadableStream({
+    pull(controller) {
+      return __async(this, null, function* () {
+        const { value, done } = yield res.next();
+        if (done) {
+          controller.close();
+          return;
+        }
+        const { text, is_finished } = JSON.parse(value);
+        if (is_finished === true) {
+          controller.close();
+        } else {
+          controller.enqueue(text);
+        }
+      });
+    }
+  });
+}
+function CohereStream(res, callbacks) {
+  return createParser3(res).pipeThrough(createCallbacksTransformer(callbacks));
+}
+
 // streams/anthropic-stream.ts
 function parseAnthropicStream() {
   let previous = "";
@@ -272,6 +297,7 @@ function LangChainStream(callbacks) {
 0 && (module.exports = {
   AIStream,
   AnthropicStream,
+  CohereStream,
   HuggingFaceStream,
   LangChainStream,
   OpenAIStream,

package/dist/index.mjs

@@ -201,6 +201,30 @@ function HuggingFaceStream(res, callbacks) {
   return createParser2(res).pipeThrough(createCallbacksTransformer(callbacks));
 }
 
+// streams/cohere-stream.ts
+function createParser3(res) {
+  return new ReadableStream({
+    pull(controller) {
+      return __async(this, null, function* () {
+        const { value, done } = yield res.next();
+        if (done) {
+          controller.close();
+          return;
+        }
+        const { text, is_finished } = JSON.parse(value);
+        if (is_finished === true) {
+          controller.close();
+        } else {
+          controller.enqueue(text);
+        }
+      });
+    }
+  });
+}
+function CohereStream(res, callbacks) {
+  return createParser3(res).pipeThrough(createCallbacksTransformer(callbacks));
+}
+
 // streams/anthropic-stream.ts
 function parseAnthropicStream() {
   let previous = "";
@@ -241,6 +265,7 @@ function LangChainStream(callbacks) {
 export {
   AIStream,
   AnthropicStream,
+  CohereStream,
   HuggingFaceStream,
   LangChainStream,
   OpenAIStream,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai",
-  "version": "2.1.8",
+  "version": "2.1.9",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -54,18 +54,18 @@
     "@edge-runtime/jest-environment": "1.1.0-beta.31",
     "@types/jest": "29.2.0",
     "@types/node": "^17.0.12",
-    "@types/react": "^18.2.0",
+    "@types/react": "^18.2.8",
     "@types/react-dom": "^18.2.0",
     "eslint": "^7.32.0",
     "jest": "29.2.1",
     "ts-jest": "29.0.3",
     "tsup": "^6.7.0",
-    "typescript": "^4.5.3",
+    "typescript": "5.1.3",
     "@vercel/ai-tsconfig": "0.0.0",
     "eslint-config-vercel-ai": "0.0.0"
   },
   "peerDependencies": {
-    "react": "^18.0.0",
+    "react": "^18.2.0",
     "svelte": "^3.29.0",
     "vue": "^3.3.4"
   },

package/react/dist/index.d.ts

@@ -1,23 +1,25 @@
+import * as react_jsx_runtime from 'react/jsx-runtime';
+
 /**
  * Shared types between the API and UI packages.
  */
-declare type Message = {
+type Message = {
     id: string;
     createdAt?: Date;
     content: string;
     role: 'system' | 'user' | 'assistant';
 };
-declare type CreateMessage = {
+type CreateMessage = {
     id?: string;
     createdAt?: Date;
     content: string;
     role: 'system' | 'user' | 'assistant';
 };
-declare type RequestOptions = {
+type RequestOptions = {
     headers?: Record<string, string> | Headers;
     body?: object;
 };
-declare type UseChatOptions = {
+type UseChatOptions = {
     /**
      * The API endpoint that accepts a `{ messages: Message[] }` object and returns
      * a stream of tokens of the AI chat response. Defaults to `/api/chat`.
@@ -73,7 +75,7 @@ declare type UseChatOptions = {
      */
     sendExtraMessageFields?: boolean;
 };
-declare type UseCompletionOptions = {
+type UseCompletionOptions = {
     /**
      * The API endpoint that accepts a `{ prompt: string }` object and returns
      * a stream of tokens of the AI completion response. Defaults to `/api/completion`.
@@ -124,7 +126,7 @@ declare type UseCompletionOptions = {
     body?: object;
 };
 
-declare type UseChatHelpers = {
+type UseChatHelpers = {
     /** Current messages in the chat */
     messages: Message[];
     /** The error object of the API request */
@@ -165,7 +167,7 @@ declare type UseChatHelpers = {
 };
 declare function useChat({ api, id, initialMessages, initialInput, sendExtraMessageFields, onResponse, onFinish, onError, headers, body }?: UseChatOptions): UseChatHelpers;
 
-declare type UseCompletionHelpers = {
+type UseCompletionHelpers = {
     /** The current completion result */
     completion: string;
     /**
@@ -209,4 +211,15 @@ declare type UseCompletionHelpers = {
 };
 declare function useCompletion({ api, id, initialCompletion, initialInput, headers, body, onResponse, onFinish, onError }?: UseCompletionOptions): UseCompletionHelpers;
 
-export { CreateMessage, Message, UseChatHelpers, UseChatOptions, UseCompletionHelpers, useChat, useCompletion };
+type Props = {
+    /**
+     * A ReadableStream produced by the AI SDK.
+     */
+    stream: ReadableStream;
+};
+/**
+ * A React Server Component that recursively renders a stream of tokens.
+ */
+declare function Tokens(props: Props): Promise<react_jsx_runtime.JSX.Element>;
+
+export { CreateMessage, Message, Tokens, UseChatHelpers, UseChatOptions, UseCompletionHelpers, useChat, useCompletion };

package/react/dist/index.js

@@ -65,6 +65,7 @@ var __async = (__this, __arguments, generator) => {
 // react/index.ts
 var react_exports = {};
 __export(react_exports, {
+  Tokens: () => Tokens,
   useChat: () => useChat,
   useCompletion: () => useCompletion
 });
@@ -444,8 +445,33 @@ function useCompletion({
     isLoading: isMutating
   };
 }
+
+// react/tokens.tsx
+var import_react3 = require("react");
+var import_jsx_runtime = require("react/jsx-runtime");
+function Tokens(props) {
+  return __async(this, null, function* () {
+    const { stream } = props;
+    const reader = stream.getReader();
+    return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(import_react3.Suspense, { children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(RecursiveTokens, { reader }) });
+  });
+}
+function RecursiveTokens(_0) {
+  return __async(this, arguments, function* ({ reader }) {
+    const { done, value } = yield reader.read();
+    if (done) {
+      return null;
+    }
+    const text = new TextDecoder().decode(value);
+    return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)(import_jsx_runtime.Fragment, { children: [
+      text,
+      /* @__PURE__ */ (0, import_jsx_runtime.jsx)(import_react3.Suspense, { fallback: null, children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(RecursiveTokens, { reader }) })
+    ] });
+  });
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  Tokens,
   useChat,
   useCompletion
 });

package/react/dist/index.mjs

@@ -410,7 +410,32 @@ function useCompletion({
     isLoading: isMutating
   };
 }
+
+// react/tokens.tsx
+import { Suspense } from "react";
+import { Fragment, jsx, jsxs } from "react/jsx-runtime";
+function Tokens(props) {
+  return __async(this, null, function* () {
+    const { stream } = props;
+    const reader = stream.getReader();
+    return /* @__PURE__ */ jsx(Suspense, { children: /* @__PURE__ */ jsx(RecursiveTokens, { reader }) });
+  });
+}
+function RecursiveTokens(_0) {
+  return __async(this, arguments, function* ({ reader }) {
+    const { done, value } = yield reader.read();
+    if (done) {
+      return null;
+    }
+    const text = new TextDecoder().decode(value);
+    return /* @__PURE__ */ jsxs(Fragment, { children: [
+      text,
+      /* @__PURE__ */ jsx(Suspense, { fallback: null, children: /* @__PURE__ */ jsx(RecursiveTokens, { reader }) })
+    ] });
+  });
+}
 export {
+  Tokens,
   useChat,
   useCompletion
 };

package/svelte/dist/index.d.ts

@@ -3,23 +3,23 @@ import { Readable, Writable } from 'svelte/store';
 /**
  * Shared types between the API and UI packages.
  */
-declare type Message = {
+type Message = {
     id: string;
     createdAt?: Date;
     content: string;
     role: 'system' | 'user' | 'assistant';
 };
-declare type CreateMessage = {
+type CreateMessage = {
     id?: string;
     createdAt?: Date;
     content: string;
     role: 'system' | 'user' | 'assistant';
 };
-declare type RequestOptions = {
+type RequestOptions = {
     headers?: Record<string, string> | Headers;
     body?: object;
 };
-declare type UseChatOptions = {
+type UseChatOptions = {
     /**
      * The API endpoint that accepts a `{ messages: Message[] }` object and returns
      * a stream of tokens of the AI chat response. Defaults to `/api/chat`.
@@ -75,7 +75,7 @@ declare type UseChatOptions = {
      */
     sendExtraMessageFields?: boolean;
 };
-declare type UseCompletionOptions = {
+type UseCompletionOptions = {
     /**
      * The API endpoint that accepts a `{ prompt: string }` object and returns
      * a stream of tokens of the AI completion response. Defaults to `/api/completion`.
@@ -126,7 +126,7 @@ declare type UseCompletionOptions = {
     body?: object;
 };
 
-declare type UseChatHelpers = {
+type UseChatHelpers = {
     /** Current messages in the chat */
     messages: Readable<Message[]>;
     /** The error object of the API request */
@@ -161,7 +161,7 @@ declare type UseChatHelpers = {
 };
 declare function useChat({ api, id, initialMessages, initialInput, sendExtraMessageFields, onResponse, onFinish, onError, headers, body }?: UseChatOptions): UseChatHelpers;
 
-declare type UseCompletionHelpers = {
+type UseCompletionHelpers = {
     /** The current completion result */
     completion: Readable<string>;
     /** The error object of the API request */

package/vue/dist/index.d.ts

@@ -3,23 +3,23 @@ import { Ref } from 'vue';
 /**
  * Shared types between the API and UI packages.
  */
-declare type Message = {
+type Message = {
     id: string;
     createdAt?: Date;
     content: string;
     role: 'system' | 'user' | 'assistant';
 };
-declare type CreateMessage = {
+type CreateMessage = {
     id?: string;
     createdAt?: Date;
     content: string;
     role: 'system' | 'user' | 'assistant';
 };
-declare type RequestOptions = {
+type RequestOptions = {
     headers?: Record<string, string> | Headers;
     body?: object;
 };
-declare type UseChatOptions = {
+type UseChatOptions = {
     /**
      * The API endpoint that accepts a `{ messages: Message[] }` object and returns
      * a stream of tokens of the AI chat response. Defaults to `/api/chat`.
@@ -75,7 +75,7 @@ declare type UseChatOptions = {
      */
     sendExtraMessageFields?: boolean;
 };
-declare type UseCompletionOptions = {
+type UseCompletionOptions = {
     /**
      * The API endpoint that accepts a `{ prompt: string }` object and returns
      * a stream of tokens of the AI completion response. Defaults to `/api/completion`.
@@ -126,7 +126,7 @@ declare type UseCompletionOptions = {
     body?: object;
 };
 
-declare type UseChatHelpers = {
+type UseChatHelpers = {
     /** Current messages in the chat */
     messages: Ref<Message[]>;
     /** The error object of the API request */
@@ -161,7 +161,7 @@ declare type UseChatHelpers = {
 };
 declare function useChat({ api, id, initialMessages, initialInput, sendExtraMessageFields, onResponse, onFinish, onError, headers, body }?: UseChatOptions): UseChatHelpers;
 
-declare type UseCompletionHelpers = {
+type UseCompletionHelpers = {
     /** The current completion result */
     completion: Ref<string>;
     /** The error object of the API request */