Skip to content

Commit f921f92

Browse files
test(langextract): make context-emitter setup test deterministic
Agent-Logs-Url: https://github.com/MervinPraison/PraisonAI/sessions/cec32ee8-0573-405f-86d1-6a28fb51b609 Co-authored-by: MervinPraison <454862+MervinPraison@users.noreply.github.com>
1 parent d0b445c commit f921f92

1 file changed

Lines changed: 11 additions & 5 deletions

File tree

src/praisonai/tests/unit/test_langextract_sink.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -410,11 +410,17 @@ def test_bridge_maps_context_events_to_action_events(self):
410410
def test_setup_observability_registers_context_emitter(self):
411411
"""`--observe langextract` must install the bridge on the context emitter."""
412412
import praisonai.cli.app as cli_app
413-
from praisonaiagents.trace.context_events import get_context_emitter
414-
415-
cli_app._setup_langextract_observability(verbose=False)
416-
emitter = get_context_emitter()
417-
assert emitter.enabled, "context emitter should be enabled after setup"
413+
from praisonaiagents.trace.context_events import get_context_emitter, set_context_emitter
414+
415+
previous_emitter = get_context_emitter()
416+
try:
417+
# Make test deterministic even when optional dependency is not installed.
418+
with patch("importlib.util.find_spec", return_value=object()):
419+
cli_app._setup_langextract_observability(verbose=False)
420+
emitter = get_context_emitter()
421+
assert emitter.enabled, "context emitter should be enabled after setup"
422+
finally:
423+
set_context_emitter(previous_emitter)
418424

419425

420426
if __name__ == "__main__":

0 commit comments

Comments
 (0)