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

Fix LibraryClient completely correctly; also make tests pass

Browse source
This commit is contained in:
Ashwin Bharambe 2024-12-16 22:16:21 -08:00
parent d4935ca439
commit 1bcc26ccd1
7 changed files with 201 additions and 100 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/client-sdk/conftest.py
View file

@ -3,13 +3,22 @@
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
import os
import pytest
from llama_stack import LlamaStackAsLibraryClient
from llama_stack.providers.tests.env import get_env_or_fail
from llama_stack_client import LlamaStackClient
@pytest.fixture
@pytest.fixture(scope="session")
def llama_stack_client():
"""Fixture to create a fresh LlamaStackClient instance for each test"""
return LlamaStackClient(base_url=get_env_or_fail("LLAMA_STACK_BASE_URL"))
if os.environ.get("LLAMA_STACK_CONFIG"):
client = LlamaStackAsLibraryClient(get_env_or_fail("LLAMA_STACK_CONFIG"))
client.initialize()
elif os.environ.get("LLAMA_STACK_BASE_URL"):
client = LlamaStackClient(base_url=get_env_or_fail("LLAMA_STACK_BASE_URL"))
else:
raise ValueError("LLAMA_STACK_CONFIG or LLAMA_STACK_BASE_URL must be set")
return client