refactor: Refactor storage and new serve template (#947)

dbgpt/serve/prompt/api/__init__.py

@@ -0,0 +1,2 @@
+# This is an auto-generated __init__.py file
+# generated by `dbgpt new serve prompt`

dbgpt/serve/prompt/api/endpoints.py

@@ -0,0 +1,114 @@
+from typing import Optional, List
+from fastapi import APIRouter, Depends, Query
+
+from dbgpt.component import SystemApp
+from dbgpt.serve.core import Result
+from dbgpt.util import PaginationResult
+from .schemas import ServeRequest, ServerResponse
+from ..service.service import Service
+from ..config import APP_NAME, SERVE_APP_NAME, ServeConfig, SERVE_SERVICE_COMPONENT_NAME
+
+router = APIRouter()
+
+# Add your API endpoints here
+
+global_system_app: Optional[SystemApp] = None
+
+
+def get_service() -> Service:
+    """Get the service instance"""
+    return global_system_app.get_component(SERVE_SERVICE_COMPONENT_NAME, Service)
+
+
+@router.get("/health")
+async def health():
+    """Health check endpoint"""
+    return {"status": "ok"}
+
+
+# TODO: Compatible with old API, will be modified in the future
+@router.post("/add", response_model=Result[ServerResponse])
+async def create(
+    request: ServeRequest, service: Service = Depends(get_service)
+) -> Result[ServerResponse]:
+    """Create a new Prompt entity
+
+    Args:
+        request (ServeRequest): The request
+        service (Service): The service
+    Returns:
+        ServerResponse: The response
+    """
+    return Result.succ(service.create(request))
+
+
+@router.post("/update", response_model=Result[ServerResponse])
+async def update(
+    request: ServeRequest, service: Service = Depends(get_service)
+) -> Result[ServerResponse]:
+    """Update a Prompt entity
+
+    Args:
+        request (ServeRequest): The request
+        service (Service): The service
+    Returns:
+        ServerResponse: The response
+    """
+    return Result.succ(service.update(request))
+
+
+@router.post("/delete", response_model=Result[None])
+async def delete(
+    request: ServeRequest, service: Service = Depends(get_service)
+) -> Result[None]:
+    """Delete a Prompt entity
+
+    Args:
+        request (ServeRequest): The request
+        service (Service): The service
+    Returns:
+        ServerResponse: The response
+    """
+    return Result.succ(service.delete(request))
+
+
+@router.post("/list", response_model=Result[List[ServerResponse]])
+async def query(
+    request: ServeRequest, service: Service = Depends(get_service)
+) -> Result[List[ServerResponse]]:
+    """Query Prompt entities
+
+    Args:
+        request (ServeRequest): The request
+        service (Service): The service
+    Returns:
+        List[ServerResponse]: The response
+    """
+    return Result.succ(service.get_list(request))
+
+
+@router.post("/query_page", response_model=Result[PaginationResult[ServerResponse]])
+async def query_page(
+    request: ServeRequest,
+    page: Optional[int] = Query(default=1, description="current page"),
+    page_size: Optional[int] = Query(default=20, description="page size"),
+    service: Service = Depends(get_service),
+) -> Result[PaginationResult[ServerResponse]]:
+    """Query Prompt entities
+
+    Args:
+        request (ServeRequest): The request
+        page (int): The page number
+        page_size (int): The page size
+        service (Service): The service
+    Returns:
+        ServerResponse: The response
+    """
+    return Result.succ(service.get_list_by_page(request, page, page_size))
+
+
+def init_endpoints(system_app: SystemApp) -> None:
+    """Initialize the endpoints"""
+    global global_system_app
+    system_app.register(Service)
+    global_system_app = system_app

dbgpt/serve/prompt/api/schemas.py

@@ -0,0 +1,73 @@
+# Define your Pydantic schemas here
+from typing import Optional
+from dbgpt._private.pydantic import BaseModel, Field
+
+
+class ServeRequest(BaseModel):
+    """Prompt request model"""
+
+    chat_scene: Optional[str] = None
+    """
+    The chat scene, e.g. chat_with_db_execute, chat_excel, chat_with_db_qa.
+    """
+
+    sub_chat_scene: Optional[str] = None
+    """
+    The sub chat scene.
+    """
+
+    prompt_type: Optional[str] = None
+    """
+    The prompt type, either common or private.
+    """
+
+    content: Optional[str] = None
+    """
+    The prompt content.
+    """
+
+    user_name: Optional[str] = None
+    """
+    The user name.
+    """
+
+    sys_code: Optional[str] = None
+    """
+    System code
+    """
+
+    prompt_name: Optional[str] = None
+    """
+    The prompt name.
+    """
+
+
+class ServerResponse(BaseModel):
+    """Prompt response model"""
+
+    id: int = None
+    """chat_scene: for example: chat_with_db_execute, chat_excel, chat_with_db_qa"""
+
+    chat_scene: str = None
+
+    """sub_chat_scene: sub chat scene"""
+    sub_chat_scene: str = None
+
+    """prompt_type: common or private"""
+    prompt_type: str = None
+
+    """content: prompt content"""
+    content: str = None
+
+    """user_name: user name"""
+    user_name: str = None
+
+    sys_code: Optional[str] = None
+    """
+    System code
+    """
+
+    """prompt_name: prompt name"""
+    prompt_name: str = None
+    gmt_created: str = None
+    gmt_modified: str = None