[fix] Allow additional properties in ModelConfigSchema and DocumentChunkSchema

- Updated ModelConfigSchema and DocumentChunkSchema to permit additional properties, enhancing flexibility in schema definitions.
- Refactored getPortData method in Dataflow class to improve clarity and maintainability by using a result variable for return values.

Author: sroussey

packages/ai/src/model/ModelSchema.ts

@@ -26,7 +26,7 @@ export const ModelConfigSchema = {
   },
   required: ["provider", "provider_config"],
   format: "model",
-  additionalProperties: false,
+  additionalProperties: true,
 } as const satisfies DataPortSchemaObject;
 
 /**

packages/dataset/src/document-chunk/DocumentChunkSchema.ts

@@ -18,7 +18,7 @@ export const DocumentChunkSchema = {
     vector: TypedArraySchema(),
     metadata: { type: "object", format: "metadata", additionalProperties: true },
   },
-  additionalProperties: false,
+  additionalProperties: true,
 } as const satisfies DataPortSchemaObject;
 export type DocumentChunkSchema = typeof DocumentChunkSchema;
 

packages/task-graph/src/task-graph/Dataflow.ts

@@ -96,13 +96,15 @@ export class Dataflow {
   }
 
   getPortData(): TaskOutput {
+    let result: TaskOutput;
     if (this.targetTaskPortId === DATAFLOW_ALL_PORTS) {
-      return this.value;
+      result = this.value;
     } else if (this.targetTaskPortId === DATAFLOW_ERROR_PORT) {
-      return { [DATAFLOW_ERROR_PORT]: this.error };
+      result = { [DATAFLOW_ERROR_PORT]: this.error };
     } else {
-      return { [this.targetTaskPortId]: this.value };
+      result = { [this.targetTaskPortId]: this.value };
     }
+    return result;
   }
 
   toJSON(): DataflowJson {