diff --git a/mlte/backend/api/endpoints/catalog_entry.py b/mlte/backend/api/endpoints/catalog_entry.py index ba94e292..a03f0225 100644 --- a/mlte/backend/api/endpoints/catalog_entry.py +++ b/mlte/backend/api/endpoints/catalog_entry.py @@ -196,7 +196,11 @@ def list_catalogs( catalog_stores.sessions try: return [ - CatalogReply(id=catalog_id, read_only=catalog.read_only) + CatalogReply( + id=catalog_id, + read_only=catalog.read_only, + type=catalog.get_uri().type.value, + ) for catalog_id, catalog in catalog_stores.sessions.items() ] except Exception as e: diff --git a/mlte/backend/api/models/catalog.py b/mlte/backend/api/models/catalog.py index a30d9d20..fd9993c7 100644 --- a/mlte/backend/api/models/catalog.py +++ b/mlte/backend/api/models/catalog.py @@ -13,3 +13,6 @@ class CatalogReply(BaseModel): read_only: bool """Whether the catalog is read only or not.""" + + type: str + """Type of catalog."""