refactor: The first refactored version for sdk release (#907)

Co-authored-by: chengfangyin2 <chengfangyin3@jd.com>
This commit is contained in:
FangYin Cheng
2023-12-08 14:45:59 +08:00
committed by GitHub
parent e7e4aff667
commit cd725db1fb
573 changed files with 2094 additions and 3571 deletions

View File

@@ -0,0 +1,25 @@
from dbgpt.core.interface.prompt import PromptTemplate
from dbgpt._private.config import Config
from dbgpt.app.scene import ChatScene
from dbgpt.app.scene.chat_normal.out_parser import NormalChatOutputParser
PROMPT_SCENE_DEFINE = None
CFG = Config()
PROMPT_NEED_STREAM_OUT = True
prompt = PromptTemplate(
template_scene=ChatScene.ChatNormal.value(),
input_variables=["input"],
response_format=None,
template_define=PROMPT_SCENE_DEFINE,
template=None,
stream_out=PROMPT_NEED_STREAM_OUT,
output_parser=NormalChatOutputParser(is_stream_out=PROMPT_NEED_STREAM_OUT),
)
# CFG.prompt_templates.update({prompt.template_scene: prompt})
CFG.prompt_template_registry.register(prompt, language=CFG.LANGUAGE, is_default=True)