Add integration with searchProject

Author: Val Brodsky

libs/labelbox/src/labelbox/client.py

@@ -2410,7 +2410,7 @@ def upsert_label_feedback(self, label_id: str, feedback: str,
 
     def get_labeling_service_dashboards(
         self,
-        from_cursor: Optional[str] = None,
+        after: Optional[str] = None,
         where: Optional[Comparison] = None,
     ) -> PaginatedCollection:
-        return LabelingServiceDashboard.get_all(self, from_cursor, where)
+        return LabelingServiceDashboard.get_all(self, after, where)

libs/labelbox/src/labelbox/schema/labeling_service_dashboard.py

@@ -1,14 +1,31 @@
 from datetime import datetime
 from string import Template
-from typing import Any, Optional
+from typing import Any, Dict, List, Optional
 
 from labelbox.exceptions import ResourceNotFoundError
 from labelbox.orm.comparison import Comparison
+from labelbox.orm import query
+from ..orm.model import Field
 from labelbox.pagination import PaginatedCollection
 from labelbox.pydantic_compat import BaseModel, root_validator
+from .organization import Organization
 from labelbox.utils import _CamelCaseMixin
 from labelbox.schema.labeling_service_status import LabelingServiceStatus
 
+GRAPHQL_QUERY_SELECTIONS = """
+                id
+                name
+                # serviceType
+                # createdAt
+                # updatedAt
+                # createdById
+                boostStatus
+                dataRowsCount
+                dataRowsInReviewCount
+                dataRowsInReworkCount
+                dataRowsDoneCount
+            """
+
 
 class LabelingServiceDashboard(BaseModel):
     id: str
@@ -69,40 +86,41 @@ def get(cls, client, project_id: str) -> 'LabelingServiceDashboard':
     def get_all(
         cls,
         client,
-        from_cursor: Optional[str] = None,
-        where: Optional[Comparison] = None,
+        after: Optional[str] = None,
+        # where: Optional[Comparison] = None,
+        search_query: Optional[List[Dict]] = None,
     ) -> PaginatedCollection:
-        page_size = 500  # hardcode to avoid overloading the server
-        # where_param = query.where_as_dict(Entity.DataRow,
-        #                                   where) if where is not None else None
-
         template = Template(
-            """query SearchProjectsPyApi($$id: ID!, $$after: ID, $$first: Int, $$where: SearchProjectsInput)  {
-                        searchProjects(id: $$id, after: $$after, first: $$first, where: $$where)
+            """query SearchProjectsPyApi($$first: Int, $$from: String) {
+                        searchProjects(input: {after: $$from, searchQuery: $search_query, size: $$first})
                             {
-                                nodes { $datarow_selections }
-                                pageInfo { hasNextPage startCursor }
+                                nodes { $labeling_dashboard_selections }
+                                pageInfo { endCursor }
                             }
                         }
                     """)
+        organization_id = client.get_organization().uid
         query_str = template.substitute(
-            datarow_selections=LabelingServiceDashboard.schema()
-            ['properties'].keys())
+            labeling_dashboard_selections=GRAPHQL_QUERY_SELECTIONS,
+            search_query=
+            f"[{{type: \"organization\", operator: \"is\",  values: [\"{organization_id}\"]}}]"
+        )
 
         params = {
-            'id': self.uid,
-            'from': from_cursor,
-            'first': page_size,
-            'where': where_param,
+            'from': after,
         }
 
+        def convert_to_labeling_service_dashboard(client, data):
+            data['client'] = client
+            return LabelingServiceDashboard(**data)
+
         return PaginatedCollection(
             client=client,
             query=query_str,
             params=params,
             dereferencing=['searchProjects', 'nodes'],
-            obj_class=LabelingServiceDashboard,
-            cursor_path=['datasetDataRows', 'pageInfo', 'endCursor'],
+            obj_class=convert_to_labeling_service_dashboard,
+            cursor_path=['searchProjects', 'pageInfo', 'endCursor'],
         )
 
     @root_validator(pre=True)

libs/labelbox/tests/integration/test_labeling_dashboard.py

@@ -0,0 +1,24 @@
+from labelbox.schema.labeling_service import LabelingServiceStatus
+
+
+def test_request_labeling_service_moe_offline_project(
+        rand_gen, offline_chat_evaluation_project, chat_evaluation_ontology,
+        offline_conversational_data_row):
+    project = offline_chat_evaluation_project
+    project.connect_ontology(chat_evaluation_ontology)
+
+    project.create_batch(
+        rand_gen(str),
+        [offline_conversational_data_row.uid],  # sample of data row objects
+    )
+    labeling_service_dashboard = project.labeling_service_dashboard()
+    assert labeling_service_dashboard.status == LabelingServiceStatus.Missing
+    assert labeling_service_dashboard.tasks_completed == 0
+    assert labeling_service_dashboard.tasks_remaining == 0
+
+    labeling_service_dashboard = [
+        ld for ld in project.client.get_labeling_service_dashboards()
+    ][0]
+    assert labeling_service_dashboard.status == LabelingServiceStatus.Missing
+    assert labeling_service_dashboard.tasks_completed == 0
+    assert labeling_service_dashboard.tasks_remaining == 0