This commit is contained in:
@@ -162,10 +162,17 @@ class BuilderPromptTests(unittest.TestCase):
|
||||
self.assertIn("caller's saved LLM credential", agent_source)
|
||||
|
||||
def test_builder_defaults_raise_timeout_budget(self) -> None:
|
||||
agent_source = Path(__file__).resolve().parents[1].joinpath("agent.py").read_text()
|
||||
config_source = Path(__file__).resolve().parents[1].joinpath(
|
||||
"agent_builder/config.py"
|
||||
).read_text()
|
||||
|
||||
self.assertIn("BUILDER_WALL_TIMEOUT_SECONDS = 60 * 60", agent_source)
|
||||
self.assertIn("timeout_seconds=BUILDER_WALL_TIMEOUT_SECONDS", agent_source)
|
||||
self.assertIn(
|
||||
"grant_run_timeout_seconds=BUILDER_GRANT_RUN_TIMEOUT_SECONDS",
|
||||
agent_source,
|
||||
)
|
||||
self.assertIn('os.environ.get("A2A_SANDBOX_TIMEOUT_S")', config_source)
|
||||
self.assertIn('os.environ.get("SANDBOX_TIMEOUT_S", "1200")', config_source)
|
||||
self.assertIn('deploy_wait_timeout_s=int(os.environ.get("DEPLOY_WAIT_TIMEOUT_S", "900"))', config_source)
|
||||
|
||||
Reference in New Issue
Block a user