_

chatbot/plugins/chat.py

@@ -167,10 +167,10 @@ gen = genai.Client(api_key=GOOGLE_API_KEY)
 
 async def search_auto_by_cohere(user_id, prompt):
     COHERE_API_KEY = os.environ.get("COHERE_API_KEY")
-    co = cohere.AsyncClient(api_key=COHERE_API_KEY)
+    co = cohere.Client(api_key=COHERE_API_KEY)
     chat_history = await db._get_cohere_chat_from_db(user_id)
     chat_history.append({"role": "User", "message": prompt})
-    stream = await co.chat_stream( 
+    stream = co.chat_stream( 
         model='command-r-08-2024',
         message=prompt,
         temperature=0.3,
@@ -180,7 +180,7 @@ async def search_auto_by_cohere(user_id, prompt):
     )
     data = []
     output = ""
-    async for event in stream:
+    for event in stream:
         if event.event_type == "search-results":
             if event.documents:
                 data.extend(event.documents)