Skip to content

Commit befa2af

Browse files
committed
Add test of string-only usage with runtime generative
1 parent c16cbd3 commit befa2af

File tree

1 file changed

+32
-1
lines changed

1 file changed

+32
-1
lines changed

src/collections/generate/integration.test.ts

Lines changed: 32 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -460,7 +460,7 @@ maybe('Testing of the collection.generate methods with runtime generative config
460460
});
461461
});
462462

463-
it('should generate using a runtime config without search', async () => {
463+
it('should generate using a runtime config without search and with extras', async () => {
464464
const query = () =>
465465
collection.generate.fetchObjects({
466466
singlePrompt: {
@@ -498,4 +498,35 @@ maybe('Testing of the collection.generate methods with runtime generative config
498498
expect(obj.generative?.debug).toBeDefined();
499499
});
500500
});
501+
502+
it('should generate using a runtime config without search nor extras', async () => {
503+
const query = () =>
504+
collection.generate.fetchObjects({
505+
singlePrompt: 'Write a haiku about ducks for {testProp}',
506+
groupedTask: 'What is the value of testProp here?',
507+
config: {
508+
name: 'generative-openai',
509+
config: {
510+
model: 'gpt-4o-mini',
511+
},
512+
},
513+
});
514+
515+
if (await client.getWeaviateVersion().then((ver) => ver.isLowerThan(1, 30, 0))) {
516+
await expect(query()).rejects.toThrow(WeaviateUnsupportedFeatureError);
517+
return;
518+
}
519+
520+
const res = await query();
521+
expect(res.objects.length).toEqual(1);
522+
expect(res.generated).toBeDefined();
523+
expect(res.generative?.text).toBeDefined();
524+
expect(res.generative?.metadata).toBeUndefined();
525+
res.objects.forEach((obj) => {
526+
expect(obj.generated).toBeDefined();
527+
expect(obj.generative?.text).toBeDefined();
528+
expect(obj.generative?.metadata).toBeUndefined();
529+
expect(obj.generative?.debug).toBeUndefined();
530+
});
531+
});
501532
});

0 commit comments

Comments
 (0)