feat: show flow schema in cli command

Author: lemorage

python/cocoindex/cli.py

@@ -54,13 +54,26 @@ def ls(show_all: bool):
 @cli.command()
 @click.argument("flow_name", type=str, required=False)
 @click.option("--color/--no-color", default=True)
-def show(flow_name: str | None, color: bool):
+@click.option(
+    "--schema", is_flag=True, show_default=True, default=False,
+    help="Also show the schema of the flow.")
+def show(flow_name: str | None, color: bool, schema: bool):
     """
     Show the flow spec in a readable format with colored output.
+    Optionally display the schema if --schema is provided.
     """
-    fl = _flow_by_name(flow_name)
+    from rich.table import Table
+    flow = _flow_by_name(flow_name)
     console = Console(no_color=not color)
-    console.print(fl._render_text())
+    console.print(flow._render_text())
+    if schema:
+        table = Table(title=f"Schema for Flow: {flow.name}", show_header=True, header_style="bold magenta")
+        table.add_column("Field", style="cyan")
+        table.add_column("Type", style="green")
+        table.add_column("Attributes", style="yellow")
+        for field_name, field_type, attr_str in flow._render_schema():
+            table.add_row(field_name, field_type, attr_str)
+        console.print(table)
 
 @cli.command()
 def setup():

python/cocoindex/flow.py

@@ -503,6 +503,12 @@ def _render_text(self) -> Text:
             return self._format_flow(flow_dict)
         except json.JSONDecodeError:
             return Text(flow_spec_str)
+    
+    def _render_schema(self) -> list[tuple[str, str, str]]:
+        """
+        Render the schema as a list of (field_name, field_type, attributes) tuples.
+        """
+        return _engine.get_flow_schema(self.name)
 
     def __str__(self):
         return str(self._render_text())