Handle edge case where it's just doing normal chat
Browse files
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):
|