dolgolyov.alexei
03afb7a075
Backend: - Skill model + migration (with FK on chats.skill_id) - Personal + general skill CRUD services with access isolation - Admin skill CRUD endpoints (POST/GET/PATCH/DELETE /admin/skills) - User skill CRUD endpoints (POST/GET/PATCH/DELETE /skills/) - Personal context GET/PUT at /users/me/context - Extended context assembly: primary + personal context + skill prompt - Chat creation/update now accepts skill_id with validation Frontend: - Skill selector dropdown in chat header (grouped: general + personal) - Reusable skill editor form component - Admin skills management page (/admin/skills) - Personal skills page (/skills) - Personal context editor page (/profile/context) - Updated sidebar: Skills, My Context nav items + admin skills link - English + Russian translations for all skill/context UI Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
108 lines
3.5 KiB
Python
108 lines
3.5 KiB
Python
import uuid
|
|
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Depends, Query, status
|
|
from fastapi.responses import StreamingResponse
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.api.deps import get_current_user
|
|
from app.database import get_db
|
|
from app.models.user import User
|
|
from app.schemas.chat import (
|
|
ChatListResponse,
|
|
ChatResponse,
|
|
CreateChatRequest,
|
|
MessageListResponse,
|
|
MessageResponse,
|
|
SendMessageRequest,
|
|
UpdateChatRequest,
|
|
)
|
|
from app.services import chat_service, skill_service
|
|
from app.services.ai_service import stream_ai_response
|
|
|
|
router = APIRouter(prefix="/chats", tags=["chats"])
|
|
|
|
|
|
@router.post("/", response_model=ChatResponse, status_code=status.HTTP_201_CREATED)
|
|
async def create_chat(
|
|
data: CreateChatRequest,
|
|
user: Annotated[User, Depends(get_current_user)],
|
|
db: Annotated[AsyncSession, Depends(get_db)],
|
|
):
|
|
if data.skill_id:
|
|
await skill_service.validate_skill_accessible(db, data.skill_id, user.id)
|
|
chat = await chat_service.create_chat(db, user, data.title, data.skill_id)
|
|
return ChatResponse.model_validate(chat)
|
|
|
|
|
|
@router.get("/", response_model=ChatListResponse)
|
|
async def list_chats(
|
|
user: Annotated[User, Depends(get_current_user)],
|
|
db: Annotated[AsyncSession, Depends(get_db)],
|
|
archived: bool | None = Query(default=None),
|
|
):
|
|
chats = await chat_service.get_user_chats(db, user.id, archived)
|
|
return ChatListResponse(chats=[ChatResponse.model_validate(c) for c in chats])
|
|
|
|
|
|
@router.get("/{chat_id}", response_model=ChatResponse)
|
|
async def get_chat(
|
|
chat_id: uuid.UUID,
|
|
user: Annotated[User, Depends(get_current_user)],
|
|
db: Annotated[AsyncSession, Depends(get_db)],
|
|
):
|
|
chat = await chat_service.get_chat(db, chat_id, user.id)
|
|
return ChatResponse.model_validate(chat)
|
|
|
|
|
|
@router.patch("/{chat_id}", response_model=ChatResponse)
|
|
async def update_chat(
|
|
chat_id: uuid.UUID,
|
|
data: UpdateChatRequest,
|
|
user: Annotated[User, Depends(get_current_user)],
|
|
db: Annotated[AsyncSession, Depends(get_db)],
|
|
):
|
|
if data.skill_id:
|
|
await skill_service.validate_skill_accessible(db, data.skill_id, user.id)
|
|
chat = await chat_service.update_chat(db, chat_id, user.id, data.title, data.is_archived, data.skill_id)
|
|
return ChatResponse.model_validate(chat)
|
|
|
|
|
|
@router.delete("/{chat_id}", status_code=status.HTTP_204_NO_CONTENT)
|
|
async def delete_chat(
|
|
chat_id: uuid.UUID,
|
|
user: Annotated[User, Depends(get_current_user)],
|
|
db: Annotated[AsyncSession, Depends(get_db)],
|
|
):
|
|
await chat_service.delete_chat(db, chat_id, user.id)
|
|
|
|
|
|
@router.get("/{chat_id}/messages", response_model=MessageListResponse)
|
|
async def list_messages(
|
|
chat_id: uuid.UUID,
|
|
user: Annotated[User, Depends(get_current_user)],
|
|
db: Annotated[AsyncSession, Depends(get_db)],
|
|
limit: int = Query(default=50, le=200),
|
|
before: uuid.UUID | None = Query(default=None),
|
|
):
|
|
messages = await chat_service.get_messages(db, chat_id, user.id, limit, before)
|
|
return MessageListResponse(messages=[MessageResponse.model_validate(m) for m in messages])
|
|
|
|
|
|
@router.post("/{chat_id}/messages")
|
|
async def send_message(
|
|
chat_id: uuid.UUID,
|
|
data: SendMessageRequest,
|
|
user: Annotated[User, Depends(get_current_user)],
|
|
db: Annotated[AsyncSession, Depends(get_db)],
|
|
):
|
|
return StreamingResponse(
|
|
stream_ai_response(db, chat_id, user.id, data.content),
|
|
media_type="text/event-stream",
|
|
headers={
|
|
"Cache-Control": "no-cache",
|
|
"Connection": "keep-alive",
|
|
"X-Accel-Buffering": "no",
|
|
},
|
|
)
|