@anthropic-ai/sdk 0.34.0-alpha.0 → 0.36.2

package/src/lib/MessageStream.ts

@@ -1,4 +1,5 @@
-import { AnthropicError, APIUserAbortError } from '../error';
+import * as Core from "../core.js";
+import { AnthropicError, APIUserAbortError } from "../error.js";
 import {
   type ContentBlock,
   Messages,
@@ -8,11 +9,10 @@ import {
   type MessageCreateParams,
   type MessageCreateParamsBase,
   type TextBlock,
-} from '../resources/messages';
-import { Stream } from '../streaming';
-import { partialParse } from '../_vendor/partial-json-parser/parser';
-import { RequestOptions } from '../internal/request-options';
-import { type ReadableStream } from '../internal/shim-types';
+} from "../resources/messages.js";
+import { type ReadableStream, type Response } from "../_shims/index.js";
+import { Stream } from "../streaming.js";
+import { partialParse } from "../_vendor/partial-json-parser/parser.js";
 
 export interface MessageStreamEvents {
   connect: () => void;
@@ -41,8 +41,8 @@ export class MessageStream implements AsyncIterable<MessageStreamEvent> {
 
   controller: AbortController = new AbortController();
 
-  #connectedPromise: Promise<void>;
-  #resolveConnectedPromise: () => void = () => {};
+  #connectedPromise: Promise<Response | null>;
+  #resolveConnectedPromise: (response: Response | null) => void = () => {};
   #rejectConnectedPromise: (error: AnthropicError) => void = () => {};
 
   #endPromise: Promise<void>;
@@ -55,9 +55,11 @@ export class MessageStream implements AsyncIterable<MessageStreamEvent> {
   #errored = false;
   #aborted = false;
   #catchingPromiseCreated = false;
+  #response: Response | null | undefined;
+  #request_id: string | null | undefined;
 
   constructor() {
-    this.#connectedPromise = new Promise<void>((resolve, reject) => {
+    this.#connectedPromise = new Promise<Response | null>((resolve, reject) => {
       this.#resolveConnectedPromise = resolve;
       this.#rejectConnectedPromise = reject;
     });
@@ -75,6 +77,41 @@ export class MessageStream implements AsyncIterable<MessageStreamEvent> {
     this.#endPromise.catch(() => {});
   }
 
+  get response(): Response | null | undefined {
+    return this.#response;
+  }
+
+  get request_id(): string | null | undefined {
+    return this.#request_id;
+  }
+
+  /**
+   * Returns the `MessageStream` data, the raw `Response` instance and the ID of the request,
+   * returned vie the `request-id` header which is useful for debugging requests and resporting
+   * issues to Anthropic.
+   *
+   * This is the same as the `APIPromise.withResponse()` method.
+   *
+   * This method will raise an error if you created the stream using `MessageStream.fromReadableStream`
+   * as no `Response` is available.
+   */
+  async withResponse(): Promise<{
+    data: MessageStream;
+    response: Response;
+    request_id: string | null | undefined;
+  }> {
+    const response = await this.#connectedPromise;
+    if (!response) {
+      throw new Error('Could not resolve a `Response` object');
+    }
+
+    return {
+      data: this,
+      response,
+      request_id: response.headers.get('request-id'),
+    };
+  }
+
   /**
    * Intended for use on the frontend, consuming a stream produced with
    * `.toReadableStream()` on the backend.
@@ -91,7 +128,7 @@ export class MessageStream implements AsyncIterable<MessageStreamEvent> {
   static createMessage(
     messages: Messages,
     params: MessageCreateParamsBase,
-    options?: RequestOptions,
+    options?: Core.RequestOptions,
   ): MessageStream {
     const runner = new MessageStream();
     for (const message of params.messages) {
@@ -128,7 +165,7 @@ export class MessageStream implements AsyncIterable<MessageStreamEvent> {
   protected async _createMessage(
     messages: Messages,
     params: MessageCreateParams,
-    options?: RequestOptions,
+    options?: Core.RequestOptions,
   ): Promise<void> {
     const signal = options?.signal;
     if (signal) {
@@ -136,11 +173,10 @@ export class MessageStream implements AsyncIterable<MessageStreamEvent> {
       signal.addEventListener('abort', () => this.controller.abort());
     }
     this.#beginRequest();
-    const stream = await messages.create(
-      { ...params, stream: true },
-      { ...options, signal: this.controller.signal },
-    );
-    this._connected();
+    const { response, data: stream } = await messages
+      .create({ ...params, stream: true }, { ...options, signal: this.controller.signal })
+      .withResponse();
+    this._connected(response);
     for await (const event of stream) {
       this.#addStreamEvent(event);
     }
@@ -150,9 +186,11 @@ export class MessageStream implements AsyncIterable<MessageStreamEvent> {
     this.#endRequest();
   }
 
-  protected _connected() {
+  protected _connected(response: Response | null) {
     if (this.ended) return;
-    this.#resolveConnectedPromise();
+    this.#response = response;
+    this.#request_id = response?.headers.get('request-id');
+    this.#resolveConnectedPromise(response);
     this._emit('connect');
   }
 
@@ -416,7 +454,7 @@ export class MessageStream implements AsyncIterable<MessageStreamEvent> {
 
   protected async _fromReadableStream(
     readableStream: ReadableStream,
-    options?: RequestOptions,
+    options?: Core.RequestOptions,
   ): Promise<void> {
     const signal = options?.signal;
     if (signal) {
@@ -424,7 +462,7 @@ export class MessageStream implements AsyncIterable<MessageStreamEvent> {
       signal.addEventListener('abort', () => this.controller.abort());
     }
     this.#beginRequest();
-    this._connected();
+    this._connected(null);
     const stream = Stream.fromReadableStream<MessageStreamEvent>(readableStream, this.controller);
     for await (const event of stream) {
       this.#addStreamEvent(event);

package/src/pagination.ts

@@ -1,116 +1,6 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-import type { Anthropic } from './client';
-import { AnthropicError } from './error';
-import { FinalRequestOptions } from './internal/request-options';
-import { defaultParseResponse, type WithRequestID } from './internal/parse';
-import { APIPromise } from './api-promise';
-import { type APIResponseProps } from './internal/parse';
-import { maybeObj } from './internal/utils/values';
-
-export type PageRequestOptions = Pick<FinalRequestOptions, 'query' | 'headers' | 'body' | 'path' | 'method'>;
-
-export abstract class AbstractPage<Item> implements AsyncIterable<Item> {
-  #client: Anthropic;
-  protected options: FinalRequestOptions;
-
-  protected response: Response;
-  protected body: unknown;
-
-  constructor(client: Anthropic, response: Response, body: unknown, options: FinalRequestOptions) {
-    this.#client = client;
-    this.options = options;
-    this.response = response;
-    this.body = body;
-  }
-
-  abstract nextPageRequestOptions(): PageRequestOptions | null;
-
-  abstract getPaginatedItems(): Item[];
-
-  hasNextPage(): boolean {
-    const items = this.getPaginatedItems();
-    if (!items.length) return false;
-    return this.nextPageRequestOptions() != null;
-  }
-
-  async getNextPage(): Promise<this> {
-    const nextOptions = this.nextPageRequestOptions();
-    if (!nextOptions) {
-      throw new AnthropicError(
-        'No next page expected; please check `.hasNextPage()` before calling `.getNextPage()`.',
-      );
-    }
-
-    return await this.#client.requestAPIList(this.constructor as any, nextOptions);
-  }
-
-  async *iterPages(): AsyncGenerator<this> {
-    // eslint-disable-next-line @typescript-eslint/no-this-alias
-    let page: this = this;
-    yield page;
-    while (page.hasNextPage()) {
-      page = await page.getNextPage();
-      yield page;
-    }
-  }
-
-  async *[Symbol.asyncIterator](): AsyncGenerator<Item> {
-    for await (const page of this.iterPages()) {
-      for (const item of page.getPaginatedItems()) {
-        yield item;
-      }
-    }
-  }
-}
-
-/**
- * This subclass of Promise will resolve to an instantiated Page once the request completes.
- *
- * It also implements AsyncIterable to allow auto-paginating iteration on an unawaited list call, eg:
- *
- *    for await (const item of client.items.list()) {
- *      console.log(item)
- *    }
- */
-export class PagePromise<
-    PageClass extends AbstractPage<Item>,
-    Item = ReturnType<PageClass['getPaginatedItems']>[number],
-  >
-  extends APIPromise<PageClass>
-  implements AsyncIterable<Item>
-{
-  constructor(
-    client: Anthropic,
-    request: Promise<APIResponseProps>,
-    Page: new (...args: ConstructorParameters<typeof AbstractPage>) => PageClass,
-  ) {
-    super(
-      request,
-      async (props) =>
-        new Page(
-          client,
-          props.response,
-          await defaultParseResponse(props),
-          props.options,
-        ) as WithRequestID<PageClass>,
-    );
-  }
-
-  /**
-   * Allow auto-paginating iteration on an unawaited list call, eg:
-   *
-   *    for await (const item of client.items.list()) {
-   *      console.log(item)
-   *    }
-   */
-  async *[Symbol.asyncIterator]() {
-    const page = await this;
-    for await (const item of page) {
-      yield item;
-    }
-  }
-}
+import { AbstractPage, Response, APIClient, FinalRequestOptions, PageInfo } from "./core.js";
 
 export interface PageResponse<Item> {
   data: Array<Item>;
@@ -142,7 +32,7 @@ export class Page<Item> extends AbstractPage<Item> implements PageResponse<Item>
 
   last_id: string | null;
 
-  constructor(client: Anthropic, response: Response, body: PageResponse<Item>, options: FinalRequestOptions) {
+  constructor(client: APIClient, response: Response, body: PageResponse<Item>, options: FinalRequestOptions) {
     super(client, response, body, options);
 
     this.data = body.data || [];
@@ -155,19 +45,27 @@ export class Page<Item> extends AbstractPage<Item> implements PageResponse<Item>
     return this.data ?? [];
   }
 
-  nextPageRequestOptions(): PageRequestOptions | null {
+  // @deprecated Please use `nextPageInfo()` instead
+  nextPageParams(): Partial<PageParams> | null {
+    const info = this.nextPageInfo();
+    if (!info) return null;
+    if ('params' in info) return info.params;
+    const params = Object.fromEntries(info.url.searchParams);
+    if (!Object.keys(params).length) return null;
+    return params;
+  }
+
+  nextPageInfo(): PageInfo | null {
     if ((this.options.query as Record<string, unknown>)?.['before_id']) {
       // in reverse
-      const first_id = this.first_id;
-      if (!first_id) {
+      const firstId = this.first_id;
+      if (!firstId) {
         return null;
       }
 
       return {
-        ...this.options,
-        query: {
-          ...maybeObj(this.options.query),
-          before_id: first_id,
+        params: {
+          before_id: firstId,
         },
       };
     }
@@ -178,9 +76,7 @@ export class Page<Item> extends AbstractPage<Item> implements PageResponse<Item>
     }
 
     return {
-      ...this.options,
-      query: {
-        ...maybeObj(this.options.query),
+      params: {
         after_id: cursor,
       },
     };

package/src/resource.ts

@@ -1,11 +1,11 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-import { BaseAnthropic } from './client';
+import * as Core from "./core.js";
 
 export class APIResource {
-  protected _client: BaseAnthropic;
+  protected _client: Core.APIClient;
 
-  constructor(client: BaseAnthropic) {
+  constructor(client: Core.APIClient) {
     this._client = client;
   }
 }

package/src/resources/beta/beta.ts

@@ -1,15 +1,25 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-import { APIResource } from '../../resource';
-import * as ModelsAPI from './models';
-import { BetaModelInfo, BetaModelInfosPage, ModelListParams, Models } from './models';
-import * as MessagesAPI from './messages/messages';
+import { APIResource } from "../../resource.js";
+import * as ModelsAPI from "./models.js";
+import { BetaModelInfo, BetaModelInfosPage, ModelListParams, Models } from "./models.js";
+import * as MessagesAPI from "./messages/messages.js";
 import {
   BetaBase64PDFBlock,
   BetaBase64PDFSource,
   BetaCacheControlEphemeral,
+  BetaCitationCharLocation,
+  BetaCitationCharLocationParam,
+  BetaCitationContentBlockLocation,
+  BetaCitationContentBlockLocationParam,
+  BetaCitationPageLocation,
+  BetaCitationPageLocationParam,
+  BetaCitationsConfigParam,
+  BetaCitationsDelta,
   BetaContentBlock,
   BetaContentBlockParam,
+  BetaContentBlockSource,
+  BetaContentBlockSourceContent,
   BetaImageBlockParam,
   BetaInputJSONDelta,
   BetaMessage,
@@ -17,6 +27,7 @@ import {
   BetaMessageParam,
   BetaMessageTokensCount,
   BetaMetadata,
+  BetaPlainTextSource,
   BetaRawContentBlockDeltaEvent,
   BetaRawContentBlockStartEvent,
   BetaRawContentBlockStopEvent,
@@ -26,6 +37,8 @@ import {
   BetaRawMessageStreamEvent,
   BetaTextBlock,
   BetaTextBlockParam,
+  BetaTextCitation,
+  BetaTextCitationParam,
   BetaTextDelta,
   BetaTool,
   BetaToolBash20241022,
@@ -45,7 +58,7 @@ import {
   MessageCreateParamsNonStreaming,
   MessageCreateParamsStreaming,
   Messages,
-} from './messages/messages';
+} from "./messages/messages.js";
 
 export class Beta extends APIResource {
   models: ModelsAPI.Models = new ModelsAPI.Models(this._client);
@@ -132,6 +145,7 @@ export interface BetaRateLimitError {
 }
 
 Beta.Models = Models;
+Beta.BetaModelInfosPage = BetaModelInfosPage;
 Beta.Messages = Messages;
 
 export declare namespace Beta {
@@ -153,7 +167,7 @@ export declare namespace Beta {
   export {
     Models as Models,
     type BetaModelInfo as BetaModelInfo,
-    type BetaModelInfosPage as BetaModelInfosPage,
+    BetaModelInfosPage as BetaModelInfosPage,
     type ModelListParams as ModelListParams,
   };
 
@@ -162,8 +176,18 @@ export declare namespace Beta {
     type BetaBase64PDFBlock as BetaBase64PDFBlock,
     type BetaBase64PDFSource as BetaBase64PDFSource,
     type BetaCacheControlEphemeral as BetaCacheControlEphemeral,
+    type BetaCitationCharLocation as BetaCitationCharLocation,
+    type BetaCitationCharLocationParam as BetaCitationCharLocationParam,
+    type BetaCitationContentBlockLocation as BetaCitationContentBlockLocation,
+    type BetaCitationContentBlockLocationParam as BetaCitationContentBlockLocationParam,
+    type BetaCitationPageLocation as BetaCitationPageLocation,
+    type BetaCitationPageLocationParam as BetaCitationPageLocationParam,
+    type BetaCitationsConfigParam as BetaCitationsConfigParam,
+    type BetaCitationsDelta as BetaCitationsDelta,
     type BetaContentBlock as BetaContentBlock,
     type BetaContentBlockParam as BetaContentBlockParam,
+    type BetaContentBlockSource as BetaContentBlockSource,
+    type BetaContentBlockSourceContent as BetaContentBlockSourceContent,
     type BetaImageBlockParam as BetaImageBlockParam,
     type BetaInputJSONDelta as BetaInputJSONDelta,
     type BetaMessage as BetaMessage,
@@ -171,6 +195,7 @@ export declare namespace Beta {
     type BetaMessageParam as BetaMessageParam,
     type BetaMessageTokensCount as BetaMessageTokensCount,
     type BetaMetadata as BetaMetadata,
+    type BetaPlainTextSource as BetaPlainTextSource,
     type BetaRawContentBlockDeltaEvent as BetaRawContentBlockDeltaEvent,
     type BetaRawContentBlockStartEvent as BetaRawContentBlockStartEvent,
     type BetaRawContentBlockStopEvent as BetaRawContentBlockStopEvent,
@@ -180,6 +205,8 @@ export declare namespace Beta {
     type BetaRawMessageStreamEvent as BetaRawMessageStreamEvent,
     type BetaTextBlock as BetaTextBlock,
     type BetaTextBlockParam as BetaTextBlockParam,
+    type BetaTextCitation as BetaTextCitation,
+    type BetaTextCitationParam as BetaTextCitationParam,
     type BetaTextDelta as BetaTextDelta,
     type BetaTool as BetaTool,
     type BetaToolBash20241022 as BetaToolBash20241022,

package/src/resources/beta/index.ts

@@ -14,14 +14,25 @@ export {
   type BetaOverloadedError,
   type BetaPermissionError,
   type BetaRateLimitError,
-} from './beta';
+} from "./beta.js";
+export { BetaModelInfosPage, Models, type BetaModelInfo, type ModelListParams } from "./models.js";
 export {
   Messages,
   type BetaBase64PDFBlock,
   type BetaBase64PDFSource,
   type BetaCacheControlEphemeral,
+  type BetaCitationCharLocation,
+  type BetaCitationCharLocationParam,
+  type BetaCitationContentBlockLocation,
+  type BetaCitationContentBlockLocationParam,
+  type BetaCitationPageLocation,
+  type BetaCitationPageLocationParam,
+  type BetaCitationsConfigParam,
+  type BetaCitationsDelta,
   type BetaContentBlock,
   type BetaContentBlockParam,
+  type BetaContentBlockSource,
+  type BetaContentBlockSourceContent,
   type BetaImageBlockParam,
   type BetaInputJSONDelta,
   type BetaMessage,
@@ -29,6 +40,7 @@ export {
   type BetaMessageParam,
   type BetaMessageTokensCount,
   type BetaMetadata,
+  type BetaPlainTextSource,
   type BetaRawContentBlockDeltaEvent,
   type BetaRawContentBlockStartEvent,
   type BetaRawContentBlockStopEvent,
@@ -38,6 +50,8 @@ export {
   type BetaRawMessageStreamEvent,
   type BetaTextBlock,
   type BetaTextBlockParam,
+  type BetaTextCitation,
+  type BetaTextCitationParam,
   type BetaTextDelta,
   type BetaTool,
   type BetaToolBash20241022,
@@ -56,5 +70,4 @@ export {
   type MessageCreateParamsNonStreaming,
   type MessageCreateParamsStreaming,
   type MessageCountTokensParams,
-} from './messages/index';
-export { Models, type BetaModelInfo, type ModelListParams, type BetaModelInfosPage } from './models';
+} from "./messages/index.js";