|
@@ -70,6 +70,8 @@ class AgentModePolicyTest(unittest.IsolatedAsyncioTestCase):
|
|
|
"AGENT_MODE": "legacy",
|
|
"AGENT_MODE": "legacy",
|
|
|
"AGENT_MAX_TOTAL_AGENTS": "not-a-number",
|
|
"AGENT_MAX_TOTAL_AGENTS": "not-a-number",
|
|
|
"AGENT_RESOURCE_BUDGET_ENABLED": "not-a-boolean",
|
|
"AGENT_RESOURCE_BUDGET_ENABLED": "not-a-boolean",
|
|
|
|
|
+ "AGENT_MAX_VALIDATION_TOOL_CALLS": "not-a-number",
|
|
|
|
|
+ "AGENT_VALIDATOR_SEARCH_PROVIDER": "unsupported",
|
|
|
},
|
|
},
|
|
|
clear=False,
|
|
clear=False,
|
|
|
):
|
|
):
|
|
@@ -77,6 +79,19 @@ class AgentModePolicyTest(unittest.IsolatedAsyncioTestCase):
|
|
|
self.assertEqual("legacy", trace.context["agent_mode"])
|
|
self.assertEqual("legacy", trace.context["agent_mode"])
|
|
|
self.assertNotIn("resource_budget", trace.context)
|
|
self.assertNotIn("resource_budget", trace.context)
|
|
|
|
|
|
|
|
|
|
+ async def test_recursive_rejects_invalid_validator_provider_before_trace(self):
|
|
|
|
|
+ with patch.dict(
|
|
|
|
|
+ os.environ,
|
|
|
|
|
+ {
|
|
|
|
|
+ "AGENT_MODE": "recursive",
|
|
|
|
|
+ "AGENT_VALIDATOR_SEARCH_PROVIDER": "unsupported",
|
|
|
|
|
+ },
|
|
|
|
|
+ clear=False,
|
|
|
|
|
+ ):
|
|
|
|
|
+ with self.assertRaisesRegex(ValueError, "search_provider"):
|
|
|
|
|
+ await self._prepare_new("recursive")
|
|
|
|
|
+ self.assertEqual([], await self.store.list_traces(limit=10))
|
|
|
|
|
+
|
|
|
async def test_invalid_mode_does_not_create_trace(self):
|
|
async def test_invalid_mode_does_not_create_trace(self):
|
|
|
with self.assertRaisesRegex(ValueError, "Invalid AGENT_MODE"):
|
|
with self.assertRaisesRegex(ValueError, "Invalid AGENT_MODE"):
|
|
|
await self._prepare_new("unsupported")
|
|
await self._prepare_new("unsupported")
|