klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Bug Fix] Handle chat sessions properly during app.chat() calls (#1084)

Browse Source
This commit is contained in:
Deshraj Yadav
2023-12-30 15:36:24 +05:30
committed by GitHub
parent 52b4577d3b
commit a54dde0509
10 changed files with 124 additions and 187 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
embedchain/memory/message.py
View File

@@ -14,16 +14,16 @@ class BaseMessage(JSONSerializable):
# The string content of the message.
content: str
# The creator of the message. AI, Human, Bot etc.
by: str
# The created_by of the message. AI, Human, Bot etc.
created_by: str
# Any additional info.
metadata: Dict[str, Any]
def __init__(self, content: str, creator: str, metadata: Optional[Dict[str, Any]] = None) -> None:
def __init__(self, content: str, created_by: str, metadata: Optional[Dict[str, Any]] = None) -> None:
super().__init__()
self.content = content
self.creator = creator
self.created_by = created_by
self.metadata = metadata
@property
@@ -36,7 +36,7 @@ class BaseMessage(JSONSerializable):
return True
def __str__(self) -> str:
return f"{self.creator}: {self.content}"
return f"{self.created_by}: {self.content}"
class ChatMessage(JSONSerializable):
@@ -57,7 +57,7 @@ class ChatMessage(JSONSerializable):
overwritting it with new message."
)
self.human_message = BaseMessage(content=message, creator="human", metadata=metadata)
self.human_message = BaseMessage(content=message, created_by="human", metadata=metadata)
def add_ai_message(self, message: str, metadata: Optional[dict] = None):
if self.ai_message:
@@ -66,7 +66,7 @@ class ChatMessage(JSONSerializable):
overwritting it with new message."
)
self.ai_message = BaseMessage(content=message, creator="ai", metadata=metadata)
self.ai_message = BaseMessage(content=message, created_by="ai", metadata=metadata)
def __str__(self) -> str:
return f"{self.human_message}\n{self.ai_message}"