ChillThrills commited on
Commit
4811382
·
1 Parent(s): 78a1b32

gemini-2.5-flash-preview-05-20

Browse files
Files changed (1) hide show
  1. app.py +2 -11
app.py CHANGED
@@ -767,16 +767,7 @@ class GaiaLevel1Agent:
767
 
768
  if genai and GOOGLE_GEMINI_API_KEY:
769
  try:
770
- # The genai.configure call is no longer needed if GOOGLE_API_KEY env var is set,
771
- # or if the API key is passed directly to genai.Client()
772
- # However, keeping it for now as it doesn't hurt if GOOGLE_GEMINI_API_KEY is set.
773
- # If GOOGLE_API_KEY (the generic one) is intended for the client, it's often picked up automatically.
774
- # For clarity, if GOOGLE_GEMINI_API_KEY is specific and different, ensure Client uses it.
775
- # The new SDK typically uses GOOGLE_API_KEY from env.
776
- # If GOOGLE_GEMINI_API_KEY is specifically for GenAI and different from a general GOOGLE_API_KEY,
777
- # it should be passed to Client() if the client supports an api_key argument,
778
- # or ensure genai.configure(api_key=GOOGLE_GEMINI_API_KEY) correctly sets it for the client.
779
- # The migration guide implies genai.configure() still works or the client picks it up.
780
  genai.configure(api_key=GOOGLE_GEMINI_API_KEY)
781
  self.genai_client = genai.Client()
782
  gaia_logger.info(f"Google GenAI Client initialized. Will use model '{self.llm_model_name}'.")
@@ -1143,7 +1134,7 @@ class GaiaLevel1Agent:
1143
 
1144
 
1145
  try:
1146
- gen_config = GenerationConfig(temperature=0.1, top_p=0.95, max_output_tokens=1024)
1147
  safety_settings = [
1148
  {"category": HarmCategory.HARM_CATEGORY_HARASSMENT, "threshold": HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE},
1149
  {"category": HarmCategory.HARM_CATEGORY_HATE_SPEECH, "threshold": HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE},
 
767
 
768
  if genai and GOOGLE_GEMINI_API_KEY:
769
  try:
770
+
 
 
 
 
 
 
 
 
 
771
  genai.configure(api_key=GOOGLE_GEMINI_API_KEY)
772
  self.genai_client = genai.Client()
773
  gaia_logger.info(f"Google GenAI Client initialized. Will use model '{self.llm_model_name}'.")
 
1134
 
1135
 
1136
  try:
1137
+ gen_config = GenerationConfig(temperature=0.1, top_p=0.95, max_output_tokens=10024)
1138
  safety_settings = [
1139
  {"category": HarmCategory.HARM_CATEGORY_HARASSMENT, "threshold": HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE},
1140
  {"category": HarmCategory.HARM_CATEGORY_HATE_SPEECH, "threshold": HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE},