Add model list subcommand

llama_toolchain/cli/model/template.py

@@ -5,21 +5,16 @@
 # the root directory of this source tree.
 
 import argparse
-import re
 import textwrap
 
 from llama_models.llama3_1.api.interface import (
     list_jinja_templates,
     render_jinja_template,
 )
-from termcolor import colored, cprint
+from termcolor import colored
 
 from llama_toolchain.cli.subcommand import Subcommand
-
-
-def strip_ansi_colors(text):
-    ansi_escape = re.compile(r"\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])")
-    return ansi_escape.sub("", text)
+from llama_toolchain.cli.table import print_table
 
 
 class ModelTemplate(Subcommand):
@@ -82,61 +77,3 @@ class ModelTemplate(Subcommand):
                 [(t.role, t.template_name) for t in templates],
                 headers,
             )
-
-
-def format_row(row, col_widths):
-    def wrap(text, width):
-        lines = []
-        for line in text.split("\n"):
-            if line.strip() == "":
-                lines.append("")
-            else:
-                line = line.strip()
-                lines.extend(
-                    textwrap.wrap(
-                        line, width, break_long_words=False, replace_whitespace=False
-                    )
-                )
-        return lines
-
-    wrapped = [wrap(item, width) for item, width in zip(row, col_widths)]
-    max_lines = max(len(subrow) for subrow in wrapped)
-
-    lines = []
-    for i in range(max_lines):
-        line = []
-        for cell_lines, width in zip(wrapped, col_widths):
-            value = cell_lines[i] if i < len(cell_lines) else ""
-            line.append(value + " " * (width - len(strip_ansi_colors(value))))
-        lines.append("| " + (" | ".join(line)) + " |")
-
-    return "\n".join(lines)
-
-
-def print_table(rows, headers=None, separate_rows: bool = False):
-    if not headers:
-        col_widths = [
-            max(len(strip_ansi_colors(item)) for item in col) for col in zip(*rows)
-        ]
-    else:
-        col_widths = [
-            max(len(header), max(len(strip_ansi_colors(item)) for item in col))
-            for header, col in zip(headers, zip(*rows))
-        ]
-    col_widths = [min(w, 80) for w in col_widths]
-
-    header_line = "+".join("-" * (width + 2) for width in col_widths)
-    header_line = f"+{header_line}+"
-
-    if headers:
-        print(header_line)
-        cprint(format_row(headers, col_widths), "white", attrs=["bold"])
-
-    print(header_line)
-    for row in rows:
-        print(format_row(row, col_widths))
-        if separate_rows:
-            print(header_line)
-
-    if not separate_rows:
-        print(header_line)