Jialun He commited on
Commit
1ff2bf8
·
1 Parent(s): b74c22e

update model

Browse files
Files changed (2) hide show
  1. agent.py +1 -1
  2. app.py +0 -1
agent.py CHANGED
@@ -164,7 +164,7 @@ def build_graph(provider: str = "google"):
164
  # Load environment variables from .env file
165
  if provider == "google":
166
  # Google Gemini
167
- llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
168
  elif provider == "groq":
169
  # Groq https://console.groq.com/docs/models
170
  llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
 
164
  # Load environment variables from .env file
165
  if provider == "google":
166
  # Google Gemini
167
+ llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash-preview-05-20", temperature=0)
168
  elif provider == "groq":
169
  # Groq https://console.groq.com/docs/models
170
  llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
app.py CHANGED
@@ -116,7 +116,6 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
116
  except Exception as e:
117
  print(f"Error running agent on task {task_id}: {e}")
118
  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
119
- break
120
 
121
  if not answers_payload:
122
  print("Agent did not produce any answers to submit.")
 
116
  except Exception as e:
117
  print(f"Error running agent on task {task_id}: {e}")
118
  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
 
119
 
120
  if not answers_payload:
121
  print("Agent did not produce any answers to submit.")