jLBUdZddlmZddlZddlmZddlmZmZddl m Z ddl m Z m Z mZe rddlmZej"eZe d Gd d Zed dZded<ddZddZdZeddddd d ddZeddZy)uGenerationContext — ambient lineage 컨텍스트 (Task A2). 파이프라인 스텝/서비스가 자기 작업을 ``with generation_context(...)`` 로 감싼다. sink 가 ambient 로 읽어 lineage(stage/still_id/scene_index/...)를 채우고, 이 scope 전용 CaptureQueue 에 enqueue 한다. scope 종료 시 ``finally`` 에서 queue.flush() (독립 non-fatal 세션)로 일괄 영속화한다. worker thread 는 contextvars 가 자동 전파되지 않으므로 ``bind_context(ctx)`` 로 명시 전파한다(image_tracer.set_context / budget bind 패턴 정합). ★ 모든 동작 non-fatal — flush 실패는 로그만(생성 파이프라인 무영향). ) annotationsN)contextmanager) ContextVarToken) dataclass) TYPE_CHECKINGIteratorOptional CaptureQueueT)frozenceZdZUdZded<ded<ded<ded<d Zded <d Zd ed <d Zd ed <d Zded<dZ ded<y )GenerationContextuE현재 capture scope 의 ambient 메타. frozen — scope 내 불변.str project_id Optional[str] episode_idstagez'CaptureQueue'queueNstill_id Optional[int] scene_index shot_index entity_idTboolcapture_enabled) __name__ __module__ __qualname____doc____annotations__rrrrrX/Users/manta/Documents/Projects/TheRoad-I1/backend/app/services/image_capture/context.pyrrsOOO J "Hm"!%K% $J $#I}# !OT r#rimage_capture_gen_ctx)defaultz'ContextVar[Optional[GenerationContext]]_gen_ctxc*tjS)u<현재 capture scope 의 GenerationContext (없으면 None).)r'getr"r#r$current_contextr*4s <<>r#c,tj|S)uworker thread 명시 전파용 — ctx 를 현재 contextvar 에 set 하고 token 반환. 호출자는 작업 종료 시 ``_gen_ctx.reset(token)`` 으로 복원한다. )r'set)ctxs r$ bind_contextr.9s << r#c*tfd}|S)u호출 시점의 current_context 를 캡처해 worker thread 에서 재설치하는 래퍼 반환 (image_call_budget.bind_current_budget 패턴 정합). ThreadPool 병렬 롤 생성에서 capture 가 skipped_no_context 로 유실되는 경로 차단 (2026-07-17 still-variants Codex BLOCKING-3). 캡처된 컨텍스트가 없으면 no-op — 비 capture 경로 무영향. worker 는 호출 종료 시 reset 으로 clean 상태를 복원한다. c|i|St} |i|tj|S#tj|wxYw)N)r.r'reset)argskwargstokencapturedfns r$_wrappedz1bind_current_generation_context.._wrappedLsN  t&v& &X& "t&v& NN5 !HNN5 !s 5A )r*)r6r7r5s` @r$bind_current_generation_contextr8As H" Or#)rrrrcapturec #6Kddlm}|j|} t|||| ||||| } | j | t j | } t| 5| ddd | jt j| y#1swY0xYw#t$rtjddYDwxYw# | jn&#t$rtjddYnwxYwt j| wxYww)ucapture scope 를 연다. with-블록 안에서 sink 가 ambient 로 enqueue. 블록 종료 시(성공/예외 무관) ``finally`` 에서 queue.flush() 로 일괄 영속화. flush 는 독립 non-fatal 세션이라 business transaction rollback 에 말려도 생성된 candidate 가 남는다(사용자 "거부 후보까지 전부" 요구). flush 실패도 non-fatal — 로그만 남기고 삼킨다. yield 값은 이 scope 의 CaptureQueue (테스트/명시 enqueue 용). rr ) rrrrrrrrrNz,generation_context: flush failed (non-fatal)T)exc_info) app.services.image_capture.queuer __new__r__init__r'r,_shot_trace_scopeflush Exceptionloggerwarningr1) rrrrrrrr9r rr-r4s r$generation_contextrDXs*> !L  .E  C NN3 LL E s #K$ Z KKM u$ #  Z NNITXN Y Z Z KKM Z NNITXN Y ZusxAD C!B&C/B!?DBC! CDCDD CD C=:D<C==DDc#Kddlm}m}m}|jr)dt |jddd|j }nD|jr)dt |jddd|j }nd|j }d |jfd |jfd |jfd |jfd |jfd|jfd|j ffDcic] \}}||| }}}|}| |jnd} ||||j || 5ddddycc}}w#1swYyxYww)u/이 capture scope 를 Opik trace 하나로 연다. ★새 경계를 만들지 않는다 — 이 scope 가 이미 샷 경계다. still_id·scene_index·shot_index 를 이미 들고 있고 worker 전파기도 있다. 설정이 꺼져 있으면 아무것도 안 한다(바이트 동일). r)build_axis_tags current_trace open_tracezstill:Nu · zentity:zstage:rrrrrrstep)rJ)nametagsmetadata thread_id) app.modules.llm.opik_tracerFrGrHrrrrrrrrrN) r-rFrGrHrKkvmetaparentrNs r$r?r?sZ44 ||CLL)"1-.d399+> S]]+BQ/0SYYK@ {#3>> * 3>> *  & COO , 3>> * #-- ( SYY    A] 1  _F$*$6  DI O$C!Y 8  8 8!   8 8s*C$E &D:57E ,E1E E E )returnzOptional[GenerationContext])r-rrTr)rrrrrrrrrrrrrrr9rrTzIterator['CaptureQueue'])r-rrTzIterator[None])r __future__rlogging contextlibr contextvarsrr dataclassesrtypingrr r r<r getLoggerrrBrr'r!r*r.r8rDr?r"r#r$r\s #%)!44=   8 $ $!!!$5?T5 1  . #!% $#222 2  2  222222j$$r#