yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-12-07 12:30:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix function tool naming to avoid overriding the name input (#5165)

Browse Source 
fix function tool naming to avoid overriding the name input
This commit is contained in:
Pierre 2025-01-23 16:42:54 +01:00 committed by GitHub
parent 141247f6d7
commit c3e84dc5ca
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
python/packages/autogen-core/src/autogen_core/_function_utils.py
View File

@ -304,18 +304,18 @@ def normalize_annotated_type(type_hint: Type[Any]) -> Type[Any]:
def args_base_model_from_signature(name: str, sig: inspect.Signature) -> Type[BaseModel]: def args_base_model_from_signature(name: str, sig: inspect.Signature) -> Type[BaseModel]:
fields: Dict[str, tuple[Type[Any], Any]] = {} fields: Dict[str, tuple[Type[Any], Any]] = {}
for name, param in sig.parameters.items(): for param_name, param in sig.parameters.items():
# This is handled externally # This is handled externally
if name == "cancellation_token": if param_name == "cancellation_token":
continue continue
if param.annotation is inspect.Parameter.empty: if param.annotation is inspect.Parameter.empty:
raise ValueError("No annotation") raise ValueError("No annotation")
type = normalize_annotated_type(param.annotation) type = normalize_annotated_type(param.annotation)
description = type2description(name, param.annotation) description = type2description(param_name, param.annotation)
default_value = param.default if param.default is not inspect.Parameter.empty else PydanticUndefined default_value = param.default if param.default is not inspect.Parameter.empty else PydanticUndefined
fields[name] = (type, Field(default=default_value, description=description)) fields[param_name] = (type, Field(default=default_value, description=description))
return cast(BaseModel, create_model(name, **fields)) # type: ignore return cast(BaseModel, create_model(name, **fields)) # type: ignore