Add upload_type to Project

Author: Val Brodsky

libs/labelbox/src/labelbox/schema/ontology_kind.py

@@ -34,16 +34,16 @@ def is_supported(cls, value):
         return isinstance(value, cls)
 
     @classmethod
-    def _missing_(cls, name) -> 'EditorTaskType':
+    def _missing_(cls, value) -> 'EditorTaskType':
         """Handle missing null new task types
             Handle upper case names for compatibility with
             the GraphQL"""
 
-        if name is None:
+        if value is None:
             return cls.Missing
 
         for name, member in cls.__members__.items():
-            if name == name.upper():
+            if name == str(value).upper():
                 return member
 
         return cls.Missing
@@ -71,3 +71,24 @@ def map_to_editor_task_type(onotology_kind: OntologyKind,
             return EditorTaskType.ModelChatEvaluation
         else:
             return EditorTaskType.Missing
+
+
+class UploadType(Enum):
+    Auto = 'AUTO',
+    Manual = 'MANUAL',
+    Missing = None
+
+    @classmethod
+    def is_supported(cls, value):
+        return isinstance(value, cls)
+
+    @classmethod
+    def _missing_(cls, value: object) -> 'UploadType':
+        if value is None:
+            return cls.Missing
+
+        for name, member in cls.__members__.items():
+            if name == str(value).upper():
+                return member
+
+        return cls.Missing

libs/labelbox/src/labelbox/schema/project.py

@@ -37,7 +37,8 @@
 from labelbox.schema.resource_tag import ResourceTag
 from labelbox.schema.task import Task
 from labelbox.schema.task_queue import TaskQueue
-from labelbox.schema.ontology_kind import (EditorTaskType, OntologyKind)
+from labelbox.schema.ontology_kind import (EditorTaskType, OntologyKind,
+                                           UploadType)
 from labelbox.schema.project_overview import ProjectOverview, ProjectOverviewDetailed
 
 if TYPE_CHECKING:
@@ -121,6 +122,7 @@ class Project(DbObject, Updateable, Deletable):
     editor_task_type = Field.Enum(EditorTaskType, "editor_task_type")
     data_row_count = Field.Int("data_row_count")
     model_setup_complete: Field = Field.Boolean("model_setup_complete")
+    upload_type: Field = Field.Enum(UploadType, "upload_type")
 
     # Relationships
     created_by = Relationship.ToOne("User", False, "created_by")
@@ -145,8 +147,7 @@ def is_chat_evaluation(self) -> bool:
         return self.media_type == MediaType.Conversational and self.editor_task_type == EditorTaskType.ModelChatEvaluation
 
     def is_auto_data_generation(self) -> bool:
-        return self.media_type == MediaType.LLMPromptCreation or self.media_type == MediaType.LLMPromptResponseCreation or self.is_chat_evaluation(
-        )
+        return (self.upload_type == UploadType.Auto) is True
 
     def project_model_configs(self):
         query_str = """query ProjectModelConfigsPyApi($id: ID!) {

libs/labelbox/tests/unit/test_project.py

@@ -32,6 +32,7 @@ def test_project_editor_task_type(api_editor_task_type,
             "queueMode": "BATCH",
             "setupComplete": "2021-06-01T00:00:00.000Z",
             "modelSetupComplete": None,
+            "uploadType": "Auto",
         })
 
     assert project.editor_task_type == expected_editor_task_type