Upload 67 files

app.py

@@ -1002,11 +1002,10 @@ def run_search(
             html_line = _fmt_event(event)
             if html_line:
                 lines.append(html_line)
+                yield _wrap(lines), final_answer, "⚙  Running…"
 
             if event.get("type") == "__final__":
                 final_answer = event.get("text", "")
-
-            if len(lines) % 3 == 0:
                 yield _wrap(lines), final_answer, "⚙  Running…"
 
     except Exception as exc:

pi_wrapper.py

@@ -599,6 +599,9 @@ def run_pi_stream(
                 et = event.get("type")
 
                 if et == "response" and event.get("id") == request_id:
+                    if not event.get("success", True):
+                        yield {"type": "error", "error": event.get("error", "RPC prompt failed")}
+                        break
                     prompt_ack = True
                     continue