Spaces:
Sleeping
Sleeping
Jialun He
commited on
Commit
·
1ff2bf8
1
Parent(s):
b74c22e
update model
Browse files
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.
|
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.")
|