@openmle/omle.js 0.1.0-rc4 → 0.1.0-rc5

package/README.md

@@ -1,11 +1,11 @@
 # omle.js
 
-TypeScript library for loading, parsing, validating, and executing [OMLE](https://github.com/openmle) models. Works in Node.js and the browser. Zero runtime dependencies.
+TypeScript library for loading, parsing, validating, and executing [OMLE](https://github.com/openmle) models. Works in Node.js and the browser. One runtime dependency (`protobufjs`, for binary `.omle` decoding).
 
 ## Features
 
 - **Full IR** — TypeScript types for every message in `omle.proto`, following the Python SDK's JSON conventions (enums as strings, flat typed-data arrays, base64 bytes).
-- **I/O** — `fromJSON` / `toJSON`, Node.js `loadFile` / `saveFile`, browser `loadBlob`.
+- **I/O** — `fromJSON` / `toJSON`, `fromProtoBinary` for binary `.omle` files, Node.js `loadFile` / `saveFile`, browser `loadBlob`.
 - **Resolution** — builds a tensor-entry index, topologically sorts the DAG, and expands `NameRange` patterns.
 - **Validation** — structural and semantic checks with per-path error and warning messages.
 - **Reference engine** — executes models in topological order with full composite-node scoping. Supports all structured bodies: decision tree, tree ensemble, linear, neural network (dense layers), naïve Bayes (Gaussian / Multinomial / Bernoulli / Categorical), prototype clustering, GMM, and linear and kernel SVM.
@@ -48,6 +48,10 @@ console.log(outputs);
 fromJSON(json: string | object): OMLEModel
 toJSON(model: OMLEModel, pretty?: boolean): string
 
+// Binary .omle files (protobuf wire format). The schema is embedded, so
+// nothing is fetched or compiled at build time.
+fromProtoBinary(data: ArrayBuffer | Uint8Array): OMLEModel
+
 // Node.js only
 loadFile(path: string): Promise<OMLEModel>
 saveFile(model: OMLEModel, path: string, pretty?: boolean): Promise<void>

package/dist/index.d.ts

@@ -1,5 +1,6 @@
 export * from './ir.js';
 export * from './io.js';
+export { fromProtoBinary } from './proto.js';
 export * from './resolve.js';
 export * from './validate.js';
 export { Engine } from './engine/executor.js';

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AASA,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,cAAc,cAAc,CAAC;AAC7B,cAAc,eAAe,CAAC;AAE9B,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAC7D,YAAY,EAAE,UAAU,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACrF,YAAY,EAAE,cAAc,EAAE,eAAe,EAAE,gBAAgB,EAAE,YAAY,EAAE,aAAa,EAAE,YAAY,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC/K,YAAY,EACV,YAAY,EAAE,mBAAmB,EAAE,eAAe,EAAE,YAAY,EAChE,aAAa,EAAE,kBAAkB,EACjC,iBAAiB,EAAE,wBAAwB,EAC3C,iBAAiB,EAAE,gBAAgB,GACpC,MAAM,qBAAqB,CAAC;AAC7B,YAAY,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AASA,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,cAAc,cAAc,CAAC;AAC7B,cAAc,eAAe,CAAC;AAE9B,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAC7D,YAAY,EAAE,UAAU,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACrF,YAAY,EAAE,cAAc,EAAE,eAAe,EAAE,gBAAgB,EAAE,YAAY,EAAE,aAAa,EAAE,YAAY,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC/K,YAAY,EACV,YAAY,EAAE,mBAAmB,EAAE,eAAe,EAAE,YAAY,EAChE,aAAa,EAAE,kBAAkB,EACjC,iBAAiB,EAAE,wBAAwB,EAC3C,iBAAiB,EAAE,gBAAgB,GACpC,MAAM,qBAAqB,CAAC;AAC7B,YAAY,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/index.js

@@ -8,6 +8,7 @@
 //   const outputs = engine.run({ X: { dtype: 'FLOAT64', shape: [1, 4], data: new Float64Array([...]) } });
 export * from './ir.js';
 export * from './io.js';
+export { fromProtoBinary } from './proto.js';
 export * from './resolve.js';
 export * from './validate.js';
 export { Engine } from './engine/executor.js';

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,uBAAuB;AACvB,EAAE;AACF,SAAS;AACT,mEAAmE;AACnE,wCAAwC;AACxC,oCAAoC;AACpC,sCAAsC;AACtC,2GAA2G;AAE3G,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,cAAc,cAAc,CAAC;AAC7B,cAAc,eAAe,CAAC;AAE9B,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAU7D,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,uBAAuB;AACvB,EAAE;AACF,SAAS;AACT,mEAAmE;AACnE,wCAAwC;AACxC,oCAAoC;AACpC,sCAAsC;AACtC,2GAA2G;AAE3G,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,cAAc,cAAc,CAAC;AAC7B,cAAc,eAAe,CAAC;AAE9B,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAU7D,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/proto.d.ts

@@ -0,0 +1,11 @@
+import type { OMLEModel } from './ir.js';
+/**
+ * Decode a binary `.omle` file into an OMLEModel.
+ *
+ * Returns a model identical to what `fromJSON` produces for the same model, so
+ * callers can treat the two encodings interchangeably. The proto encoding wraps
+ * repeated typed arrays in list messages (Float32List and friends); fromJSON
+ * already unwraps those, so decoding hands its output straight to it.
+ */
+export declare function fromProtoBinary(data: ArrayBuffer | Uint8Array): OMLEModel;
+//# sourceMappingURL=proto.d.ts.map

package/dist/proto.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"proto.d.ts","sourceRoot":"","sources":["../src/proto.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAWzC;;;;;;;GAOG;AACH,wBAAgB,eAAe,CAAC,IAAI,EAAE,WAAW,GAAG,UAAU,GAAG,SAAS,CAWzE"}

package/dist/proto.js

@@ -0,0 +1,38 @@
+// Binary .omle loader — decodes the protobuf wire format.
+//
+// Wire format is not self-describing: the bytes carry field numbers, not names
+// or types, so decoding needs the schema. It is embedded as a protobufjs JSON
+// descriptor (src/proto_descriptor.ts, generated from omle.proto) rather than
+// the .proto text, which is ~4x larger and would also drag in the .proto
+// grammar parser. protobufjs/light provides reflection without that parser.
+import protobufLight from 'protobufjs/light.js';
+import { protoDescriptor } from './proto_descriptor.js';
+import { fromJSON } from './io.js';
+let _root = null;
+function getRoot() {
+    if (!_root) {
+        _root = protobufLight.Root.fromJSON(protoDescriptor);
+    }
+    return _root;
+}
+/**
+ * Decode a binary `.omle` file into an OMLEModel.
+ *
+ * Returns a model identical to what `fromJSON` produces for the same model, so
+ * callers can treat the two encodings interchangeably. The proto encoding wraps
+ * repeated typed arrays in list messages (Float32List and friends); fromJSON
+ * already unwraps those, so decoding hands its output straight to it.
+ */
+export function fromProtoBinary(data) {
+    const bytes = data instanceof Uint8Array ? data : new Uint8Array(data);
+    const ModelType = getRoot().lookupType('omle.OMLEModel');
+    const decoded = ModelType.decode(bytes);
+    const obj = ModelType.toObject(decoded, {
+        longs: Number,
+        enums: String,
+        bytes: String, // base64 — matches the JSON convention
+        defaults: false,
+    });
+    return fromJSON(obj);
+}
+//# sourceMappingURL=proto.js.map

package/dist/proto.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"proto.js","sourceRoot":"","sources":["../src/proto.ts"],"names":[],"mappings":"AAAA,0DAA0D;AAC1D,EAAE;AACF,+EAA+E;AAC/E,8EAA8E;AAC9E,8EAA8E;AAC9E,yEAAyE;AACzE,4EAA4E;AAE5E,OAAO,aAAa,MAAM,qBAAqB,CAAC;AAEhD,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AAGnC,IAAI,KAAK,GAAgB,IAAI,CAAC;AAE9B,SAAS,OAAO;IACd,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,KAAK,GAAG,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC;IACvD,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,eAAe,CAAC,IAA8B;IAC5D,MAAM,KAAK,GAAG,IAAI,YAAY,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC;IACvE,MAAM,SAAS,GAAG,OAAO,EAAE,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC;IACzD,MAAM,OAAO,GAAG,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACxC,MAAM,GAAG,GAAG,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE;QACtC,KAAK,EAAE,MAAM;QACb,KAAK,EAAE,MAAM;QACb,KAAK,EAAE,MAAM,EAAI,uCAAuC;QACxD,QAAQ,EAAE,KAAK;KAChB,CAAC,CAAC;IACH,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC;AACvB,CAAC"}

package/dist/proto_descriptor.d.ts

@@ -0,0 +1,4 @@
+import type { INamespace } from 'protobufjs/light.js';
+export declare const protoDescriptor: INamespace;
+export default protoDescriptor;
+//# sourceMappingURL=proto_descriptor.d.ts.map

package/dist/proto_descriptor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"proto_descriptor.d.ts","sourceRoot":"","sources":["../src/proto_descriptor.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAEtD,eAAO,MAAM,eAAe,EAi0EZ,UAAU,CAAC;AAE3B,eAAe,eAAe,CAAC"}