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

default text model

Browse source
This commit is contained in:
Xi Yan 2025-03-05 16:24:43 -08:00
parent 5d43b9157e
commit 54abeeebce
1 changed files with 19 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tests/integration/conftest.py
View file

@ -86,7 +86,7 @@ def pytest_addoption(parser):
)
parser.addoption(
"--judge-model",
default=None,
default=TEXT_MODEL,
help="Specify the judge model to use for testing",
)
parser.addoption(
@ -230,10 +230,16 @@ def llama_stack_client_with_mocked_inference(llama_stack_client, request):
# Replace the methods with recordable mocks
inference_mock.chat_completion = RecordableMock(
original_inference.chat_completion, cache_dir, "chat_completion", record=record_responses
original_inference.chat_completion,
cache_dir,
"chat_completion",
record=record_responses,
)
inference_mock.completion = RecordableMock(
original_inference.completion, cache_dir, "text_completion", record=record_responses
original_inference.completion,
cache_dir,
"text_completion",
record=record_responses,
)
inference_mock.embeddings = RecordableMock(
original_inference.embeddings, cache_dir, "embeddings", record=record_responses
@ -247,7 +253,10 @@ def llama_stack_client_with_mocked_inference(llama_stack_client, request):
# Replace the methods with recordable mocks
tool_runtime_mock.invoke_tool = RecordableMock(
original_tool_runtime_api.invoke_tool, cache_dir, "invoke_tool", record=record_responses
original_tool_runtime_api.invoke_tool,
cache_dir,
"invoke_tool",
record=record_responses,
)
agents_impl.tool_runtime_api = tool_runtime_mock
@ -267,7 +276,12 @@ def inference_provider_type(llama_stack_client):
@pytest.fixture(scope="session")
def client_with_models(
llama_stack_client, text_model_id, vision_model_id, embedding_model_id, embedding_dimension, judge_model_id
llama_stack_client,
text_model_id,
vision_model_id,
embedding_model_id,
embedding_dimension,
judge_model_id,
):
client = llama_stack_client