From ba8afa92359b2dbfadae826baa8f6e910edf7a9e Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Mon, 11 Nov 2024 23:10:12 -0800
Subject: [PATCH] print resources when booting up

---
 llama_stack/distribution/stack.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/llama_stack/distribution/stack.py b/llama_stack/distribution/stack.py
index 15bb213c5..7fe7d3ca7 100644
--- a/llama_stack/distribution/stack.py
+++ b/llama_stack/distribution/stack.py
@@ -6,6 +6,8 @@
 
 from typing import Any, Dict
 
+from termcolor import colored
+
 from llama_models.llama3.api.datatypes import *  # noqa: F403
 from llama_stack.apis.agents import *  # noqa: F403
 from llama_stack.apis.datasets import *  # noqa: F403
@@ -76,4 +78,23 @@ async def construct_stack(run_config: StackRunConfig) -> Dict[Api, Any]:
     for obj in objects:
         await dist_registry.register(obj)
 
+    resources = [
+        ("models", Api.models),
+        ("shields", Api.shields),
+        ("memory_banks", Api.memory_banks),
+        ("datasets", Api.datasets),
+        ("scoring_fns", Api.scoring_functions),
+        ("eval_tasks", Api.eval_tasks),
+    ]
+    for rsrc, api in resources:
+        if api not in impls:
+            continue
+
+        method = getattr(impls[api], f"list_{api.value}")
+        for obj in await method():
+            print(
+                f"{rsrc.capitalize()}: {colored(obj.identifier, 'white', attrs=['bold'])} served by {colored(obj.provider_id, 'white', attrs=['bold'])}",
+            )
+
+    print("")
     return impls