aquin 0.1.0 → 0.2.2

package/dist/client.d.ts

@@ -6,6 +6,7 @@ export interface CompletionResponse {
 export interface CompletionOptions {
     max_tokens?: number;
     temperature?: number;
+    system_prompt?: string;
 }
 export declare class AquinClient {
     private apiKey;

package/dist/client.js

@@ -11,7 +11,12 @@ class AquinClient {
         this.baseUrl = baseUrl.replace(/\/$/, "");
     }
     async complete(prompt, options = {}) {
-        const { max_tokens = 512, temperature = 0.7 } = options;
+        const { max_tokens = 512, temperature = 0.7, system_prompt } = options;
+        const messages = [];
+        if (system_prompt)
+            messages.push({ role: "system", content: system_prompt });
+        messages.push({ role: "user", content: prompt });
+        const body = { prompt, messages, max_tokens, temperature };
         let res;
         try {
             res = await fetch(`${this.baseUrl}/api/v1/model`, {
@@ -20,7 +25,7 @@ class AquinClient {
                     Authorization: `Bearer ${this.apiKey}`,
                     "Content-Type": "application/json",
                 },
-                body: JSON.stringify({ prompt, max_tokens, temperature }),
+                body: JSON.stringify(body),
                 redirect: "follow",
             });
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aquin",
-  "version": "0.1.0",
+  "version": "0.2.2",
   "description": "JavaScript/TypeScript SDK for the Aquin model API",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",