[PTDT-1552] Add LLM media types (#1192)

Author: kkim-labelbox

labelbox/schema/media_type.py

@@ -10,14 +10,16 @@ class MediaType(Enum):
     Html = "HTML"
     Image = "IMAGE"
     Json = "JSON"
+    LLMPromptCreation = "LLM_PROMPT_CREATION"
+    LLMPromptResponseCreation = "LLM_PROMPT_RESPONSE_CREATION"
     Pdf = "PDF"
     Simple_Tile = "TMS_SIMPLE"
     Text = "TEXT"
     Tms_Geo = "TMS_GEO"
     Tms_Simple = "TMS_SIMPLE"
-    Video = "VIDEO"
     Unknown = "UNKNOWN"
     Unsupported = "UNSUPPORTED"
+    Video = "VIDEO"
 
     @classmethod
     def _missing_(cls, name):

tests/integration/test_project.py

@@ -394,6 +394,11 @@ def test_media_type(client, configured_project: Project, rand_gen):
     project.delete()
 
     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
+        ]:
+            continue
 
         project = client.create_project(name=rand_gen(str),
                                         media_type=MediaType[media_type])