소스 검색

调整gemini prompt限制json格式

supeng 1 년 전
부모
커밋
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={