diff --git a/llama_stack/distribution/stack.py b/llama_stack/distribution/stack.py index 65bb6935b..b5b87e579 100644 --- a/llama_stack/distribution/stack.py +++ b/llama_stack/distribution/stack.py @@ -186,6 +186,7 @@ async def construct_stack( run_config, provider_registry or get_provider_registry(), dist_registry ) print("impls", impls) + print("run_config", run_config) await register_resources(run_config, impls) return impls diff --git a/llama_stack/providers/tests/resolver.py b/llama_stack/providers/tests/resolver.py index 0b30c6240..2f8a7412d 100644 --- a/llama_stack/providers/tests/resolver.py +++ b/llama_stack/providers/tests/resolver.py @@ -9,6 +9,7 @@ import tempfile from typing import Any, Dict, List, Optional from pydantic import BaseModel +from termcolor import cprint from llama_stack.apis.datasets import DatasetInput from llama_stack.apis.eval_tasks import EvalTaskInput @@ -57,11 +58,8 @@ async def construct_stack_for_test( scoring_fns=scoring_fns or [], eval_tasks=eval_tasks or [], ) - from rich.pretty import pprint - - pprint(f"run_config BEFORE: {run_config}") run_config = parse_and_maybe_upgrade_config(run_config) - pprint(f"run_config AFTER: {run_config}") + cprint(f"run_config AFTER: {run_config}", "red") try: remote_config = remote_provider_config(run_config) if not remote_config: