fix: use platform llm grants for builder
All checks were successful
build / build (push) Successful in 2s
All checks were successful
build / build (push) Successful in 2s
This commit is contained in:
@@ -24,6 +24,13 @@ class BuilderPromptTests(unittest.TestCase):
|
||||
)
|
||||
self.assertIn("stream=True", agent_source)
|
||||
|
||||
def test_outer_builder_uses_platform_llm_grants(self) -> None:
|
||||
agent_source = Path(__file__).resolve().parents[1].joinpath("agent.py").read_text()
|
||||
|
||||
self.assertIn("llm_provisioning = LLMProvisioning.PLATFORM", agent_source)
|
||||
self.assertIn("caller_pays_llm=False", agent_source)
|
||||
self.assertIn("platform-scoped LiteLLM grant", agent_source)
|
||||
|
||||
def test_builder_defaults_raise_timeout_budget(self) -> None:
|
||||
config_source = Path(__file__).resolve().parents[1].joinpath(
|
||||
"agent_builder/config.py"
|
||||
|
||||
Reference in New Issue
Block a user