feat(core): APP use new SDK component (#1050)

dbgpt/serve/conversation/api/schemas.py

@@ -1,4 +1,6 @@
 # Define your Pydantic schemas here
+from typing import Any, Optional
+
 from dbgpt._private.pydantic import BaseModel, Field
 
 from ..config import SERVE_APP_NAME_HUMP
@@ -7,15 +9,133 @@ from ..config import SERVE_APP_NAME_HUMP
 class ServeRequest(BaseModel):
     """Conversation request model"""
 
-    # TODO define your own fields here
-
     class Config:
         title = f"ServeRequest for {SERVE_APP_NAME_HUMP}"
 
+    # Just for query
+    chat_mode: str = Field(
+        default=None,
+        description="The chat mode.",
+        examples=[
+            "chat_normal",
+        ],
+    )
+    conv_uid: Optional[str] = Field(
+        default=None,
+        description="The conversation uid.",
+        examples=[
+            "5e7100bc-9017-11ee-9876-8fe019728d79",
+        ],
+    )
+    user_name: Optional[str] = Field(
+        default=None,
+        description="The user name.",
+        examples=[
+            "zhangsan",
+        ],
+    )
+    sys_code: Optional[str] = Field(
+        default=None,
+        description="The system code.",
+        examples=[
+            "dbgpt",
+        ],
+    )
+
 
 class ServerResponse(BaseModel):
     """Conversation response model"""
 
-    # TODO define your own fields here
     class Config:
         title = f"ServerResponse for {SERVE_APP_NAME_HUMP}"
+
+    conv_uid: str = Field(
+        ...,
+        description="The conversation uid.",
+        examples=[
+            "5e7100bc-9017-11ee-9876-8fe019728d79",
+        ],
+    )
+    user_input: str = Field(
+        ...,
+        description="The user input, we return it as the summary the conversation.",
+        examples=[
+            "Hello world",
+        ],
+    )
+    chat_mode: str = Field(
+        ...,
+        description="The chat mode.",
+        examples=[
+            "chat_normal",
+        ],
+    )
+    select_param: Optional[str] = Field(
+        default=None,
+        description="The select param.",
+        examples=[
+            "my_knowledge_space_name",
+        ],
+    )
+    model_name: Optional[str] = Field(
+        default=None,
+        description="The model name.",
+        examples=[
+            "vicuna-13b-v1.5",
+        ],
+    )
+    user_name: Optional[str] = Field(
+        default=None,
+        description="The user name.",
+        examples=[
+            "zhangsan",
+        ],
+    )
+    sys_code: Optional[str] = Field(
+        default=None,
+        description="The system code.",
+        examples=[
+            "dbgpt",
+        ],
+    )
+
+
+class MessageVo(BaseModel):
+    role: str = Field(
+        ...,
+        description="The role that sends out the current message.",
+        examples=["human", "ai", "view"],
+    )
+    context: str = Field(
+        ...,
+        description="The current message content.",
+        examples=[
+            "Hello",
+            "Hi, how are you?",
+        ],
+    )
+
+    order: int = Field(
+        ...,
+        description="The current message order.",
+        examples=[
+            1,
+            2,
+        ],
+    )
+
+    time_stamp: Optional[Any] = Field(
+        default=None,
+        description="The current message time stamp.",
+        examples=[
+            "2023-01-07 09:00:00",
+        ],
+    )
+
+    model_name: Optional[str] = Field(
+        default=None,
+        description="The model name.",
+        examples=[
+            "vicuna-13b-v1.5",
+        ],
+    )