@anthropic-ai/sdk 0.3.0 → 0.4.0

package/build/src/index.d.ts

@@ -0,0 +1,30 @@
+export type SamplingParameters = {
+    prompt: string;
+    model: string;
+    stop_sequences: string[];
+    max_tokens_to_sample: number;
+};
+export type OnOpen = (response: Response) => void | Promise<void>;
+export type OnUpdate = (completion: CompletionResponse) => void | Promise<void>;
+export declare const HUMAN_PROMPT = "\n\nHuman:";
+export declare const AI_PROMPT = "\n\nAssistant:";
+export type CompletionResponse = {
+    completion: string;
+    stop: string | null;
+    stop_reason: string | null;
+    truncated: boolean;
+    exception: string | null;
+    log_id: string;
+};
+export declare class Client {
+    private apiKey;
+    private apiUrl;
+    constructor(apiKey: string, options?: {
+        apiUrl?: string;
+    });
+    complete(params: SamplingParameters): Promise<CompletionResponse>;
+    completeStream(params: SamplingParameters, { onOpen, onUpdate }: {
+        onOpen?: OnOpen;
+        onUpdate?: OnUpdate;
+    }): Promise<CompletionResponse>;
+}

package/build/src/index.js

@@ -0,0 +1,100 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Client = exports.AI_PROMPT = exports.HUMAN_PROMPT = void 0;
+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.0";
+const DEFAULT_API_URL = "https://api.anthropic.com";
+var Event;
+(function (Event) {
+    Event["Ping"] = "ping";
+})(Event || (Event = {}));
+class Client {
+    constructor(apiKey, options) {
+        var _a;
+        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) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const response = yield (0, cross_fetch_1.default)(`${this.apiUrl}/v1/complete`, {
+                method: "POST",
+                headers: {
+                    Accept: "application/json",
+                    "Content-Type": "application/json",
+                    Client: CLIENT_ID,
+                    "X-API-Key": this.apiKey,
+                },
+                body: JSON.stringify(params),
+            });
+            if (!response.ok) {
+                const error = new Error(`Sampling error: ${response.status} ${response.statusText}`);
+                console.error(error);
+                throw error;
+            }
+            const completion = (yield response.json());
+            return completion;
+        });
+    }
+    completeStream(params, { onOpen, onUpdate }) {
+        const abortController = new AbortController();
+        return new Promise((resolve, reject) => {
+            (0, fetch_event_source_1.fetchEventSource)(`${this.apiUrl}/v1/complete`, {
+                method: "POST",
+                headers: {
+                    Accept: "application/json",
+                    "Content-Type": "application/json",
+                    Client: CLIENT_ID,
+                    "X-API-Key": this.apiKey,
+                },
+                body: JSON.stringify(Object.assign(Object.assign({}, params), { stream: true })),
+                signal: abortController.signal,
+                onopen: (response) => __awaiter(this, void 0, void 0, function* () {
+                    if (!response.ok) {
+                        abortController.abort();
+                        return reject(Error(`Failed to open sampling stream, HTTP status code ${response.status}: ${response.statusText}`));
+                    }
+                    if (onOpen) {
+                        yield Promise.resolve(onOpen(response));
+                    }
+                }),
+                onmessage: (ev) => {
+                    if (ev.event === Event.Ping) {
+                        return;
+                    }
+                    const completion = JSON.parse(ev.data);
+                    if (completion.stop_reason !== null) {
+                        abortController.abort();
+                        return resolve(completion);
+                    }
+                    if (onUpdate) {
+                        Promise.resolve(onUpdate(completion)).catch((error) => {
+                            abortController.abort();
+                            reject(error);
+                        });
+                    }
+                },
+                onerror: (error) => {
+                    console.error("Sampling error:", error);
+                    abortController.abort();
+                    return reject(error);
+                },
+            });
+        });
+    }
+}
+exports.Client = Client;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@anthropic-ai/sdk",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "description": "Library for accessing the Anthropic API",
   "repository": "https://github.com/anthropics/anthropic-sdk-typescript",
   "license": "MIT",
@@ -18,7 +18,8 @@
     "example:basic_sync": "tsc --build && node ./build/examples/basic_sync.js"
   },
   "dependencies": {
-    "@fortaine/fetch-event-source": "^3.0.6"
+    "@fortaine/fetch-event-source": "^3.0.6",
+    "cross-fetch": "^3.1.5"
   },
   "devDependencies": {
     "@types/node": "^18.11.18",