Update Akeno/plugins/cohere.py

Akeno/plugins/cohere.py

@@ -5,10 +5,10 @@ from Akeno.utils.database import db
 
 from pyrogram import Client, filters, enums
 from pyrogram.types import Message
-import cohere
+from cohere import Client as Client_cohere
 from config import cohere_key, CMD_HANDLER
 
-co = cohere.Client(cohere_key)
+co_pro = Client_cohere(cohere_key)
 
 @Akeno(filters.command("cohere", CMD_HANDLER) & filters.me)
 async def cohere(c: Client, message: Message):
@@ -26,7 +26,7 @@ async def cohere(c: Client, message: Message):
             return
         chat_history.append({"role": "USER", "message": prompt})
         await message.edit_text("<code>Umm, lemme think...</code>")
-        response = co.chat_stream(
+        response = co_pro.chat_stream(
             chat_history=chat_history,
             model="command-r-plus",
             message=prompt,
@@ -82,4 +82,4 @@ async def cohere(c: Client, message: Message):
             )
 
     except Exception as e:
-        await message.edit_text(f"An error occurred: {e}")
+        await message.edit_text(f"An error occurred: {e}")