phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added a "--raw" option for model template printing

Browse source
This commit is contained in:
Ashwin Bharambe 2024-09-03 22:04:43 -07:00
parent 85d56ed3f2
commit 86059af5af
1 changed files with 23 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

33
llama_toolchain/cli/model/template.py
View file

@ -59,10 +59,15 @@ class ModelTemplate(Subcommand):
self.parser.add_argument(
"--format",
type=str,
help="ToolPromptFormat ( json or functino_tag). This flag is used to print the template in a specific formats.",
help="ToolPromptFormat (json or function_tag). This flag is used to print the template in a specific formats.",
required=False,
default="json",
)
self.parser.add_argument(
"--raw",
action="store_true",
help="If set to true, don't pretty-print into a table. Useful to copy-paste.",
)
def _run_model_template_cmd(self, args: argparse.Namespace) -> None:
from llama_models.llama3.api.interface import (
@ -82,15 +87,23 @@ class ModelTemplate(Subcommand):
else:
rendered += tok
rendered = rendered.replace("\n", "↵\n")
print_table(
[
("Name", colored(template.template_name, "white", attrs=["bold"])),
("Template", rendered),
("Notes", template.notes),
],
separate_rows=True,
)
if not args.raw:
rendered = rendered.replace("\n", "↵\n")
print_table(
[
(
"Name",
colored(template.template_name, "white", attrs=["bold"]),
),
("Template", rendered),
("Notes", template.notes),
],
separate_rows=True,
)
else:
print("Template: ", template.template_name)
print("=" * 40)
print(rendered)
else:
templates = list_jinja_templates()
headers = ["Role", "Template Name"]