From 1df696c1e1f570410f46f9069cda24d1ed8ff020 Mon Sep 17 00:00:00 2001 From: Charlie Doern Date: Fri, 19 Dec 2025 09:24:30 -0500 Subject: [PATCH] fix: post training algorithm param MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit was getting: ``` ╰─ llama-stack-clien--t version Traceback (most recent call last): File "/Users/charliedoern/projects/Documents/llama-stack/venv/bin/llama-stack-client", line 4, in from llama_stack_client.lib.cli.llama_stack_client import main File "/Users/charliedoern/projects/Documents/llama-stack-client-python/src/llama_stack_client/lib/cli/llama_stack_client.py", line 23, in from .post_training import post_training File "/Users/charliedoern/projects/Documents/llama-stack-client-python/src/llama_stack_client/lib/cli/post_training/__init__.py", line 7, in from .post_training import post_training File "/Users/charliedoern/projects/Documents/llama-stack-client-python/src/llama_stack_client/lib/cli/post_training/post_training.py", line 12, in from llama_stack_client.types.alpha.post_training_supervised_fine_tune_params import ( ImportError: cannot import name 'AlgorithmConfigParam' from 'llama_stack_client.types.alpha.post_training_supervised_fine_tune_params' (/Users/charliedoern/projects/Documents/llama-stack-client-python/src/llama_stack_client/types/alpha/post_training_supervised_fine_tune_params.py). Did you mean: 'AlgorithmConfig'? ``` fixes that. Signed-off-by: Charlie Doern --- src/llama_stack_client/lib/cli/post_training/post_training.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/llama_stack_client/lib/cli/post_training/post_training.py b/src/llama_stack_client/lib/cli/post_training/post_training.py index 3a890b80..2d766652 100644 --- a/src/llama_stack_client/lib/cli/post_training/post_training.py +++ b/src/llama_stack_client/lib/cli/post_training/post_training.py @@ -10,7 +10,7 @@ from rich.console import Console from llama_stack_client.types.alpha.post_training_supervised_fine_tune_params import ( - AlgorithmConfigParam, + AlgorithmConfig, TrainingConfig, ) @@ -36,7 +36,7 @@ def supervised_fine_tune( ctx, job_uuid: str, model: str, - algorithm_config: AlgorithmConfigParam, + algorithm_config: AlgorithmConfig, training_config: TrainingConfig, checkpoint_dir: Optional[str], ):