| from abc import ABC | |||||
| from typing import Optional | from typing import Optional | ||||
| from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk | from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk | ||||
| } | } | ||||
| class Callback(ABC): | |||||
| class Callback: | |||||
| """ | """ | ||||
| Base class for callbacks. | Base class for callbacks. | ||||
| Only for LLM. | Only for LLM. |
| :return: the label of the tool | :return: the label of the tool | ||||
| """ | """ | ||||
| cls._builtin_tools_labels | |||||
| if len(cls._builtin_tools_labels) == 0: | if len(cls._builtin_tools_labels) == 0: | ||||
| # init the builtin providers | # init the builtin providers | ||||
| cls.load_builtin_providers_cache() | cls.load_builtin_providers_cache() |
| parameters = self._generate_parameters(variable_pool, node_data) | parameters = self._generate_parameters(variable_pool, node_data) | ||||
| # get tool runtime | # get tool runtime | ||||
| try: | try: | ||||
| self.app_id | |||||
| tool_runtime = ToolManager.get_workflow_tool_runtime(self.tenant_id, self.app_id, self.node_id, node_data) | tool_runtime = ToolManager.get_workflow_tool_runtime(self.tenant_id, self.app_id, self.node_id, node_data) | ||||
| except Exception as e: | except Exception as e: | ||||
| return NodeRunResult( | return NodeRunResult( |
| [tool.ruff.lint] | [tool.ruff.lint] | ||||
| ignore-init-module-imports = true | ignore-init-module-imports = true | ||||
| select = [ | select = [ | ||||
| "B", # flake8-bugbear rules | |||||
| "F", # pyflakes rules | "F", # pyflakes rules | ||||
| "I001", # unsorted-imports | |||||
| "I002", # missing-required-import | |||||
| "I", # isort rules | |||||
| "UP", # pyupgrade rules | "UP", # pyupgrade rules | ||||
| "RUF019", # unnecessary-key-check | "RUF019", # unnecessary-key-check | ||||
| ] | ] | ||||
| "F841", # unused-variable | "F841", # unused-variable | ||||
| "UP007", # non-pep604-annotation | "UP007", # non-pep604-annotation | ||||
| "UP032", # f-string | "UP032", # f-string | ||||
| "B005", # strip-with-multi-characters | |||||
| "B006", # mutable-argument-default | |||||
| "B007", # unused-loop-control-variable | |||||
| "B026", # star-arg-unpacking-after-keyword-arg | |||||
| "B904", # raise-without-from-inside-except | |||||
| "B905", # zip-without-explicit-strict | |||||
| ] | ] | ||||
| [tool.ruff.lint.per-file-ignores] | [tool.ruff.lint.per-file-ignores] |