There was an error while loading. Please reload this page.
1 parent 8f69ee8 commit d52d0baCopy full SHA for d52d0ba
1 file changed
flo_ai/flo_ai/llm/openai_vllm.py
@@ -37,14 +37,13 @@ async def generate(
37
if messages and messages[0]['role'] == 'system':
38
messages[0]['content'] = (
39
messages[0]['content']
40
- + '\n\nPlease provide your response in JSON format according to the specified schema.'
+ + f'\n\nPlease provide your response in JSON format according to the specified schema. \n\n {output_schema}'
41
)
42
else:
43
messages.insert(
44
- 0,
45
{
46
'role': 'system',
47
- 'content': 'Please provide your response in JSON format according to the specified schema.',
+ 'content': f'Please provide your response in JSON format according to the specified schema.\n \n {output_schema}',
48
},
49
50
0 commit comments