Просмотр исходного кода

调整gemini prompt限制json格式

supeng 6 дней назад
Родитель
Сommit
a8a9a3e4ea
1 измененных файлов с 2 добавлено и 1 удалено
  1. 2 1
      utils/google_ai_studio.py

+ 2 - 1
utils/google_ai_studio.py

@@ -116,7 +116,8 @@ class GoogleAI(object):
             response = model.generate_content(
                 contents=[
                     video,
-                    f'{prompt}\n输出返回格式样例json输出:\n{SAMPLE_DATA}',
+                    # f'{prompt}\n输出返回格式样例json输出:\n{SAMPLE_DATA}',
+                    f'{prompt}\n除了无法分析的情况,尽量保证每一项都有结果,且严格按照以下Schema输出一个有效的JSON对象,不要包含任何额外的解释、注释或非JSON文本:\n{SAMPLE_DATA}\n\n请确保输出是且仅是一个合法的JSON字符串。',
                 ],
                 stream=False,
                 request_options={