feat: Merge geekan:env_refactore

This commit is contained in:
莘权 马 2023-11-22 16:57:42 +08:00
commit 20ae74e99d
117 changed files with 657 additions and 671 deletions

View file

@ -83,6 +83,7 @@ class Message(BaseModel):
new_val = val
super().__setattr__(key, new_val)
def __str__(self):
# prefix = '-'.join([self.role, str(self.cause_by)])
return f"{self.role}: {self.content}"
@ -109,6 +110,7 @@ class Message(BaseModel):
return None
class UserMessage(Message):
"""便于支持OpenAI的消息
Facilitate support for OpenAI messages
@ -118,6 +120,7 @@ class UserMessage(Message):
super().__init__(content=content, role="user")
class SystemMessage(Message):
"""便于支持OpenAI的消息
Facilitate support for OpenAI messages
@ -200,3 +203,4 @@ class MessageQueue:
logger.warning(f"JSON load failed: {v}, error:{e}")
return q