fix: [DEV-13100] add dataset_id to Workflow type (#383)

Co-authored-by: Anmol Sandhu <anmolsandhu@Mac.cable.rcn.com>

Author: arsandhu

indico/queries/gallery.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, List
 
 from indico.client.request import GraphQLRequest, PagedRequestV2
 from indico.types.component_blueprint import BlueprintPage, BlueprintTags
@@ -112,7 +112,7 @@ class GetGalleryTags(GraphQLRequest[BlueprintTags]):
     def __init__(
         self,
         component_family: "Optional[str]" = None,
-        tag_categories: "Optional[list[str]]" = None,
+        tag_categories: "Optional[List[str]]" = None,
     ):
         self.component_family = component_family
         self.tag_categories = tag_categories

indico/queries/workflow.py

@@ -76,6 +76,7 @@ class ListWorkflows(GraphQLRequest["List[Workflow]"]):
                     }
 
                   }
+                  datasetId
                 }
             }
         }

indico/types/workflow.py

@@ -45,6 +45,7 @@ class Workflow(BaseType):
     id: int
     name: str
     status: str
+    dataset_id: int
     review_enabled: bool
     auto_review_enabled: bool
     components: List[WorkflowComponent]

tests/integration/queries/test_gallery.py

@@ -1,6 +1,7 @@
 import pytest
+
 from indico.client import IndicoClient
-from indico.queries.gallery import ListGallery, GetGalleryTags
+from indico.queries.gallery import GetGalleryTags, ListGallery
 from indico.types.component_blueprint import BlueprintPage, BlueprintTags
 
 
@@ -76,9 +77,9 @@ def test_list_gallery_pagination(indico):
     if len(blueprints) >= 2:
         # Check that we have unique blueprints
         blueprint_ids = [bp.id for bp in blueprints]
-        assert len(blueprint_ids) == len(set(blueprint_ids)), (
-            "Duplicate blueprints found in pagination"
-        )
+        assert len(blueprint_ids) == len(
+            set(blueprint_ids)
+        ), "Duplicate blueprints found in pagination"
 
 
 def test_get_gallery_tags(indico):