@api-client/core 0.19.3 → 0.19.5

package/data/models/example-generator-api.json

@@ -42062,6 +42062,9 @@
 "@id": "#209"
 },
 {
+"@id": "#206"
+},
+{
 "@id": "#191"
 },
 {
@@ -42077,9 +42080,6 @@
 "@id": "#203"
 },
 {
-"@id": "#206"
-},
-{
 "@id": "#209"
 }
 ],
@@ -43436,7 +43436,7 @@
 "doc:ExternalDomainElement",
 "doc:DomainElement"
 ],
-"doc:raw": "countryCode: \"BE\"\ngraydonEnterpriseId: 1057155523\nregistrationId: \"0422319093\"\nvatNumber: \"BE0422319093\"\ngraydonCompanyId: \"0422319093\"\nisBranchOffice: false\n",
+"doc:raw": "addressType: 'REGISTERED-OFFICE-ADDRESS'\nstreetName: 'UITBREIDINGSTRAAT'\nhouseNumber: '84'\nhouseNumberAddition: '/1'\npostalCode: '2600'\ncity: 'BERCHEM (ANTWERPEN)'\ncountry: 'Belgium'\ncountryCode: 'BE'\nfullFormatedAddress: \"UITBREIDINGSTRAAT 84 /1, 2600 BERCHEM (ANTWERPEN), BELIUM\"\n",
 "core:mediaType": "application/yaml",
 "sourcemaps:sources": [
 {
@@ -43457,7 +43457,7 @@
 "doc:ExternalDomainElement",
 "doc:DomainElement"
 ],
-"doc:raw": "addressType: 'REGISTERED-OFFICE-ADDRESS'\nstreetName: 'UITBREIDINGSTRAAT'\nhouseNumber: '84'\nhouseNumberAddition: '/1'\npostalCode: '2600'\ncity: 'BERCHEM (ANTWERPEN)'\ncountry: 'Belgium'\ncountryCode: 'BE'\nfullFormatedAddress: \"UITBREIDINGSTRAAT 84 /1, 2600 BERCHEM (ANTWERPEN), BELIUM\"\n",
+"doc:raw": "code: '5'\ndescription: 'Limited company'\n",
 "core:mediaType": "application/yaml",
 "sourcemaps:sources": [
 {
@@ -43478,7 +43478,7 @@
 "doc:ExternalDomainElement",
 "doc:DomainElement"
 ],
-"doc:raw": "code: '5'\ndescription: 'Limited company'\n",
+"doc:raw": "class: '3'\ndescription: '150 - 300'\nnumberOfFte: 5500\nnumberOfEmployees: 5232\n",
 "core:mediaType": "application/yaml",
 "sourcemaps:sources": [
 {
@@ -43499,7 +43499,7 @@
 "doc:ExternalDomainElement",
 "doc:DomainElement"
 ],
-"doc:raw": "class: '3'\ndescription: '150 - 300'\nnumberOfFte: 5500\nnumberOfEmployees: 5232\n",
+"doc:raw": "code: 'J'\ndescription: 'Information and communication'\n",
 "core:mediaType": "application/yaml",
 "sourcemaps:sources": [
 {
@@ -43520,7 +43520,7 @@
 "doc:ExternalDomainElement",
 "doc:DomainElement"
 ],
-"doc:raw": "code: 'J'\ndescription: 'Information and communication'\n",
+"doc:raw": "code: '7487'\ndescription: 'Financial and insurance activities'\ntype: \"PRIMARY\"\nclassificationCode: 'BE_NACEBEL2008'\nactivityGroupCode: 'ABCDE'\n",
 "core:mediaType": "application/yaml",
 "sourcemaps:sources": [
 {
@@ -43541,7 +43541,7 @@
 "doc:ExternalDomainElement",
 "doc:DomainElement"
 ],
-"doc:raw": "code: '7487'\ndescription: 'Financial and insurance activities'\ntype: \"PRIMARY\"\nclassificationCode: 'BE_NACEBEL2008'\nactivityGroupCode: 'ABCDE'\n",
+"doc:raw": "countryCode: \"BE\"\ngraydonEnterpriseId: 1057155523\nregistrationId: \"0422319093\"\nvatNumber: \"BE0422319093\"\ngraydonCompanyId: \"0422319093\"\nisBranchOffice: false\n",
 "core:mediaType": "application/yaml",
 "sourcemaps:sources": [
 {
@@ -44756,32 +44756,32 @@
 {
 "@id": "#193/source-map/lexical/element_0",
 "sourcemaps:element": "amf://id#193",
-"sourcemaps:value": "[(1,0)-(7,0)]"
+"sourcemaps:value": "[(1,0)-(10,0)]"
 },
 {
 "@id": "#196/source-map/lexical/element_0",
 "sourcemaps:element": "amf://id#196",
-"sourcemaps:value": "[(1,0)-(10,0)]"
+"sourcemaps:value": "[(1,0)-(3,0)]"
 },
 {
 "@id": "#199/source-map/lexical/element_0",
 "sourcemaps:element": "amf://id#199",
-"sourcemaps:value": "[(1,0)-(3,0)]"
+"sourcemaps:value": "[(1,0)-(5,0)]"
 },
 {
 "@id": "#202/source-map/lexical/element_0",
 "sourcemaps:element": "amf://id#202",
-"sourcemaps:value": "[(1,0)-(5,0)]"
+"sourcemaps:value": "[(1,0)-(3,0)]"
 },
 {
 "@id": "#205/source-map/lexical/element_0",
 "sourcemaps:element": "amf://id#205",
-"sourcemaps:value": "[(1,0)-(3,0)]"
+"sourcemaps:value": "[(1,0)-(6,0)]"
 },
 {
 "@id": "#208/source-map/lexical/element_0",
 "sourcemaps:element": "amf://id#208",
-"sourcemaps:value": "[(1,0)-(6,0)]"
+"sourcemaps:value": "[(1,0)-(7,0)]"
 },
 {
 "@id": "#223/source-map/lexical/element_0",

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@api-client/core",
   "description": "The API Client's core client library. Works in NodeJS and in a ES enabled browser.",
-  "version": "0.19.3",
+  "version": "0.19.5",
   "license": "Apache-2.0",
   "exports": {
     "./browser.js": {

package/src/modeling/ai/DataDomainDelta.ts

@@ -119,7 +119,7 @@ export class DataDomainDelta {
 
     if (delta.addedEntities) {
       for (const e of delta.addedEntities) {
-        this.handleAddEntity(this.domain, e)
+        this.handleAddEntity(e)
       }
     }
 
@@ -150,6 +150,9 @@ export class DataDomainDelta {
         if (mod.name) entity.info.name = mod.name
         if (mod.displayName) entity.info.displayName = mod.displayName
         if (mod.description) entity.info.description = mod.description
+        if (mod.tags !== undefined) {
+          entity.tags = mod.tags === null ? [] : mod.tags
+        }
 
         this.applySemantics(entity, mod.addedSemantics, mod.modifiedSemantics, mod.deletedSemanticIds)
 
@@ -181,8 +184,8 @@ export class DataDomainDelta {
    * @param domain The data domain where the entity will reside.
    * @param e The schema definition of the entity to be added.
    */
-  protected handleAddEntity(domain: DataDomain, e: AiDomainEntityResponseSchema): void {
-    const targetModel = this.getModel(domain, e.modelKey || 'ai_generated')
+  protected handleAddEntity(e: AiDomainEntityResponseSchema): void {
+    const targetModel = this.getModel(this.domain, e.modelKey || 'ai_generated')
     const entityInput = { ...e }
     delete entityInput.semantics
     let entityKey = this.keyMap.get(e.key)
@@ -261,7 +264,7 @@ export class DataDomainDelta {
       writeOnly: p.constraints?.writeOnly,
       deprecated: p.deprecated,
       schema: p.schema,
-      tags: p.tags,
+      tags: p.tags || undefined,
     })
     this.applySemantics(prop, p.semantics)
   }
@@ -273,7 +276,7 @@ export class DataDomainDelta {
    * @param domain The target data domain.
    * @param keys The array of property keys (AI-generated or actual) to remove.
    */
-  handlePropertyDeletes(domain: DataDomain, keys?: string[]): void {
+  protected handlePropertyDeletes(domain: DataDomain, keys?: string[]): void {
     if (!keys) {
       return
     }
@@ -294,7 +297,7 @@ export class DataDomainDelta {
    * @param domain The target data domain.
    * @param mods The list of property modifications to apply.
    */
-  handlePropertyMods(domain: DataDomain, mods?: AiDomainPropertyDelta[]): void {
+  protected handlePropertyMods(domain: DataDomain, mods?: AiDomainPropertyDelta[]): void {
     if (!mods) {
       return
     }
@@ -317,72 +320,76 @@ export class DataDomainDelta {
         prop.type = mod.type
       }
       if (mod.deprecated !== undefined) {
-        prop.deprecated = mod.deprecated
+        prop.deprecated = mod.deprecated === null ? undefined : mod.deprecated
       }
       if (mod.constraints) {
         if (mod.constraints.required !== undefined) {
-          prop.required = mod.constraints.required
+          prop.required = mod.constraints.required === null ? undefined : mod.constraints.required
         }
         if (mod.constraints.unique !== undefined) {
-          prop.unique = mod.constraints.unique
+          prop.unique = mod.constraints.unique === null ? undefined : mod.constraints.unique
         }
         if (mod.constraints.index !== undefined) {
-          prop.index = mod.constraints.index
+          prop.index = mod.constraints.index === null ? undefined : mod.constraints.index
         }
         if (mod.constraints.primary !== undefined) {
-          prop.primary = mod.constraints.primary
+          prop.primary = mod.constraints.primary === null ? undefined : mod.constraints.primary
         }
         if (mod.constraints.multiple !== undefined) {
-          prop.multiple = mod.constraints.multiple
+          prop.multiple = mod.constraints.multiple === null ? undefined : mod.constraints.multiple
         }
         if (mod.constraints.readOnly !== undefined) {
-          prop.readOnly = mod.constraints.readOnly
+          prop.readOnly = mod.constraints.readOnly === null ? undefined : mod.constraints.readOnly
         }
         if (mod.constraints.writeOnly !== undefined) {
-          prop.writeOnly = mod.constraints.writeOnly
+          prop.writeOnly = mod.constraints.writeOnly === null ? undefined : mod.constraints.writeOnly
         }
       }
       if (mod.schema) {
         if (!prop.schema) {
           prop.schema = {}
         }
-        if (mod.schema.defaultValue) {
-          if (!prop.schema.defaultValue) {
-            prop.schema.defaultValue = {
-              value: '',
-              type: 'literal',
+        if (mod.schema.defaultValue !== undefined) {
+          if (mod.schema.defaultValue === null) {
+            prop.schema.defaultValue = undefined
+          } else {
+            if (!prop.schema.defaultValue) {
+              prop.schema.defaultValue = {
+                value: '',
+                type: 'literal',
+              }
+            }
+            if (mod.schema.defaultValue.type) {
+              prop.schema.defaultValue.type = mod.schema.defaultValue.type
+            }
+            if (mod.schema.defaultValue.value) {
+              prop.schema.defaultValue.value = mod.schema.defaultValue.value
             }
-          }
-          if (mod.schema.defaultValue.type) {
-            prop.schema.defaultValue.type = mod.schema.defaultValue.type
-          }
-          if (mod.schema.defaultValue.value) {
-            prop.schema.defaultValue.value = mod.schema.defaultValue.value
           }
         }
         if (mod.schema.pattern !== undefined) {
-          prop.schema.pattern = mod.schema.pattern
+          prop.schema.pattern = mod.schema.pattern === null ? undefined : mod.schema.pattern
         }
         if (mod.schema.minimum !== undefined) {
-          prop.schema.minimum = mod.schema.minimum
+          prop.schema.minimum = mod.schema.minimum === null ? undefined : mod.schema.minimum
         }
         if (mod.schema.maximum !== undefined) {
-          prop.schema.maximum = mod.schema.maximum
+          prop.schema.maximum = mod.schema.maximum === null ? undefined : mod.schema.maximum
         }
         if (mod.schema.exclusiveMinimum !== undefined) {
-          prop.schema.exclusiveMinimum = mod.schema.exclusiveMinimum
+          prop.schema.exclusiveMinimum = mod.schema.exclusiveMinimum === null ? undefined : mod.schema.exclusiveMinimum
         }
         if (mod.schema.exclusiveMaximum !== undefined) {
-          prop.schema.exclusiveMaximum = mod.schema.exclusiveMaximum
+          prop.schema.exclusiveMaximum = mod.schema.exclusiveMaximum === null ? undefined : mod.schema.exclusiveMaximum
         }
         if (mod.schema.multipleOf !== undefined) {
-          prop.schema.multipleOf = mod.schema.multipleOf
+          prop.schema.multipleOf = mod.schema.multipleOf === null ? undefined : mod.schema.multipleOf
         }
         if (mod.schema.enum !== undefined) {
-          prop.schema.enum = [...mod.schema.enum]
+          prop.schema.enum = mod.schema.enum === null ? undefined : [...mod.schema.enum]
         }
         if (mod.schema.examples !== undefined) {
-          prop.schema.examples = [...mod.schema.examples]
+          prop.schema.examples = mod.schema.examples === null ? undefined : [...mod.schema.examples]
         }
       }
       this.applySemantics(prop, mod.addedSemantics, mod.modifiedSemantics, mod.deletedSemanticIds)
@@ -395,7 +402,7 @@ export class DataDomainDelta {
    * @param entity The target domain entity referencing the newly created associations.
    * @param adds An array of association definitions to add, if any.
    */
-  handleAssociationAdds(entity: DomainEntity, adds: AiDomainAssociation[] | undefined): void {
+  protected handleAssociationAdds(entity: DomainEntity, adds: AiDomainAssociation[] | undefined): void {
     if (!adds) {
       return
     }
@@ -412,7 +419,7 @@ export class DataDomainDelta {
    * @param entity The source domain entity creating the association.
    * @param a The defining schema for the new association.
    */
-  handleAddAssociation(entity: DomainEntity, a: AiDomainAssociation): void {
+  protected handleAddAssociation(entity: DomainEntity, a: AiDomainAssociation): void {
     let assocKey = this.keyMap.get(a.key)
     if (!assocKey) {
       assocKey = nanoid()
@@ -467,7 +474,7 @@ export class DataDomainDelta {
    * @param domain The target data domain.
    * @param keys An array of association keys to safely remove.
    */
-  handleAssociationDeletes(domain: DataDomain, keys?: string[]): void {
+  protected handleAssociationDeletes(domain: DataDomain, keys?: string[]): void {
     if (!keys) {
       return
     }
@@ -488,7 +495,7 @@ export class DataDomainDelta {
    * @param domain The target data domain containing the associations.
    * @param mods An array of requested association modifications.
    */
-  handleAssociationMods(domain: DataDomain, mods: AiDomainAssociationDelta[] | undefined): void {
+  protected handleAssociationMods(domain: DataDomain, mods: AiDomainAssociationDelta[] | undefined): void {
     if (!mods) {
       return
     }
@@ -508,10 +515,13 @@ export class DataDomainDelta {
         assoc.info.description = mod.description
       }
       if (mod.required !== undefined) {
-        assoc.required = mod.required
+        assoc.required = mod.required === null ? undefined : mod.required
+      }
+      if (mod.multiple !== undefined) {
+        assoc.multiple = mod.multiple === null ? undefined : mod.multiple
       }
       if (mod.onDelete !== undefined) {
-        assoc.onDelete = mod.onDelete
+        assoc.onDelete = mod.onDelete === null ? undefined : mod.onDelete
       }
 
       if (mod.targets) {

package/src/modeling/ai/prompts/domain_system.ts

@@ -76,4 +76,10 @@ You are an enterprise Data Architect. You must actively defend against malicious
 - **Conciseness:** If an operation requires more than 10 properties, suggest splitting the task into two parts rather than outputting one massive JSON array.
 - **No Encoding:** Never attempt to output Base64, Hex, or encoded strings unless explicitly requested for a \`binary\` type.
 - **Loop Prevention:** If you find yourself repeating the same key or value more than 5 times in a single array, stop and ask the user for clarification.
+
+# PARTIAL UPDATES & DELETIONS
+When modifying existing objects in the \`modifiedEntities\` or \`modifiedModels\` arrays, use strict partial-update logic:
+- **To update a value:** Provide the key and the new value.
+- **To leave a value unchanged:** Completely omit the key from the JSON object. Do not output it.
+- **To delete an existing value:** You MUST explicitly output the key with a value of \`null\` (e.g., \`"description": null\`).
 `

package/src/modeling/ai/types.ts

@@ -132,7 +132,7 @@ export interface AiDomainProperty {
   deprecated?: boolean
   schema?: PropertySchema
   semantics?: AiDomainSemantic[]
-  tags?: string[]
+  tags?: string[] | null
 }
 
 /**
@@ -226,7 +226,7 @@ export interface AiDomainEntityDelta {
   name?: string
   displayName?: string
   description?: string
-  tags?: string[]
+  tags?: string[] | null
   // Entity semantics delta
   addedSemantics?: AiDomainSemantic[]
   modifiedSemantics?: AiDomainSemantic[]
@@ -240,6 +240,10 @@ export interface AiDomainEntityDelta {
   deletedAssociationKeys?: string[]
 }
 
+export type NullablePropertySchema = {
+  [K in keyof PropertySchema]?: PropertySchema[K] | null
+}
+
 /**
  * Represents a targeted modification to an existing property generated by the AI.
  * It captures changes to superficial properties (name, description), type changes,
@@ -252,20 +256,20 @@ export interface AiDomainPropertyDelta {
   description?: string
   type?: DomainPropertyType
   constraints?: {
-    required?: boolean
-    unique?: boolean
-    index?: boolean
-    primary?: boolean
-    multiple?: boolean
-    readOnly?: boolean
-    writeOnly?: boolean
-  }
-  deprecated?: boolean
+    required?: boolean | null
+    unique?: boolean | null
+    index?: boolean | null
+    primary?: boolean | null
+    multiple?: boolean | null
+    readOnly?: boolean | null
+    writeOnly?: boolean | null
+  } | null
+  deprecated?: boolean | null
   // Property semantics delta
   addedSemantics?: AiDomainSemantic[]
   modifiedSemantics?: AiDomainSemantic[]
   deletedSemanticIds?: SemanticType[]
-  schema?: PropertySchema
+  schema?: NullablePropertySchema
 }
 
 /**
@@ -279,8 +283,9 @@ export interface AiDomainAssociationDelta {
   displayName?: string
   description?: string
   targets?: AssociationTarget[]
-  required?: boolean
-  onDelete?: OnDeleteRule
+  required?: boolean | null
+  multiple?: boolean | null
+  onDelete?: OnDeleteRule | null
   // Association semantics delta
   addedSemantics?: AiDomainSemantic[]
   modifiedSemantics?: AiDomainSemantic[]

package/src/models/AiSession.ts

@@ -3,6 +3,13 @@ import { AiSessionKind } from './kinds.js'
 import { nanoid } from '../nanoid.js'
 import type { AiModelMessage } from './AiMessage.js'
 
+/**
+ * The application the session is for.
+ *
+ * - `domain` - The session is for data domain modeling.
+ * - `api` - The session is for API modeling.
+ * - `general` - The session is for general AI chat.
+ */
 export type AiSessionApp = 'domain' | 'api' | 'general'
 
 export interface AiSessionSchema {
@@ -51,6 +58,11 @@ export interface AiSessionSchema {
    * Always set when the `delete` is true.
    */
   deletedInfo?: IDeletion
+  /**
+   * If the session is related to a file (data domain, API model, etc.), this is the key of the file.
+   * It is useful to filter session by specific object.
+   */
+  fileKey?: string
 }
 
 export class AiSession implements AiSessionSchema {
@@ -71,6 +83,7 @@ export class AiSession implements AiSessionSchema {
   updatedAt: number
   deleted: boolean
   deletedInfo?: IDeletion
+  fileKey?: string
 
   /**
    * Checks if the given value is a valid AiSessionApp.
@@ -101,6 +114,9 @@ export class AiSession implements AiSessionSchema {
       result.deleted = input.deleted
       result.deletedInfo = input.deletedInfo
     }
+    if (input.fileKey) {
+      result.fileKey = input.fileKey
+    }
     return result
   }
 
@@ -115,6 +131,7 @@ export class AiSession implements AiSessionSchema {
     this.updatedAt = init.updatedAt
     this.deleted = init.deleted || false
     this.deletedInfo = init.deletedInfo
+    this.fileKey = init.fileKey
   }
 
   toJSON(): AiSessionSchema {
@@ -132,6 +149,9 @@ export class AiSession implements AiSessionSchema {
       result.deleted = this.deleted
       result.deletedInfo = { ...this.deletedInfo }
     }
+    if (this.fileKey) {
+      result.fileKey = this.fileKey
+    }
     return result
   }
 

package/tests/unit/modeling/ai/DataDomainDelta.spec.ts

@@ -189,6 +189,116 @@ test.group('DataDomainDelta.apply()', () => {
     assert.equal(prop2!.info.name, 'New Prop')
   })
 
+  test('unsets properties, constraints, schema, and association options using null values', ({ assert }) => {
+    const domain = new DataDomain()
+    const model = domain.addModel()
+    const entity = domain.addEntity(model.key, { info: { name: 'unset-entity' }, tags: ['tag1'] })
+
+    entity.addProperty({
+      key: 'prop1',
+      type: 'string',
+      deprecated: true,
+      required: true,
+      unique: true,
+      index: true,
+      primary: true,
+      multiple: true,
+      readOnly: true,
+      writeOnly: true,
+      schema: {
+        defaultValue: { type: 'literal', value: 'foo' },
+        pattern: '^[a-z]+$',
+        minimum: 1,
+        maximum: 10,
+        exclusiveMinimum: true,
+        exclusiveMaximum: true,
+        multipleOf: 2,
+        enum: ['foo', 'bar'],
+        examples: ['foo'],
+      },
+    })
+
+    entity.addAssociation({
+      key: 'assoc1',
+      required: true,
+      multiple: true,
+      onDelete: 'cascade',
+    })
+
+    const delta = {
+      modifiedEntities: [
+        {
+          key: entity.key,
+          tags: null,
+          modifiedProperties: [
+            {
+              key: 'prop1',
+              deprecated: null,
+              constraints: {
+                required: null,
+                unique: null,
+                index: null,
+                primary: null,
+                multiple: null,
+                readOnly: null,
+                writeOnly: null,
+              },
+              schema: {
+                defaultValue: null,
+                pattern: null,
+                minimum: null,
+                maximum: null,
+                exclusiveMinimum: null,
+                exclusiveMaximum: null,
+                multipleOf: null,
+                enum: null,
+                examples: null,
+              },
+            },
+          ],
+          modifiedAssociations: [
+            {
+              key: 'assoc1',
+              required: null,
+              multiple: null,
+              onDelete: null,
+            },
+          ],
+        },
+      ],
+    }
+
+    const instance = new DataDomainDelta(domain)
+    instance.apply(delta)
+
+    assert.lengthOf(entity.tags, 0)
+
+    const prop1 = domain.findProperty('prop1')!
+    assert.isUndefined(prop1.deprecated)
+    assert.isUndefined(prop1.required)
+    assert.isUndefined(prop1.unique)
+    assert.isUndefined(prop1.index)
+    assert.isUndefined(prop1.primary)
+    assert.isUndefined(prop1.multiple)
+    assert.isUndefined(prop1.readOnly)
+    assert.isUndefined(prop1.writeOnly)
+
+    assert.isUndefined(prop1.schema?.defaultValue)
+    assert.isUndefined(prop1.schema?.pattern)
+    assert.isUndefined(prop1.schema?.minimum)
+    assert.isUndefined(prop1.schema?.maximum)
+    assert.isUndefined(prop1.schema?.exclusiveMinimum)
+    assert.isUndefined(prop1.schema?.exclusiveMaximum)
+    assert.isUndefined(prop1.schema?.multipleOf)
+    assert.isUndefined(prop1.schema?.enum)
+    assert.isUndefined(prop1.schema?.examples)
+
+    const assoc1 = domain.findAssociation('assoc1')!
+    assert.isUndefined(assoc1.required)
+    assert.isUndefined(assoc1.multiple)
+    assert.isUndefined(assoc1.onDelete)
+  })
+
   test('deletes entity properties', ({ assert }) => {
     const domain = new DataDomain()
     const model = domain.addModel()

package/tests/unit/models/AiSession.spec.ts

@@ -39,6 +39,7 @@ test.group('AiSession', () => {
     assert.equal(schema.updatedAt, 0)
     assert.isUndefined(schema.deleted)
     assert.isUndefined(schema.deletedInfo)
+    assert.isUndefined(schema.fileKey)
   })
 
   test('constructor builds AiSession instance', ({ assert }) => {
@@ -50,6 +51,7 @@ test.group('AiSession', () => {
       userRenamed: true,
       createdAt: 100,
       updatedAt: 200,
+      fileKey: 'file-123',
     })
 
     assert.equal(session.key, 'test-key')
@@ -61,6 +63,7 @@ test.group('AiSession', () => {
     assert.equal(session.updatedAt, 200)
     assert.isFalse(session.deleted)
     assert.isUndefined(session.deletedInfo)
+    assert.equal(session.fileKey, 'file-123')
   })
 
   test('toJSON correctly serializes an AiSession', ({ assert }) => {
@@ -75,6 +78,7 @@ test.group('AiSession', () => {
         byMe: true,
         user: 'user1',
       },
+      fileKey: 'file-123',
     })
     session.title = 'Test Title'
 
@@ -94,6 +98,7 @@ test.group('AiSession', () => {
         byMe: true,
         user: 'user1',
       },
+      fileKey: 'file-123',
     })
   })