diff --git a/.github/workflows/contrib-openai.yml b/.github/workflows/contrib-openai.yml index 63abe36b891..f0d497b3cd6 100644 --- a/.github/workflows/contrib-openai.yml +++ b/.github/workflows/contrib-openai.yml @@ -161,7 +161,7 @@ jobs: python -m pip install --upgrade pip wheel pip install -e .[teachable] python -c "import autogen" - pip install coverage + pip install coverage pytest - name: Coverage env: OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }} diff --git a/autogen/agentchat/contrib/teachable_agent.py b/autogen/agentchat/contrib/teachable_agent.py index 29d7f197ffc..e746548862c 100644 --- a/autogen/agentchat/contrib/teachable_agent.py +++ b/autogen/agentchat/contrib/teachable_agent.py @@ -1,5 +1,4 @@ import os -from autogen import oai from autogen.agentchat.agent import Agent from autogen.agentchat.assistant_agent import ConversableAgent from autogen.agentchat.contrib.text_analyzer_agent import TextAnalyzerAgent @@ -7,6 +6,7 @@ import chromadb from chromadb.config import Settings import pickle +from tqdm import tqdm try: @@ -133,7 +133,7 @@ def learn_from_user_feedback(self): print(colored("\nREVIEWING CHAT FOR USER TEACHINGS TO REMEMBER", "light_yellow")) # Look at each user turn. if len(self.user_comments) > 0: - for comment in self.user_comments: + for comment in tqdm(self.user_comments): # Consider whether to store something from this user turn in the DB. self.consider_memo_storage(comment) self.user_comments = []