fix (ai/core): streamObject fixes partial json with empty objects (#1529)

Author: lgrammel

.changeset/few-bobcats-mate.md

@@ -0,0 +1,5 @@
+---
+'ai': patch
+---
+
+fix (ai/core): streamObject fixes partial json with empty objects correctly

packages/core/core/util/fix-json.test.ts

@@ -168,9 +168,13 @@ describe('object', () => {
     assert.strictEqual(fixJson('{"k1": 1, "k2":'), '{"k1": 1}');
   });
 
-  test('should handle trailing whitespaces', () => {
+  test('should handle trailing whitespace', () => {
     assert.strictEqual(fixJson('{"key": "value"  '), '{"key": "value"}');
   });
+
+  test('should handle closing after empty object', () => {
+    assert.strictEqual(fixJson('{"a": {"b": {}'), '{"a": {"b": {}}}');
+  });
 });
 
 describe('nesting', () => {
@@ -265,4 +269,11 @@ describe('regression', () => {
       ].join('\n'),
     );
   });
+
+  test('should handle empty objects inside nested objects and arrays', () => {
+    assert.strictEqual(
+      fixJson(`{"type":"div","children":[{"type":"Card","props":{}`),
+      `{"type":"div","children":[{"type":"Card","props":{}}]}`,
+    );
+  });
 });

packages/core/core/util/fix-json.ts

@@ -140,6 +140,7 @@ export function fixJson(input: string): string {
             break;
           }
           case '}': {
+            lastValidIndex = i;
             stack.pop();
             break;
           }