Source code for langchain_community.tools.gitlab.tool

"""
这个工具允许代理与python-gitlab库进行交互，并在GitLab存储库上操作。

要使用这个工具，你必须首先设置环境变量：
    GITLAB_PRIVATE_ACCESS_TOKEN
    GITLAB_REPOSITORY -> 格式: {owner}/{repo}

"""
from typing import Optional

from langchain_core.callbacks import CallbackManagerForToolRun
from langchain_core.pydantic_v1 import Field
from langchain_core.tools import BaseTool

from langchain_community.utilities.gitlab import GitLabAPIWrapper


class GitLabAction(BaseTool):
    """与GitLab API交互的工具。"""

    api_wrapper: GitLabAPIWrapper = Field(default_factory=GitLabAPIWrapper)  # type: ignore[arg-type]
    mode: str
    name: str = ""
    description: str = ""

    def _run(
        self,
        instructions: str,
        run_manager: Optional[CallbackManagerForToolRun] = None,
    ) -> str:
        """使用GitLab API来运行一个操作。"""
        return self.api_wrapper.run(self.mode, instructions)