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

debug

Browse source
This commit is contained in:
Xi Yan 2024-12-30 14:45:53 -08:00
parent 07806f85c7
commit 52776ca897
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/tests/agents/fixtures.py
View file

@ -67,6 +67,7 @@ async def agents_stack(request, inference_model, safety_shield):
for key in ["inference", "safety", "memory", "agents"]: for key in ["inference", "safety", "memory", "agents"]:
fixture = request.getfixturevalue(f"{key}_{fixture_dict[key]}") fixture = request.getfixturevalue(f"{key}_{fixture_dict[key]}")
providers[key] = fixture.providers providers[key] = fixture.providers
print(f"fixture {key} providers: {providers[key]}")
if key == "inference": if key == "inference":
providers[key].append( providers[key].append(
Provider( Provider(
@ -81,6 +82,9 @@ async def agents_stack(request, inference_model, safety_shield):
inference_models = ( inference_models = (
inference_model if isinstance(inference_model, list) else [inference_model] inference_model if isinstance(inference_model, list) else [inference_model]
) )
print(providers)
print(inference_models, safety_shield)
models = [ models = [
ModelInput( ModelInput(
model_id=model, model_id=model,