feat: (0.6)New UI (#1855)

Co-authored-by: 夏姜 <wenfengjiang.jwf@digital-engine.com>
Co-authored-by: aries_ckt <916701291@qq.com>
Co-authored-by: wb-lh513319 <wb-lh513319@alibaba-inc.com>
Co-authored-by: csunny <cfqsunny@163.com>

dbgpt/app/openapi/api_view_model.py

@@ -1,4 +1,8 @@
-from typing import Any, Dict, Generic, Optional, TypeVar
+import socket
+import time
+import uuid
+from enum import Enum
+from typing import Any, Dict, Generic, List, Literal, Optional, TypeVar
 
 from dbgpt._private.pydantic import BaseModel, ConfigDict, Field, model_to_dict
 
@@ -10,9 +14,10 @@ class Result(BaseModel, Generic[T]):
     err_code: Optional[str] = None
     err_msg: Optional[str] = None
     data: Optional[T] = None
+    host_name: Optional[str] = socket.gethostname()
 
     @classmethod
-    def succ(cls, data: T):
+    def succ(cls, data: T = None):
         return Result(success=True, err_code=None, err_msg=None, data=data)
 
     @classmethod
@@ -24,11 +29,14 @@ class Result(BaseModel, Generic[T]):
 
 
 class ChatSceneVo(BaseModel):
-    chat_scene: str = Field(..., description="chat_scene")
-    scene_name: str = Field(..., description="chat_scene name show for user")
-    scene_describe: str = Field("", description="chat_scene describe ")
-    param_title: str = Field("", description="chat_scene required parameter title")
-    show_disable: bool = Field(False, description="chat_scene show disable")
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+    chat_scene: Optional[str] = Field(..., description="chat_scene")
+    scene_name: Optional[str] = Field(..., description="chat_scene name show for user")
+    scene_describe: Optional[str] = Field("", description="chat_scene describe ")
+    param_title: Optional[str] = Field(
+        "", description="chat_scene required parameter title"
+    )
+    show_disable: Optional[bool] = Field(False, description="chat_scene show disable")
 
 
 class ConversationVo(BaseModel):
@@ -50,12 +58,17 @@ class ConversationVo(BaseModel):
     """ 
     the scene of chat 
     """
-    chat_mode: str = ""
+    chat_mode: Optional[str] = ""
+    """
+    the app of  chat
+    """
+    app_code: Optional[str] = ""
 
+    temperature: Optional[float] = 0.5
     """
     chat scene select param 
     """
-    select_param: Optional[str] = Field(None, description="chat scene select param")
+    select_param: Optional[Any] = Field(None, description="chat scene select param")
     """
     llm model name
     """
@@ -68,6 +81,8 @@ class ConversationVo(BaseModel):
 
     sys_code: Optional[str] = Field(None, description="System code")
 
+    ext_info: Optional[dict] = {}
+
 
 class MessageVo(BaseModel):
     model_config = ConfigDict(protected_namespaces=())
@@ -96,3 +111,69 @@ class MessageVo(BaseModel):
     model_name
     """
     model_name: str
+
+
+class DeltaMessage(BaseModel):
+    role: Optional[str] = None
+    content: Optional[str] = None
+
+
+class ChatCompletionResponseStreamChoice(BaseModel):
+    index: Optional[int]
+    delta: Optional[DeltaMessage]
+    finish_reason: Optional[Literal["stop", "length"]] = None
+
+
+class ChatCompletionStreamResponse(BaseModel):
+    id: Optional[str] = Field(default_factory=lambda: f"chatcmpl-{str(uuid.uuid1())}")
+    created: Optional[int] = Field(default_factory=lambda: int(time.time()))
+    model: Optional[str] = None
+    choices: Optional[List[ChatCompletionResponseStreamChoice]]
+
+
+class ChatContext(BaseModel):
+    model: Optional[str] = Field(
+        ..., description="The model name", examples=["gpt-3.5-turbo", "proxyllm"]
+    )
+
+
+class ChatAppContext(ChatContext):
+    app_code: Optional[str]
+
+
+class ChatKnowledgeContext(ChatContext):
+    space: Optional[str]
+
+
+class ChatFlowContext(ChatContext):
+    flow_id: Optional[str]
+
+
+class OpenAPIChatCompletionRequest(BaseModel):
+    conv_uid: Optional[str] = Field(None, description="conversation uid")
+    user_input: Optional[str] = Field(
+        None, description="User input messages", examples=["Hello", "How are you?"]
+    )
+    chat_type: Optional[str] = Field(
+        "normal",
+        description="chat type",
+        examples=["normal", "app", "flow", "knowledge"],
+    )
+    stream: Optional[bool] = Field(default=True, description="Whether return stream")
+    context: Optional[Dict[str, Any]] = Field(
+        None,
+        description="The context of the model inference",
+    )
+
+
+class ChatType(Enum):
+    NORMAL = "normal"
+    APP = "app"
+    FLOW = "flow"
+    KNOWLEDGE = "knowledge"
+
+
+class APIToken(BaseModel):
+    api_key: Optional[str] = Field(..., description="ant app name")
+    api_token: Optional[str] = Field(..., description="api token")
+    user_id: Optional[str] = Field(..., description="user id")