From efc75aebc92b34fdcfb3248fdb2a36c80633aec0 Mon Sep 17 00:00:00 2001
From: Kaveen Kumarasinghe <ocfinancesmc@gmail.com>
Date: Tue, 18 Apr 2023 00:48:37 -0400
Subject: [PATCH] fix bugs

---
 cogs/commands.py           | 2 +-
 cogs/search_service_cog.py | 9 +++++++--
 gpt3discord.py             | 2 +-
 3 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/cogs/commands.py b/cogs/commands.py
index d841abc7..15ef0093 100644
--- a/cogs/commands.py
+++ b/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",
diff --git a/cogs/search_service_cog.py b/cogs/search_service_cog.py
index 060c7f5c..2f47308e 100644
--- a/cogs/search_service_cog.py
+++ b/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
diff --git a/gpt3discord.py b/gpt3discord.py
index 3b71f569..ff71a474 100644
--- a/gpt3discord.py
+++ b/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")