Add wasm binding for values (#214)

Co-authored-by: grf53 <[email protected]>

Author: khj809

Cargo.lock

@@ -26,9 +26,9 @@ python = [
     "dep:pyo3-async-runtimes",
     "dep:pyo3-stub-gen",
     "dep:pyo3-stub-gen-derive",
-    "ailoy-model-cli",
 ]
-nodejs = ["dep:napi", "dep:napi-derive", "ailoy-model-cli"]
+nodejs = ["dep:napi", "dep:napi-derive"]
+wasm = []
 ailoy-model-cli = ["dep:aws-config", "dep:aws-sdk-s3", "dep:clap", "dep:indicatif"]
 default = []
 
@@ -54,10 +54,10 @@ jsonschema = { version = "0.33.0", default-features = false }
 mime-infer = "4.0.0"
 minijinja = { version = "2.11.0", features = ["loader", "custom_syntax", "json", "preserve_order"] }
 minijinja-contrib = { version = "2.11.0", features = ["pycompat"] }
-napi = { version = "3", features = ["tokio_rt", "napi8", "serde-json"], optional = true }
-napi-derive = { version = "3", optional = true }
+napi = { git = "https://github.com/brekkylab/napi-rs.git", package = "napi", rev = "5c8991b91072b5e468b1781de5703dc8449e000a", features = ["tokio_rt", "napi8", "serde-json"], optional = true }
+napi-derive = { git = "https://github.com/brekkylab/napi-rs.git", package = "napi-derive", rev = "5c8991b91072b5e468b1781de5703dc8449e000a", optional = true }
 ordered-float = { version = "5.0.0", features = ["serde"] }
-pyo3 = { version = "0.26", features = ["experimental-async", "macros"], optional = true }
+pyo3 = { version = "0.26", features = ["experimental-async", "macros", "anyhow"], optional = true }
 pyo3-async-runtimes = { version = "0.26", features = ["attributes", "tokio-runtime"], optional = true }
 pyo3-stub-gen = { version = "0.12.2", optional = true }
 pyo3-stub-gen-derive = { version = "0.12.2", optional = true }
@@ -88,6 +88,7 @@ js-sys = "0.3"
 sse-stream = "0.2"
 tokenizers = { version = "0.21.4", default-features = false, features = ["fancy-regex"] }
 tokio = { version = "1.0", default-features = false, features = ["macros", "rt", "sync"] }
+tsify = { version = "0.5.5", default-features = false, features = ["js"] }
 uuid = { version = "1.18.0", features = ["v4", "js"] }
 wasm-bindgen = "0.2"
 wasm-bindgen-futures = "0.4"

v2/Cargo.toml

@@ -4,12 +4,12 @@
     "main": "index.js",
     "napi": {
         "binaryName": "ailoy_core",
-        "dtsHeaderFile": "./src/header.d.ts",
         "targets": [
             "x86_64-pc-windows-msvc",
             "x86_64-unknown-linux-gnu",
             "aarch64-apple-darwin"
-        ]
+        ],
+        "constEnum": false
     },
     "scripts": {
         "build:napi": "napi build --platform --release --package ailoy --features nodejs --manifest-path ../../../Cargo.toml --js ailoy_core.js --dts ailoy_core.d.ts --esm -o ./src && prettier -w ./src/ailoy_core.d.ts",

v2/bindings/nodejs/package.json

@@ -1,3 +1,5 @@
+/* auto-generated by NAPI-RS */
+/* eslint-disable */
 export declare class LangModel {
   static createLocal(
     modelName: string,
@@ -19,14 +21,13 @@ export declare class LangModelRunIterator {
   next(): Promise<LanguageModelIteratorResult>;
 }
 
-export declare const enum APISpecification {
-  ChatCompletion = "ChatCompletion",
-  OpenAI = "OpenAI",
-  Gemini = "Gemini",
-  Claude = "Claude",
-  Responses = "Responses",
-  Grok = "Grok",
-}
+export type APISpecification =
+  | "ChatCompletion"
+  | "OpenAI"
+  | "Gemini"
+  | "Claude"
+  | "Responses"
+  | "Grok";
 
 export interface CacheProgress {
   comment: string;
@@ -35,24 +36,24 @@ export interface CacheProgress {
 }
 
 export type FinishReason =
-  | { type: "Stop" }
-  | { type: "Length" }
-  | { type: "ToolCall" }
-  | { type: "Refusal"; field0: string };
+  | { type: "stop" }
+  | { type: "length" }
+  | { type: "tool_call" }
+  | { type: "refusal"; reason: string };
 
 export type Grammar =
-  | { type: "Plain" }
-  | { type: "JSON" }
-  | { type: "JSONSchema"; field0: string }
-  | { type: "Regex"; field0: string }
-  | { type: "CFG"; field0: string };
+  | { type: "plain" }
+  | { type: "json" }
+  | { type: "jsonschema"; schema: string }
+  | { type: "regex"; regex: string }
+  | { type: "cfg"; cfg: string };
 
 export interface InferenceConfig {
-  thinkEffort: ThinkEffort;
+  thinkEffort?: ThinkEffort;
   temperature?: number;
   topP?: number;
   maxTokens?: number;
-  grammar: Grammar;
+  grammar?: Grammar;
 }
 
 export interface LanguageModelIteratorResult {
@@ -62,82 +63,71 @@ export interface LanguageModelIteratorResult {
 
 export interface Message {
   role: Role;
-  id?: string;
-  thinking: string;
   contents: Array<Part>;
-  toolCalls: Array<Part>;
+  id?: string;
+  thinking?: string;
+  tool_calls?: Array<Part>;
   signature?: string;
 }
 
 export interface MessageDelta {
   role?: Role;
   id?: string;
-  thinking: string;
+  thinking?: string;
   contents: Array<PartDelta>;
-  toolCalls: Array<PartDelta>;
+  tool_calls: Array<PartDelta>;
   signature?: string;
 }
 
 export interface MessageOutput {
   delta: MessageDelta;
-  finishReason?: FinishReason;
+  finish_reason?: FinishReason;
 }
 
 export type Part =
-  | { type: "Text"; text: string }
-  | { type: "Function"; id?: string; f: PartFunction }
-  | { type: "Value"; value: any }
-  | { type: "Image"; image: PartImage };
+  | { type: "text"; text: string }
+  | { type: "function"; id?: string; function: PartFunction }
+  | { type: "value"; value: any }
+  | { type: "image"; image: PartImage };
 
 export type PartDelta =
-  | { type: "Text"; text: string }
-  | { type: "Function"; id?: string; f: PartDeltaFunction }
-  | { type: "Value"; value: any }
-  | { type: "Null" };
+  | { type: "text"; text: string }
+  | { type: "function"; id?: string; function: PartDeltaFunction }
+  | { type: "value"; value: any }
+  | { type: "null" };
 
 export type PartDeltaFunction =
-  | { type: "Verbatim"; field0: string }
-  | { type: "WithStringArgs"; name: string; args: string }
-  | { type: "WithParsedArgs"; name: string; args: any };
+  | { type: "verbatim"; text: string }
+  | { type: "with_string_args"; name: string; arguments: string }
+  | { type: "with_parsed_args"; name: string; arguments: any };
 
 export interface PartFunction {
   name: string;
-  args: any;
+  arguments: any;
 }
 
 export type PartImage = {
-  type: "Binary";
-  h: number;
-  w: number;
-  c: PartImageColorspace;
+  type: "binary";
+  height: number;
+  width: number;
+  colorspace: PartImageColorspace;
   data: Buffer;
 };
 
-export declare const enum PartImageColorspace {
-  Grayscale = "Grayscale",
-  RGB = "RGB",
-  RGBA = "RGBA",
-}
+export type PartImageColorspace = "grayscale" | "rgb" | "rgba";
 
 /** The author of a message (or streaming delta) in a chat. */
-export declare const enum Role {
+export type Role =
   /** System instructions and constraints provided to the assistant. */
-  System = "System",
+  | "system"
   /** Content authored by the end user. */
-  User = "User",
+  | "user"
   /** Content authored by the assistant/model. */
-  Assistant = "Assistant",
+  | "assistant"
   /** Outputs produced by external tools/functions */
-  Tool = "Tool",
-}
+  | "tool";
 
-export declare const enum ThinkEffort {
-  Disable = "Disable",
-  Enable = "Enable",
-  Low = "Low",
-  Medium = "Medium",
-  High = "High",
-}
+export type ThinkEffort = "disable" | "enable" | "low" | "medium" | "high";
 
 export interface ToolDesc {
   name: string;

v2/bindings/nodejs/src/ailoy_core.d.ts

@@ -10,5 +10,5 @@
     "allowSyntheticDefaultImports": true,
   },
   "include": ["./src"],
-  "exclude": ["node_modules", "./src/header.d.ts"]
+  "exclude": ["node_modules"]
 }