@anthropic-ai/sdk 0.4.1 → 0.4.3

package/build/src/index.d.ts

@@ -6,6 +6,9 @@ export type SamplingParameters = {
     top_k?: number;
     top_p?: number;
     model: string;
+    tags?: {
+        [key: string]: string;
+    };
 };
 export type OnOpen = (response: Response) => void | Promise<void>;
 export type OnUpdate = (completion: CompletionResponse) => void | Promise<void>;
@@ -25,9 +28,12 @@ export declare class Client {
     constructor(apiKey: string, options?: {
         apiUrl?: string;
     });
-    complete(params: SamplingParameters): Promise<CompletionResponse>;
-    completeStream(params: SamplingParameters, { onOpen, onUpdate }: {
+    complete(params: SamplingParameters, options?: {
+        signal?: AbortSignal;
+    }): Promise<CompletionResponse>;
+    completeStream(params: SamplingParameters, { onOpen, onUpdate, signal, }: {
         onOpen?: OnOpen;
         onUpdate?: OnUpdate;
+        signal?: AbortSignal;
     }): Promise<CompletionResponse>;
 }

package/build/src/index.js

@@ -17,7 +17,7 @@ const fetch_event_source_1 = require("@fortaine/fetch-event-source");
 const cross_fetch_1 = __importDefault(require("cross-fetch"));
 exports.HUMAN_PROMPT = "\n\nHuman:";
 exports.AI_PROMPT = "\n\nAssistant:";
-const CLIENT_ID = "anthropic-typescript/0.4.1";
+const CLIENT_ID = "anthropic-typescript/0.4.3";
 const DEFAULT_API_URL = "https://api.anthropic.com";
 var Event;
 (function (Event) {
@@ -30,7 +30,7 @@ class Client {
         this.apiKey = apiKey;
         this.apiUrl = (_a = options === null || options === void 0 ? void 0 : options.apiUrl) !== null && _a !== void 0 ? _a : DEFAULT_API_URL;
     }
-    complete(params) {
+    complete(params, options) {
         return __awaiter(this, void 0, void 0, function* () {
             const response = yield (0, cross_fetch_1.default)(`${this.apiUrl}/v1/complete`, {
                 method: "POST",
@@ -41,6 +41,7 @@ class Client {
                     "X-API-Key": this.apiKey,
                 },
                 body: JSON.stringify(Object.assign(Object.assign({}, params), { stream: false })),
+                signal: options === null || options === void 0 ? void 0 : options.signal,
             });
             if (!response.ok) {
                 const error = new Error(`Sampling error: ${response.status} ${response.statusText}`);
@@ -51,9 +52,16 @@ class Client {
             return completion;
         });
     }
-    completeStream(params, { onOpen, onUpdate }) {
+    completeStream(params, { onOpen, onUpdate, signal, }) {
         const abortController = new AbortController();
         return new Promise((resolve, reject) => {
+            signal === null || signal === void 0 ? void 0 : signal.addEventListener("abort", (event) => {
+                abortController.abort(event);
+                reject({
+                    name: "AbortError",
+                    message: "Caller aborted completeStream",
+                });
+            });
             (0, fetch_event_source_1.fetchEventSource)(`${this.apiUrl}/v1/complete`, {
                 method: "POST",
                 headers: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@anthropic-ai/sdk",
-  "version": "0.4.1",
+  "version": "0.4.3",
   "description": "Library for accessing the Anthropic API",
   "repository": "https://github.com/anthropics/anthropic-sdk-typescript",
   "license": "MIT",
@@ -15,7 +15,8 @@
     "lint": "eslint src && prettier --check .",
     "format": "prettier --write .",
     "example:basic_stream": "tsc --build && node ./build/examples/basic_stream.js",
-    "example:basic_sync": "tsc --build && node ./build/examples/basic_sync.js"
+    "example:basic_sync": "tsc --build && node ./build/examples/basic_sync.js",
+    "example:cancellation": "tsc --build && node ./build/examples/cancellation.js"
   },
   "dependencies": {
     "@fortaine/fetch-event-source": "^3.0.6",