Spaces:
Sleeping
Sleeping
File size: 2,281 Bytes
b4c9cb7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
from fastapi import APIRouter, status, Depends
from fastapi.responses import JSONResponse
from typing import Annotated
from src.apis.models.user_models import User
from src.apis.controllers.auth_controller import (
login_control,
)
from src.apis.middlewares.auth_middleware import get_current_user
from pydantic import BaseModel, Field
from src.utils.logger import logger
from src.config.mongo import PromptCRUD
from bson import ObjectId
router = APIRouter(prefix="/prompt", tags=["Prompt Editor"])
user_dependency = Annotated[User, Depends(get_current_user)]
@router.get("/list", status_code=status.HTTP_200_OK)
async def list_prompt(user: user_dependency):
try:
logger.info(f"User {user['username']} is listing prompts.")
prompts = await PromptCRUD.read({})
for prompt in prompts:
prompt.pop("created_at")
prompt.pop("updated_at")
prompt.pop("expire_at")
return JSONResponse(content=prompts, status_code=status.HTTP_200_OK)
except Exception as e:
return JSONResponse(content={"message": str(e)}, status_code=500)
class UpdatePromptRequest(BaseModel):
id: str = Field(..., description="Prompt's id")
script_writer_prompt: str = Field(None, description="Script Writer Prompt")
script_re_outline_prompt: str = Field(None, description="Script Re-Outline Prompt")
scientific_fact_finder_prompt: str = Field(None, description="Scientific Fact Finder Prompt")
comment_insight_extractor_prompt: str = Field(None, description="Comment Insight Extractor Prompt")
script_structure_analyzer_prompt: str = Field(None, description="Script Structure Analyzer Prompt")
@router.put("/update", status_code=status.HTTP_200_OK)
async def update_prompt(user: user_dependency, body: UpdatePromptRequest):
try:
logger.info(f"User {user['username']} is updating prompt {body.id}.")
await PromptCRUD.update(
{"_id": ObjectId(body.id)},
{"$set": body.model_dump(exclude_none=True)},
)
return JSONResponse(
content={"message": "Prompt updated successfully"},
status_code=status.HTTP_200_OK,
)
except Exception as e:
return JSONResponse(content={"message": str(e)}, status_code=500)
|