github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-08 03:44:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

代码清理

Browse Source
This commit is contained in:
yhjun1026 2023-07-24 14:53:34 +08:00
parent 6e505860be
commit 439d23c29c
3 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pilot/openapi/api_v1/api_v1.py
View File

@ -151,6 +151,7 @@ async def dialogue_scenes():
scene_name=scene.scene_name(), scene_name=scene.scene_name(),
scene_describe=scene.describe(), scene_describe=scene.describe(),
param_title=",".join(scene.param_types()), param_title=",".join(scene.param_types()),
show_disable=scene.show_disable(),
) )
scene_vos.append(scene_vo) scene_vos.append(scene_vo)
return Result.succ(scene_vos) return Result.succ(scene_vos)

1
pilot/openapi/api_v1/api_view_model.py
View File

@ -28,6 +28,7 @@ class ChatSceneVo(BaseModel):
scene_name: str = Field(..., description="chat_scene name show for user") scene_name: str = Field(..., description="chat_scene name show for user")
scene_describe: str = Field("", description="chat_scene describe ") scene_describe: str = Field("", description="chat_scene describe ")
param_title: str = Field("", description="chat_scene required parameter title") param_title: str = Field("", description="chat_scene required parameter title")
show_disable: bool = Field(False, description="chat_scene show disable")
class ConversationVo(BaseModel): class ConversationVo(BaseModel):

8
pilot/scene/base.py
View File

@ -3,12 +3,13 @@ from typing import List
class Scene: 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.code = code
self.name = name self.name = name
self.describe = describe self.describe = describe
self.param_types = param_types self.param_types = param_types
self.is_inner = is_inner self.is_inner = is_inner
self.show_disable = show_disable
class ChatScene(Enum): class ChatScene(Enum):
@ -29,6 +30,8 @@ class ChatScene(Enum):
"Plugin", "Plugin",
"Use tools through dialogue to accomplish your goals.", "Use tools through dialogue to accomplish your goals.",
["Plugin Select"], ["Plugin Select"],
False,
True
) )
ChatDefaultKnowledge = Scene( ChatDefaultKnowledge = Scene(
"chat_default_knowledge", "chat_default_knowledge",
@ -82,3 +85,6 @@ class ChatScene(Enum):
def param_types(self): def param_types(self):
return self._value_.param_types return self._value_.param_types
def show_disable(self):
return self._value_.show_disable