Fix reaction handling.

fjerkroa_bot/ai_responder.py

@@ -248,7 +248,7 @@ class AIResponder(AIResponderBase):
     async def fix(self, answer: str) -> str:
         raise NotImplementedError()
 
-    async def memory_rewrite(self, memory: str, user: str, question: str, answer: str) -> str:
+    async def memory_rewrite(self, memory: str, message_user: str, answer_user: str, question: str, answer: str) -> str:
         raise NotImplementedError()
 
     async def translate(self, text: str, language: str = "english") -> str:
@@ -292,13 +292,14 @@ class AIResponder(AIResponderBase):
             response["picture"] = await self.translate(response["picture"])
         return True
 
-    async def memoize(self, user: str, message: str, answer: str) -> None:
-        self.memory = await self.memory_rewrite(self.memory, user, message, answer)
+    async def memoize(self, message_user: str, answer_user: str, message: str, answer: str) -> None:
+        self.memory = await self.memory_rewrite(self.memory, message_user, answer_user, message, answer)
         self.update_memory(self.memory)
 
     async def memoize_reaction(self, message_user: str, reaction_user: str, operation: str, reaction: str, message: str) -> None:
         quoted_message = message.replace('\n', '\n> ')
-        await self.memoize(reaction_user, f'{message_user}:\n> {quoted_message}',
+        await self.memoize(message_user, reaction_user,
+                           f'> {quoted_message}',
                            f'User {reaction_user} has {operation} this raction: {reaction}')
 
     async def send(self, message: AIMessage) -> AIResponse:
@@ -352,7 +353,7 @@ class AIResponder(AIResponderBase):
 
             # Update memory
             if answer_message.answer is not None:
-                await self.memoize(message.user, message.message, answer_message.answer)
+                await self.memoize(message.user, 'assistant', message.message, answer_message.answer)
 
             # Return the updated answer message
             return answer_message