Message(msg) -> Message(content=msg)

This commit is contained in:
刘棒棒 2023-12-21 14:17:05 +08:00
parent bd3d5fe1f3
commit 44e648eabf

View file

@ -302,7 +302,7 @@ class OpenAIGPTAPI(BaseGPTAPI, RateLimiter):
def _process_message(self, messages: Union[str, Message, list[dict], list[Message], list[str]]) -> list[dict]:
"""convert messages to list[dict]."""
if isinstance(messages, list):
messages = [Message(msg) if isinstance(msg, str) else msg for msg in messages]
messages = [Message(content=msg) if isinstance(msg, str) else msg for msg in messages]
return [msg if isinstance(msg, dict) else msg.to_dict() for msg in messages]
if isinstance(messages, Message):