Skip to content

Commit d34e748

Browse files
committed
fix response
1 parent 2c8fb7f commit d34e748

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

packages/core/src/generation.ts

+8-2
Original file line numberDiff line numberDiff line change
@@ -371,6 +371,8 @@ export const generateObject = async ({
371371
schemaName,
372372
schemaDescription,
373373
stop,
374+
mode = 'json',
375+
enum: enumValues,
374376
verifiableInference,
375377
verifiableInferenceAdapter,
376378
verifiableInferenceOptions,
@@ -385,15 +387,19 @@ export const generateObject = async ({
385387
elizaLogger.debug(`Generating object with ${runtime.modelProvider} model. for ${schemaName}`);
386388
const { client, model } = initializeModelClient(runtime, modelClass);
387389

390+
if (output === 'enum' && !enumValues) {
391+
throw new Error('Enum values are required when output type is enum');
392+
}
388393

389394
const {object} = await aiGenerateObject({
390395
model: client.languageModel(model),
391396
prompt: context.toString(),
392397
system: runtime.character.system ?? settings.SYSTEM_PROMPT ?? undefined,
393398
output: output as never,
394399
...(schema ? { schema, schemaName, schemaDescription } : {}),
395-
mode: 'json'
396-
})
400+
...(enumValues ? { enum: enumValues } : {}),
401+
mode: mode as never
402+
});
397403

398404
elizaLogger.debug(`Received Object response from ${model} model.`);
399405

0 commit comments

Comments
 (0)