@anthropic-ai/sdk 0.25.2 → 0.26.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (65) hide show
  1. package/CHANGELOG.md +23 -0
  2. package/index.d.mts +2 -0
  3. package/index.d.ts +2 -0
  4. package/index.d.ts.map +1 -1
  5. package/index.js +2 -0
  6. package/index.js.map +1 -1
  7. package/index.mjs +2 -0
  8. package/index.mjs.map +1 -1
  9. package/lib/PromptCachingBetaMessageStream.d.ts +96 -0
  10. package/lib/PromptCachingBetaMessageStream.d.ts.map +1 -0
  11. package/lib/PromptCachingBetaMessageStream.js +459 -0
  12. package/lib/PromptCachingBetaMessageStream.js.map +1 -0
  13. package/lib/PromptCachingBetaMessageStream.mjs +455 -0
  14. package/lib/PromptCachingBetaMessageStream.mjs.map +1 -0
  15. package/package.json +1 -1
  16. package/resources/beta/beta.d.ts +9 -0
  17. package/resources/beta/beta.d.ts.map +1 -0
  18. package/resources/beta/beta.js +40 -0
  19. package/resources/beta/beta.js.map +1 -0
  20. package/resources/beta/beta.mjs +13 -0
  21. package/resources/beta/beta.mjs.map +1 -0
  22. package/resources/beta/index.d.ts +3 -0
  23. package/resources/beta/index.d.ts.map +1 -0
  24. package/resources/beta/index.js +9 -0
  25. package/resources/beta/index.js.map +1 -0
  26. package/resources/beta/index.mjs +4 -0
  27. package/resources/beta/index.mjs.map +1 -0
  28. package/resources/beta/prompt-caching/index.d.ts +3 -0
  29. package/resources/beta/prompt-caching/index.d.ts.map +1 -0
  30. package/resources/beta/prompt-caching/index.js +9 -0
  31. package/resources/beta/prompt-caching/index.js.map +1 -0
  32. package/resources/beta/prompt-caching/index.mjs +4 -0
  33. package/resources/beta/prompt-caching/index.mjs.map +1 -0
  34. package/resources/beta/prompt-caching/messages.d.ts +538 -0
  35. package/resources/beta/prompt-caching/messages.d.ts.map +1 -0
  36. package/resources/beta/prompt-caching/messages.js +27 -0
  37. package/resources/beta/prompt-caching/messages.js.map +1 -0
  38. package/resources/beta/prompt-caching/messages.mjs +23 -0
  39. package/resources/beta/prompt-caching/messages.mjs.map +1 -0
  40. package/resources/beta/prompt-caching/prompt-caching.d.ts +23 -0
  41. package/resources/beta/prompt-caching/prompt-caching.d.ts.map +1 -0
  42. package/resources/beta/prompt-caching/prompt-caching.js +40 -0
  43. package/resources/beta/prompt-caching/prompt-caching.js.map +1 -0
  44. package/resources/beta/prompt-caching/prompt-caching.mjs +13 -0
  45. package/resources/beta/prompt-caching/prompt-caching.mjs.map +1 -0
  46. package/resources/index.d.ts +1 -0
  47. package/resources/index.d.ts.map +1 -1
  48. package/resources/index.js +3 -1
  49. package/resources/index.js.map +1 -1
  50. package/resources/index.mjs +1 -0
  51. package/resources/index.mjs.map +1 -1
  52. package/resources/messages.d.ts +2 -2
  53. package/src/index.ts +3 -0
  54. package/src/lib/PromptCachingBetaMessageStream.ts +579 -0
  55. package/src/resources/beta/beta.ts +12 -0
  56. package/src/resources/beta/index.ts +4 -0
  57. package/src/resources/beta/prompt-caching/index.ts +20 -0
  58. package/src/resources/beta/prompt-caching/messages.ts +646 -0
  59. package/src/resources/beta/prompt-caching/prompt-caching.ts +26 -0
  60. package/src/resources/index.ts +1 -0
  61. package/src/resources/messages.ts +2 -2
  62. package/src/version.ts +1 -1
  63. package/version.d.ts +1 -1
  64. package/version.js +1 -1
  65. package/version.mjs +1 -1
@@ -0,0 +1 @@
1
+ {"version":3,"file":"prompt-caching.js","sourceRoot":"","sources":["../../../src/resources/beta/prompt-caching/prompt-caching.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;;;;;;;;;;;;;;;;;;;;;;;;AAEtF,mDAAgD;AAChD,2DAA0C;AAE1C,MAAa,aAAc,SAAQ,sBAAW;IAA9C;;QACE,aAAQ,GAAyB,IAAI,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1E,CAAC;CAAA;AAFD,sCAEC;AAED,WAAiB,aAAa;IACd,sBAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC;AAehD,CAAC,EAhBgB,aAAa,GAAb,qBAAa,KAAb,qBAAa,QAgB7B"}
@@ -0,0 +1,13 @@
1
+ // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+ import { APIResource } from "../../../resource.mjs";
3
+ import * as MessagesAPI from "./messages.mjs";
4
+ export class PromptCaching extends APIResource {
5
+ constructor() {
6
+ super(...arguments);
7
+ this.messages = new MessagesAPI.Messages(this._client);
8
+ }
9
+ }
10
+ (function (PromptCaching) {
11
+ PromptCaching.Messages = MessagesAPI.Messages;
12
+ })(PromptCaching || (PromptCaching = {}));
13
+ //# sourceMappingURL=prompt-caching.mjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"prompt-caching.mjs","sourceRoot":"","sources":["../../../src/resources/beta/prompt-caching/prompt-caching.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;OACf,KAAK,WAAW;AAEvB,MAAM,OAAO,aAAc,SAAQ,WAAW;IAA9C;;QACE,aAAQ,GAAyB,IAAI,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1E,CAAC;CAAA;AAED,WAAiB,aAAa;IACd,sBAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC;AAehD,CAAC,EAhBgB,aAAa,KAAb,aAAa,QAgB7B"}
@@ -1,3 +1,4 @@
1
+ export { Beta } from "./beta/beta.js";
1
2
  export { Completion, CompletionCreateParams, CompletionCreateParamsNonStreaming, CompletionCreateParamsStreaming, Completions, } from "./completions.js";
2
3
  export { ContentBlock, ContentBlockDeltaEvent, ContentBlockStartEvent, ContentBlockStopEvent, ImageBlockParam, InputJsonDelta, InputJSONDelta, Message, MessageDeltaEvent, MessageDeltaUsage, MessageParam, MessageStartEvent, MessageStopEvent, MessageStreamEvent, Model, RawContentBlockDeltaEvent, RawContentBlockStartEvent, RawContentBlockStopEvent, RawMessageDeltaEvent, RawMessageStartEvent, RawMessageStopEvent, RawMessageStreamEvent, TextBlock, TextBlockParam, TextDelta, Tool, ToolResultBlockParam, ToolUseBlock, ToolUseBlockParam, Usage, MessageCreateParams, MessageCreateParamsNonStreaming, MessageCreateParamsStreaming, MessageStreamParams, Messages, } from "./messages.js";
3
4
  //# sourceMappingURL=index.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,UAAU,EACV,sBAAsB,EACtB,kCAAkC,EAClC,+BAA+B,EAC/B,WAAW,GACZ,MAAM,eAAe,CAAC;AACvB,OAAO,EACL,YAAY,EACZ,sBAAsB,EACtB,sBAAsB,EACtB,qBAAqB,EACrB,eAAe,EACf,cAAc,EACd,cAAc,EACd,OAAO,EACP,iBAAiB,EACjB,iBAAiB,EACjB,YAAY,EACZ,iBAAiB,EACjB,gBAAgB,EAChB,kBAAkB,EAClB,KAAK,EACL,yBAAyB,EACzB,yBAAyB,EACzB,wBAAwB,EACxB,oBAAoB,EACpB,oBAAoB,EACpB,mBAAmB,EACnB,qBAAqB,EACrB,SAAS,EACT,cAAc,EACd,SAAS,EACT,IAAI,EACJ,oBAAoB,EACpB,YAAY,EACZ,iBAAiB,EACjB,KAAK,EACL,mBAAmB,EACnB,+BAA+B,EAC/B,4BAA4B,EAC5B,mBAAmB,EACnB,QAAQ,GACT,MAAM,YAAY,CAAC"}
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AACnC,OAAO,EACL,UAAU,EACV,sBAAsB,EACtB,kCAAkC,EAClC,+BAA+B,EAC/B,WAAW,GACZ,MAAM,eAAe,CAAC;AACvB,OAAO,EACL,YAAY,EACZ,sBAAsB,EACtB,sBAAsB,EACtB,qBAAqB,EACrB,eAAe,EACf,cAAc,EACd,cAAc,EACd,OAAO,EACP,iBAAiB,EACjB,iBAAiB,EACjB,YAAY,EACZ,iBAAiB,EACjB,gBAAgB,EAChB,kBAAkB,EAClB,KAAK,EACL,yBAAyB,EACzB,yBAAyB,EACzB,wBAAwB,EACxB,oBAAoB,EACpB,oBAAoB,EACpB,mBAAmB,EACnB,qBAAqB,EACrB,SAAS,EACT,cAAc,EACd,SAAS,EACT,IAAI,EACJ,oBAAoB,EACpB,YAAY,EACZ,iBAAiB,EACjB,KAAK,EACL,mBAAmB,EACnB,+BAA+B,EAC/B,4BAA4B,EAC5B,mBAAmB,EACnB,QAAQ,GACT,MAAM,YAAY,CAAC"}
@@ -1,7 +1,9 @@
1
1
  "use strict";
2
2
  // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
3
3
  Object.defineProperty(exports, "__esModule", { value: true });
4
- exports.Messages = exports.Completions = void 0;
4
+ exports.Messages = exports.Completions = exports.Beta = void 0;
5
+ var beta_1 = require("./beta/beta.js");
6
+ Object.defineProperty(exports, "Beta", { enumerable: true, get: function () { return beta_1.Beta; } });
5
7
  var completions_1 = require("./completions.js");
6
8
  Object.defineProperty(exports, "Completions", { enumerable: true, get: function () { return completions_1.Completions; } });
7
9
  var messages_1 = require("./messages.js");
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,gDAMuB;AADrB,0GAAA,WAAW,OAAA;AAEb,0CAoCoB;AADlB,oGAAA,QAAQ,OAAA"}
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,uCAAmC;AAA1B,4FAAA,IAAI,OAAA;AACb,gDAMuB;AADrB,0GAAA,WAAW,OAAA;AAEb,0CAoCoB;AADlB,oGAAA,QAAQ,OAAA"}
@@ -1,4 +1,5 @@
1
1
  // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+ export { Beta } from "./beta/beta.mjs";
2
3
  export { Completions, } from "./completions.mjs";
3
4
  export { Messages, } from "./messages.mjs";
4
5
  //# sourceMappingURL=index.mjs.map
@@ -1 +1 @@
1
- {"version":3,"file":"index.mjs","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAKL,WAAW,GACZ;OACM,EAmCL,QAAQ,GACT"}
1
+ {"version":3,"file":"index.mjs","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,IAAI,EAAE;OACR,EAKL,WAAW,GACZ;OACM,EAmCL,QAAQ,GACT"}
@@ -370,8 +370,8 @@ export interface MessageCreateParamsBase {
370
370
  * We currently support the `base64` source type for images, and the `image/jpeg`,
371
371
  * `image/png`, `image/gif`, and `image/webp` media types.
372
372
  *
373
- * See [examples](https://docs.anthropic.com/en/api/messages-examples) for more
374
- * input examples.
373
+ * See [examples](https://docs.anthropic.com/en/api/messages-examples#vision) for
374
+ * more input examples.
375
375
  *
376
376
  * Note that if you want to include a
377
377
  * [system prompt](https://docs.anthropic.com/en/docs/system-prompts), you can use
package/src/index.ts CHANGED
@@ -131,6 +131,7 @@ export class Anthropic extends Core.APIClient {
131
131
 
132
132
  completions: API.Completions = new API.Completions(this);
133
133
  messages: API.Messages = new API.Messages(this);
134
+ beta: API.Beta = new API.Beta(this);
134
135
 
135
136
  protected override defaultQuery(): Core.DefaultQuery | undefined {
136
137
  return this._options.defaultQuery;
@@ -279,6 +280,8 @@ export namespace Anthropic {
279
280
  export import MessageCreateParamsNonStreaming = API.MessageCreateParamsNonStreaming;
280
281
  export import MessageCreateParamsStreaming = API.MessageCreateParamsStreaming;
281
282
  export import MessageStreamParams = API.MessageStreamParams;
283
+
284
+ export import Beta = API.Beta;
282
285
  }
283
286
 
284
287
  export default Anthropic;
@@ -0,0 +1,579 @@
1
+ import * as Core from "../core.js";
2
+ import { AnthropicError, APIUserAbortError } from "../error.js";
3
+ import { type ContentBlock, type TextBlock } from "../resources/messages.js";
4
+ import {
5
+ Messages,
6
+ type PromptCachingBetaMessage,
7
+ type RawPromptCachingBetaMessageStreamEvent,
8
+ type PromptCachingBetaMessageParam,
9
+ type MessageCreateParams,
10
+ type MessageCreateParamsBase,
11
+ } from "../resources/beta/prompt-caching/messages.js";
12
+ import { type ReadableStream } from "../_shims/index.js";
13
+ import { Stream } from "../streaming.js";
14
+ import { partialParse } from "../_vendor/partial-json-parser/parser.js";
15
+
16
+ export interface PromptCachingBetaMessageStreamEvents {
17
+ connect: () => void;
18
+ streamEvent: (event: RawPromptCachingBetaMessageStreamEvent, snapshot: PromptCachingBetaMessage) => void;
19
+ text: (textDelta: string, textSnapshot: string) => void;
20
+ inputJson: (partialJson: string, jsonSnapshot: unknown) => void;
21
+ message: (message: PromptCachingBetaMessage) => void;
22
+ contentBlock: (content: ContentBlock) => void;
23
+ finalPromptCachingBetaMessage: (message: PromptCachingBetaMessage) => void;
24
+ error: (error: AnthropicError) => void;
25
+ abort: (error: APIUserAbortError) => void;
26
+ end: () => void;
27
+ }
28
+
29
+ type PromptCachingBetaMessageStreamEventListeners<Event extends keyof PromptCachingBetaMessageStreamEvents> =
30
+ {
31
+ listener: PromptCachingBetaMessageStreamEvents[Event];
32
+ once?: boolean;
33
+ }[];
34
+
35
+ const JSON_BUF_PROPERTY = '__json_buf';
36
+
37
+ export class PromptCachingBetaMessageStream implements AsyncIterable<RawPromptCachingBetaMessageStreamEvent> {
38
+ messages: PromptCachingBetaMessageParam[] = [];
39
+ receivedMessages: PromptCachingBetaMessage[] = [];
40
+ #currentMessageSnapshot: PromptCachingBetaMessage | undefined;
41
+
42
+ controller: AbortController = new AbortController();
43
+
44
+ #connectedPromise: Promise<void>;
45
+ #resolveConnectedPromise: () => void = () => {};
46
+ #rejectConnectedPromise: (error: AnthropicError) => void = () => {};
47
+
48
+ #endPromise: Promise<void>;
49
+ #resolveEndPromise: () => void = () => {};
50
+ #rejectEndPromise: (error: AnthropicError) => void = () => {};
51
+
52
+ #listeners: {
53
+ [Event in keyof PromptCachingBetaMessageStreamEvents]?: PromptCachingBetaMessageStreamEventListeners<Event>;
54
+ } = {};
55
+
56
+ #ended = false;
57
+ #errored = false;
58
+ #aborted = false;
59
+ #catchingPromiseCreated = false;
60
+
61
+ constructor() {
62
+ this.#connectedPromise = new Promise<void>((resolve, reject) => {
63
+ this.#resolveConnectedPromise = resolve;
64
+ this.#rejectConnectedPromise = reject;
65
+ });
66
+
67
+ this.#endPromise = new Promise<void>((resolve, reject) => {
68
+ this.#resolveEndPromise = resolve;
69
+ this.#rejectEndPromise = reject;
70
+ });
71
+
72
+ // Don't let these promises cause unhandled rejection errors.
73
+ // we will manually cause an unhandled rejection error later
74
+ // if the user hasn't registered any error listener or called
75
+ // any promise-returning method.
76
+ this.#connectedPromise.catch(() => {});
77
+ this.#endPromise.catch(() => {});
78
+ }
79
+
80
+ /**
81
+ * Intended for use on the frontend, consuming a stream produced with
82
+ * `.toReadableStream()` on the backend.
83
+ *
84
+ * Note that messages sent to the model do not appear in `.on('message')`
85
+ * in this context.
86
+ */
87
+ static fromReadableStream(stream: ReadableStream): PromptCachingBetaMessageStream {
88
+ const runner = new PromptCachingBetaMessageStream();
89
+ runner._run(() => runner._fromReadableStream(stream));
90
+ return runner;
91
+ }
92
+
93
+ static createMessage(
94
+ messages: Messages,
95
+ params: MessageCreateParamsBase,
96
+ options?: Core.RequestOptions,
97
+ ): PromptCachingBetaMessageStream {
98
+ const runner = new PromptCachingBetaMessageStream();
99
+ for (const message of params.messages) {
100
+ runner._addPromptCachingBetaMessageParam(message);
101
+ }
102
+ runner._run(() =>
103
+ runner._createPromptCachingBetaMessage(
104
+ messages,
105
+ { ...params, stream: true },
106
+ { ...options, headers: { ...options?.headers, 'X-Stainless-Helper-Method': 'stream' } },
107
+ ),
108
+ );
109
+ return runner;
110
+ }
111
+
112
+ protected _run(executor: () => Promise<any>) {
113
+ executor().then(() => {
114
+ this._emitFinal();
115
+ this._emit('end');
116
+ }, this.#handleError);
117
+ }
118
+
119
+ protected _addPromptCachingBetaMessageParam(message: PromptCachingBetaMessageParam) {
120
+ this.messages.push(message);
121
+ }
122
+
123
+ protected _addPromptCachingBetaMessage(message: PromptCachingBetaMessage, emit = true) {
124
+ this.receivedMessages.push(message);
125
+ if (emit) {
126
+ this._emit('message', message);
127
+ }
128
+ }
129
+
130
+ protected async _createPromptCachingBetaMessage(
131
+ messages: Messages,
132
+ params: MessageCreateParams,
133
+ options?: Core.RequestOptions,
134
+ ): Promise<void> {
135
+ const signal = options?.signal;
136
+ if (signal) {
137
+ if (signal.aborted) this.controller.abort();
138
+ signal.addEventListener('abort', () => this.controller.abort());
139
+ }
140
+ this.#beginRequest();
141
+ const stream = await messages.create(
142
+ { ...params, stream: true },
143
+ { ...options, signal: this.controller.signal },
144
+ );
145
+ this._connected();
146
+ for await (const event of stream) {
147
+ this.#addStreamEvent(event);
148
+ }
149
+ if (stream.controller.signal?.aborted) {
150
+ throw new APIUserAbortError();
151
+ }
152
+ this.#endRequest();
153
+ }
154
+
155
+ protected _connected() {
156
+ if (this.ended) return;
157
+ this.#resolveConnectedPromise();
158
+ this._emit('connect');
159
+ }
160
+
161
+ get ended(): boolean {
162
+ return this.#ended;
163
+ }
164
+
165
+ get errored(): boolean {
166
+ return this.#errored;
167
+ }
168
+
169
+ get aborted(): boolean {
170
+ return this.#aborted;
171
+ }
172
+
173
+ abort() {
174
+ this.controller.abort();
175
+ }
176
+
177
+ /**
178
+ * Adds the listener function to the end of the listeners array for the event.
179
+ * No checks are made to see if the listener has already been added. Multiple calls passing
180
+ * the same combination of event and listener will result in the listener being added, and
181
+ * called, multiple times.
182
+ * @returns this PromptCachingBetaMessageStream, so that calls can be chained
183
+ */
184
+ on<Event extends keyof PromptCachingBetaMessageStreamEvents>(
185
+ event: Event,
186
+ listener: PromptCachingBetaMessageStreamEvents[Event],
187
+ ): this {
188
+ const listeners: PromptCachingBetaMessageStreamEventListeners<Event> =
189
+ this.#listeners[event] || (this.#listeners[event] = []);
190
+ listeners.push({ listener });
191
+ return this;
192
+ }
193
+
194
+ /**
195
+ * Removes the specified listener from the listener array for the event.
196
+ * off() will remove, at most, one instance of a listener from the listener array. If any single
197
+ * listener has been added multiple times to the listener array for the specified event, then
198
+ * off() must be called multiple times to remove each instance.
199
+ * @returns this PromptCachingBetaMessageStream, so that calls can be chained
200
+ */
201
+ off<Event extends keyof PromptCachingBetaMessageStreamEvents>(
202
+ event: Event,
203
+ listener: PromptCachingBetaMessageStreamEvents[Event],
204
+ ): this {
205
+ const listeners = this.#listeners[event];
206
+ if (!listeners) return this;
207
+ const index = listeners.findIndex((l) => l.listener === listener);
208
+ if (index >= 0) listeners.splice(index, 1);
209
+ return this;
210
+ }
211
+
212
+ /**
213
+ * Adds a one-time listener function for the event. The next time the event is triggered,
214
+ * this listener is removed and then invoked.
215
+ * @returns this PromptCachingBetaMessageStream, so that calls can be chained
216
+ */
217
+ once<Event extends keyof PromptCachingBetaMessageStreamEvents>(
218
+ event: Event,
219
+ listener: PromptCachingBetaMessageStreamEvents[Event],
220
+ ): this {
221
+ const listeners: PromptCachingBetaMessageStreamEventListeners<Event> =
222
+ this.#listeners[event] || (this.#listeners[event] = []);
223
+ listeners.push({ listener, once: true });
224
+ return this;
225
+ }
226
+
227
+ /**
228
+ * This is similar to `.once()`, but returns a Promise that resolves the next time
229
+ * the event is triggered, instead of calling a listener callback.
230
+ * @returns a Promise that resolves the next time given event is triggered,
231
+ * or rejects if an error is emitted. (If you request the 'error' event,
232
+ * returns a promise that resolves with the error).
233
+ *
234
+ * Example:
235
+ *
236
+ * const message = await stream.emitted('message') // rejects if the stream errors
237
+ */
238
+ emitted<Event extends keyof PromptCachingBetaMessageStreamEvents>(
239
+ event: Event,
240
+ ): Promise<
241
+ Parameters<PromptCachingBetaMessageStreamEvents[Event]> extends [infer Param] ? Param
242
+ : Parameters<PromptCachingBetaMessageStreamEvents[Event]> extends [] ? void
243
+ : Parameters<PromptCachingBetaMessageStreamEvents[Event]>
244
+ > {
245
+ return new Promise((resolve, reject) => {
246
+ this.#catchingPromiseCreated = true;
247
+ if (event !== 'error') this.once('error', reject);
248
+ this.once(event, resolve as any);
249
+ });
250
+ }
251
+
252
+ async done(): Promise<void> {
253
+ this.#catchingPromiseCreated = true;
254
+ await this.#endPromise;
255
+ }
256
+
257
+ get currentMessage(): PromptCachingBetaMessage | undefined {
258
+ return this.#currentMessageSnapshot;
259
+ }
260
+
261
+ #getFinalMessage(): PromptCachingBetaMessage {
262
+ if (this.receivedMessages.length === 0) {
263
+ throw new AnthropicError(
264
+ 'stream ended without producing a PromptCachingBetaMessage with role=assistant',
265
+ );
266
+ }
267
+ return this.receivedMessages.at(-1)!;
268
+ }
269
+
270
+ /**
271
+ * @returns a promise that resolves with the the final assistant PromptCachingBetaMessage response,
272
+ * or rejects if an error occurred or the stream ended prematurely without producing a PromptCachingBetaMessage.
273
+ */
274
+ async finalMessage(): Promise<PromptCachingBetaMessage> {
275
+ await this.done();
276
+ return this.#getFinalMessage();
277
+ }
278
+
279
+ #getFinalText(): string {
280
+ if (this.receivedMessages.length === 0) {
281
+ throw new AnthropicError(
282
+ 'stream ended without producing a PromptCachingBetaMessage with role=assistant',
283
+ );
284
+ }
285
+ const textBlocks = this.receivedMessages
286
+ .at(-1)!
287
+ .content.filter((block): block is TextBlock => block.type === 'text')
288
+ .map((block) => block.text);
289
+ if (textBlocks.length === 0) {
290
+ throw new AnthropicError('stream ended without producing a content block with type=text');
291
+ }
292
+ return textBlocks.join(' ');
293
+ }
294
+
295
+ /**
296
+ * @returns a promise that resolves with the the final assistant PromptCachingBetaMessage's text response, concatenated
297
+ * together if there are more than one text blocks.
298
+ * Rejects if an error occurred or the stream ended prematurely without producing a PromptCachingBetaMessage.
299
+ */
300
+ async finalText(): Promise<string> {
301
+ await this.done();
302
+ return this.#getFinalText();
303
+ }
304
+
305
+ #handleError = (error: unknown) => {
306
+ this.#errored = true;
307
+ if (error instanceof Error && error.name === 'AbortError') {
308
+ error = new APIUserAbortError();
309
+ }
310
+ if (error instanceof APIUserAbortError) {
311
+ this.#aborted = true;
312
+ return this._emit('abort', error);
313
+ }
314
+ if (error instanceof AnthropicError) {
315
+ return this._emit('error', error);
316
+ }
317
+ if (error instanceof Error) {
318
+ const anthropicError: AnthropicError = new AnthropicError(error.message);
319
+ // @ts-ignore
320
+ anthropicError.cause = error;
321
+ return this._emit('error', anthropicError);
322
+ }
323
+ return this._emit('error', new AnthropicError(String(error)));
324
+ };
325
+
326
+ protected _emit<Event extends keyof PromptCachingBetaMessageStreamEvents>(
327
+ event: Event,
328
+ ...args: Parameters<PromptCachingBetaMessageStreamEvents[Event]>
329
+ ) {
330
+ // make sure we don't emit any PromptCachingBetaMessageStreamEvents after end
331
+ if (this.#ended) return;
332
+
333
+ if (event === 'end') {
334
+ this.#ended = true;
335
+ this.#resolveEndPromise();
336
+ }
337
+
338
+ const listeners: PromptCachingBetaMessageStreamEventListeners<Event> | undefined = this.#listeners[event];
339
+ if (listeners) {
340
+ this.#listeners[event] = listeners.filter((l) => !l.once) as any;
341
+ listeners.forEach(({ listener }: any) => listener(...args));
342
+ }
343
+
344
+ if (event === 'abort') {
345
+ const error = args[0] as APIUserAbortError;
346
+ if (!this.#catchingPromiseCreated && !listeners?.length) {
347
+ Promise.reject(error);
348
+ }
349
+ this.#rejectConnectedPromise(error);
350
+ this.#rejectEndPromise(error);
351
+ this._emit('end');
352
+ return;
353
+ }
354
+
355
+ if (event === 'error') {
356
+ // NOTE: _emit('error', error) should only be called from #handleError().
357
+
358
+ const error = args[0] as AnthropicError;
359
+ if (!this.#catchingPromiseCreated && !listeners?.length) {
360
+ // Trigger an unhandled rejection if the user hasn't registered any error handlers.
361
+ // If you are seeing stack traces here, make sure to handle errors via either:
362
+ // - runner.on('error', () => ...)
363
+ // - await runner.done()
364
+ // - await runner.final...()
365
+ // - etc.
366
+ Promise.reject(error);
367
+ }
368
+ this.#rejectConnectedPromise(error);
369
+ this.#rejectEndPromise(error);
370
+ this._emit('end');
371
+ }
372
+ }
373
+
374
+ protected _emitFinal() {
375
+ const finalPromptCachingBetaMessage = this.receivedMessages.at(-1);
376
+ if (finalPromptCachingBetaMessage) {
377
+ this._emit('finalPromptCachingBetaMessage', this.#getFinalMessage());
378
+ }
379
+ }
380
+
381
+ #beginRequest() {
382
+ if (this.ended) return;
383
+ this.#currentMessageSnapshot = undefined;
384
+ }
385
+ #addStreamEvent(event: RawPromptCachingBetaMessageStreamEvent) {
386
+ if (this.ended) return;
387
+ const messageSnapshot = this.#accumulateMessage(event);
388
+ this._emit('streamEvent', event, messageSnapshot);
389
+
390
+ switch (event.type) {
391
+ case 'content_block_delta': {
392
+ const content = messageSnapshot.content.at(-1)!;
393
+ if (event.delta.type === 'text_delta' && content.type === 'text') {
394
+ this._emit('text', event.delta.text, content.text || '');
395
+ } else if (event.delta.type === 'input_json_delta' && content.type === 'tool_use') {
396
+ if (content.input) {
397
+ this._emit('inputJson', event.delta.partial_json, content.input);
398
+ }
399
+ }
400
+ break;
401
+ }
402
+ case 'message_stop': {
403
+ this._addPromptCachingBetaMessageParam(messageSnapshot);
404
+ this._addPromptCachingBetaMessage(messageSnapshot, true);
405
+ break;
406
+ }
407
+ case 'content_block_stop': {
408
+ this._emit('contentBlock', messageSnapshot.content.at(-1)!);
409
+ break;
410
+ }
411
+ case 'message_start': {
412
+ this.#currentMessageSnapshot = messageSnapshot;
413
+ break;
414
+ }
415
+ case 'content_block_start':
416
+ case 'message_delta':
417
+ break;
418
+ }
419
+ }
420
+ #endRequest(): PromptCachingBetaMessage {
421
+ if (this.ended) {
422
+ throw new AnthropicError(`stream has ended, this shouldn't happen`);
423
+ }
424
+ const snapshot = this.#currentMessageSnapshot;
425
+ if (!snapshot) {
426
+ throw new AnthropicError(`request ended without sending any chunks`);
427
+ }
428
+ this.#currentMessageSnapshot = undefined;
429
+ return snapshot;
430
+ }
431
+
432
+ protected async _fromReadableStream(
433
+ readableStream: ReadableStream,
434
+ options?: Core.RequestOptions,
435
+ ): Promise<void> {
436
+ const signal = options?.signal;
437
+ if (signal) {
438
+ if (signal.aborted) this.controller.abort();
439
+ signal.addEventListener('abort', () => this.controller.abort());
440
+ }
441
+ this.#beginRequest();
442
+ this._connected();
443
+ const stream = Stream.fromReadableStream<RawPromptCachingBetaMessageStreamEvent>(
444
+ readableStream,
445
+ this.controller,
446
+ );
447
+ for await (const event of stream) {
448
+ this.#addStreamEvent(event);
449
+ }
450
+ if (stream.controller.signal?.aborted) {
451
+ throw new APIUserAbortError();
452
+ }
453
+ this.#endRequest();
454
+ }
455
+
456
+ /**
457
+ * Mutates this.#currentPromptCachingBetaMessage with the current event. Handling the accumulation of multiple messages
458
+ * will be needed to be handled by the caller, this method will throw if you try to accumulate for multiple
459
+ * messages.
460
+ */
461
+ #accumulateMessage(event: RawPromptCachingBetaMessageStreamEvent): PromptCachingBetaMessage {
462
+ let snapshot = this.#currentMessageSnapshot;
463
+
464
+ if (event.type === 'message_start') {
465
+ if (snapshot) {
466
+ throw new AnthropicError(`Unexpected event order, got ${event.type} before receiving "message_stop"`);
467
+ }
468
+ return event.message;
469
+ }
470
+
471
+ if (!snapshot) {
472
+ throw new AnthropicError(`Unexpected event order, got ${event.type} before "message_start"`);
473
+ }
474
+
475
+ switch (event.type) {
476
+ case 'message_stop':
477
+ return snapshot;
478
+ case 'message_delta':
479
+ snapshot.stop_reason = event.delta.stop_reason;
480
+ snapshot.stop_sequence = event.delta.stop_sequence;
481
+ snapshot.usage.output_tokens = event.usage.output_tokens;
482
+ return snapshot;
483
+ case 'content_block_start':
484
+ snapshot.content.push(event.content_block);
485
+ return snapshot;
486
+ case 'content_block_delta': {
487
+ const snapshotContent = snapshot.content.at(event.index);
488
+ if (snapshotContent?.type === 'text' && event.delta.type === 'text_delta') {
489
+ snapshotContent.text += event.delta.text;
490
+ } else if (snapshotContent?.type === 'tool_use' && event.delta.type === 'input_json_delta') {
491
+ // we need to keep track of the raw JSON string as well so that we can
492
+ // re-parse it for each delta, for now we just store it as an untyped
493
+ // non-enumerable property on the snapshot
494
+ let jsonBuf = (snapshotContent as any)[JSON_BUF_PROPERTY] || '';
495
+ jsonBuf += event.delta.partial_json;
496
+
497
+ Object.defineProperty(snapshotContent, JSON_BUF_PROPERTY, {
498
+ value: jsonBuf,
499
+ enumerable: false,
500
+ writable: true,
501
+ });
502
+
503
+ if (jsonBuf) {
504
+ snapshotContent.input = partialParse(jsonBuf);
505
+ }
506
+ }
507
+ return snapshot;
508
+ }
509
+ case 'content_block_stop':
510
+ return snapshot;
511
+ }
512
+ }
513
+
514
+ [Symbol.asyncIterator](): AsyncIterator<RawPromptCachingBetaMessageStreamEvent> {
515
+ const pushQueue: RawPromptCachingBetaMessageStreamEvent[] = [];
516
+ const readQueue: {
517
+ resolve: (chunk: RawPromptCachingBetaMessageStreamEvent | undefined) => void;
518
+ reject: (error: unknown) => void;
519
+ }[] = [];
520
+ let done = false;
521
+
522
+ this.on('streamEvent', (event) => {
523
+ const reader = readQueue.shift();
524
+ if (reader) {
525
+ reader.resolve(event);
526
+ } else {
527
+ pushQueue.push(event);
528
+ }
529
+ });
530
+
531
+ this.on('end', () => {
532
+ done = true;
533
+ for (const reader of readQueue) {
534
+ reader.resolve(undefined);
535
+ }
536
+ readQueue.length = 0;
537
+ });
538
+
539
+ this.on('abort', (err) => {
540
+ done = true;
541
+ for (const reader of readQueue) {
542
+ reader.reject(err);
543
+ }
544
+ readQueue.length = 0;
545
+ });
546
+
547
+ this.on('error', (err) => {
548
+ done = true;
549
+ for (const reader of readQueue) {
550
+ reader.reject(err);
551
+ }
552
+ readQueue.length = 0;
553
+ });
554
+
555
+ return {
556
+ next: async (): Promise<IteratorResult<RawPromptCachingBetaMessageStreamEvent>> => {
557
+ if (!pushQueue.length) {
558
+ if (done) {
559
+ return { value: undefined, done: true };
560
+ }
561
+ return new Promise<RawPromptCachingBetaMessageStreamEvent | undefined>((resolve, reject) =>
562
+ readQueue.push({ resolve, reject }),
563
+ ).then((chunk) => (chunk ? { value: chunk, done: false } : { value: undefined, done: true }));
564
+ }
565
+ const chunk = pushQueue.shift()!;
566
+ return { value: chunk, done: false };
567
+ },
568
+ return: async () => {
569
+ this.abort();
570
+ return { value: undefined, done: true };
571
+ },
572
+ };
573
+ }
574
+
575
+ toReadableStream(): ReadableStream {
576
+ const stream = new Stream(this[Symbol.asyncIterator].bind(this), this.controller);
577
+ return stream.toReadableStream();
578
+ }
579
+ }
@@ -0,0 +1,12 @@
1
+ // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ import { APIResource } from "../../resource.js";
4
+ import * as PromptCachingAPI from "./prompt-caching/prompt-caching.js";
5
+
6
+ export class Beta extends APIResource {
7
+ promptCaching: PromptCachingAPI.PromptCaching = new PromptCachingAPI.PromptCaching(this._client);
8
+ }
9
+
10
+ export namespace Beta {
11
+ export import PromptCaching = PromptCachingAPI.PromptCaching;
12
+ }