Quellcode durchsuchen

调整gemini prompt限制json格式

supeng vor 6 Tagen
Ursprung
Commit
a8a9a3e4ea
1 geänderte Dateien mit 2 neuen und 1 gelöschten Zeilen
  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={