github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-05-13 03:00:55 -04:00
Code Issues Packages Projects Releases Wiki Activity

replace assertion with valueerror (#3974)

Browse Source 
Co-authored-by: Leonardo Pinheiro <lpinheiro@microsoft.com>
This commit is contained in:
Leonardo Pinheiro
2024-10-27 15:13:52 +10:00
committed by GitHub
parent 3fe0f9e97d
commit a626d7cf18
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/packages/autogen-core/src/autogen_core/components/tools/_function_tool.py
View File

@@ -46,5 +46,6 @@ class FunctionTool(BaseTool[BaseModel, BaseModel]):
cancellation_token.link_future(future)
result = await future
assert isinstance(result, self.return_type())
if not isinstance(result, self.return_type()):
raise ValueError(f"Expected return type {self.return_type()}, got {type(result)}")
return result