File tree Expand file tree Collapse file tree 1 file changed +16
-1
lines changed
Expand file tree Collapse file tree 1 file changed +16
-1
lines changed Original file line number Diff line number Diff line change @@ -415,12 +415,27 @@ def test_setup_observability_registers_context_emitter(self):
415415 previous_emitter = get_context_emitter ()
416416 try :
417417 # Make test deterministic even when optional dependency is not installed.
418- with patch ("importlib.util.find_spec" , return_value = object ()):
418+ with patch ("importlib.util.find_spec" , return_value = object ()), \
419+ patch ("atexit.register" ):
419420 cli_app ._setup_langextract_observability (verbose = False )
420421 emitter = get_context_emitter ()
421422 assert emitter .enabled , "context emitter should be enabled after setup"
422423 finally :
423424 set_context_emitter (previous_emitter )
425+ assert get_context_emitter () is previous_emitter
426+
427+ def test_setup_observability_without_langextract_leaves_context_emitter_unchanged (self ):
428+ """Setup should be a no-op when optional langextract dependency is unavailable."""
429+ import praisonai .cli .app as cli_app
430+ from praisonaiagents .trace .context_events import get_context_emitter , set_context_emitter
431+
432+ previous_emitter = get_context_emitter ()
433+ try :
434+ with patch ("importlib.util.find_spec" , return_value = None ):
435+ cli_app ._setup_langextract_observability (verbose = False )
436+ assert get_context_emitter () is previous_emitter
437+ finally :
438+ set_context_emitter (previous_emitter )
424439
425440
426441if __name__ == "__main__" :
You can’t perform that action at this time.
0 commit comments