refactor: adjust default and maximum limits for plugin candidates and torrent results; enhance result formatting for agents

tests/test_agent_query_workflows_tool.py

@@ -0,0 +1,56 @@
+import asyncio
+import json
+import unittest
+from types import SimpleNamespace
+from unittest.mock import AsyncMock, MagicMock, patch
+
+from app.agent.tools.impl.query_workflows import QueryWorkflowsTool
+
+
+class _AsyncSessionContext:
+    """为工作流查询工具提供最小异步 DB 上下文。"""
+
+    async def __aenter__(self):
+        return object()
+
+    async def __aexit__(self, exc_type, exc, tb):
+        return False
+
+
+class TestQueryWorkflowsTool(unittest.TestCase):
+    def test_query_workflows_omits_large_result_field(self):
+        tool = QueryWorkflowsTool(session_id="session-1", user_id="10001")
+        workflow = SimpleNamespace(
+            id=1,
+            name="demo",
+            description="demo workflow",
+            state="S",
+            trigger_type="manual",
+            run_count=1,
+            timer=None,
+            event_type=None,
+            add_time="2026-05-08 10:00:00",
+            last_time="2026-05-08 10:01:00",
+            current_action=None,
+            result="x" * 10000,
+        )
+        workflow_oper = MagicMock()
+        workflow_oper.async_list = AsyncMock(return_value=[workflow])
+
+        with patch(
+            "app.agent.tools.impl.query_workflows.AsyncSessionFactory",
+            return_value=_AsyncSessionContext(),
+        ), patch(
+            "app.agent.tools.impl.query_workflows.WorkflowOper",
+            return_value=workflow_oper,
+        ):
+            result = asyncio.run(tool.run())
+
+        payload = json.loads(result)
+        self.assertEqual(len(payload), 1)
+        self.assertEqual(payload[0]["name"], "demo")
+        self.assertNotIn("result", payload[0])
+
+
+if __name__ == "__main__":
+    unittest.main()

tests/test_agent_tool_result_limits.py

@@ -0,0 +1,51 @@
+import asyncio
+import json
+import unittest
+
+from app.agent.tools.base import (
+    DEFAULT_TOOL_RESULT_MAX_CHARS,
+    MoviePilotTool,
+    format_tool_result_for_agent,
+)
+
+
+class OversizedResultTool(MoviePilotTool):
+    name: str = "oversized_result_tool"
+    description: str = "Tool used to verify result truncation."
+
+    async def run(self, **kwargs) -> str:
+        return "x" * (DEFAULT_TOOL_RESULT_MAX_CHARS + 100)
+
+
+class TestAgentToolResultLimits(unittest.TestCase):
+    def test_arun_truncates_oversized_tool_result(self):
+        tool = OversizedResultTool(session_id="session-1", user_id="10001")
+
+        result = asyncio.run(tool._arun())
+        payload = json.loads(result)
+
+        self.assertTrue(payload["tool_result_truncated"])
+        self.assertEqual(payload["tool_name"], "oversized_result_tool")
+        self.assertEqual(payload["returned_chars"], DEFAULT_TOOL_RESULT_MAX_CHARS)
+        self.assertGreater(payload["total_chars"], payload["returned_chars"])
+
+    def test_formatter_preserves_sensitive_json_fields_for_agent_use(self):
+        result = format_tool_result_for_agent(
+            {
+                "cookie": "uid=abc; token=secret",
+                "nested": {
+                    "api_key": "secret-key",
+                    "plugin_author": "MoviePilot",
+                },
+            },
+            tool_name="sensitive_tool",
+        )
+        payload = json.loads(result)
+
+        self.assertEqual(payload["cookie"], "uid=abc; token=secret")
+        self.assertEqual(payload["nested"]["api_key"], "secret-key")
+        self.assertEqual(payload["nested"]["plugin_author"], "MoviePilot")
+
+
+if __name__ == "__main__":
+    unittest.main()