Patch json-bigint to preserve MLflow parsing behavior

Keep json-bigint using plain objects and numeric floats so frontend record
parsing and metric formatting keep working without the Databricks fork.

Signed-off-by: mprahl <mprahl@users.noreply.github.com>

Author: mprahl

mlflow/server/js/package.json

@@ -58,7 +58,7 @@
     "immutable": "3.8.1",
     "invariant": "^2.2.4",
     "js-yaml": "^3.14.0",
-    "json-bigint": "^1.0.0",
+    "json-bigint": "patch:json-bigint@npm%3A1.0.0#yarn/patches/json-bigint-npm-1.0.0-mlflow.patch",
     "leaflet": "^1.5.1",
     "lodash": "^4.17.21",
     "mod-arch-core": "^1.6.0",

mlflow/server/js/src/common/utils/FetchUtils.test.ts

@@ -136,6 +136,26 @@ describe('FetchUtils', () => {
       await new Promise(setImmediate);
       expect(mockResolve).toHaveBeenCalledWith({ a: '11111111222222223333333344444445555555555' });
     });
+    it('jsonBigIntResponseParser preserves plain object prototypes', async () => {
+      const mockResponse = {
+        text: () => Promise.resolve('{"a": {"b": 1}}'),
+      };
+      jsonBigIntResponseParser({ resolve: mockResolve, response: mockResponse });
+      await new Promise(setImmediate);
+      const parsed = mockResolve.mock.calls[0][0];
+      expect(parsed.hasOwnProperty('a')).toBe(true);
+      expect(parsed.a.hasOwnProperty('b')).toBe(true);
+    });
+    it('jsonBigIntResponseParser preserves high precision floats as numbers', async () => {
+      const mockResponse = {
+        text: () => Promise.resolve('{"a": 0.00011124613492593128}'),
+      };
+      jsonBigIntResponseParser({ resolve: mockResolve, response: mockResponse });
+      await new Promise(setImmediate);
+      const parsed = mockResolve.mock.calls[0][0];
+      expect(typeof parsed.a).toBe('number');
+      expect(parsed.a.toExponential(3)).toBe('1.112e-4');
+    });
     it('yamlResponseParser', async () => {
       const mockResponse = {
         text: () => Promise.resolve('artifact_path: model_signature\nflavors:\n  keras:\n    data: data\n'),

mlflow/server/js/yarn.lock

@@ -4892,7 +4892,7 @@ __metadata:
     jest-canvas-mock: "npm:^2.2.0"
     jest-localstorage-mock: "npm:^2.3.0"
     js-yaml: "npm:^3.14.0"
-    json-bigint: "npm:^1.0.0"
+    json-bigint: "patch:json-bigint@npm%3A1.0.0#yarn/patches/json-bigint-npm-1.0.0-mlflow.patch"
     knip: "npm:^5.30.2"
     leaflet: "npm:^1.5.1"
     lodash: "npm:^4.17.21"
@@ -23405,7 +23405,7 @@ __metadata:
   languageName: node
   linkType: hard
 
-"json-bigint@npm:^1.0.0":
+"json-bigint@npm:1.0.0":
   version: 1.0.0
   resolution: "json-bigint@npm:1.0.0"
   dependencies:
@@ -23414,6 +23414,15 @@ __metadata:
   languageName: node
   linkType: hard
 
+"json-bigint@patch:json-bigint@npm%3A1.0.0#yarn/patches/json-bigint-npm-1.0.0-mlflow.patch::locator=%40mlflow%2Fmlflow%40workspace%3A.":
+  version: 1.0.0
+  resolution: "json-bigint@patch:json-bigint@npm%3A1.0.0#yarn/patches/json-bigint-npm-1.0.0-mlflow.patch::version=1.0.0&hash=61ac52&locator=%40mlflow%2Fmlflow%40workspace%3A."
+  dependencies:
+    bignumber.js: "npm:^9.0.0"
+  checksum: 10c0/d5f02a7423bf44af46adbe4e604fd295445ee4681249ad485997bb16209475869b44f6fbaec1309ee42b79606f6a3d9a6a787c64993e67cfb6d16fa6012f59cc
+  languageName: node
+  linkType: hard
+
 "json-parse-better-errors@npm:^1.0.2":
   version: 1.0.2
   resolution: "json-parse-better-errors@npm:1.0.2"

mlflow/server/js/yarn/patches/json-bigint-npm-1.0.0-mlflow.patch

@@ -0,0 +1,49 @@
+diff --git a/lib/parse.js b/lib/parse.js
+index bb4e5eb..39363cd 100644
+--- a/lib/parse.js
++++ b/lib/parse.js
+@@ -207,19 +207,22 @@ var json_parse = function (options) {
+       } else {
+         if (BigNumber == null) BigNumber = require('bignumber.js');
+         //if (number > 9007199254740992 || number < -9007199254740992)
+-        // Bignumber has stricter check: everything with length > 15 digits disallowed
+-        if (string.length > 15)
+-          return _options.storeAsString
+-            ? string
+-            : _options.useNativeBigInt
+-            ? BigInt(string)
+-            : new BigNumber(string);
+-        else
+-          return !_options.alwaysParseAsBig
+-            ? number
+-            : _options.useNativeBigInt
+-            ? BigInt(number)
+-            : new BigNumber(number);
++        if (_options.storeAsString) {
++          if (number >= Number.MAX_SAFE_INTEGER || number <= Number.MIN_SAFE_INTEGER) {
++            return string;
++          }
++        } else {
++          // Bignumber has stricter check: everything with length > 15 digits disallowed
++          if (string.length > 15)
++            return _options.useNativeBigInt
++              ? BigInt(string)
++              : new BigNumber(string);
++        }
++        return !_options.alwaysParseAsBig
++          ? number
++          : _options.useNativeBigInt
++          ? BigInt(number)
++          : new BigNumber(number);
+       }
+     },
+     string = function () {
+@@ -327,7 +330,7 @@ var json_parse = function (options) {
+       // Parse an object value.
+ 
+       var key,
+-        object = Object.create(null);
++        object = {};
+ 
+       if (ch === '{') {
+         next('{');