Adjust temperature and max output tokens for content generation

py_files/gemini_analysis.py

@@ -122,10 +122,10 @@ def few_shots_generator(eval_dir='./eval', files=None, api_key=None):
                     contents=data,
                     config=types.GenerateContentConfig(
                         system_instruction=textsi_1,
-                        temperature=0,
+                        temperature=1,
                         top_p=0.1,
                         top_k=1,
-                        max_output_tokens=12288,
+                        max_output_tokens=24*1024,
                         safety_settings=[
                             types.SafetySetting(category='HARM_CATEGORY_HARASSMENT', threshold='BLOCK_NONE'),
                             types.SafetySetting(category='HARM_CATEGORY_HATE_SPEECH', threshold='BLOCK_NONE'),
@@ -321,10 +321,10 @@ def thinking_generator(eval_dir="./eval", files=None, api_key=None):
                         contents=data,
                         config=types.GenerateContentConfig(
                             system_instruction=textsi_1,
-                            temperature=0,
+                            temperature=1,
                             top_p=0.1,
                             top_k=1,
-                            max_output_tokens=65536,
+                            max_output_tokens=24*1024,
                             thinking_config=types.ThinkingConfig(include_thoughts=True),
                             safety_settings=[
                                 types.SafetySetting(category='HARM_CATEGORY_HARASSMENT', threshold='BLOCK_NONE'),