From c39bfcaa6ed283e84b017cdeaa947f4eedff0023 Mon Sep 17 00:00:00 2001 From: Chi Wang Date: Sun, 1 Oct 2023 04:57:21 -0700 Subject: [PATCH] typing & docstr update (#59) * typing & docstr update * bump version to 0.1.5 --- autogen/agentchat/conversable_agent.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/autogen/agentchat/conversable_agent.py b/autogen/agentchat/conversable_agent.py index fdf42f67b6a3..84e7dea8dec7 100644 --- a/autogen/agentchat/conversable_agent.py +++ b/autogen/agentchat/conversable_agent.py @@ -210,8 +210,8 @@ def max_consecutive_auto_reply(self, sender: Optional[Agent] = None) -> int: return self._max_consecutive_auto_reply if sender is None else self._max_consecutive_auto_reply_dict[sender] @property - def chat_messages(self) -> Dict[str, List[Dict]]: - """A dictionary of conversations from name to list of ChatCompletion messages.""" + def chat_messages(self) -> Dict[Agent, List[Dict]]: + """A dictionary of conversations from agent to list of messages.""" return self._oai_messages def last_message(self, agent: Optional[Agent] = None) -> Dict: