observe context manager

[eliorc]

Describe the feature or potential improvement

The observe decorators is really useful, and on many occasions using it as a context manager makes a lot of sense.
For example, I might have in the same function multiple LLM calls which I would like to name for better auditibility, and it would be so convinient to just

@observe(name='my_container')
def func(...)
    
      with observe(as_type='generation', name='step 1'):
             # First LLM call

      with observe(as_type='generation', name='step 2'):
             # Second LLM call

instead of having to put a single LLM call code line inside a function just to be able to use the observe decorator.

Since Python 3.2, decorators implemented using @contextlib.contextmanager can be used both as context managers and as decorators. I've looked at the implementation of observe and it is quite complex for me to try and do it on my own.
If it's implementation could utilize @contextlib.contextmanager that would be awesome or alternatively implement a separate observe_context context manager.

Additional information

No response

[marcklingen]

Thanks a lot for sharing this! We track this idea here, I'll copy your note to this thread: https://github.com/orgs/langfuse/discussions/2488

[marcklingen]

@eliorc please add your upvote to the other thread and subscribe to it to follow any updates on this

[hassiebp]

Agree this would be a good DX improvement, thanks for raising this @eliorc