mirror of
https://github.com/leonvanzyl/autocoder.git
synced 2026-01-30 06:12:06 +00:00
feat: Add conversational AI assistant panel for project codebase Q&A
Implement a slide-in chat panel that allows users to ask questions about their codebase using Claude Opus 4.5 with read-only access to project files. Backend changes: - Add SQLAlchemy models for conversation persistence (assistant_database.py) - Create AssistantChatSession with read-only Claude SDK client - Add WebSocket endpoint for real-time chat streaming - Include read-only MCP tools: feature_get_stats, feature_get_next, etc. Frontend changes: - Add floating action button (bottom-right) to toggle panel - Create slide-in panel component (400px width) - Implement WebSocket hook with reconnection logic - Add keyboard shortcut 'A' to toggle assistant Key features: - Read-only access: Only Read, Glob, Grep, WebFetch, WebSearch tools - Persistent history: Conversations saved to SQLite per project - Real-time streaming: Text chunks streamed as Claude generates response - Tool visibility: Shows when assistant is using tools to explore code 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Auto
340
server/routers/assistant_chat.py
Normal file
340
server/routers/assistant_chat.py
Normal file
@@ -0,0 +1,340 @@
|
||||
"""
|
||||
Assistant Chat Router
|
||||
=====================
|
||||
|
||||
WebSocket and REST endpoints for the read-only project assistant.
|
||||
"""
|
||||
|
||||
import json
|
||||
import logging
|
||||
import re
|
||||
from pathlib import Path
|
||||
from typing import Optional
|
||||
|
||||
from fastapi import APIRouter, WebSocket, WebSocketDisconnect, HTTPException
|
||||
from pydantic import BaseModel
|
||||
|
||||
from ..services.assistant_chat_session import (
|
||||
AssistantChatSession,
|
||||
get_session,
|
||||
create_session,
|
||||
remove_session,
|
||||
list_sessions,
|
||||
)
|
||||
from ..services.assistant_database import (
|
||||
get_conversations,
|
||||
get_conversation,
|
||||
delete_conversation,
|
||||
create_conversation,
|
||||
)
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
router = APIRouter(prefix="/api/assistant", tags=["assistant-chat"])
|
||||
|
||||
# Root directory
|
||||
ROOT_DIR = Path(__file__).parent.parent.parent
|
||||
|
||||
|
||||
def _get_project_path(project_name: str) -> Optional[Path]:
|
||||
"""Get project path from registry."""
|
||||
import sys
|
||||
root = Path(__file__).parent.parent.parent
|
||||
if str(root) not in sys.path:
|
||||
sys.path.insert(0, str(root))
|
||||
|
||||
from registry import get_project_path
|
||||
return get_project_path(project_name)
|
||||
|
||||
|
||||
def validate_project_name(name: str) -> bool:
|
||||
"""Validate project name to prevent path traversal."""
|
||||
return bool(re.match(r'^[a-zA-Z0-9_-]{1,50}$', name))
|
||||
|
||||
|
||||
# ============================================================================
|
||||
# Pydantic Models
|
||||
# ============================================================================
|
||||
|
||||
class ConversationSummary(BaseModel):
|
||||
"""Summary of a conversation."""
|
||||
id: int
|
||||
project_name: str
|
||||
title: Optional[str]
|
||||
created_at: Optional[str]
|
||||
updated_at: Optional[str]
|
||||
message_count: int
|
||||
|
||||
|
||||
class ConversationMessageModel(BaseModel):
|
||||
"""A message within a conversation."""
|
||||
id: int
|
||||
role: str
|
||||
content: str
|
||||
timestamp: Optional[str]
|
||||
|
||||
|
||||
class ConversationDetail(BaseModel):
|
||||
"""Full conversation with messages."""
|
||||
id: int
|
||||
project_name: str
|
||||
title: Optional[str]
|
||||
created_at: Optional[str]
|
||||
updated_at: Optional[str]
|
||||
messages: list[ConversationMessageModel]
|
||||
|
||||
|
||||
class SessionInfo(BaseModel):
|
||||
"""Active session information."""
|
||||
project_name: str
|
||||
conversation_id: Optional[int]
|
||||
is_active: bool
|
||||
|
||||
|
||||
# ============================================================================
|
||||
# REST Endpoints - Conversation Management
|
||||
# ============================================================================
|
||||
|
||||
@router.get("/conversations/{project_name}", response_model=list[ConversationSummary])
|
||||
async def list_project_conversations(project_name: str):
|
||||
"""List all conversations for a project."""
|
||||
if not validate_project_name(project_name):
|
||||
raise HTTPException(status_code=400, detail="Invalid project name")
|
||||
|
||||
project_dir = _get_project_path(project_name)
|
||||
if not project_dir or not project_dir.exists():
|
||||
raise HTTPException(status_code=404, detail="Project not found")
|
||||
|
||||
conversations = get_conversations(project_dir, project_name)
|
||||
return [ConversationSummary(**c) for c in conversations]
|
||||
|
||||
|
||||
@router.get("/conversations/{project_name}/{conversation_id}", response_model=ConversationDetail)
|
||||
async def get_project_conversation(project_name: str, conversation_id: int):
|
||||
"""Get a specific conversation with all messages."""
|
||||
if not validate_project_name(project_name):
|
||||
raise HTTPException(status_code=400, detail="Invalid project name")
|
||||
|
||||
project_dir = _get_project_path(project_name)
|
||||
if not project_dir or not project_dir.exists():
|
||||
raise HTTPException(status_code=404, detail="Project not found")
|
||||
|
||||
conversation = get_conversation(project_dir, conversation_id)
|
||||
if not conversation:
|
||||
raise HTTPException(status_code=404, detail="Conversation not found")
|
||||
|
||||
return ConversationDetail(
|
||||
id=conversation["id"],
|
||||
project_name=conversation["project_name"],
|
||||
title=conversation["title"],
|
||||
created_at=conversation["created_at"],
|
||||
updated_at=conversation["updated_at"],
|
||||
messages=[ConversationMessageModel(**m) for m in conversation["messages"]],
|
||||
)
|
||||
|
||||
|
||||
@router.post("/conversations/{project_name}", response_model=ConversationSummary)
|
||||
async def create_project_conversation(project_name: str):
|
||||
"""Create a new conversation for a project."""
|
||||
if not validate_project_name(project_name):
|
||||
raise HTTPException(status_code=400, detail="Invalid project name")
|
||||
|
||||
project_dir = _get_project_path(project_name)
|
||||
if not project_dir or not project_dir.exists():
|
||||
raise HTTPException(status_code=404, detail="Project not found")
|
||||
|
||||
conversation = create_conversation(project_dir, project_name)
|
||||
return ConversationSummary(
|
||||
id=conversation.id,
|
||||
project_name=conversation.project_name,
|
||||
title=conversation.title,
|
||||
created_at=conversation.created_at.isoformat() if conversation.created_at else None,
|
||||
updated_at=conversation.updated_at.isoformat() if conversation.updated_at else None,
|
||||
message_count=0,
|
||||
)
|
||||
|
||||
|
||||
@router.delete("/conversations/{project_name}/{conversation_id}")
|
||||
async def delete_project_conversation(project_name: str, conversation_id: int):
|
||||
"""Delete a conversation."""
|
||||
if not validate_project_name(project_name):
|
||||
raise HTTPException(status_code=400, detail="Invalid project name")
|
||||
|
||||
project_dir = _get_project_path(project_name)
|
||||
if not project_dir or not project_dir.exists():
|
||||
raise HTTPException(status_code=404, detail="Project not found")
|
||||
|
||||
success = delete_conversation(project_dir, conversation_id)
|
||||
if not success:
|
||||
raise HTTPException(status_code=404, detail="Conversation not found")
|
||||
|
||||
return {"success": True, "message": "Conversation deleted"}
|
||||
|
||||
|
||||
# ============================================================================
|
||||
# REST Endpoints - Session Management
|
||||
# ============================================================================
|
||||
|
||||
@router.get("/sessions", response_model=list[str])
|
||||
async def list_active_sessions():
|
||||
"""List all active assistant sessions."""
|
||||
return list_sessions()
|
||||
|
||||
|
||||
@router.get("/sessions/{project_name}", response_model=SessionInfo)
|
||||
async def get_session_info(project_name: str):
|
||||
"""Get information about an active session."""
|
||||
if not validate_project_name(project_name):
|
||||
raise HTTPException(status_code=400, detail="Invalid project name")
|
||||
|
||||
session = get_session(project_name)
|
||||
if not session:
|
||||
raise HTTPException(status_code=404, detail="No active session for this project")
|
||||
|
||||
return SessionInfo(
|
||||
project_name=project_name,
|
||||
conversation_id=session.get_conversation_id(),
|
||||
is_active=True,
|
||||
)
|
||||
|
||||
|
||||
@router.delete("/sessions/{project_name}")
|
||||
async def close_session(project_name: str):
|
||||
"""Close an active session."""
|
||||
if not validate_project_name(project_name):
|
||||
raise HTTPException(status_code=400, detail="Invalid project name")
|
||||
|
||||
session = get_session(project_name)
|
||||
if not session:
|
||||
raise HTTPException(status_code=404, detail="No active session for this project")
|
||||
|
||||
await remove_session(project_name)
|
||||
return {"success": True, "message": "Session closed"}
|
||||
|
||||
|
||||
# ============================================================================
|
||||
# WebSocket Endpoint
|
||||
# ============================================================================
|
||||
|
||||
@router.websocket("/ws/{project_name}")
|
||||
async def assistant_chat_websocket(websocket: WebSocket, project_name: str):
|
||||
"""
|
||||
WebSocket endpoint for assistant chat.
|
||||
|
||||
Message protocol:
|
||||
|
||||
Client -> Server:
|
||||
- {"type": "start", "conversation_id": int | null} - Start/resume session
|
||||
- {"type": "message", "content": "..."} - Send user message
|
||||
- {"type": "ping"} - Keep-alive ping
|
||||
|
||||
Server -> Client:
|
||||
- {"type": "conversation_created", "conversation_id": int} - New conversation created
|
||||
- {"type": "text", "content": "..."} - Text chunk from Claude
|
||||
- {"type": "tool_call", "tool": "...", "input": {...}} - Tool being called
|
||||
- {"type": "response_done"} - Response complete
|
||||
- {"type": "error", "content": "..."} - Error message
|
||||
- {"type": "pong"} - Keep-alive pong
|
||||
"""
|
||||
if not validate_project_name(project_name):
|
||||
await websocket.close(code=4000, reason="Invalid project name")
|
||||
return
|
||||
|
||||
project_dir = _get_project_path(project_name)
|
||||
if not project_dir:
|
||||
await websocket.close(code=4004, reason="Project not found in registry")
|
||||
return
|
||||
|
||||
if not project_dir.exists():
|
||||
await websocket.close(code=4004, reason="Project directory not found")
|
||||
return
|
||||
|
||||
await websocket.accept()
|
||||
logger.info(f"Assistant WebSocket connected for project: {project_name}")
|
||||
|
||||
session: Optional[AssistantChatSession] = None
|
||||
|
||||
try:
|
||||
while True:
|
||||
try:
|
||||
data = await websocket.receive_text()
|
||||
message = json.loads(data)
|
||||
msg_type = message.get("type")
|
||||
logger.info(f"Assistant received message type: {msg_type}")
|
||||
|
||||
if msg_type == "ping":
|
||||
await websocket.send_json({"type": "pong"})
|
||||
continue
|
||||
|
||||
elif msg_type == "start":
|
||||
# Get optional conversation_id to resume
|
||||
conversation_id = message.get("conversation_id")
|
||||
|
||||
try:
|
||||
# Create a new session
|
||||
session = await create_session(
|
||||
project_name,
|
||||
project_dir,
|
||||
conversation_id=conversation_id,
|
||||
)
|
||||
|
||||
# Stream the initial greeting
|
||||
async for chunk in session.start():
|
||||
await websocket.send_json(chunk)
|
||||
except Exception as e:
|
||||
logger.exception(f"Error starting assistant session for {project_name}")
|
||||
await websocket.send_json({
|
||||
"type": "error",
|
||||
"content": f"Failed to start session: {str(e)}"
|
||||
})
|
||||
|
||||
elif msg_type == "message":
|
||||
if not session:
|
||||
session = get_session(project_name)
|
||||
if not session:
|
||||
await websocket.send_json({
|
||||
"type": "error",
|
||||
"content": "No active session. Send 'start' first."
|
||||
})
|
||||
continue
|
||||
|
||||
user_content = message.get("content", "").strip()
|
||||
if not user_content:
|
||||
await websocket.send_json({
|
||||
"type": "error",
|
||||
"content": "Empty message"
|
||||
})
|
||||
continue
|
||||
|
||||
# Stream Claude's response
|
||||
async for chunk in session.send_message(user_content):
|
||||
await websocket.send_json(chunk)
|
||||
|
||||
else:
|
||||
await websocket.send_json({
|
||||
"type": "error",
|
||||
"content": f"Unknown message type: {msg_type}"
|
||||
})
|
||||
|
||||
except json.JSONDecodeError:
|
||||
await websocket.send_json({
|
||||
"type": "error",
|
||||
"content": "Invalid JSON"
|
||||
})
|
||||
|
||||
except WebSocketDisconnect:
|
||||
logger.info(f"Assistant chat WebSocket disconnected for {project_name}")
|
||||
|
||||
except Exception as e:
|
||||
logger.exception(f"Assistant chat WebSocket error for {project_name}")
|
||||
try:
|
||||
await websocket.send_json({
|
||||
"type": "error",
|
||||
"content": f"Server error: {str(e)}"
|
||||
})
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
finally:
|
||||
# Don't remove session on disconnect - allow resume
|
||||
pass
|
||||
Reference in New Issue
Block a user