Add last message query generator

src/canopy/chat_engine/query_generator/__init__.py

@@ -1,2 +1,3 @@
 from .base import QueryGenerator
 from .function_calling import FunctionCallingQueryGenerator
+from .last_message import LastMessageQueryGenerator

src/canopy/chat_engine/query_generator/last_message.py

@@ -0,0 +1,17 @@
+from typing import List
+
+from canopy.chat_engine.query_generator import QueryGenerator
+from canopy.models.data_models import Messages, Query
+
+
+class LastMessageQueryGenerator(QueryGenerator):
+
+    def generate(self,
+                 messages: Messages,
+                 max_prompt_tokens: int) -> List[Query]:
+        return [Query(text=messages[-1].content)]
+
+    async def agenerate(self,
+                        messages: Messages,
+                        max_prompt_tokens: int) -> List[Query]:
+        return self.generate(messages, max_prompt_tokens)

tests/unit/query_generators/test_last_message_query_generator.py

@@ -0,0 +1,29 @@
+import pytest
+
+from canopy.chat_engine.query_generator import LastMessageQueryGenerator
+from canopy.models.data_models import UserMessage, Query
+
+
+@pytest.fixture
+def sample_messages():
+    return [
+        UserMessage(content="What is photosynthesis?")
+    ]
+
+
+@pytest.fixture
+def query_generator():
+    return LastMessageQueryGenerator()
+
+
+def test_generate(query_generator, sample_messages):
+    expected = [Query(text=sample_messages[-1].content)]
+    actual = query_generator.generate(sample_messages, 0)
+    assert actual == expected
+
+
+@pytest.mark.asyncio
+async def test_agenerate(query_generator, sample_messages):
+    expected = [Query(text=sample_messages[-1].content)]
+    actual = await query_generator.agenerate(sample_messages, 0)
+    assert actual == expected