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

Merge branch 'main' into meta-ref-vision-fix-2

Browse source
This commit is contained in:
Xi Yan 2024-12-30 16:26:15 -08:00 • committed by GitHub
commit 2543b54a6a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 23 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -81,14 +81,28 @@ async def agents_stack(request, inference_model, safety_shield):
inference_models = (
inference_model if isinstance(inference_model, list) else [inference_model]
)
models = [
ModelInput(
model_id=model,
model_type=ModelType.llm,
provider_id=providers["inference"][i].provider_id,
# NOTE: meta-reference provider needs 1 provider per model, lookup provider_id from provider config
model_to_provider_id = {}
for provider in providers["inference"]:
if "model" in provider.config:
model_to_provider_id[provider.config["model"]] = provider.provider_id
models = []
for model in inference_models:
if model in model_to_provider_id:
provider_id = model_to_provider_id[model]
else:
provider_id = providers["inference"][0].provider_id
models.append(
ModelInput(
model_id=model,
model_type=ModelType.llm,
provider_id=provider_id,
)
)
for i, model in enumerate(inference_models)
]
models.append(
ModelInput(
model_id="all-MiniLM-L6-v2",