mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-08-08 03:44:14 +00:00
代码清理
This commit is contained in:
parent
6e505860be
commit
439d23c29c
@ -151,6 +151,7 @@ async def dialogue_scenes():
|
||||
scene_name=scene.scene_name(),
|
||||
scene_describe=scene.describe(),
|
||||
param_title=",".join(scene.param_types()),
|
||||
show_disable=scene.show_disable(),
|
||||
)
|
||||
scene_vos.append(scene_vo)
|
||||
return Result.succ(scene_vos)
|
||||
|
@ -28,6 +28,7 @@ class ChatSceneVo(BaseModel):
|
||||
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")
|
||||
|
||||
|
||||
class ConversationVo(BaseModel):
|
||||
|
@ -3,12 +3,13 @@ from typing import List
|
||||
|
||||
|
||||
class Scene:
|
||||
def __init__(self, code, name, describe, param_types: List = [], is_inner: bool = False):
|
||||
def __init__(self, code, name, describe, param_types: List = [], is_inner: bool = False, show_disable=False):
|
||||
self.code = code
|
||||
self.name = name
|
||||
self.describe = describe
|
||||
self.param_types = param_types
|
||||
self.is_inner = is_inner
|
||||
self.show_disable = show_disable
|
||||
|
||||
|
||||
class ChatScene(Enum):
|
||||
@ -29,6 +30,8 @@ class ChatScene(Enum):
|
||||
"Plugin",
|
||||
"Use tools through dialogue to accomplish your goals.",
|
||||
["Plugin Select"],
|
||||
False,
|
||||
True
|
||||
)
|
||||
ChatDefaultKnowledge = Scene(
|
||||
"chat_default_knowledge",
|
||||
@ -82,3 +85,6 @@ class ChatScene(Enum):
|
||||
|
||||
def param_types(self):
|
||||
return self._value_.param_types
|
||||
|
||||
def show_disable(self):
|
||||
return self._value_.show_disable
|
Loading…
Reference in New Issue
Block a user