From 4bbb4ddeaed83d94dddeb27810690a7ffb2e3230 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?S=C3=A9bastien=20Han?= Date: Thu, 6 Mar 2025 20:27:47 +0100 Subject: [PATCH] fix: resolve pydantic warning on .dict() usage (#1445) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit # What does this PR do? The method "dict" in class "BaseModel" is deprecated we should use model_dump instead. Signed-off-by: Sébastien Han --- llama_stack/cli/model/describe.py | 2 +- llama_stack/distribution/configure.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/llama_stack/cli/model/describe.py b/llama_stack/cli/model/describe.py index 593fb9715..f347bdf8d 100644 --- a/llama_stack/cli/model/describe.py +++ b/llama_stack/cli/model/describe.py @@ -64,7 +64,7 @@ class ModelDescribe(Subcommand): ] if model.recommended_sampling_params is not None: - sampling_params = model.recommended_sampling_params.dict() + sampling_params = model.recommended_sampling_params.model_dump() for k in ("max_tokens", "repetition_penalty"): del sampling_params[k] rows.append( diff --git a/llama_stack/distribution/configure.py b/llama_stack/distribution/configure.py index 825846a23..715bb5db4 100644 --- a/llama_stack/distribution/configure.py +++ b/llama_stack/distribution/configure.py @@ -39,7 +39,7 @@ def configure_single_provider(registry: Dict[str, ProviderSpec], provider: Provi return Provider( provider_id=provider.provider_id, provider_type=provider.provider_type, - config=cfg.dict(), + config=cfg.model_dump(), )