forust
fb43306571
Dockerfile fixes: - edu_master/phpsessid-bot: add --no-install-recommends, pin pip versions - edu_master/webinar-checker: pin pip versions with --no-cache-dir - userbot: add SHELL with pipefail for pipe operations Ruff fixes (173 → 0): - W293/W291/W292: whitespace clean via ruff format - N806: camelCase → snake_case (anilist, safone, hearts, flux, etc.) - ARG001/ARG002: prefix unused params with _ - SIM115: use context managers for file I/O - SIM117: combine nested with statements - S608: noqa on SQL f-strings (module name is validated) - E402/N812/N817: import fixes - B023: pass loop variable as argument - I001: auto-sorted imports - syntax: fixed = vs == in dtek_notif/main.py
123 lines
4.3 KiB
Python
123 lines
4.3 KiB
Python
import asyncio
|
|
|
|
from utils.config import cohere_key
|
|
from utils.scripts import import_library
|
|
|
|
cohere = import_library('cohere')
|
|
|
|
import cohere # noqa: F811, E402
|
|
|
|
co = cohere.Client(cohere_key)
|
|
|
|
from pyrogram import Client, enums, filters # noqa: E402
|
|
from pyrogram.errors import MessageTooLong # noqa: E402
|
|
from pyrogram.types import Message # noqa: E402
|
|
from utils.db import db # noqa: E402
|
|
from utils.misc import modules_help, prefix # noqa: E402
|
|
from utils.rentry import paste as rentry_paste # noqa: E402
|
|
from utils.scripts import format_exc # noqa: E402
|
|
|
|
|
|
@Client.on_message(filters.command('cohere', prefix) & filters.me)
|
|
async def cohere(c: Client, message: Message):
|
|
try:
|
|
user_id = message.from_user.id
|
|
chat_history = db.get_chat_history(user_id)
|
|
|
|
if len(message.command) > 1:
|
|
prompt = message.text.split(maxsplit=1)[1]
|
|
elif message.reply_to_message:
|
|
prompt = message.reply_to_message.text
|
|
else:
|
|
await message.edit_text(f'<b>Usage: </b><code>{prefix}cohere [prompt/reply to message]</code>')
|
|
return
|
|
|
|
db.add_chat_history(user_id, {'role': 'USER', 'message': prompt})
|
|
|
|
await message.edit_text('<code>Umm, lemme think...</code>')
|
|
|
|
response = co.chat_stream(
|
|
chat_history=chat_history,
|
|
model='command-r-plus',
|
|
message=prompt,
|
|
temperature=0.8,
|
|
tools=[{'name': 'internet_search'}],
|
|
connectors=[],
|
|
prompt_truncation='OFF',
|
|
)
|
|
output = ''
|
|
tool_message = ''
|
|
data = []
|
|
for event in response:
|
|
if event.event_type == 'tool-calls-chunk':
|
|
if event.tool_call_delta and event.tool_call_delta.text is None:
|
|
tool_message += ''
|
|
else:
|
|
tool_message += event.text
|
|
if event.event_type == 'search-results':
|
|
data.append(event.documents)
|
|
if event.event_type == 'text-generation':
|
|
output += event.text
|
|
if output == '':
|
|
output = "I can't seem to find an answer to that"
|
|
|
|
db.add_chat_history(user_id, {'role': 'CHATBOT', 'message': output})
|
|
|
|
await message.edit_text(f'<code>{tool_message}</code>')
|
|
|
|
await asyncio.sleep(5)
|
|
|
|
try:
|
|
data = data[0]
|
|
references = ''
|
|
reference_dict = {}
|
|
for item in data:
|
|
title = item['title']
|
|
url = item['url']
|
|
if title not in reference_dict:
|
|
reference_dict[title] = url
|
|
|
|
i = 1
|
|
for title, url in reference_dict.items():
|
|
references += f'**{i}.** [{title}]({url})\n'
|
|
i += 1
|
|
|
|
await message.edit_text(
|
|
f'**Question:**`{prompt}`\n**Answer:** {output}\n\n**References:**\n{references}',
|
|
parse_mode=enums.ParseMode.MARKDOWN,
|
|
disable_web_page_preview=True,
|
|
)
|
|
|
|
except IndexError:
|
|
references = ''
|
|
await message.edit_text(
|
|
f'**Question:**`{prompt}`\n**Answer:** {output}\n',
|
|
parse_mode=enums.ParseMode.MARKDOWN,
|
|
disable_web_page_preview=True,
|
|
)
|
|
except MessageTooLong:
|
|
await message.edit_text('<code>Output is too long... Pasting to rentry...</code>')
|
|
try:
|
|
output = output + '\n\n' + references if references else output
|
|
rentry_url, edit_code = await rentry_paste(text=output, return_edit=True)
|
|
except RuntimeError:
|
|
await message.edit_text('<b>Error:</b> <code>Failed to paste to rentry</code>')
|
|
return
|
|
await c.send_message(
|
|
'me',
|
|
f"Here's your edit code for Url: {rentry_url}\nEdit code: <code>{edit_code}</code>",
|
|
disable_web_page_preview=True,
|
|
)
|
|
await message.edit_text(
|
|
f'<b>Output:</b> {rentry_url}\n<b>Note:</b> <code>Edit Code has been sent to your saved messages</code>',
|
|
disable_web_page_preview=True,
|
|
)
|
|
|
|
except Exception as e:
|
|
await message.edit_text(f'An error occurred: {format_exc(e)}')
|
|
|
|
|
|
modules_help['cohere'] = {
|
|
'cohere': 'Chat with cohere ai' + '\nSupports Chat History\n' + 'Supports real time internet search'
|
|
}
|