Pārlūkot izejas kodu

fix(tool): validate return type from _invoke method to ensure ToolInv… (#12079)

Signed-off-by: -LAN- <laipz8200@outlook.com>
tags/0.15.0
-LAN- pirms 10 mēnešiem
vecāks
revīzija
7ae417b4b6
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
1 mainītis faili ar 6 papildinājumiem un 0 dzēšanām
  1. 6
    0
      api/core/tools/tool/tool.py

+ 6
- 0
api/core/tools/tool/tool.py Parādīt failu

@@ -222,6 +222,12 @@ class Tool(BaseModel, ABC):
if not isinstance(result, list):
result = [result]

if not all(isinstance(message, ToolInvokeMessage) for message in result):
raise ValueError(
f"Invalid return type from {self.__class__.__name__}._invoke method. "
"Expected ToolInvokeMessage or list of ToolInvokeMessage."
)

return result

def _transform_tool_parameters_type(self, tool_parameters: Mapping[str, Any]) -> dict[str, Any]:

Notiek ielāde…
Atcelt
Saglabāt