@anthropic-ai/sdk 0.3.1 → 0.4.1

package/README.md

@@ -4,6 +4,8 @@ This package provides access to Anthropic's safety-first language model APIs.
 
 For more information on our APIs, please check the confidential documentation portal.
 
+**Note that this package only officially supports Node 16+.**
+
 ## Installation
 
 Install using your package manager of choice. For example:

package/build/src/index.d.ts

@@ -1,8 +1,11 @@
 export type SamplingParameters = {
     prompt: string;
-    model: string;
-    stop_sequences: string[];
+    temperature?: number;
     max_tokens_to_sample: number;
+    stop_sequences: string[];
+    top_k?: number;
+    top_p?: number;
+    model: string;
 };
 export type OnOpen = (response: Response) => void | Promise<void>;
 export type OnUpdate = (completion: CompletionResponse) => void | Promise<void>;
@@ -11,7 +14,7 @@ export declare const AI_PROMPT = "\n\nAssistant:";
 export type CompletionResponse = {
     completion: string;
     stop: string | null;
-    stop_reason: string | null;
+    stop_reason: "stop_sequence" | "max_tokens";
     truncated: boolean;
     exception: string | null;
     log_id: string;

package/build/src/index.js

@@ -8,17 +8,22 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         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.3.1";
+const CLIENT_ID = "anthropic-typescript/0.4.1";
 const DEFAULT_API_URL = "https://api.anthropic.com";
 var Event;
 (function (Event) {
     Event["Ping"] = "ping";
 })(Event || (Event = {}));
+const DONE_MESSAGE = "[DONE]";
 class Client {
     constructor(apiKey, options) {
         var _a;
@@ -27,7 +32,7 @@ class Client {
     }
     complete(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            const response = yield fetch(`${this.apiUrl}/v1/complete`, {
+            const response = yield (0, cross_fetch_1.default)(`${this.apiUrl}/v1/complete`, {
                 method: "POST",
                 headers: {
                     Accept: "application/json",
@@ -35,7 +40,7 @@ class Client {
                     Client: CLIENT_ID,
                     "X-API-Key": this.apiKey,
                 },
-                body: JSON.stringify(params),
+                body: JSON.stringify(Object.assign(Object.assign({}, params), { stream: false })),
             });
             if (!response.ok) {
                 const error = new Error(`Sampling error: ${response.status} ${response.statusText}`);
@@ -72,17 +77,21 @@ class Client {
                     if (ev.event === Event.Ping) {
                         return;
                     }
-                    const completion = JSON.parse(ev.data);
-                    if (completion.stop_reason !== null) {
-                        abortController.abort();
-                        return resolve(completion);
+                    if (ev.data === DONE_MESSAGE) {
+                        console.error("Unexpected done message before stop_reason has been issued");
+                        return;
                     }
+                    const completion = JSON.parse(ev.data);
                     if (onUpdate) {
                         Promise.resolve(onUpdate(completion)).catch((error) => {
                             abortController.abort();
                             reject(error);
                         });
                     }
+                    if (completion.stop_reason !== null) {
+                        abortController.abort();
+                        return resolve(completion);
+                    }
                 },
                 onerror: (error) => {
                     console.error("Sampling error:", error);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@anthropic-ai/sdk",
-  "version": "0.3.1",
+  "version": "0.4.1",
   "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",