Enhance tests for workflow router builtins by validating specific instructional text in the generated prompt and removing cost assertions.

This commit is contained in:
2026-06-15 21:27:11 +02:00
parent d8c6dc06c6
commit e35a29baef

View File

@@ -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()