camel-ai 0.1.5.1__py3-none-any.whl → 0.1.5.3__py3-none-any.whl

camel/prompts/task_prompt_template.py

@@ -18,11 +18,19 @@ 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.generate_text_embedding_data import (
+    GenerateTextEmbeddingDataPromptTemplateDict,
+)
 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 +60,10 @@ 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.GENERATE_TEXT_EMBEDDING_DATA: GenerateTextEmbeddingDataPromptTemplateDict(),  # 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

@@ -22,9 +22,10 @@ from camel.agents import (
 from camel.generators import SystemMessageGenerator
 from camel.human import Human
 from camel.messages import BaseMessage
+from camel.models import BaseModelBackend
 from camel.prompts import TextPrompt
 from camel.responses import ChatAgentResponse
-from camel.types import ModelType, RoleType, TaskType
+from camel.types import RoleType, TaskType
 
 
 class RolePlaying:
@@ -48,9 +49,9 @@ class RolePlaying:
             in the loop. (default: :obj:`False`)
         critic_criteria (str, optional): Critic criteria for the critic agent.
             If not specified, set the criteria to improve task performance.
-        model_type (ModelType, optional): Model type that will be used for
-            role playing. If specified, it will override the model in all
-            agents. (default: :obj:`None`)
+        model (BaseModelBackend, optional): The model backend to use for
+            generating responses. If specified, it will override the model in
+            all agents. (default: :obj:`None`)
         task_type (TaskType, optional): The type of task to perform.
             (default: :obj:`TaskType.AI_SOCIETY`)
         assistant_agent_kwargs (Dict, optional): Additional arguments to pass
@@ -84,7 +85,7 @@ class RolePlaying:
         with_task_planner: bool = False,
         with_critic_in_the_loop: bool = False,
         critic_criteria: Optional[str] = None,
-        model_type: Optional[ModelType] = None,
+        model: Optional[BaseModelBackend] = None,
         task_type: TaskType = TaskType.AI_SOCIETY,
         assistant_agent_kwargs: Optional[Dict] = None,
         user_agent_kwargs: Optional[Dict] = None,
@@ -99,7 +100,7 @@ class RolePlaying:
         self.with_task_specify = with_task_specify
         self.with_task_planner = with_task_planner
         self.with_critic_in_the_loop = with_critic_in_the_loop
-        self.model_type = model_type
+        self.model = model
         self.task_type = task_type
         self.task_prompt = task_prompt
 
@@ -189,12 +190,10 @@ class RolePlaying:
                     )
                 )
             task_specify_meta_dict.update(extend_task_specify_meta_dict or {})
-            if self.model_type is not None:
+            if self.model is not None:
                 if task_specify_agent_kwargs is None:
                     task_specify_agent_kwargs = {}
-                task_specify_agent_kwargs.update(
-                    dict(model_type=self.model_type)
-                )
+                task_specify_agent_kwargs.update(dict(model=self.model))
             task_specify_agent = TaskSpecifyAgent(
                 task_type=self.task_type,
                 output_language=output_language,
@@ -224,12 +223,10 @@ class RolePlaying:
                 agents. (default: :obj:`None`)
         """
         if self.with_task_planner:
-            if self.model_type is not None:
+            if self.model is not None:
                 if task_planner_agent_kwargs is None:
                     task_planner_agent_kwargs = {}
-                task_planner_agent_kwargs.update(
-                    dict(model_type=self.model_type)
-                )
+                task_planner_agent_kwargs.update(dict(model=self.model))
             task_planner_agent = TaskPlannerAgent(
                 output_language=output_language,
                 **(task_planner_agent_kwargs or {}),
@@ -274,7 +271,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)
             ]
@@ -320,13 +318,13 @@ class RolePlaying:
             output_language (str, optional): The language to be output by the
                 agents. (default: :obj:`None`)
         """
-        if self.model_type is not None:
+        if self.model is not None:
             if assistant_agent_kwargs is None:
                 assistant_agent_kwargs = {}
-            assistant_agent_kwargs.update(dict(model_type=self.model_type))
+            assistant_agent_kwargs.update(dict(model=self.model))
             if user_agent_kwargs is None:
                 user_agent_kwargs = {}
-            user_agent_kwargs.update(dict(model_type=self.model_type))
+            user_agent_kwargs.update(dict(model=self.model))
 
         self.assistant_agent = ChatAgent(
             init_assistant_sys_msg,
@@ -382,10 +380,10 @@ class RolePlaying:
                     critic_msg_meta_dict,
                     role_tuple=(critic_role_name, RoleType.CRITIC),
                 )
-                if self.model_type is not None:
+                if self.model is not None:
                     if critic_kwargs is None:
                         critic_kwargs = {}
-                    critic_kwargs.update(dict(model_type=self.model_type))
+                    critic_kwargs.update(dict(model=self.model))
                 self.critic = CriticAgent(
                     self.critic_sys_msg,
                     **(critic_kwargs or {}),

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

@@ -14,10 +14,11 @@
 from .enums import (
     AudioModelType,
     EmbeddingModelType,
+    ModelPlatformType,
     ModelType,
     OpenAIBackendRole,
-    OpenAIImageDetailType,
     OpenAIImageType,
+    OpenAIVisionDetailType,
     OpenAPIName,
     RoleType,
     StorageType,
@@ -58,9 +59,11 @@ __all__ = [
     'ChatCompletionAssistantMessageParam',
     'ChatCompletionFunctionMessageParam',
     'CompletionUsage',
+    'OpenAIVideoType',
     'OpenAIImageType',
-    'OpenAIImageDetailType',
+    'OpenAIVisionDetailType',
     'OpenAPIName',
+    'ModelPlatformType',
     'AudioModelType',
     'VoiceType',
 ]