uPj "UdZddlZddlmZddlmZmZmZmZddl m Z dZ ee d<ee zZee d<d e d ee efdzfd Zd ee efd e ded efdZd ee efdeeee effd dfdZd ee efdeeee effdeeee effd dfdZy)aO Shared helpers for recovering Responses API output items from raw SSE chunks. The same recovery logic is needed in multiple places (e.g. the ChatGPT Responses transformation and the LiteLLM Responses-to-Chat-Completions bridge). Keep the implementation in a single module so a fix in one caller automatically applies to all of them. N)Mapping)Final SupportsInt TypeAliascast)STREAM_SSE_DONE_STRINGi_MAX_CONTENT_INDEX_ConvertibleToIntchunkreturnc.ddlm}|j|jxsdj}|r|tk(s|j dry t j|}t|tsy|S#t j$rYywxYw)a8Parse a single raw SSE line into a JSON object dict. Returns ``None`` for empty lines, ``event:`` lines, ``[DONE]`` markers, invalid JSON, or non-dict payloads. Centralizes the parsing step that feeds into the recovery helpers in this module so behavior stays consistent across all callers. r)CustomStreamWrapperzevent:N) ,litellm.litellm_core_utils.streaming_handlerr_strip_sse_data_from_chunkstripr startswithjsonloadsJSONDecodeError isinstancedict)r rstripped_chunk parsed_chunks j/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/responses/sse_output_recovery.pyparse_sse_json_chunkrsQ0KKEKKMZ`^`ggiN ^/EEIbIbckIl&*jj&@  lD )    sA>>BBrkeyfallbackc|j|}||S ttt|S#tt f$r|cYSwxYw)N)getintrr TypeError ValueError)rrr raw_indexs r _chunk_indexr%,sN#'',I4)9566 z "s0AA output_itemsc~|jd}t|tsyt|dt |}|||<y)zRecord an OUTPUT_ITEM_DONE chunk into ``output_items`` keyed by ``output_index`` (falling back to the next free slot when missing). itemN output_index)r rrr%len)rr&r(r)s rrecord_output_item_chunkr+6s>""6*D dD !&|^SEVWL!%Ltext_only_itemsc|jd}t|tsyt|dt |}||vry|j|}|#d|jdxsd|ddgd }|||<|j d g}t|t syt|d t |}|d ks |tkDryt ||kr$|jd dgdt ||kr$||}t|tsi}|||<d |d<||d<|jd |d|d<y|j dgy)zRecord an OUTPUT_TEXT_DONE chunk as a synthetic message item in ``text_only_items``. Real OUTPUT_ITEM_DONE events already captured in ``output_items`` take precedence at the same ``output_index``. textNr)messageitem_idmsg_ assistant completed)typeidrolestatuscontentr9 content_indexr output_textr)r5r/ annotationsr5r<) r rstrr%r* setdefaultlistr appendr) rr&r-r/r)r(r9r: content_items rrecord_output_text_chunkrBDsl""6*D dC &|^SEYZL|#   | ,D |""9-F4 ~1F!  )- %__Y3G gt $' os7|TMqM,>> g,- '%!   g,- '=)L lD ) !- (LL &2&2=&A ]# r2r,)__doc__rcollections.abcrtypingrrrrlitellm.constantsrr __annotations__r=r robjectrr!r%r+rBr,rrJs #664 E *S090S&[(9D(@.wsF{3#QT &#v+& &sDf--. &  &93#v+&93#tCK00193#tCK00193 93r,