UPDATE

rag/tests/test_chat_context.py

@@ -0,0 +1,52 @@
+from unittest.mock import patch
+
+from django.test import SimpleTestCase
+
+from rag.chat import build_chat_context
+
+
+class ChatContextTests(SimpleTestCase):
+    @patch("rag.chat.search_with_query")
+    @patch("rag.chat._rank_text_chunks_by_query")
+    @patch("rag.chat.chunk_text")
+    def test_build_chat_context_combines_farm_and_kb_context(
+        self,
+        mock_chunk_text,
+        mock_rank_text_chunks_by_query,
+        mock_search_with_query,
+    ):
+        mock_chunk_text.return_value = ["chunk-a", "chunk-b"]
+        mock_rank_text_chunks_by_query.return_value = ["chunk-b"]
+        mock_search_with_query.return_value = [
+            {"text": "kb text 1"},
+            {"text": "kb text 2"},
+        ]
+
+        context = build_chat_context(
+            query="وضعیت مزرعه چطور است؟",
+            farm_uuid="farm-123",
+            farm_details={"sensor_payload": {"sensor-7-1": {"soil_moisture": 30}}},
+        )
+
+        self.assertIn("[بخش‌های مرتبط بازیابی‌شده از اطلاعات مزرعه]", context)
+        self.assertIn("chunk-b", context)
+        self.assertIn("[اطلاعات بازیابی‌شده از پایگاه دانش]", context)
+        self.assertIn("kb text 1", context)
+        self.assertIn("kb text 2", context)
+
+    @patch("rag.chat.search_with_query", return_value=[])
+    @patch("rag.chat._rank_text_chunks_by_query", return_value=[])
+    @patch("rag.chat.chunk_text", return_value=["farm chunk"])
+    def test_build_chat_context_falls_back_to_full_farm_context(
+        self,
+        _mock_chunk_text,
+        _mock_rank_text_chunks_by_query,
+        _mock_search_with_query,
+    ):
+        context = build_chat_context(
+            query="رطوبت چقدر است؟",
+            farm_uuid="farm-123",
+            farm_details={"sensor_payload": {"sensor-7-1": {"soil_moisture": 30}}},
+        )
+
+        self.assertEqual(context, "")