feat: support new item types

.changes/unreleased/new-items-20260128-120006.yaml

@@ -0,0 +1,6 @@
+kind: new-items
+body: Add support of CosmosDBDatabase item type in mkdir, get, set, rm, ls, cp, mv, import and export commands
+time: 2026-01-28T12:00:06.840541+02:00
+custom:
+    Author: v-alexmoraru
+    AuthorLink: https://github.com/v-alexmoraru

.changes/unreleased/new-items-20260129-143011.yaml

@@ -0,0 +1,6 @@
+kind: new-items
+body: Add support of UserDataFunction item type in mkdir, get, set, rm, cp, mv, import and export commands
+time: 2026-01-29T14:30:11.9516836+02:00
+custom:
+    Author: v-alexmoraru
+    AuthorLink: https://github.com/v-alexmoraru

.changes/unreleased/new-items-20260129-143030.yaml

@@ -0,0 +1,6 @@
+kind: new-items
+body: Add support of GraphQuerySet item type in mkdir, get, rm and export commands
+time: 2026-01-29T14:30:30.9936987+02:00
+custom:
+    Author: v-alexmoraru
+    AuthorLink: https://github.com/v-alexmoraru

src/fabric_cli/core/fab_config/command_support.yaml

@@ -112,6 +112,7 @@ commands:
           - connection
           - gateway
           - sql_database
+          - cosmos_db_database
           - graph_ql_api
       rm:
         supported_elements: 
@@ -142,7 +143,6 @@ commands:
           - retail_data_manager
           - healthcare_data_solution
           - sustainability_data_solution
-          - user_data_function
           - ai_skill
       mv:
         supported_elements:
@@ -160,6 +160,7 @@ commands:
           # - eventhouse
           # - kql_database
           - mirrored_database
+          - cosmos_db_database
           - reflex
           # - eventstream
           - mounted_data_factory
@@ -169,6 +170,7 @@ commands:
           #- lakehouse # this is included just to support underlying /Files
           - dataflow
           - sql_database
+          - user_data_function
       cp:
         supported_elements:
           - workspace
@@ -186,6 +188,7 @@ commands:
           # - eventhouse
           # - kql_database
           - mirrored_database
+          - cosmos_db_database
           - reflex
           # - eventstream
           - mounted_data_factory
@@ -195,6 +198,7 @@ commands:
           #- lakehouse # this is included just to support underlying /Files
           - dataflow
           - sql_database
+          - user_data_function
       ln:
         supported_elements:
           - onelake
@@ -246,6 +250,9 @@ commands:
           - graph_ql_api
           - dataflow
           - sql_database
+          - cosmos_db_database
+          - user_data_function
+          - graph_query_set
       import:
         supported_items:
           - report
@@ -267,6 +274,10 @@ commands:
           - graph_ql_api
           - dataflow
           - sql_database
+          - cosmos_db_database
+          - user_data_function
+        unsupported_items:
+          - graph_query_set
       get:
         supported_elements:
           - workspace
@@ -281,6 +292,9 @@ commands:
           - connection
           - gateway
           - external_data_share
+          - cosmos_db_database
+          - user_data_function
+          - graph_query_set
         unsupported_items:
           - dashboard
           - paginated_report
@@ -294,7 +308,6 @@ commands:
           - retail_data_manager
           - healthcare_data_solution
           - sustainability_data_solution
-          - user_data_function
           - ai_skill
       set:
         supported_elements:
@@ -321,8 +334,8 @@ commands:
           - retail_data_manager
           - healthcare_data_solution
           - sustainability_data_solution
-          - user_data_function
           - ai_skill
+          - graph_query_set
       open:
         supported_elements: 
           - workspace
@@ -355,7 +368,6 @@ commands:
           - retail_data_manager
           - healthcare_data_solution
           - sustainability_data_solution
-          - user_data_function
           - ai_skill
       start:
         supported_items:

src/fabric_cli/core/fab_types.py

@@ -140,7 +140,8 @@ def from_string(cls, vws_type_str):
 class _BaseItemType(Enum):
     @classmethod
     def from_string(cls, item_type_str):
-        raise NotImplementedError("This method must be implemented in the subclass")
+        raise NotImplementedError(
+            "This method must be implemented in the subclass")
 
 
 ##################################
@@ -199,7 +200,8 @@ def from_string(cls, vws_type_str):
             if item.value.lower() == vws_type_str.lower():
                 return item
         raise FabricCLIError(
-            ErrorMessages.Common.invalid_virtual_item_container_type(vws_type_str),
+            ErrorMessages.Common.invalid_virtual_item_container_type(
+                vws_type_str),
             fab_constant.ERROR_INVALID_ITEM_TYPE,
         )
 
@@ -253,10 +255,10 @@ class ItemType(_BaseItemType):
     METRIC_SET = "MetricSet"
     ORG_APP = "OrgApp"
     SUSTAINABILITY_DATA_SOLUTION = "SustainabilityDataSolution"
-    USER_DATA_FUNCTION = "UserDataFunction"
     VARIABLE_LIBRARY = "VariableLibrary"
 
     # API
+    COSMOS_DB_DATABASE = "CosmosDBDatabase"
     DASHBOARD = "Dashboard"
     DATAMART = "Datamart"
     DATA_PIPELINE = "DataPipeline"
@@ -281,6 +283,8 @@ class ItemType(_BaseItemType):
     WAREHOUSE = "Warehouse"
     COPYJOB = "CopyJob"
     GRAPHQLAPI = "GraphQLApi"
+    GRAPH_QUERY_SET = "GraphQuerySet"
+    USER_DATA_FUNCTION = "UserDataFunction"
     MOUNTED_DATA_FACTORY = "MountedDataFactory"
     SQL_DATABASE = "SQLDatabase"
     DATAFLOW = "Dataflow"
@@ -482,9 +486,9 @@ class MirroredDatabaseFolders(Enum):
     ItemType.SUSTAINABILITY_DATA_SOLUTION: "sustainabilitydatasolutions",
     ItemType.METRIC_SET: "metricsets",
     ItemType.ORG_APP: "orgapps",
-    ItemType.USER_DATA_FUNCTION: "userdatafunctions",
     ItemType.VARIABLE_LIBRARY: "variablelibraries",
     # API
+    ItemType.COSMOS_DB_DATABASE: "cosmosDbDatabases",
     ItemType.DASHBOARD: "dashboards",
     ItemType.DATA_PIPELINE: "dataPipelines",
     ItemType.DATAMART: "datamarts",
@@ -510,6 +514,8 @@ class MirroredDatabaseFolders(Enum):
     ItemType.WAREHOUSE: "warehouses",
     ItemType.COPYJOB: "copyJobs",
     ItemType.GRAPHQLAPI: "graphqlapis",
+    ItemType.GRAPH_QUERY_SET: "GraphQuerySets",
+    ItemType.USER_DATA_FUNCTION: "userdatafunctions",
     ItemType.MOUNTED_DATA_FACTORY: "mounteddatafactories",
     ItemType.DATAFLOW: "dataflows",
 }
@@ -526,9 +532,9 @@ class MirroredDatabaseFolders(Enum):
     ItemType.SUSTAINABILITY_DATA_SOLUTION: "sustainability-data-manager",
     ItemType.METRIC_SET: "metricsets",
     ItemType.ORG_APP: "orgapps",
-    ItemType.USER_DATA_FUNCTION: "userdatafunctions",
     ItemType.VARIABLE_LIBRARY: "variable-libraries",
     # API
+    ItemType.COSMOS_DB_DATABASE: "cosmosdbdatabases",
     ItemType.DASHBOARD: "dashboards",
     ItemType.DATAMART: "datamarts",
     ItemType.DATA_PIPELINE: "pipelines",
@@ -552,6 +558,8 @@ class MirroredDatabaseFolders(Enum):
     ItemType.SQL_ENDPOINT: "lakewarehouses",
     ItemType.WAREHOUSE: "datawarehouses",
     ItemType.COPYJOB: "copyjobs",
+    ItemType.GRAPH_QUERY_SET: "graph-queryset",
+    ItemType.USER_DATA_FUNCTION: "userdatafunctions",
     ItemType.GRAPHQLAPI: "graphql",
     ItemType.MOUNTED_DATA_FACTORY: "mounteddatafactories",
     ItemType.DATAFLOW: "dataflows-gen2",
@@ -566,4 +574,7 @@ class MirroredDatabaseFolders(Enum):
         ".py": "?format=fabricGitSource",
         ".ipynb": "?format=ipynb",
     },
+    ItemType.COSMOS_DB_DATABASE: {"default": ""},
+    ItemType.USER_DATA_FUNCTION: {"default": ""},
+    ItemType.GRAPH_QUERY_SET: {"default": ""},
 }

src/fabric_cli/core/hiearchy/fab_item.py

@@ -39,7 +39,8 @@ def item_type(self) -> ItemType:
             return _item_type
         else:
             raise FabricCLIError(
-                ErrorMessages.Hierarchy.item_type_not_valid(str(super().item_type)),
+                ErrorMessages.Hierarchy.item_type_not_valid(
+                    str(super().item_type)),
                 fab_constant.ERROR_INVALID_ITEM_TYPE,
             )
 
@@ -119,6 +120,9 @@ def get_payload(self, definition, input_format=None) -> dict:
                 | ItemType.GRAPHQLAPI
                 | ItemType.DATAFLOW
                 | ItemType.SQL_DATABASE
+                | ItemType.COSMOS_DB_DATABASE
+                | ItemType.GRAPH_QUERY_SET
+                | ItemType.USER_DATA_FUNCTION
             ):
                 return {
                     "type": str(self.item_type),