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

fix(tests): move llama stack client init back to fixture

Browse source
This commit is contained in:
Ashwin Bharambe 2025-08-07 15:24:26 -07:00
parent 5f1ddd35e4
commit 1b076910ac
2 changed files with 7 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/integration/fixtures/common.py
View file

@ -178,8 +178,13 @@ def skip_if_no_model(request):
@pytest.fixture(scope="session")
def llama_stack_client(request):
client = request.session._llama_stack_client
assert client is not None, "llama_stack_client not found in session cache"
# ideally, we could do this in session start given all the complex logs during initialization
# don't clobber the test one-liner outputs. however, this also means all tests in a sub-directory
# would be forced to use llama_stack_client, which is not what we want.
print("\ninstantiating llama_stack_client")
start_time = time.time()
client = instantiate_llama_stack_client(request.session)
print(f"llama_stack_client instantiated in {time.time() - start_time:.3f}s")
return client