github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-13 05:01:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: (0.6)New UI (#1855)

Browse Source 
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>
This commit is contained in:
明天
2024-08-21 17:37:45 +08:00
committed by GitHub
parent 3fc82693ba
commit b124ecc10b
824 changed files with 93371 additions and 2515 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
dbgpt/serve/conversation/service/service.py
View File

@@ -14,6 +14,7 @@ from dbgpt.storage.metadata import BaseDao
from dbgpt.storage.metadata._base_dao import REQ, RES
from dbgpt.util.pagination_utils import PaginationResult
from ...feedback.api.endpoints import get_service
from ..api.schemas import MessageVo, ServeRequest, ServerResponse
from ..config import SERVE_CONFIG_KEY_PREFIX, SERVE_SERVICE_COMPONENT_NAME, ServeConfig
from ..models.models import ServeDao, ServeEntity
@@ -149,6 +150,15 @@ class Service(BaseService[ServeEntity, ServeRequest, ServerResponse]):
conv: StorageConversation = self.create_storage_conv(request)
conv.delete()
def clear(self, request: ServeRequest) -> None:
"""Clear current conversation and its messages
Args:
request (ServeRequest): The request
"""
conv: StorageConversation = self.create_storage_conv(request)
conv.clear()
def get_list(self, request: ServeRequest) -> List[ServerResponse]:
"""Get a list of Conversation entities
@@ -192,13 +202,27 @@ class Service(BaseService[ServeEntity, ServeRequest, ServerResponse]):
conv: StorageConversation = self.create_storage_conv(request)
result = []
messages = _append_view_messages(conv.messages)
feedback_service = get_service()
feedbacks = feedback_service.list_conv_feedbacks(conv_uid=request.conv_uid)
fb_map = {fb.message_id: fb.to_dict() for fb in feedbacks}
for msg in messages:
feedback = {}
if (
msg.round_index is not None
and fb_map.get(str(msg.round_index)) is not None
):
feedback = fb_map.get(str(msg.round_index))
result.append(
MessageVo(
role=msg.type,
context=msg.content,
order=msg.round_index,
model_name=self.config.default_model,
feedback=feedback,
)
)
return result