from collections.abc import Generatorfrom typing import Anyfrom dify_plugin import Toolfrom dify_plugin.entities.tool import ToolInvokeMessage# 导入 logging 和自定义处理器import loggingfrom dify_plugin.config.logger_format import plugin_logger_handler# 使用自定义处理器设置日志logger = logging.getLogger(__name__)logger.setLevel(logging.INFO)logger.addHandler(plugin_logger_handler)class LoggerDemoTool(Tool): def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessage]: # 以不同级别输出日志信息 logger.info("This is an INFO log message.") logger.warning("This is a WARNING log message.") logger.error("This is an ERROR log message.") yield self.create_text_message("Hello, Dify!")
