Progress bar for learn_from_user_feedback

.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 }}

autogen/agentchat/contrib/teachable_agent.py

@@ -1,12 +1,12 @@
 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
 from typing import Callable, Dict, Optional, Union, List, Tuple, Any
 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 = []