Erwin Jacimino 8 месяцев назад
Родитель
Сommit
461ded2973
2 измененных файлов с 2 добавлено и 0 удалено
  1. 1 0
      main.py
  2. 1 0
      routes/chat.py

+ 1 - 0
main.py

@@ -29,6 +29,7 @@ def main():
         redis_client = redis.Redis(host='localhost', port=6379, db=1 if DEVELOPMENT else 0)
         # eliminar la lista de usuarios conectados al iniciar
         redis_client.delete("connected_users")
+        redis_client.delete("chat_history")
         initialize_db()
         # Display startup information
         logger.info(f"Server starting on http://localhost:{PORT}")

+ 1 - 0
routes/chat.py

@@ -102,6 +102,7 @@ async def chat_irc_endpoint(websocket: WebSocket):
                     logger.debug(f"IA Message from {payload['username']}")
                     message_username = payload["username"]
                     response_content = await generate_completion(parsed_messages, current_user)
+                    redis_client.rpush("chat_history", json.dumps({"type": "ai_message", "username": "IAKlein", "message": response_content}))
                     response = {"type": "message", "username": "IAKlein", "message": response_content}
                 elif event_type == "mention":
                     logger.debug(f"Mention to {payload['username']}")