Added chat session
Some checks failed
Docker Build and Publish / build (push) Has been cancelled

This commit is contained in:
Alessandro Santini
2025-02-02 11:30:41 +01:00
parent 47f5d4921b
commit 3569dd1ac7

11
main.py
View File

@@ -14,15 +14,19 @@ async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
fr"Hi {user.mention_markdown_v2()}\!", fr"Hi {user.mention_markdown_v2()}\!",
) )
async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None: async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
"""Handle a message.""" """Handle a message."""
response = flowiseClient.create_prediction(PredictionData(question=update.message.text, chatflowId=chatflowId, streaming=False))
# Extract the session id from the message
session_id = update.message.from_user.id
# Create a prediction
response = flowiseClient.create_prediction(PredictionData(question=update.message.text, chatflowId=chatflowId, streaming=False, chatId=session_id))
# Extract text from the response # Extract text from the response
response_text = "" response_text = ""
for item in response: for item in response:
print(item)
response_text += item["text"] response_text += item["text"]
# Escape special characters for markdown v2
await update.message.reply_text(response_text) await update.message.reply_text(response_text)
if __name__ == "__main__": if __name__ == "__main__":
@@ -49,7 +53,6 @@ if __name__ == "__main__":
raise ValueError("TELEGRAM_API_KEY not found in environment variables") raise ValueError("TELEGRAM_API_KEY not found in environment variables")
application = Application.builder().token(TELEGRAM_API_KEY).build() application = Application.builder().token(TELEGRAM_API_KEY).build()
application.add_handler(MessageHandler(filters.TEXT, handle_message)) application.add_handler(MessageHandler(filters.TEXT, handle_message))
application.add_handler(CommandHandler("start", start))
# Run the bot # Run the bot
application.run_polling() application.run_polling()