chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,22 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from yuxi.utils.thread_utils import extract_thread_id
|
||||
|
||||
|
||||
def test_extract_thread_id_reads_stable_event_paths():
|
||||
assert extract_thread_id({"thread_id": " parent-thread "}) == "parent-thread"
|
||||
assert extract_thread_id({"configurable": {"thread_id": "child-thread"}}) == "child-thread"
|
||||
assert extract_thread_id({"metadata": {"thread_id": "meta-thread"}}) == "meta-thread"
|
||||
assert extract_thread_id({"stream_event": {"thread_id": "event-thread"}}) == "event-thread"
|
||||
assert extract_thread_id({"meta": {"thread_id": "run-thread"}}) == "run-thread"
|
||||
|
||||
|
||||
def test_extract_thread_id_uses_fallback_for_missing_or_unstable_paths():
|
||||
assert extract_thread_id(None, "fallback-thread") == "fallback-thread"
|
||||
assert extract_thread_id({"thread_id": " "}, "fallback-thread") == "fallback-thread"
|
||||
assert extract_thread_id({"data": {"metadata": {"thread_id": "nested-thread"}}}, "fallback-thread") == (
|
||||
"fallback-thread"
|
||||
)
|
||||
assert extract_thread_id({"metadata": {"configurable": {"thread_id": "nested-thread"}}}, "fallback-thread") == (
|
||||
"fallback-thread"
|
||||
)
|
||||
Reference in New Issue
Block a user