fix provider

This commit is contained in:
Xi Yan 2024-12-30 14:12:22 -08:00
parent 51457b0750
commit 57340e7a51

View file

@ -86,9 +86,9 @@ async def agents_stack(request, inference_model, safety_shield):
ModelInput( ModelInput(
model_id=model, model_id=model,
model_type=ModelType.llm, model_type=ModelType.llm,
provider_id=providers["inference"][0].provider_id, provider_id=providers["inference"][i].provider_id,
) )
for model in inference_models for i, model in enumerate(inference_models)
] ]
models.append( models.append(
ModelInput( ModelInput(