新增功能

2026-07-28 09:46:53 +08:00
parent 45b3bc0852
commit 980795b4da
57 changed files with 18421 additions and 1051 deletions
@@ -0,0 +1,106 @@
+# -*- coding: utf-8 -*-
+"""Routing tests for the local Grok Build customer-service provider."""
+
+from __future__ import annotations
+
+import unittest
+from unittest import mock
+
+import ai_chat
+import ai_config
+from grok_customer_agent import GrokCustomerAgentError
+
+
+class GrokCustomerProviderRoutingTest(unittest.TestCase):
+    def test_stable_wecom_session_is_forwarded_to_local_agent(self) -> None:
+        session_id = "00112233445566778899aabbccddeeff"
+        with (
+            mock.patch.object(
+                ai_config,
+                "GROK_CUSTOMER_SERVICE_ENABLED",
+                True,
+                create=True,
+            ),
+            mock.patch(
+                "grok_customer_agent.generate_customer_reply",
+                return_value="收到，请问有什么可以帮您？",
+            ) as generate,
+        ):
+            result = ai_chat.call_ai_text(
+                "客户最新消息",
+                history=[
+                    {"role": "user", "content": "本地历史由工具读取"},
+                ],
+                session_id=session_id,
+            )
+
+        self.assertEqual("收到，请问有什么可以帮您？", result)
+        generate.assert_called_once_with(
+            "客户最新消息",
+            session_id=session_id,
+        )
+
+    def test_ad_hoc_draft_gets_an_isolated_valid_scope(self) -> None:
+        with (
+            mock.patch.object(
+                ai_config,
+                "GROK_CUSTOMER_SERVICE_ENABLED",
+                True,
+                create=True,
+            ),
+            mock.patch(
+                "grok_customer_agent.generate_customer_reply",
+                return_value="草稿",
+            ) as generate,
+        ):
+            ai_chat.call_ai_text("临时草稿")
+
+        generated_session = generate.call_args.kwargs["session_id"]
+        self.assertRegex(generated_session, r"^[0-9a-f]{32}$")
+
+    def test_disabled_agent_never_falls_back_to_http_model(self) -> None:
+        with (
+            mock.patch.object(
+                ai_config,
+                "GROK_CUSTOMER_SERVICE_ENABLED",
+                False,
+                create=True,
+            ),
+            mock.patch(
+                "grok_customer_agent.generate_customer_reply",
+            ) as generate,
+            mock.patch.object(ai_chat.requests, "post") as post,
+            self.assertRaisesRegex(RuntimeError, "已关闭"),
+        ):
+            ai_chat.call_ai_text(
+                "客户消息",
+                session_id="0" * 32,
+            )
+
+        generate.assert_not_called()
+        post.assert_not_called()
+
+    def test_unfinished_agent_reply_is_not_returned_by_unified_entry(self) -> None:
+        with (
+            mock.patch.object(
+                ai_config,
+                "GROK_CUSTOMER_SERVICE_ENABLED",
+                True,
+                create=True,
+            ),
+            mock.patch.object(ai_config, "AI_USE_VISION", False),
+            mock.patch(
+                "grok_customer_agent.generate_customer_reply",
+                side_effect=GrokCustomerAgentError("未完整结束"),
+            ),
+        ):
+            result = ai_chat.get_ai_reply(
+                chat_text="客户消息",
+                session_id="0" * 32,
+            )
+
+        self.assertEqual("", result)
+
+
+if __name__ == "__main__":
+    unittest.main()