aquin 0.1.0

package/README.md

@@ -0,0 +1,2 @@
+Aquin SDK - JS/TS
+npm install aquin

package/dist/client.d.ts

@@ -0,0 +1,16 @@
+export interface CompletionResponse {
+    text: string;
+    tokens_in: number;
+    tokens_out: number;
+}
+export interface CompletionOptions {
+    max_tokens?: number;
+    temperature?: number;
+}
+export declare class AquinClient {
+    private apiKey;
+    private baseUrl;
+    constructor(apiKey: string, baseUrl?: string);
+    complete(prompt: string, options?: CompletionOptions): Promise<CompletionResponse>;
+    private handleResponse;
+}

package/dist/client.js

@@ -0,0 +1,60 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AquinClient = void 0;
+const exceptions_1 = require("./exceptions");
+const DEFAULT_BASE_URL = "https://www.aquin.app";
+class AquinClient {
+    constructor(apiKey, baseUrl = DEFAULT_BASE_URL) {
+        if (!apiKey)
+            throw new Error("apiKey is required");
+        this.apiKey = apiKey;
+        this.baseUrl = baseUrl.replace(/\/$/, "");
+    }
+    async complete(prompt, options = {}) {
+        const { max_tokens = 512, temperature = 0.7 } = options;
+        let res;
+        try {
+            res = await fetch(`${this.baseUrl}/api/v1/model`, {
+                method: "POST",
+                headers: {
+                    Authorization: `Bearer ${this.apiKey}`,
+                    "Content-Type": "application/json",
+                },
+                body: JSON.stringify({ prompt, max_tokens, temperature }),
+                redirect: "follow",
+            });
+        }
+        catch (err) {
+            throw new exceptions_1.InferenceError(`Network error: ${err.message}`);
+        }
+        return this.handleResponse(res);
+    }
+    async handleResponse(res) {
+        if (res.ok) {
+            const data = await res.json();
+            return {
+                text: data.text,
+                tokens_in: data.tokens_in,
+                tokens_out: data.tokens_out,
+            };
+        }
+        let errorMsg = `HTTP ${res.status}`;
+        try {
+            const body = await res.json();
+            errorMsg = body.error ?? errorMsg;
+        }
+        catch {
+            // empty body
+        }
+        switch (res.status) {
+            case 401: throw new exceptions_1.InvalidKeyError(errorMsg);
+            case 402: throw new exceptions_1.InsufficientCreditsError(errorMsg);
+            case 404: throw new exceptions_1.ModelNotFoundError(errorMsg);
+            case 429: throw new exceptions_1.RateLimitError(errorMsg);
+            case 502:
+            case 503: throw new exceptions_1.InferenceError(errorMsg, res.status);
+            default: throw new exceptions_1.AquinError(errorMsg, res.status);
+        }
+    }
+}
+exports.AquinClient = AquinClient;

package/dist/exceptions.d.ts

@@ -0,0 +1,19 @@
+export declare class AquinError extends Error {
+    statusCode?: number;
+    constructor(message: string, statusCode?: number);
+}
+export declare class InvalidKeyError extends AquinError {
+    constructor(message: string);
+}
+export declare class InsufficientCreditsError extends AquinError {
+    constructor(message: string);
+}
+export declare class RateLimitError extends AquinError {
+    constructor(message: string);
+}
+export declare class ModelNotFoundError extends AquinError {
+    constructor(message: string);
+}
+export declare class InferenceError extends AquinError {
+    constructor(message: string, statusCode?: number);
+}

package/dist/exceptions.js

@@ -0,0 +1,46 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InferenceError = exports.ModelNotFoundError = exports.RateLimitError = exports.InsufficientCreditsError = exports.InvalidKeyError = exports.AquinError = void 0;
+class AquinError extends Error {
+    constructor(message, statusCode) {
+        super(message);
+        this.name = "AquinError";
+        this.statusCode = statusCode;
+    }
+}
+exports.AquinError = AquinError;
+class InvalidKeyError extends AquinError {
+    constructor(message) {
+        super(message, 401);
+        this.name = "InvalidKeyError";
+    }
+}
+exports.InvalidKeyError = InvalidKeyError;
+class InsufficientCreditsError extends AquinError {
+    constructor(message) {
+        super(message, 402);
+        this.name = "InsufficientCreditsError";
+    }
+}
+exports.InsufficientCreditsError = InsufficientCreditsError;
+class RateLimitError extends AquinError {
+    constructor(message) {
+        super(message, 429);
+        this.name = "RateLimitError";
+    }
+}
+exports.RateLimitError = RateLimitError;
+class ModelNotFoundError extends AquinError {
+    constructor(message) {
+        super(message, 404);
+        this.name = "ModelNotFoundError";
+    }
+}
+exports.ModelNotFoundError = ModelNotFoundError;
+class InferenceError extends AquinError {
+    constructor(message, statusCode) {
+        super(message, statusCode);
+        this.name = "InferenceError";
+    }
+}
+exports.InferenceError = InferenceError;

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+export { AquinClient } from "./client";
+export type { CompletionResponse, CompletionOptions } from "./client";
+export { AquinError, InvalidKeyError, InsufficientCreditsError, RateLimitError, ModelNotFoundError, InferenceError, } from "./exceptions";

package/dist/index.js

@@ -0,0 +1,12 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InferenceError = exports.ModelNotFoundError = exports.RateLimitError = exports.InsufficientCreditsError = exports.InvalidKeyError = exports.AquinError = exports.AquinClient = void 0;
+var client_1 = require("./client");
+Object.defineProperty(exports, "AquinClient", { enumerable: true, get: function () { return client_1.AquinClient; } });
+var exceptions_1 = require("./exceptions");
+Object.defineProperty(exports, "AquinError", { enumerable: true, get: function () { return exceptions_1.AquinError; } });
+Object.defineProperty(exports, "InvalidKeyError", { enumerable: true, get: function () { return exceptions_1.InvalidKeyError; } });
+Object.defineProperty(exports, "InsufficientCreditsError", { enumerable: true, get: function () { return exceptions_1.InsufficientCreditsError; } });
+Object.defineProperty(exports, "RateLimitError", { enumerable: true, get: function () { return exceptions_1.RateLimitError; } });
+Object.defineProperty(exports, "ModelNotFoundError", { enumerable: true, get: function () { return exceptions_1.ModelNotFoundError; } });
+Object.defineProperty(exports, "InferenceError", { enumerable: true, get: function () { return exceptions_1.InferenceError; } });

package/package.json

@@ -0,0 +1,17 @@
+{
+  "name": "aquin",
+  "version": "0.1.0",
+  "description": "JavaScript/TypeScript SDK for the Aquin model API",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": ["dist"],
+  "scripts": {
+    "build": "tsc",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": ["aquin", "llm", "fine-tuning", "ai"],
+  "license": "MIT",
+  "devDependencies": {
+    "typescript": "^5.0.0"
+  }
+}