Backend: added storage type to catalog info returned by catalogs endp…

…oint
mlte-team · Sep 19, 2024 · 790e48f · 790e48f
1 parent 9cc1a17
commit 790e48f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git 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
@@ -13,3 +13,6 @@ class CatalogReply(BaseModel):
 
     read_only: bool
     """Whether the catalog is read only or not."""
+
+    type: str
+    """Type of catalog."""