Fix Live lifecycle bugs in query.py streaming

- Drop eager Live restart after a tool call, mirroring the lazy pattern
  used in chat.py (commit 8708fc4 applied the fix only to chat.py)
- Start Live inside the try block so a synchronous raise cannot leak it

Author: rejojer

openkb/agent/query.py

@@ -137,20 +137,22 @@ def _start_live() -> Live | None:
         lv.start()
         return lv
 
-    live = _start_live()
-
+    live: Live | None = None
     result = Runner.run_streamed(agent, question, max_turns=MAX_TURNS)
     collected: list[str] = []
     segment: list[str] = []
     try:
+        live = _start_live()
         async for event in result.stream_events():
             if isinstance(event, RawResponsesStreamEvent):
                 if isinstance(event.data, ResponseTextDeltaEvent):
                     text = event.data.delta
                     if text:
                         collected.append(text)
                         segment.append(text)
-                        if live:
+                        if console is not None:
+                            if live is None:
+                                live = _start_live()
                             live.update(_make_markdown("".join(segment)))
                         else:
                             sys.stdout.write(text)
@@ -166,7 +168,6 @@ def _start_live() -> Live | None:
                     sys.stdout.write(f"\n[tool call] {raw.name}({args})\n\n")
                     sys.stdout.flush()
                     segment = []
-                    live = _start_live()
                 elif item.type == "tool_call_output_item":
                     pass
     finally: