kenqia commited on
Commit
58a54f8
·
verified ·
1 Parent(s): d36b5fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -23
app.py CHANGED
@@ -37,7 +37,7 @@ class BasicAgent:
37
  def __init__(self):
38
  print("BasicAgent initialized.")
39
 
40
- self.MODEL_NAME = "qwen-plus"
41
  # export DASHSCOPE_API_KEY="sk-**"
42
  self.api_key = os.getenv("DASHSCOPE_API_KEY")
43
  if not self.api_key:
@@ -145,31 +145,31 @@ class BasicAgent:
145
  "messages": [self.chat_with_tools.invoke([sys_msg] + state["messages"])],
146
  }
147
  def _safe_parse_json(self, text: str) -> dict:
148
- """
149
- Parse JSON from model output safely.
150
- The model may sometimes wrap JSON with extra text.
151
- """
152
- text = text.strip()
153
-
154
- try:
155
- return json.loads(text)
156
- except Exception:
157
- pass
158
-
159
- start = text.find("{")
160
- end = text.rfind("}")
161
-
162
- if start != -1 and end != -1 and end > start:
163
  try:
164
- return json.loads(text[start:end + 1])
165
  except Exception:
166
  pass
167
-
168
- return {
169
- "verdict": "PASS",
170
- "revised_answer": text,
171
- "issue": "JSON parse failed, using raw verifier output.",
172
- }
 
 
 
 
 
 
 
 
 
173
 
174
 
175
  def verify_answer(self, state: AgentState):
 
37
  def __init__(self):
38
  print("BasicAgent initialized.")
39
 
40
+ self.MODEL_NAME = "qwen3.6-flash"
41
  # export DASHSCOPE_API_KEY="sk-**"
42
  self.api_key = os.getenv("DASHSCOPE_API_KEY")
43
  if not self.api_key:
 
145
  "messages": [self.chat_with_tools.invoke([sys_msg] + state["messages"])],
146
  }
147
  def _safe_parse_json(self, text: str) -> dict:
148
+ """
149
+ Parse JSON from model output safely.
150
+ The model may sometimes wrap JSON with extra text.
151
+ """
152
+ text = text.strip()
153
+
 
 
 
 
 
 
 
 
 
154
  try:
155
+ return json.loads(text)
156
  except Exception:
157
  pass
158
+
159
+ start = text.find("{")
160
+ end = text.rfind("}")
161
+
162
+ if start != -1 and end != -1 and end > start:
163
+ try:
164
+ return json.loads(text[start:end + 1])
165
+ except Exception:
166
+ pass
167
+
168
+ return {
169
+ "verdict": "PASS",
170
+ "revised_answer": text,
171
+ "issue": "JSON parse failed, using raw verifier output.",
172
+ }
173
 
174
 
175
  def verify_answer(self, state: AgentState):