@ariacompute/engine-ts 1.6.2 → 1.6.3

package/dist/index.js

@@ -175,9 +175,9 @@ class Engine {
         this.lib = loadLib(opts.ffiLib);
         this.fnInit = this.lib.func("aria_model_init", "void*", ["str"]);
         this.fnDestroy = this.lib.func("aria_model_destroy", "void", ["void*"]);
-        this.fnComplete = this.lib.func("aria_complete", "int", ["void*", "str", "str", "str", "str", "size_t"]);
-        this.fnEmbed = this.lib.func("aria_embed", "int", ["void*", "str", "str", "size_t"]);
-        this.fnTranscribe = this.lib.func("aria_transcribe", "int", ["void*", "void*", "size_t", "str", "str", "size_t"]);
+        this.fnComplete = this.lib.func("aria_complete", "int", ["void*", "str", "str", "str", "void*", "size_t"]);
+        this.fnEmbed = this.lib.func("aria_embed", "int", ["void*", "str", "void*", "size_t"]);
+        this.fnTranscribe = this.lib.func("aria_transcribe", "int", ["void*", "void*", "size_t", "str", "void*", "size_t"]);
         this.fnLastError = this.lib.func("aria_last_error", "str", []);
         this.handle = this.fnInit(bundle);
         if (!this.handle) {
@@ -198,9 +198,11 @@ class Engine {
         return new Engine(bundle, opts);
     }
     complete(messages, opts = {}) {
-        const payload = JSON.stringify({ messages, opts });
+        const messagesJson = JSON.stringify(messages);
+        const optionsJson = JSON.stringify(opts || { max_tokens: 16 });
+        const toolsJson = JSON.stringify([]);
         const buf = Buffer.alloc(1 << 16);
-        const rc = this.fnComplete(this.handle, payload, "", "", buf, buf.length);
+        const rc = this.fnComplete(this.handle, messagesJson, optionsJson, toolsJson, buf, buf.length);
         if (rc !== 0) {
             const err = this.fnLastError();
             return { success: false, response: "", error: err || "complete failed" };
@@ -215,7 +217,7 @@ class Engine {
     }
     embed(text) {
         const buf = Buffer.alloc(1 << 20);
-        const rc = this.fnEmbed(this.handle, text, buf, buf.length);
+        const rc = this.fnEmbed(this.handle, JSON.stringify({ input: text }), buf, buf.length);
         if (rc !== 0) {
             const err = this.fnLastError();
             throw new Error(err || "embed failed");
@@ -224,7 +226,7 @@ class Engine {
     }
     transcribe(pcm) {
         const buf = Buffer.alloc(1 << 16);
-        const rc = this.fnTranscribe(this.handle, pcm, pcm.length, "", buf, buf.length);
+        const rc = this.fnTranscribe(this.handle, Buffer.from(pcm), pcm.length, null, buf, buf.length);
         if (rc !== 0) {
             const err = this.fnLastError();
             throw new Error(err || "transcribe failed");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ariacompute/engine-ts",
-  "version": "1.6.2",
+  "version": "1.6.3",
   "description": "TypeScript/Node SDK for Aria Engine",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",