Skip to content

Commit 5953bb7

Browse files
committed
remove customization for enum fields
1 parent 3bb993e commit 5953bb7

File tree

2 files changed

+0
-30
lines changed

2 files changed

+0
-30
lines changed

src/providers/google-vertex-ai/utils.test.ts

Lines changed: 0 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -503,15 +503,6 @@ describe('transformGeminiToolParameters', () => {
503503
expect(transformed.$defs).toBeUndefined();
504504
});
505505

506-
it('adds format: "enum" for enum fields (status)', () => {
507-
expect(transformed.properties.status.enum).toEqual([
508-
'ACTIVE',
509-
'INACTIVE',
510-
'BANNED',
511-
]);
512-
expect(transformed.properties.status.format).toBe('enum');
513-
});
514-
515506
it('flattens anyOf [string, null] to { type: string, nullable: true } and preserves metadata (notes)', () => {
516507
expect(transformed.properties.notes).toEqual({
517508
type: 'string',
@@ -551,21 +542,6 @@ describe('transformGeminiToolParameters', () => {
551542
expect(union[1].type).toBe('object');
552543
});
553544

554-
it('adds format: "enum" for nested enums (preferences.notification_frequency, pet.species, social.platform)', () => {
555-
const nf =
556-
transformed.properties.preferences.properties.notification_frequency;
557-
expect(nf.enum).toEqual(['daily', 'weekly', 'monthly']);
558-
expect(nf.format).toBe('enum');
559-
560-
const species = transformed.properties.pets.items.properties.species;
561-
expect(species.enum).toEqual(['dog', 'cat', 'bird', 'other']);
562-
expect(species.format).toBe('enum');
563-
564-
const platform = transformed.properties.social.items.properties.platform;
565-
expect(platform.enum).toEqual(['twitter', 'linkedin', 'github', 'other']);
566-
expect(platform.format).toBe('enum');
567-
});
568-
569545
it('retains default values/titles when flattening (notes, contact.phone)', () => {
570546
expect(transformed.properties.notes.default).toBe(null);
571547
expect(transformed.properties.notes.title).toBe('Notes');

src/providers/google-vertex-ai/utils.ts

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -264,12 +264,6 @@ export const transformGeminiToolParameters = (
264264
const transformed: JsonSchema = {};
265265

266266
for (const [key, value] of Object.entries(node)) {
267-
if (key === 'enum' && Array.isArray(value)) {
268-
transformed.enum = value;
269-
transformed.format = 'enum';
270-
continue;
271-
}
272-
273267
if ((key === 'anyOf' || key === 'oneOf') && Array.isArray(value)) {
274268
const nonNullItems = value.filter((item) => !isNullTypeNode(item));
275269
const hadNull = nonNullItems.length < value.length;

0 commit comments

Comments
 (0)