OpenBMB/ChatDev
1
1
Fork 0
mirror of https://github.com/OpenBMB/ChatDev.git synced 2024-09-16 04:37:11 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #419 from TheChronicMonster/refusal-fix

Browse Source 
fix: unexpected keyword arg refusal
This commit is contained in:
Thinkwee 2024-08-28 10:31:34 +08:00 committed by GitHub
commit 0e5cfc81a9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
camel/messages/chat_messages.py
View File

@ -38,12 +38,14 @@ class ChatMessage(BaseMessage):
for the message.
role (str): The role of the message in OpenAI chat system.
content (str): The content of the message. (default: :obj:`""`)
refusal (str): The refusal to build argument.
"""
role_name: str
role_type: RoleType
meta_dict: Optional[Dict[str, str]]
role: str
content: str = ""
refusal: str = None
if openai_new_api:
function_call: Optional[FunctionCall] = None
tool_calls: Optional[ChatCompletionMessageToolCall] = None
@ -55,6 +57,7 @@ class ChatMessage(BaseMessage):
meta_dict=self.meta_dict,
role="user",
content=self.content,
refusal=self.refusal,
)
@ -72,12 +75,14 @@ class AssistantChatMessage(ChatMessage):
role (str): The role of the message in OpenAI chat system.
(default: :obj:`"assistant"`)
content (str): The content of the message. (default: :obj:`""`)
refusal (str): The refusal to build argument.
"""
role_name: str
role_type: RoleType = RoleType.ASSISTANT
meta_dict: Optional[Dict[str, str]] = None
role: str = "user"
content: str = ""
refusal: str = None
@dataclass
@ -92,9 +97,11 @@ class UserChatMessage(ChatMessage):
role (str): The role of the message in OpenAI chat system.
(default: :obj:`"user"`)
content (str): The content of the message. (default: :obj:`""`)
refusal (str): The refusal to build argument.
"""
role_name: str
role_type: RoleType = RoleType.USER
meta_dict: Optional[Dict[str, str]] = None
role: str = "user"
content: str = ""
refusal: str = None