ai 2.2.14 → 2.2.16

package/dist/index.d.ts

@@ -284,6 +284,10 @@ interface Completion {
      * The object type, which is always "text_completion"
      */
     object: string;
+    /**
+     * Usage statistics for the completion request.
+     */
+    usage?: CompletionUsage;
 }
 interface CompletionChoice {
     /**
@@ -291,11 +295,28 @@ interface CompletionChoice {
      * hit a natural stop point or a provided stop sequence, or `length` if the maximum
      * number of tokens specified in the request was reached.
      */
-    finish_reason: 'stop' | 'length';
+    finish_reason: 'stop' | 'length' | 'content_filter';
     index: number;
     logprobs: any | null;
     text: string;
 }
+interface CompletionUsage {
+    /**
+     * Usage statistics for the completion request.
+     */
+    /**
+     * Number of tokens in the generated completion.
+     */
+    completion_tokens: number;
+    /**
+     * Number of tokens in the prompt.
+     */
+    prompt_tokens: number;
+    /**
+     * Total number of tokens used in the request (prompt + completion).
+     */
+    total_tokens: number;
+}
 type AsyncIterableOpenAIStreamReturnTypes = AsyncIterable<ChatCompletionChunk> | AsyncIterable<Completion>;
 declare function OpenAIStream(res: Response | AsyncIterableOpenAIStreamReturnTypes, callbacks?: OpenAIStreamCallbacks): ReadableStream;
 
@@ -499,7 +520,7 @@ interface Prediction {
     webhook?: string;
     webhook_events_filter?: ('start' | 'output' | 'logs' | 'completed')[];
     created_at: string;
-    updated_at: string;
+    updated_at?: string;
     completed_at?: string;
     urls: {
         get: string;
@@ -571,4 +592,4 @@ declare const getStreamStringTypeAndValue: (line: string) => {
  */
 declare const COMPLEX_HEADER = "X-Experimental-Stream-Data";
 
-export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AnthropicStream, COMPLEX_HEADER, ChatRequest, ChatRequestOptions, CohereStream, CreateMessage, FunctionCall, FunctionCallHandler, FunctionCallPayload, HuggingFaceStream, JSONValue, LangChainStream, Message, OpenAIStream, OpenAIStreamCallbacks, ReplicateStream, RequestOptions, StreamString, StreamStringPrefixes, StreamingTextResponse, UseChatOptions, UseCompletionOptions, createCallbacksTransformer, createChunkDecoder, createEventStreamTransformer, createStreamDataTransformer, experimental_StreamData, getStreamString, getStreamStringTypeAndValue, isStreamStringEqualToType, nanoid, readableFromAsyncIterable, streamToResponse, trimStartOfStreamHelper };
+export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AnthropicStream, COMPLEX_HEADER, ChatRequest, ChatRequestOptions, CohereStream, CompletionUsage, CreateMessage, FunctionCall, FunctionCallHandler, FunctionCallPayload, HuggingFaceStream, JSONValue, LangChainStream, Message, OpenAIStream, OpenAIStreamCallbacks, ReplicateStream, RequestOptions, StreamString, StreamStringPrefixes, StreamingTextResponse, UseChatOptions, UseCompletionOptions, createCallbacksTransformer, createChunkDecoder, createEventStreamTransformer, createStreamDataTransformer, experimental_StreamData, getStreamString, getStreamStringTypeAndValue, isStreamStringEqualToType, nanoid, readableFromAsyncIterable, streamToResponse, trimStartOfStreamHelper };

package/dist/index.js

@@ -84,10 +84,9 @@ function createCallbacksTransformer(cb) {
     },
     async transform(message, controller) {
       controller.enqueue(textEncoder.encode(message));
+      aggregatedResponse += message;
       if (callbacks.onToken)
         await callbacks.onToken(message);
-      if (callbacks.onCompletion)
-        aggregatedResponse += message;
     },
     async flush() {
       const isOpenAICallbacks = isOfTypeOpenAIStreamCallbacks(callbacks);

package/dist/index.mjs

@@ -39,10 +39,9 @@ function createCallbacksTransformer(cb) {
     },
     async transform(message, controller) {
       controller.enqueue(textEncoder.encode(message));
+      aggregatedResponse += message;
       if (callbacks.onToken)
         await callbacks.onToken(message);
-      if (callbacks.onCompletion)
-        aggregatedResponse += message;
     },
     async flush() {
       const isOpenAICallbacks = isOfTypeOpenAIStreamCallbacks(callbacks);

package/package.json

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