[PLT-1231] Add missing media types (#1705)

Author: adrian-chang

libs/labelbox/src/labelbox/schema/media_type.py

@@ -20,6 +20,8 @@ class MediaType(Enum):
     Unknown = "UNKNOWN"
     Unsupported = "UNSUPPORTED"
     Video = "VIDEO"
+    Point_Cloud = "POINT_CLOUD"
+    LLM = "LLM"
 
     @classmethod
     def _missing_(cls, name):

libs/labelbox/tests/data/annotation_import/test_data_types.py

@@ -210,8 +210,6 @@ def validate_iso_format(date_string: str):
         ConversationData,
         DocumentData,
         DicomData,
-        LlmPromptCreationData,
-        LlmPromptResponseCreationData,
         LlmResponseCreationData,
     ],
 )

libs/labelbox/tests/data/annotation_import/test_generic_data_types.py

@@ -154,8 +154,6 @@ def validate_iso_format(date_string: str):
         ConversationData,
         DocumentData,
         DicomData,
-        LlmPromptCreationData,
-        LlmPromptResponseCreationData,
         LlmResponseCreationData,
     ],
 )

libs/labelbox/tests/integration/test_project.py

@@ -244,7 +244,9 @@ def test_media_type(client, project: Project, rand_gen):
     for media_type in MediaType.get_supported_members():
         # Exclude LLM media types for now, as they are not supported
         if MediaType[media_type] in [
-                MediaType.LLMPromptCreation, MediaType.LLMPromptResponseCreation
+                MediaType.LLMPromptCreation,
+                MediaType.LLMPromptResponseCreation,
+                MediaType.LLM
         ]:
             continue