make TGI_API_TOKEN optional in fixture

This commit is contained in:
Dinesh Yeduguru 2024-11-25 10:06:39 -08:00
parent 7e6a11d17b
commit b7b764f8c8

View file

@ -166,7 +166,7 @@ def inference_tgi() -> ProviderFixture:
provider_type="remote::tgi",
config=TGIImplConfig(
url=get_env_or_fail("TGI_URL"),
api_token=get_env_or_fail("TGI_API_TOKEN"),
api_token=os.getenv("TGI_API_TOKEN", None),
).model_dump(),
)
],