andras/great-ai
andras/great-ai
Watch
1
0
Fork
You've already forked great-ai
 0
Code Issues Pull requests 2 Projects Releases 4 Packages 1 Wiki Activity Actions

Restore outer tracing contexts after nested predictions

Browse source
This commit is contained in:
Andras Schmelczer 2026-10-27 19:55:16 +00:00
commit d10eb75d8d
1 changed files with 29 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
great_ai/tracing/tracing_context.py
View file

@ -74,7 +74,7 @@ class TracingContext(Generic[T]):
return _current_tracing_context.get()
def __enter__(self) -> "TracingContext":
_current_tracing_context.set(self)
self._token = _current_tracing_context.set(self)
return self
def __exit__(
@ -83,7 +83,7 @@ class TracingContext(Generic[T]):
exception: Optional[BaseException],
traceback: Optional[TracebackType],
) -> Literal[False]:
_current_tracing_context.set(None)
_current_tracing_context.reset(self._token)
if exception is not None and type is not None:
self.finalise(exception=exception)