step B.2 complete: all 23 tools use execute_on_client(); add embed() to llm

app/agents/checkpoint_agent.py

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-import json
 from typing import Any
 
 from langchain_core.messages import HumanMessage, SystemMessage
@@ -10,6 +9,7 @@ from langchain_core.tools import tool
 
 from app.core.agent_registry import ChatAgent, registry
 from app.core.llm import get_llm
+from app.core.ws_context import execute_on_client
 
 _SYSTEM_PROMPT = (
     "You are a project checkpoint assistant. Checkpoints are milestone dates that\n"
@@ -28,11 +28,16 @@ _SYSTEM_PROMPT = (
 @tool
 async def list_checkpoints(project_id: str = "") -> str:
     """List checkpoints. Provide project_id to scope to a specific project."""
-    return json.dumps({
-        "action": "list",
-        "table": "checkpoints",
-        "filters": {"projectId": project_id or None},
-    })
+    result = await execute_on_client(
+        action="select",
+        table="checkpoints",
+        filters={"projectId": project_id or None},
+    )
+    rows = result.get("rows", [])
+    if not rows:
+        return "No checkpoints found."
+    lines = [f"- {r['title']} (date: {r['date']}, id: {r['id']})" for r in rows]
+    return f"Found {len(rows)} checkpoint(s):\n" + "\n".join(lines)
 
 
 @tool
@@ -50,17 +55,19 @@ async def create_checkpoint(
     is_ai_suggested: 1 if proactively suggested, 0 if user-requested
     is_approved: 0 until the user confirms
     """
-    return json.dumps({
-        "action": "create_record",
-        "table": "checkpoints",
-        "data": {
+    result = await execute_on_client(
+        action="insert",
+        table="checkpoints",
+        data={
             "projectId": project_id,
             "title": title,
             "date": date,
             "isAiSuggested": is_ai_suggested,
             "isApproved": is_approved,
         },
-    })
+    )
+    row = result["row"]
+    return f"Checkpoint created: '{row['title']}' (id: {row['id']}, date: {row['date']})"
 
 
 @tool
@@ -82,21 +89,20 @@ async def update_checkpoint(
         updates["date"] = date
     if is_approved != -1:
         updates["isApproved"] = is_approved
-    return json.dumps({
-        "action": "update_record",
-        "table": "checkpoints",
-        "data": {"id": checkpoint_id, "updates": updates},
-    })
+    result = await execute_on_client(
+        action="update",
+        table="checkpoints",
+        data={"id": checkpoint_id, "updates": updates},
+    )
+    row = result["row"]
+    return f"Checkpoint updated: '{row['title']}' (id: {row['id']})"
 
 
 @tool
 async def delete_checkpoint(checkpoint_id: str) -> str:
     """Delete a checkpoint permanently by its UUID."""
-    return json.dumps({
-        "action": "delete_record",
-        "table": "checkpoints",
-        "data": {"id": checkpoint_id},
-    })
+    await execute_on_client(action="delete", table="checkpoints", data={"id": checkpoint_id})
+    return f"Checkpoint {checkpoint_id} deleted."
 
 
 @registry.register