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

多场景对架构一期0525

Browse Source
This commit is contained in:
yhjun1026
2023-05-25 10:04:25 +08:00
parent 2fc62c16ef
commit ff99723014
12 changed files with 190 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
pilot/scene/message.py
View File

@@ -9,8 +9,7 @@ from typing import (
List,
)
from pilot.scene.base_message import BaseMessage, AIMessage, HumanMessage, SystemMessage, messages_to_dict, \
messages_from_dict
from pilot.scene.base_message import BaseMessage, AIMessage, HumanMessage, SystemMessage, ViewMessage, messages_to_dict, messages_from_dict
class OnceConversation:
@@ -27,12 +26,23 @@ class OnceConversation:
def add_user_message(self, message: str) -> None:
"""Add a user message to the store"""
has_message = any(isinstance(instance, HumanMessage) for instance in self.messages)
if has_message:
raise ValueError("Already Have Human message")
self.messages.append(HumanMessage(content=message))
def add_ai_message(self, message: str) -> None:
"""Add an AI message to the store"""
has_message = any(isinstance(instance, AIMessage) for instance in self.messages)
if has_message:
raise ValueError("Already Have Ai message")
self.messages.append(AIMessage(content=message))
""" """
def add_view_message(self, message: str) -> None:
"""Add an AI message to the store"""
self.messages.append(ViewMessage(content=message))
""" """
def add_system_message(self, message: str) -> None:
"""Add an AI message to the store"""