from langchain.agents.tools import BaseTool
from pydantic import BaseModel, Field
from typing import Type

class DelSubstrToolInput(BaseModel):
    text: str = Field(description="原字符串")
    content: str = Field(description="去除的内容")


class DelSubstrTool(BaseTool):
    name: str = "remove_str"

    description: str = (
        "去除字符串的指定的内容"
    )
    args_schema: Type[BaseModel] = DelSubstrToolInput

    def _run(self, text: str, content: str):
        return text.replace(r"{content}","")