implemented copilot proposed changes

Author: qchapp

src/ai_agent/agent/models.py

@@ -3,7 +3,7 @@
 from typing import List, Optional, Any, Dict
 from pydantic import BaseModel, Field
 
-from ai_agent.generator.schema import ToolChoice, ToolSelection, Conversation, ConversationStatus, CandidateDoc
+from ai_agent.generator.schema import ToolSelection, CandidateDoc
 
 class ToolRunLog(BaseModel):
     tool: str

src/ai_agent/cli.py

@@ -65,6 +65,7 @@ def run_chat():
             ok = pipe.reload_index()
             if ok:
                 log.info("[startup-refresh] reloaded FAISS index")
+                refresh_ui_docs_from_index()
             else:
                 log.warning("[startup-refresh] reload failed; serving previous index")
         else:

src/ai_agent/ui/components.py

@@ -288,7 +288,8 @@ def handle_chat(message: str, history: List[dict], files: List, state_dict: dict
                 )
 
                 # Remove thinking indicator
-                history.pop()
+                if history and history[-1] == thinking_msg:
+                    history.pop()
 
                 # Add assistant response with rich media
                 # Build text content first

src/ai_agent/ui/handlers.py

@@ -168,8 +168,12 @@ def respond(
     if state.last_preview_path:
         try:
             data_url = _to_supported_png_dataurl(state.last_preview_path)
-        except Exception:
-            pass
+        except Exception as e:
+            log.debug(
+                "Failed to build PNG data URL from preview %r: %r",
+                state.last_preview_path,
+                e,
+            )
 
     # Extract original formats
     original_formats = []

src/ai_agent/ui/state.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import json
+import os
 from typing import List, Dict, Any, Optional, Tuple
 from dataclasses import dataclass, field
 
@@ -69,7 +70,6 @@ def to_markdown(self) -> str:
 
         # Render file links
         for file_path, label in self.files:
-            import os
             if os.path.exists(file_path):
                 parts.append(f"\n📎 [{label}]({file_path})")