feat(closes OPEN-9557): add 'promote' parameter to the trace decorator

[gustavocidornelas]

Pull Request

Summary

Introduces a promote kwarg in the @trace() decorator. This kwarg can be used to surface any of the traced functions kwargs to the row-level. This feature is useful for users who want to filter or create tests that target these promoted values.

There are two ways to use it:

1. Passing a list of kwargs to promote:

# In this case, `tool_call_count` and `user_query` will appear in the row-level, even if `my_func` is deeply nested
@trace(promote=["tool_call_count", "user_query"])
def my_func(user_query: str, tool_call_count: int, other_kwarg: bool):
    pass

2. Passing a dict to promote, to alias the columns

# In this case, `tool_call_count` is aliased to `tools` and `user_query` is aliased to `query`
@trace(promote={"tool_call_count": "tools", "user_query": "query"})
def my_func(user_query: str, tool_call_count: int, other_kwarg: bool):
    pass

Changes

• Added the promote to all trace decorators (e.g., trace, trace_async, etc.) and piped the data through

Context

OPEN-9557: Add promote parameter to the trace decorator

Testing

• Manual testing