From c4656c9eca9333f7eb0aebc42e0e5795dae64615 Mon Sep 17 00:00:00 2001 From: Zecheng Zhang Date: Mon, 17 Jun 2024 08:04:49 -0700 Subject: [PATCH] Fix video description naming (#654) --- camel/prompts/task_prompt_template.py | 2 +- camel/types/enums.py | 2 +- examples/vision/{descripte_video.py => video_description.py} | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) rename examples/vision/{descripte_video.py => video_description.py} (98%) diff --git a/camel/prompts/task_prompt_template.py b/camel/prompts/task_prompt_template.py index b7fbd80b66..3d67daf1f8 100644 --- 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 index 4d51e890fa..e360c1a051 100644 --- 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 b/examples/vision/video_description.py similarity index 98% rename from examples/vision/descripte_video.py rename to examples/vision/video_description.py index 322d8a7577..6e1657fb22 100644 --- a/examples/vision/descripte_video.py +++ b/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",