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

add tests

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-13 21:36:24 -08:00
parent 89342d352c
commit 43af05d851
1 changed files with 19 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

19
llama_stack/providers/tests/inference/test_model_registration.py
View file

@ -33,3 +33,22 @@ class TestModelRegistration:
await models_impl.register_model( await models_impl.register_model(
model_id="Llama3-NonExistent-Model", model_id="Llama3-NonExistent-Model",
) )
@pytest.mark.asyncio
async def test_update_model(self, inference_stack):
_, models_impl = inference_stack
# Register a model to update
model_id = "Llama3.1-8B-Instruct"
await models_impl.register_model(model_id=model_id)
# Update the model
new_provider_id = "updated_provider"
await models_impl.update_model(model_id=model_id, provider_id=new_provider_id)
# Retrieve the updated model to verify changes
updated_model = await models_impl.get_model(model_id)
assert updated_model.provider_id == new_provider_id
# Cleanup
await models_impl.delete_model(model_id=model_id)