diff --git a/backend/tests/test_workflow_router_builtins.py b/backend/tests/test_workflow_router_builtins.py index ddcf010..59a49ca 100644 --- a/backend/tests/test_workflow_router_builtins.py +++ b/backend/tests/test_workflow_router_builtins.py @@ -168,9 +168,13 @@ class RouterAndBuiltinTests(unittest.IsolatedAsyncioTestCase): self.assertEqual(result["inputs"]["web_search_query"], "Iran political news today") self.assertIn("Recent conversation:\n1. User: what's the weather in Iran today", prompt) self.assertIn("search for 'Iran political news today'", prompt) + self.assertIn("Never choose Input -> Output for weather, news, politics today", prompt) + self.assertIn("Web Answer is for straightforward current lookups", prompt) + self.assertIn("Research is for explicit deeper investigation", prompt) self.assertIn("Allowed workflows. This list is already filtered", prompt) self.assertIn("web-answer", prompt) self.assertNotIn("input-output", prompt) + self.assertNotIn("Cost:", prompt) self.assertNotIn("Web search mode:", prompt) self.assertNotIn("Selected knowledge database:", prompt) db.close()