typing & docstr update (#59)

* typing & docstr update * bump version to 0.1.5
microsoft · Oct 1, 2023 · c39bfca · c39bfca
1 parent e4eb7ae
commit c39bfca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: