phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 19:04:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Allow simpler initialization of RemoteProviderConfig; fix issue in httpx client

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-06 19:18:58 -08:00
parent 064d2a5287
commit 489f74a70b
3 changed files with 28 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

12
llama_stack/providers/tests/conftest.py
View file

@ -25,15 +25,19 @@ class ProviderFixture(BaseModel):
def remote_stack_fixture() -> ProviderFixture:
if url := os.getenv("REMOTE_STACK_URL", None):
config = RemoteProviderConfig.from_url(url)
else:
config = RemoteProviderConfig(
host=get_env_or_fail("REMOTE_STACK_HOST"),
port=int(get_env_or_fail("REMOTE_STACK_PORT")),
)
return ProviderFixture(
providers=[
Provider(
provider_id="remote",
provider_type="remote",
config=RemoteProviderConfig(
host=get_env_or_fail("REMOTE_STACK_HOST"),
port=int(get_env_or_fail("REMOTE_STACK_PORT")),
).model_dump(),
config=config.model_dump(),
)
],
)