def chatbot_pvt(client, message):
msg = message
user_id = msg.from_user.id
if not user_id in db.USERS:
ses = api_client.create_session()
ses_id = str(ses.id)
expires = str(ses.expires)
db.set_ses(user_id, ses_id, expires)
sesh: str
sesh, exp = db.get_ses(user_id)
query = msg.text
if int(exp) < time():
ses = api_client.create_session()
ses_id = str(ses.id)
expires = str(ses.expires)
db.set_ses(user_id, ses_id, expires)
sesh, exp = ses_id, expires
try:
app.send_chat_action(msg.chat.id, "typing")
response = api_client.think_thought(sesh, query)
msg.reply_text(response)
except CFError as e:
app.send_message(chat_id=msg.chat.id,
text=f"An error occurred:\n`{e}`",
parse_mode="md")
async def chatbot(app: Client, message: Message) -> None:
msg = message
if not check_message(msg):
return
user_id = msg.from_user.id
if user_id not in db.USERS:
return
sesh, exp = db.get_ses(user_id)
query = msg.text
if int(exp) < time():
ses = api_client.create_session()
ses_id = str(ses.id)
expires = str(ses.expires)
db.set_ses(user_id, ses_id, expires)
sesh, exp = ses_id, expires
try:
await msg.reply_chat_action("typing")
response = api_client.think_thought(sesh, query)
await msg.reply_text(response)
except CFError as e:
await app.send_message(
chat_id=msg.chat.id,
text=f"An error occurred:\n`{e}`",
parse_mode="md"
)
def add_user(client, message):
if not message.reply_to_message:
message.edit_text(
"Reply to someone to enable chatbot for that person!")
return
user_id = message.reply_to_message.from_user.id
is_user = db.is_user(user_id)
if not is_user:
ses = api_client.create_session()
ses_id = str(ses.id)
expires = str(ses.expires)
db.set_ses(user_id, ses_id, expires)
message.edit_text("AI enabled for user successfully!")
LOGGER.info(f"AI enabled for user - {user_id}")
else:
message.edit_text("AI is already enabled for this user!")