fill out Schema builder

hsubox76 · hsubox76 · commit 6290b0356896 · 2024-09-25T09:18:53.000-07:00
diff --git a/packages/vertexai/src/requests/schema-builder.ts b/packages/vertexai/src/requests/schema-builder.ts
@@ -211,6 +211,15 @@ export class StringSchema extends Schema {
     }
     return obj as SchemaRequest;
   }
+
+  toJSON(): Record<string, unknown> {
+    const obj = super.toJSON();
+    if (this.enumValues) {
+      obj['enum'] = this.enumValues;
+      delete obj.enumValues;
+    }
+    return obj;
+  }
 }
 
 /**

Original file line number	Diff line number	Diff line change
`@@ -211,6 +211,15 @@ export class StringSchema extends Schema {`
`211`	`211`	`}`
`212`	`212`	`return obj as SchemaRequest;`
`213`	`213`	`}`
	`214`	`+`
	`215`	`+ toJSON(): Record<string, unknown> {`
	`216`	`+ const obj = super.toJSON();`
	`217`	`+ if (this.enumValues) {`
	`218`	`+ obj['enum'] = this.enumValues;`
	`219`	`+ delete obj.enumValues;`
	`220`	`+ }`
	`221`	`+ return obj;`
	`222`	`+ }`
`214`	`223`	`}`
`215`	`224`
`216`	`225`	`/**`