Enhance tests for workflow router builtins by validating specific instructional text in the generated prompt and removing cost assertions.
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user