camel-ai 0.1.5__py3-none-any.whl → 0.1.5.2__py3-none-any.whl

camel/prompts/base.py

@@ -208,7 +208,8 @@ class TextPromptDict(Dict[Any, TextPrompt]):
     EMBODIMENT_PROMPT = TextPrompt(
         "System information :"
         + "\n".join(
-            f"{key}: {value}" for key, value in get_system_information().items()
+            f"{key}: {value}"
+            for key, value in get_system_information().items()
         )
         + "\n"
         + """You are the physical embodiment of the {role} who is working on solving a task: {task}.

camel/prompts/descripte_video_prompt.py

@@ -0,0 +1,33 @@
+# =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. ===========
+# Licensed under the Apache License, Version 2.0 (the “License”);
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an “AS IS” BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. ===========
+from typing import Any
+
+from camel.prompts.base import TextPrompt, TextPromptDict
+from camel.types import RoleType
+
+
+# flake8: noqa :E501
+class DescriptionVideoPromptTemplateDict(TextPromptDict):
+    ASSISTANT_PROMPT = TextPrompt(
+        """You are a master of video analysis. 
+        Please provide a shot description of the content of the current video."""
+    )
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+        self.update(
+            {
+                RoleType.ASSISTANT: self.ASSISTANT_PROMPT,
+            }
+        )

camel/prompts/task_prompt_template.py

@@ -18,11 +18,16 @@ from camel.prompts.ai_society import (
     TextPromptDict,
 )
 from camel.prompts.code import CodePromptTemplateDict
+from camel.prompts.descripte_video_prompt import (
+    DescriptionVideoPromptTemplateDict,
+)
 from camel.prompts.evaluation import (
     EvaluationPromptTemplateDict,
 )
 from camel.prompts.misalignment import MisalignmentPromptTemplateDict
-from camel.prompts.object_recognition import ObjectRecognitionPromptTemplateDict
+from camel.prompts.object_recognition import (
+    ObjectRecognitionPromptTemplateDict,
+)
 from camel.prompts.role_description_prompt_template import (
     RoleDescriptionPromptTemplateDict,
 )
@@ -52,8 +57,9 @@ class TaskPromptTemplateDict(Dict[Any, TextPromptDict]):
                 TaskType.MISALIGNMENT: MisalignmentPromptTemplateDict(),
                 TaskType.TRANSLATION: TranslationPromptTemplateDict(),
                 TaskType.EVALUATION: EvaluationPromptTemplateDict(),
-                TaskType.SOLUTION_EXTRACTION: SolutionExtractionPromptTemplateDict(),
+                TaskType.SOLUTION_EXTRACTION: SolutionExtractionPromptTemplateDict(),  # noqa: E501
                 TaskType.ROLE_DESCRIPTION: RoleDescriptionPromptTemplateDict(),
-                TaskType.OBJECT_RECOGNITION: ObjectRecognitionPromptTemplateDict(),
+                TaskType.OBJECT_RECOGNITION: ObjectRecognitionPromptTemplateDict(),  # noqa: E501
+                TaskType.VIDEO_DESCRIPTION: DescriptionVideoPromptTemplateDict(),  # noqa: E501
             }
         )

camel/retrievers/auto_retriever.py

@@ -42,7 +42,7 @@ class AutoRetriever:
         vector_storage_local_path (Optional[str]): Local path for vector
             storage, if applicable.
         storage_type (Optional[StorageType]): The type of vector storage to
-            use. Defaults to `StorageType.MILVUS`.
+            use. Defaults to `StorageType.QDRANT`.
         embedding_model (Optional[BaseEmbedding]): Model used for embedding
             queries and documents. Defaults to `OpenAIEmbedding()`.
     """
@@ -54,7 +54,7 @@ class AutoRetriever:
         storage_type: Optional[StorageType] = None,
         embedding_model: Optional[BaseEmbedding] = None,
     ):
-        self.storage_type = storage_type or StorageType.MILVUS
+        self.storage_type = storage_type or StorageType.QDRANT
         self.embedding_model = embedding_model or OpenAIEmbedding()
         self.vector_storage_local_path = vector_storage_local_path
         self.url_and_api_key = url_and_api_key
@@ -101,8 +101,8 @@ class AutoRetriever:
         r"""Generates a valid collection name from a given file path or URL.
 
         Args:
-        content_input_path: str. The input URL or file path from which to
-            generate the collection name.
+            content_input_path: str. The input URL or file path from which to
+                generate the collection name.
 
         Returns:
             str: A sanitized, valid collection name suitable for use.
@@ -134,7 +134,9 @@ class AutoRetriever:
         collection_name = collection_name[:30]
         return collection_name
 
-    def _get_file_modified_date_from_file(self, content_input_path: str) -> str:
+    def _get_file_modified_date_from_file(
+        self, content_input_path: str
+    ) -> str:
         r"""Retrieves the last modified date and time of a given file. This
         function takes a file path as input and returns the last modified date
         and time of that file.
@@ -147,9 +149,9 @@ class AutoRetriever:
             str: The last modified time from file.
         """
         mod_time = os.path.getmtime(content_input_path)
-        readable_mod_time = datetime.datetime.fromtimestamp(mod_time).isoformat(
-            timespec='seconds'
-        )
+        readable_mod_time = datetime.datetime.fromtimestamp(
+            mod_time
+        ).isoformat(timespec='seconds')
         return readable_mod_time
 
     def _get_file_modified_date_from_storage(
@@ -211,7 +213,7 @@ class AutoRetriever:
                 `DEFAULT_SIMILARITY_THRESHOLD`.
             return_detailed_info (bool, optional): Whether to return detailed
                 information including similarity score, content path and
-                metadata. Defaults to False.
+                metadata. Defaults to `False`.
 
         Returns:
             string: By default, returns only the text information. If
@@ -321,8 +323,15 @@ class AutoRetriever:
             info['text'] for info in all_retrieved_info if 'text' in info
         )
 
-        detailed_info = f"Original Query:\n{{ {query} }}\nRetrieved Context:\n{retrieved_infos}"
-        text_info = f"Original Query:\n{{ {query} }}\nRetrieved Context:\n{retrieved_infos_text}"
+        detailed_info = (
+            f"Original Query:\n{{ {query} }}\n"
+            f"Retrieved Context:\n{retrieved_infos}"
+        )
+
+        text_info = (
+            f"Original Query:\n{{ {query} }}\n"
+            f"Retrieved Context:\n{retrieved_infos_text}"
+        )
 
         if return_detailed_info:
             return detailed_info

camel/retrievers/base.py

@@ -32,7 +32,8 @@ def _query_unimplemented(self, *input: Any) -> None:
         registered hooks while the latter silently ignores them.
     """
     raise NotImplementedError(
-        f"Retriever [{type(self).__name__}] is missing the required \"query\" function"
+        f"Retriever [{type(self).__name__}] is missing the required"
+        " \"query\" function"
     )
 
 
@@ -52,7 +53,8 @@ def _process_unimplemented(self, *input: Any) -> None:
         registered hooks while the latter silently ignores them.
     """
     raise NotImplementedError(
-        f"Retriever [{type(self).__name__}] is missing the required \"process\" function"
+        f"Retriever [{type(self).__name__}] is missing the required "
+        "\"process\" function"
     )
 
 

camel/retrievers/bm25_retriever.py

@@ -47,7 +47,8 @@ class BM25Retriever(BaseRetriever):
             from rank_bm25 import BM25Okapi
         except ImportError as e:
             raise ImportError(
-                "Package `rank_bm25` not installed, install by running 'pip install rank_bm25'"
+                "Package `rank_bm25` not installed, install by running 'pip "
+                "install rank_bm25'"
             ) from e
 
         self.bm25: BM25Okapi = None

camel/retrievers/cohere_rerank_retriever.py

@@ -66,7 +66,8 @@ class CohereRerankRetriever(BaseRetriever):
             self.api_key = api_key or os.environ["COHERE_API_KEY"]
         except ValueError as e:
             raise ValueError(
-                "Must pass in cohere api key or specify via COHERE_API_KEY environment variable."
+                "Must pass in cohere api key or specify via COHERE_API_KEY"
+                " environment variable."
             ) from e
 
         self.co = cohere.Client(self.api_key)

camel/retrievers/vector_retriever.py

@@ -64,7 +64,9 @@ class VectorRetriever(BaseRetriever):
         self.storage = (
             storage
             if storage is not None
-            else QdrantStorage(vector_dim=self.embedding_model.get_output_dim())
+            else QdrantStorage(
+                vector_dim=self.embedding_model.get_output_dim()
+            )
         )
         self.similarity_threshold = similarity_threshold
         self.unstructured_modules: UnstructuredIO = UnstructuredIO()
@@ -151,7 +153,8 @@ class VectorRetriever(BaseRetriever):
 
         if query_results[0].record.payload is None:
             raise ValueError(
-                "Payload of vector storage is None, please check the collection."
+                "Payload of vector storage is None, please check the "
+                "collection."
             )
 
         # format the results
@@ -176,8 +179,11 @@ class VectorRetriever(BaseRetriever):
         if not formatted_results:
             return [
                 {
-                    'text': f"""No suitable information retrieved from {content_path} \
-                with similarity_threshold = {self.similarity_threshold}."""
+                    'text': (
+                        f"No suitable information retrieved "
+                        f"from {content_path} with similarity_threshold"
+                        f" = {self.similarity_threshold}."
+                    )
                 }
             ]
         return formatted_results

camel/societies/babyagi_playing.py

@@ -150,7 +150,8 @@ class BabyAGI:
         if self.task_type in [TaskType.AI_SOCIETY, TaskType.MISALIGNMENT]:
             task_specify_meta_dict.update(
                 dict(
-                    assistant_role=assistant_role_name, user_role=user_role_name
+                    assistant_role=assistant_role_name,
+                    user_role=user_role_name,
                 )
             )
         task_specify_meta_dict.update(extend_task_specify_meta_dict or {})

camel/societies/role_playing.py

@@ -274,7 +274,8 @@ class RolePlaying:
         ]:
             extend_sys_msg_meta_dicts = [
                 dict(
-                    assistant_role=assistant_role_name, user_role=user_role_name
+                    assistant_role=assistant_role_name,
+                    user_role=user_role_name,
                 )
                 for _ in range(2)
             ]

camel/storages/graph_storages/base.py

@@ -70,6 +70,7 @@ class BaseGraphStorage(ABC):
         self, query: str, params: Optional[Dict[str, Any]] = None
     ) -> List[Dict[str, Any]]:
         r"""Query the graph store with statement and parameters.
+
         Args:
             query (str): The query to be executed.
             params (Optional[Dict[str, Any]]): A dictionary of parameters to

camel/storages/graph_storages/neo4j_graph.py

@@ -99,7 +99,9 @@ class Neo4jGraph(BaseGraphStorage):
                 "Please install it with `pip install neo4j`."
             )
 
-        self.driver = neo4j.GraphDatabase.driver(url, auth=(username, password))
+        self.driver = neo4j.GraphDatabase.driver(
+            url, auth=(username, password)
+        )
         self.database = database
         self.timeout = timeout
         self.truncate = truncate
@@ -373,8 +375,8 @@ class Neo4jGraph(BaseGraphStorage):
         with self.driver.session(database=self.database) as session:
             session.run(
                 (
-                    "MATCH (n1:{})-[r:{}]->(n2:{}) WHERE n1.id = $subj AND n2.id"
-                    " = $obj DELETE r"
+                    "MATCH (n1:{})-[r:{}]->(n2:{}) WHERE n1.id = $subj AND"
+                    " n2.id = $obj DELETE r"
                 ).format(
                     BASE_ENTITY_LABEL.replace("_", ""),
                     rel,

camel/storages/vectordb_storages/base.py

@@ -147,7 +147,8 @@ class BaseVectorStorage(ABC):
         query: VectorDBQuery,
         **kwargs: Any,
     ) -> List[VectorDBQueryResult]:
-        r"""Searches for similar vectors in the storage based on the provided query.
+        r"""Searches for similar vectors in the storage based on the provided
+        query.
 
         Args:
             query (VectorDBQuery): The query object containing the search

camel/storages/vectordb_storages/milvus.py

@@ -155,7 +155,8 @@ class MilvusStorage(BaseVectorStorage):
             field_name="payload",
             datatype=DataType.JSON,
             description=(
-                'Any additional metadata or information related' 'to the vector'
+                'Any additional metadata or information related'
+                'to the vector'
             ),
         )
 
@@ -267,7 +268,9 @@ class MilvusStorage(BaseVectorStorage):
         for record in records:
             record_dict = {
                 "id": record.id,
-                "payload": record.payload if record.payload is not None else '',
+                "payload": record.payload
+                if record.payload is not None
+                else '',
                 "vector": record.vector,
             }
             validated_data.append(record_dict)

camel/toolkits/github_toolkit.py

@@ -14,6 +14,7 @@
 
 import os
 from dataclasses import dataclass
+from datetime import datetime, timedelta
 from typing import List, Optional
 
 from camel.functions import OpenAIFunction
@@ -48,7 +49,8 @@ class GithubIssue:
             body (str): The body/content of the GitHub issue.
             number (int): The issue number.
             file_path (str): The path of the file associated with the issue.
-            file_content (str): The content of the file associated with the issue.
+            file_content (str): The content of the file associated with the
+                issue.
         """
         self.title = title
         self.body = body
@@ -56,11 +58,12 @@ class GithubIssue:
         self.file_path = file_path
         self.file_content = file_content
 
-    def summary(self) -> str:
-        r"""Returns a summary of the issue.
+    def __str__(self) -> str:
+        r"""Returns a string representation of the issue.
 
         Returns:
-            str: A string containing the title, body, number, file path, and file content of the issue.
+            str: A string containing the title, body, number, file path, and
+                file content of the issue.
         """
         return (
             f"Title: {self.title}\n"
@@ -71,16 +74,60 @@ class GithubIssue:
         )
 
 
+@dataclass
+class GithubPullRequestDiff:
+    r"""Represents a single diff of a pull request on Github.
+
+    Attributes:
+        filename (str): The name of the file that was changed.
+        patch (str): The diff patch for the file.
+    """
+
+    filename: str
+    patch: str
+
+    def __str__(self) -> str:
+        r"""Returns a string representation of this diff."""
+        return f"Filename: {self.filename}\nPatch: {self.patch}"
+
+
+@dataclass
+class GithubPullRequest:
+    r"""Represents a pull request on Github.
+
+    Attributes:
+        title (str): The title of the GitHub pull request.
+        body (str): The body/content of the GitHub pull request.
+        diffs (List[GithubPullRequestDiff]): A list of diffs for the pull
+            request.
+    """
+
+    title: str
+    body: str
+    diffs: List[GithubPullRequestDiff]
+
+    def __str__(self) -> str:
+        r"""Returns a string representation of the pull request."""
+        diff_summaries = '\n'.join(str(diff) for diff in self.diffs)
+        return (
+            f"Title: {self.title}\n"
+            f"Body: {self.body}\n"
+            f"Diffs: {diff_summaries}\n"
+        )
+
+
 class GithubToolkit(BaseToolkit):
-    r"""A class representing a toolkit for interacting with GitHub repositories.
+    r"""A class representing a toolkit for interacting with GitHub
+    repositories.
 
-    This class provides methods for retrieving open issues, retrieving specific issues,
-    and creating pull requests in a GitHub repository.
+    This class provides methods for retrieving open issues, retrieving
+        specific issues, and creating pull requests in a GitHub repository.
 
     Args:
         repo_name (str): The name of the GitHub repository.
-        access_token (str, optional): The access token to authenticate with GitHub.
-            If not provided, it will be obtained using the `get_github_access_token` method.
+        access_token (str, optional): The access token to authenticate with
+            GitHub. If not provided, it will be obtained using the
+            `get_github_access_token` method.
     """
 
     def __init__(
@@ -90,8 +137,9 @@ class GithubToolkit(BaseToolkit):
 
         Args:
             repo_name (str): The name of the GitHub repository.
-            access_token (str, optional): The access token to authenticate with GitHub.
-                If not provided, it will be obtained using the `get_github_access_token` method.
+            access_token (str, optional): The access token to authenticate
+                with GitHub. If not provided, it will be obtained using the
+                `get_github_access_token` method.
         """
         if access_token is None:
             access_token = self.get_github_access_token()
@@ -101,21 +149,24 @@ class GithubToolkit(BaseToolkit):
         except ImportError:
             raise ImportError(
                 "Please install `github` first. You can install it by running "
-                "`pip install wikipedia`."
+                "`pip install pygithub`."
             )
         self.github = Github(auth=Auth.Token(access_token))
         self.repo = self.github.get_repo(repo_name)
 
     def get_tools(self) -> List[OpenAIFunction]:
-        r"""Returns a list of OpenAIFunction objects representing the functions in the toolkit.
+        r"""Returns a list of OpenAIFunction objects representing the
+        functions in the toolkit.
 
         Returns:
-            List[OpenAIFunction]: A list of OpenAIFunction objects representing the functions in the toolkit.
+            List[OpenAIFunction]: A list of OpenAIFunction objects
+                representing the functions in the toolkit.
         """
         return [
             OpenAIFunction(self.retrieve_issue_list),
             OpenAIFunction(self.retrieve_issue),
             OpenAIFunction(self.create_pull_request),
+            OpenAIFunction(self.retrieve_pull_requests),
         ]
 
     def get_github_access_token(self) -> str:
@@ -125,15 +176,16 @@ class GithubToolkit(BaseToolkit):
             str: A string containing the GitHub access token.
 
         Raises:
-            ValueError: If the API key or secret is not found in the environment variables.
+            ValueError: If the API key or secret is not found in the
+                environment variables.
         """
         # Get `GITHUB_ACCESS_TOKEN` here: https://github.com/settings/tokens
         GITHUB_ACCESS_TOKEN = os.environ.get("GITHUB_ACCESS_TOKEN")
 
         if not GITHUB_ACCESS_TOKEN:
             raise ValueError(
-                "`GITHUB_ACCESS_TOKEN` not found in environment variables. Get it "
-                "here: `https://github.com/settings/tokens`."
+                "`GITHUB_ACCESS_TOKEN` not found in environment variables. Get"
+                " it here: `https://github.com/settings/tokens`."
             )
         return GITHUB_ACCESS_TOKEN
 
@@ -151,7 +203,7 @@ class GithubToolkit(BaseToolkit):
                 number=issue.number,
                 file_path=issue.labels[
                     0
-                ].name,  # for now we require file path to be the first label in the PR
+                ].name,  # we require file path to be the first label in the PR
                 file_content=self.retrieve_file_content(issue.labels[0].name),
             )
             for issue in issues
@@ -173,9 +225,47 @@ class GithubToolkit(BaseToolkit):
         issues = self.retrieve_issue_list()
         for issue in issues:
             if issue.number == issue_number:
-                return issue.summary()
+                return str(issue)
         return None
 
+    def retrieve_pull_requests(
+        self, days: int, state: str, max_prs: int
+    ) -> List[str]:
+        r"""Retrieves a summary of merged pull requests from the repository.
+        The summary will be provided for the last specified number of days.
+
+        Args:
+            days (int): The number of days to retrieve merged pull requests
+                for.
+            state (str): A specific state of PRs to retrieve. Can be open or
+                closed.
+            max_prs (int): The maximum number of PRs to retrieve.
+
+        Returns:
+             List[str]: A list of merged pull request summaries.
+        """
+        pull_requests = self.repo.get_pulls(state=state)
+        merged_prs = []
+        earliest_date: datetime = datetime.utcnow() - timedelta(days=days)
+
+        for pr in pull_requests[:max_prs]:
+            if (
+                pr.merged
+                and pr.merged_at is not None
+                and pr.merged_at.timestamp() > earliest_date.timestamp()
+            ):
+                pr_details = GithubPullRequest(pr.title, pr.body, [])
+
+                # Get files changed in the PR
+                files = pr.get_files()
+
+                for file in files:
+                    diff = GithubPullRequestDiff(file.filename, file.patch)
+                    pr_details.diffs.append(diff)
+
+                merged_prs.append(str(pr_details))
+        return merged_prs
+
     def create_pull_request(
         self,
         file_path: str,
@@ -186,19 +276,23 @@ class GithubToolkit(BaseToolkit):
     ) -> str:
         r"""Creates a pull request.
 
-        This function creates a pull request in specified repository, which updates a
-        file in the specific path with new content. The pull request description
-        contains information about the issue title and number.
+        This function creates a pull request in specified repository, which
+        updates a file in the specific path with new content. The pull request
+        description contains information about the issue title and number.
 
         Args:
-            file_path (str): The path of the file to be updated in the repository.
+            file_path (str): The path of the file to be updated in the
+                repository.
             new_content (str): The specified new content of the specified file.
-            pr_title (str): The title of the issue that is solved by this pull request.
+            pr_title (str): The title of the issue that is solved by this pull
+                request.
             body (str): The commit message for the pull request.
-            branch_name (str): The name of the branch to create and submit the pull request from.
+            branch_name (str): The name of the branch to create and submit the
+                pull request from.
 
         Returns:
-            str: A formatted report of whether the pull request was created successfully or not.
+            str: A formatted report of whether the pull request was created
+                successfully or not.
         """
         sb = self.repo.get_branch(self.repo.default_branch)
         self.repo.create_git_ref(

camel/types/__init__.py

@@ -16,8 +16,8 @@ from .enums import (
     EmbeddingModelType,
     ModelType,
     OpenAIBackendRole,
-    OpenAIImageDetailType,
     OpenAIImageType,
+    OpenAIVisionDetailType,
     OpenAPIName,
     RoleType,
     StorageType,
@@ -58,8 +58,9 @@ __all__ = [
     'ChatCompletionAssistantMessageParam',
     'ChatCompletionFunctionMessageParam',
     'CompletionUsage',
+    'OpenAIVideoType',
     'OpenAIImageType',
-    'OpenAIImageDetailType',
+    'OpenAIVisionDetailType',
     'OpenAPIName',
     'AudioModelType',
     'VoiceType',

camel/types/enums.py

@@ -29,6 +29,9 @@ class ModelType(Enum):
     GPT_4_32K = "gpt-4-32k"
     GPT_4_TURBO = "gpt-4-turbo"
     GPT_4O = "gpt-4o"
+    GLM_4 = "glm-4"
+    GLM_4V = 'glm-4v'
+    GLM_3_TURBO = "glm-3-turbo"
 
     STUB = "stub"
 
@@ -62,6 +65,15 @@ class ModelType(Enum):
             ModelType.GPT_4O,
         }
 
+    @property
+    def is_zhipuai(self) -> bool:
+        r"""Returns whether this type of models is an ZhipuAI model."""
+        return self in {
+            ModelType.GLM_3_TURBO,
+            ModelType.GLM_4,
+            ModelType.GLM_4V,
+        }
+
     @property
     def is_open_source(self) -> bool:
         r"""Returns whether this type of models is open-source."""
@@ -103,6 +115,12 @@ class ModelType(Enum):
             return 128000
         elif self is ModelType.GPT_4O:
             return 128000
+        elif self == ModelType.GLM_4:
+            return 8192
+        elif self == ModelType.GLM_3_TURBO:
+            return 8192
+        elif self == ModelType.GLM_4V:
+            return 1024
         elif self is ModelType.STUB:
             return 4096
         elif self is ModelType.LLAMA_2:
@@ -191,6 +209,7 @@ class TaskType(Enum):
     ROLE_DESCRIPTION = "role_description"
     OBJECT_RECOGNITION = "object_recognition"
     DEFAULT = "default"
+    VIDEO_DESCRIPTION = "video_description"
 
 
 class VectorDistance(Enum):
@@ -238,7 +257,7 @@ class OpenAIImageType(Enum, metaclass=OpenAIImageTypeMeta):
     GIF = "gif"
 
 
-class OpenAIImageDetailType(Enum):
+class OpenAIVisionDetailType(Enum):
     AUTO = "auto"
     LOW = "low"
     HIGH = "high"
@@ -253,6 +272,11 @@ class OpenAPIName(Enum):
     COURSERA = "coursera"
     KLARNA = "klarna"
     SPEAK = "speak"
+    NASA_APOD = "nasa_apod"
+    BIZTOC = "biztoc"
+    CREATE_QR_CODE = "create_qr_code"
+    OUTSCHOOL = "outschool"
+    WEB_SCRAPER = "web_scraper"
 
 
 class AudioModelType(Enum):