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

refactor: Refactor for core SDK (#1092)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-01-21 09:57:57 +08:00
committed by GitHub
parent ba7248adbb
commit 2d905191f8
45 changed files with 236 additions and 133 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
dbgpt/app/chat_adapter.py
View File

@@ -9,7 +9,7 @@ from functools import cache
from typing import Dict, List, Tuple
from dbgpt.core.interface.message import ModelMessage, ModelMessageRoleType
from dbgpt.model.conversation import Conversation, get_conv_template
from dbgpt.model.llm.conversation import Conversation, get_conv_template
class BaseChatAdpter:
@@ -21,7 +21,7 @@ class BaseChatAdpter:
def get_generate_stream_func(self, model_path: str):
"""Return the generate stream handler func"""
from dbgpt.model.inference import generate_stream
from dbgpt.model.llm.inference import generate_stream
return generate_stream