Thanks for raising the issue and propose a solution.

I have added some regression tests and it seems there are some case that are not properly fixed:

@tool
@dummy_decorator
def multi_decorator_tool(text: str) -> str:
    ...

becomes

def forward(self, text: str) -> str:
    def multi_decorator_tool(text: str) -> str:
        ...

and
2.

@tool
@dummy_decorator_1
@dummy_decorator_2
def complex_tool(
    text: str,
    multiplier: int = 2,
    separator: str = " ",
    multiline_parameter_1: int = 1_000,
    multiline_parameter_2: int = 2_000,
) -> str:
    ...

becomes

def forward(self, text: str, multiplier: int = 2, separator: str = ' ', multiline_parameter_1: int = 1000, multiline_parameter_2: int = 2000) -> str:
    @dummy_decorator_2
    def complex_tool(
        text: str,
        multiplier: int = 2,
        separator: str = " ",
        multiline_parameter_1: int = 1_000,
        multiline_parameter_2: int = 2_000,
    ) -> str:
        ...

Thanks again for addressing the issues in the new tests. That said, it looks like the proposed changes are causing some other tests to fail.

Pull Request Overview

This PR fixes the function body extraction mechanism in the @tool decorator to handle more complex function definitions, including functions with multiple decorators and multiline signatures. The fix addresses issue #1626 where the previous simple line-splitting approach failed to correctly extract function bodies in these cases.

Key changes:

• Replaces naive line-splitting approach with AST-based parsing for robust function body extraction
• Adds warning for functions with decorators other than @tool to alert users about potential remote execution issues
• Adds comprehensive tests covering multiple decorators and multiline function signatures

Reviewed Changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 4 comments.

Thanks again! Great!

I just made some simplifications and rephrasing of the error/warning messages.