Custom function descriptions about magentic HOT 2 CLOSED

from functools import wraps


def use_custom_docstring(docstring: str):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            return func(*args, **kwargs)
        wrapper.__doc__ = docstring
        return wrapper
    return decorator

from magentic.chat_model.function_schema import FunctionCallFunctionSchema

def plus(a: int, b: int) -> int:
    """Original docstring"""
    return a + b

s = FunctionCallFunctionSchema(plus)
print()
print(s.description)
print(s.parameters)

plus_with_new_docstring = use_custom_docstring("new docstring")(plus)

s = FunctionCallFunctionSchema(plus_with_new_docstring)
print()
print(s.description)
print(s.parameters)

from functools import wraps


def use_custom_docstring(func, docstring: str):
    @wraps(func)
    def wrapper(*args, **kwargs):
        return func(*args, **kwargs)
    wrapper.__doc__ = docstring
    return wrapper

# usage
@prompt(
    PROMPT,
    functions=[use_custom_docstring(my_function, "my function description")],
    model=OpenaiChatModel(model="gpt-4-1106-preview"),
)
def answer_question(world_state: str, question_id: str) -> str:
    ...