Fix OpenAI chat message selection

2026-10-12 02:19:52 +00:00 · 2026-08-11 13:26:29 +08:00
parent a0857e980a
commit 01b456900a
2 changed files with 20 additions and 2 deletions
@@ -122,6 +122,24 @@ class TestOpenAI(unittest.TestCase):

        self.assertIsNotNone(response["choices"][0]["message"]["content"])

+    def testChatLatestMessage(self):
+        """
+        Test a chat completion with multiple messages
+        """
+
+        response = self.client.post(
+            "/v1/chat/completions",
+            json={
+                "messages": [
+                    {"role": "system", "content": "You are a helpful assistant"},
+                    {"role": "user", "content": "Hello"},
+                ],
+                "model": "segmentation",
+            },
+        ).json()
+
+        self.assertEqual(response["choices"][0]["message"]["content"], "Hello")
+
    def testChatPipeline(self):
        """
        Test a chat completion with a pipeline