fix bugs

cogs/commands.py

@@ -1082,7 +1082,7 @@ async def summarize_action(self, ctx, message: discord.Message):
         name="chat",
         description="Chat with GPT connected to the internet!",
         guild_ids=ALLOWED_GUILDS,
-        checks=[Check.check_gpt_roles()],
+        checks=[Check.check_search_roles()],
     )
     @discord.option(
         name="search_scope",

cogs/search_service_cog.py

@@ -178,7 +178,11 @@ async def on_message(self, message):
                 pass
 
             agent = self.chat_agents[message.channel.id]
-            response = await self.bot.loop.run_in_executor(None, agent.run, prompt)
+            try:
+                response = await self.bot.loop.run_in_executor(None, agent.run, prompt)
+            except Exception as e:
+                response = f"Error: {e}"
+                traceback.print_exc()
             if len(response) > 2000:
                 embed_pages = await self.paginate_chat_embed(response)
                 paginator = pages.Paginator(
@@ -245,7 +249,6 @@ async def search_chat_command(
         )
 
         if use_gpt4:
-            print("using GPT4")
             llm = ChatOpenAI(
                 model="gpt-4", temperature=0.7, openai_api_key=OPENAI_API_KEY
             )
@@ -260,6 +263,8 @@ async def search_chat_command(
             agent=AgentType.CHAT_CONVERSATIONAL_REACT_DESCRIPTION,
             verbose=True,
             memory=memory,
+            max_execution_time=120,
+            early_stopping_method="generate",
         )
 
         self.chat_agents[thread.id] = agent_chain

gpt3discord.py

@@ -33,7 +33,7 @@
 from models.openai_model import Model
 
 
-__version__ = "11.3.1"
+__version__ = "11.3.2"
 
 
 PID_FILE = Path("bot.pid")