From b2f7c7780569d210a11c29ff2b6bb1c2c1fc923f Mon Sep 17 00:00:00 2001
From: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
Date: Fri, 23 Jun 2023 12:38:56 -0400
Subject: [PATCH] chore(cli): better command recommendation (#56)

---
 src/openllm/cli.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/openllm/cli.py b/src/openllm/cli.py
index 6a11e65b..751acd85 100644
--- a/src/openllm/cli.py
+++ b/src/openllm/cli.py
@@ -825,8 +825,19 @@ Available model_id(s): {llm_config['model_ids']} [default: {llm_config['default_
             raise
         else:
             if not get_debug_mode():
+                cmd_name = f"openllm build {model_name}"
+                if adapter_map is not None:
+                    cmd_name += " " + " ".join(
+                        [
+                            f"--adapter-id {s}"
+                            for s in [
+                                f"{p}:{name}" if name not in (None, "default") else p
+                                for p, name in adapter_map.items()
+                            ]
+                        ]
+                    )
                 _echo(
-                    f"\n🚀 Next step: run 'openllm build {model_name}' to create a Bento for {model_name}",
+                    f"\n🚀 Next step: run '{cmd_name}' to create a Bento for {model_name}",
                     fg="blue",
                 )