test: Change tests for new response in ai_extract_structured (box/b…

…ox-codegen#641)

.codegen.json

@@ -1 +1 @@
-{ "engineHash": "cf82faa", "specHash": "1fdcbef", "version": "1.11.0" }
+{ "engineHash": "a74691d", "specHash": "1fdcbef", "version": "1.11.0" }

src/internal/utils.ts

@@ -453,10 +453,10 @@ export async function delayInSeconds(seconds: number): Promise<void> {
  * @returns Value from object raw data.
  */
 export function getValueFromObjectRawData(obj: any, key: string): any {
-  if (obj && typeof obj === 'object' && obj.rawData) {
-    return obj.rawData[key];
+  if (!obj || typeof obj !== 'object' || !obj.rawData) {
+    return undefined;
   }
-  return undefined;
+  return key.split('.').reduce((value, k) => value?.[k], obj.rawData);
 }
 
 /**

src/test/ai.generated.test.ts

@@ -384,42 +384,51 @@ test('testAIExtractStructuredWithFields', async function testAIExtractStructured
     } satisfies AiExtractStructured);
   if (
     !(
-      (toString(getValueFromObjectRawData(response, 'firstName')) as string) ==
-      'John'
+      (toString(
+        getValueFromObjectRawData(response, 'answer.hobby'),
+      ) as string) == (['guitar'].map(toString).join(',') as string)
     )
   ) {
     throw new Error('Assertion failed');
   }
   if (
     !(
-      (toString(getValueFromObjectRawData(response, 'lastName')) as string) ==
-      'Doe'
+      (toString(
+        getValueFromObjectRawData(response, 'answer.firstName'),
+      ) as string) == 'John'
     )
   ) {
     throw new Error('Assertion failed');
   }
   if (
     !(
       (toString(
-        getValueFromObjectRawData(response, 'dateOfBirth'),
-      ) as string) == '1990-07-04'
+        getValueFromObjectRawData(response, 'answer.lastName'),
+      ) as string) == 'Doe'
     )
   ) {
     throw new Error('Assertion failed');
   }
   if (
-    !((toString(getValueFromObjectRawData(response, 'age')) as string) == '34')
+    !(
+      (toString(
+        getValueFromObjectRawData(response, 'answer.dateOfBirth'),
+      ) as string) == '1990-07-04'
+    )
   ) {
     throw new Error('Assertion failed');
   }
   if (
     !(
-      (toString(getValueFromObjectRawData(response, 'hobby')) as string) ==
-      (['guitar'].map(toString).join(',') as string)
+      (toString(getValueFromObjectRawData(response, 'answer.age')) as string) ==
+      '34'
     )
   ) {
     throw new Error('Assertion failed');
   }
+  if (!(response.completionReason == 'done')) {
+    throw new Error('Assertion failed');
+  }
   await client.files.deleteFileById(file.id);
 });
 test('testAIExtractStructuredWithMetadataTemplate', async function testAIExtractStructuredWithMetadataTemplate(): Promise<any> {
@@ -491,42 +500,51 @@ test('testAIExtractStructuredWithMetadataTemplate', async function testAIExtract
     } satisfies AiExtractStructured);
   if (
     !(
-      (toString(getValueFromObjectRawData(response, 'firstName')) as string) ==
-      'John'
+      (toString(
+        getValueFromObjectRawData(response, 'answer.firstName'),
+      ) as string) == 'John'
     )
   ) {
     throw new Error('Assertion failed');
   }
   if (
     !(
-      (toString(getValueFromObjectRawData(response, 'lastName')) as string) ==
-      'Doe'
+      (toString(
+        getValueFromObjectRawData(response, 'answer.lastName'),
+      ) as string) == 'Doe'
     )
   ) {
     throw new Error('Assertion failed');
   }
   if (
     !(
       (toString(
-        getValueFromObjectRawData(response, 'dateOfBirth'),
+        getValueFromObjectRawData(response, 'answer.dateOfBirth'),
       ) as string) == '1990-07-04T00:00:00Z'
     )
   ) {
     throw new Error('Assertion failed');
   }
   if (
-    !((toString(getValueFromObjectRawData(response, 'age')) as string) == '34')
+    !(
+      (toString(getValueFromObjectRawData(response, 'answer.age')) as string) ==
+      '34'
+    )
   ) {
     throw new Error('Assertion failed');
   }
   if (
     !(
-      (toString(getValueFromObjectRawData(response, 'hobby')) as string) ==
-      (['guitar'].map(toString).join(',') as string)
+      (toString(
+        getValueFromObjectRawData(response, 'answer.hobby'),
+      ) as string) == (['guitar'].map(toString).join(',') as string)
     )
   ) {
     throw new Error('Assertion failed');
   }
+  if (!(response.completionReason == 'done')) {
+    throw new Error('Assertion failed');
+  }
   await client.metadataTemplates.deleteMetadataTemplate(
     'enterprise' as DeleteMetadataTemplateScope,
     template.templateKey!,