Handle messages as direct, when bot is mentioned.

tests/test_main.py

@@ -8,7 +8,7 @@ import logging
 import pytest
 from unittest.mock import Mock, PropertyMock, MagicMock, AsyncMock, patch, mock_open, ANY
 from fjerkroa_bot import FjerkroaBot
-from fjerkroa_bot.ai_responder import parse_maybe_json, AIResponse
+from fjerkroa_bot.ai_responder import parse_maybe_json, AIResponse, AIMessage
 from discord import User, Message, TextChannel
 
 
@@ -20,7 +20,7 @@ class TestBotBase(unittest.IsolatedAsyncioTestCase):
             Mock(text="Nice day today!")
         ]
         self.config_data = {
-            "openai-token": os.environ['OPENAI_TOKEN'],
+            "openai-token": os.environ.get('OPENAI_TOKEN', 'test'),
             "model": "gpt-4",
             "max-tokens": 1024,
             "temperature": 0.9,
@@ -79,18 +79,19 @@ class TestFunctionality(TestBotBase):
         self.assertEqual(parse_maybe_json(json_struct), expected_output)
 
     async def test_message_lings(self) -> None:
+        request = AIMessage('Lala', 'Hello there!', 'chat', False,)
         message = {'answer': 'Test [Link](https://www.example.com/test)',
                    'answer_needed': True, 'staff': None, 'picture': None, 'hack': False}
         expected = AIResponse('Test https://www.example.com/test', True, None, None, False)
-        self.assertEqual(str(await self.bot.airesponder.post_process(message)), str(expected))
+        self.assertEqual(str(await self.bot.airesponder.post_process(request, message)), str(expected))
         message = {'answer': 'Test @[Link](https://www.example.com/test)',
                    'answer_needed': True, 'staff': None, 'picture': None, 'hack': False}
         expected = AIResponse('Test Link', True, None, None, False)
-        self.assertEqual(str(await self.bot.airesponder.post_process(message)), str(expected))
+        self.assertEqual(str(await self.bot.airesponder.post_process(request, message)), str(expected))
         message = {'answer': 'Test [Link](https://www.example.com/test) and [Link2](https://xxx) lala',
                    'answer_needed': True, 'staff': None, 'picture': None, 'hack': False}
         expected = AIResponse('Test https://www.example.com/test and https://xxx lala', True, None, None, False)
-        self.assertEqual(str(await self.bot.airesponder.post_process(message)), str(expected))
+        self.assertEqual(str(await self.bot.airesponder.post_process(request, message)), str(expected))
 
     async def test_on_message_event(self) -> None:
         async def acreate(*a, **kw):