Fix video description naming (#654)

camel-ai · Jun 17, 2024 · c4656c9 · c4656c9
1 parent 99a11fa
commit c4656c9
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/camel/prompts/task_prompt_template.py b/camel/prompts/task_prompt_template.py
@@ -60,6 +60,6 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
                 TaskType.SOLUTION_EXTRACTION: SolutionExtractionPromptTemplateDict(),  # noqa: E501
                 TaskType.ROLE_DESCRIPTION: RoleDescriptionPromptTemplateDict(),
                 TaskType.OBJECT_RECOGNITION: ObjectRecognitionPromptTemplateDict(),  # noqa: E501
-                TaskType.DESCRIPTE_VIDEO: DescriptionVideoPromptTemplateDict(),
+                TaskType.VIDEO_DESCRIPTION: DescriptionVideoPromptTemplateDict(),  # noqa: E501
             }
         )
diff --git a/camel/types/enums.py b/camel/types/enums.py
@@ -209,7 +209,7 @@ class TaskType(Enum):
     ROLE_DESCRIPTION = "role_description"
     OBJECT_RECOGNITION = "object_recognition"
     DEFAULT = "default"
-    DESCRIPTE_VIDEO = "descripte_video"
+    VIDEO_DESCRIPTION = "video_description"
 
 
 class VectorDistance(Enum):

diff --git a/examples/vision/descripte_video.py → examples/vision/video_description.py b/examples/vision/descripte_video.py → examples/vision/video_description.py
@@ -21,7 +21,7 @@
 
 # Define system message
 sys_msg_prompt = PromptTemplateGenerator().get_prompt_from_key(
-    TaskType.DESCRIPTE_VIDEO, RoleType.ASSISTANT
+    TaskType.VIDEO_DESCRIPTION, RoleType.ASSISTANT
 )
 sys_msg = BaseMessage.make_assistant_message(
     role_name="Assistant",