Yuzhiran Dev
f6e5e90736
- Redesign index.html from system-monitor to user-centric workspace (welcome banner, summary cards, 今日待办, AI 智能建议, collapsible admin section) - Add fetchTodos/fetchSuggestions/getPriorityClass/isOverdue methods - Update uni-nav current-page from 'dashboard' to 'workspace' - Add require_role to 10 admin router imports (RBAC foundation) - Update ai-assistant.js: add factory/insights pages, org_id in chat requests
117 lines
3.9 KiB
Python
117 lines
3.9 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.orm import Session
|
|
from typing import List, Optional
|
|
from pydantic import BaseModel, ConfigDict
|
|
from datetime import datetime
|
|
|
|
from ..database import get_db
|
|
from ..models import PromptConfig
|
|
from .auth import get_current_admin, require_role
|
|
|
|
router = APIRouter(prefix="/api/admin/prompt-configs", tags=["admin"])
|
|
|
|
|
|
class PromptConfigBase(BaseModel):
|
|
key: str
|
|
module_id: Optional[str] = None
|
|
category: str = "prompt"
|
|
version: str = "v1"
|
|
content: str
|
|
variables: List[dict] = []
|
|
description: Optional[str] = None
|
|
enabled: bool = True
|
|
temperature: Optional[float] = None
|
|
max_tokens: Optional[int] = None
|
|
created_by: Optional[str] = None
|
|
|
|
|
|
class PromptConfigCreate(PromptConfigBase):
|
|
pass
|
|
|
|
|
|
class PromptConfigUpdate(BaseModel):
|
|
content: Optional[str] = None
|
|
variables: Optional[List[dict]] = None
|
|
description: Optional[str] = None
|
|
enabled: Optional[bool] = None
|
|
temperature: Optional[float] = None
|
|
max_tokens: Optional[int] = None
|
|
|
|
|
|
class PromptConfigResponse(PromptConfigBase):
|
|
id: int
|
|
created_at: Optional[datetime] = None
|
|
updated_at: Optional[datetime] = None
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
@router.get("", response_model=List[PromptConfigResponse])
|
|
def list_prompts(
|
|
module_id: Optional[str] = None,
|
|
category: Optional[str] = None,
|
|
db: Session = Depends(get_db),
|
|
admin_user=Depends(get_current_admin),
|
|
):
|
|
q = db.query(PromptConfig)
|
|
if module_id:
|
|
q = q.filter(PromptConfig.module_id.in_([module_id, "all"]))
|
|
if category:
|
|
q = q.filter(PromptConfig.category == category)
|
|
prompts = q.order_by(PromptConfig.module_id, PromptConfig.key).all()
|
|
if not prompts:
|
|
_ensure_defaults(db)
|
|
prompts = q.order_by(PromptConfig.module_id, PromptConfig.key).all()
|
|
return [PromptConfigResponse.model_validate(p) for p in prompts]
|
|
|
|
|
|
@router.get("/{prompt_id}", response_model=PromptConfigResponse)
|
|
def get_prompt(prompt_id: int, db: Session = Depends(get_db), admin_user=Depends(get_current_admin)):
|
|
prompt = db.query(PromptConfig).filter(PromptConfig.id == prompt_id).first()
|
|
if not prompt:
|
|
raise HTTPException(status_code=404, detail="未找到该配置")
|
|
return prompt
|
|
|
|
|
|
@router.post("", response_model=PromptConfigResponse)
|
|
def create_prompt(data: PromptConfigCreate, db: Session = Depends(get_db), admin_user=Depends(get_current_admin)):
|
|
existing = db.query(PromptConfig).filter(PromptConfig.key == data.key).first()
|
|
if existing:
|
|
raise HTTPException(status_code=400, detail=f"key '{data.key}' 已存在")
|
|
prompt = PromptConfig(**data.model_dump())
|
|
db.add(prompt)
|
|
db.commit()
|
|
db.refresh(prompt)
|
|
return prompt
|
|
|
|
|
|
@router.put("/{prompt_id}", response_model=PromptConfigResponse)
|
|
def update_prompt(prompt_id: int, data: PromptConfigUpdate, db: Session = Depends(get_db), admin_user=Depends(get_current_admin)):
|
|
prompt = db.query(PromptConfig).filter(PromptConfig.id == prompt_id).first()
|
|
if not prompt:
|
|
raise HTTPException(status_code=404, detail="未找到该配置")
|
|
for field, value in data.model_dump(exclude_unset=True).items():
|
|
if value is not None:
|
|
setattr(prompt, field, value)
|
|
db.commit()
|
|
db.refresh(prompt)
|
|
return prompt
|
|
|
|
|
|
@router.delete("/{prompt_id}")
|
|
def delete_prompt(prompt_id: int, db: Session = Depends(get_db), admin_user=Depends(get_current_admin)):
|
|
prompt = db.query(PromptConfig).filter(PromptConfig.id == prompt_id).first()
|
|
if not prompt:
|
|
raise HTTPException(status_code=404, detail="未找到该配置")
|
|
db.delete(prompt)
|
|
db.commit()
|
|
return {"message": "删除成功"}
|
|
|
|
|
|
def _ensure_defaults(db: Session):
|
|
existing_keys = {p.key for p in db.query(PromptConfig).all()}
|
|
from .task_configs import DEFAULT_PROMPTS
|
|
for p in DEFAULT_PROMPTS:
|
|
if p["key"] not in existing_keys:
|
|
db.add(PromptConfig(**p))
|
|
db.commit() |