kz919 commited on
Commit
6a09bf1
·
verified ·
1 Parent(s): 67f9994

Handle edge case where it's just doing normal chat

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -60,6 +60,9 @@ def parse_response(response):
60
  reflection = reflection_match.group(1).strip() if reflection_match else ""
61
  steps = re.findall(r'<step>(.*?)</step>', response, re.DOTALL)
62
 
 
 
 
63
  return answer, reflection, steps
64
 
65
  def generate(message, history, model, system_prompt, thinking_budget, api_key):
 
60
  reflection = reflection_match.group(1).strip() if reflection_match else ""
61
  steps = re.findall(r'<step>(.*?)</step>', response, re.DOTALL)
62
 
63
+ if answer == "":
64
+ return response, "", ""
65
+
66
  return answer, reflection, steps
67
 
68
  def generate(message, history, model, system_prompt, thinking_budget, api_key):