@@ -251,12 +251,12 @@ def _handle_in_thinking_mode(
251251 yield self .handle_thinking_delta (vendor_part_id = vendor_part_id , content = before_end )
252252
253253 # Close the thinking part
254- if vendor_part_id is not None :
254+ if vendor_part_id is not None : # pragma: lax no cover
255255 self ._vendor_id_to_part_index .pop (vendor_part_id , None )
256256 self ._tag_buffer .pop (vendor_part_id , None )
257257
258258 # Process any remaining content after the end tag
259- if after_end :
259+ if after_end : # pragma: lax no cover
260260 yield from self ._handle_text_delta_with_thinking_tags (
261261 vendor_part_id = vendor_part_id ,
262262 content = after_end ,
@@ -266,11 +266,11 @@ def _handle_in_thinking_mode(
266266 )
267267 elif self ._could_be_tag_start (combined_content , end_tag ):
268268 # Might be start of end tag, buffer it
269- if vendor_part_id is not None :
269+ if vendor_part_id is not None : # pragma: lax no cover
270270 self ._tag_buffer [vendor_part_id ] = combined_content
271271 else :
272272 # Not an end tag, add to thinking content
273- if vendor_part_id is not None :
273+ if vendor_part_id is not None : # pragma: lax no cover
274274 self ._tag_buffer .pop (vendor_part_id , None )
275275 yield self .handle_thinking_delta (vendor_part_id = vendor_part_id , content = combined_content )
276276
@@ -290,7 +290,7 @@ def _handle_start_of_response(
290290 after_start = combined_content [len (start_tag ) :]
291291
292292 # Clear any state for this vendor_part_id and start thinking part
293- if vendor_part_id is not None :
293+ if vendor_part_id is not None : # pragma: lax no cover
294294 self ._vendor_id_to_part_index .pop (vendor_part_id , None )
295295 self ._tag_buffer .pop (vendor_part_id , None )
296296 yield self .handle_thinking_delta (vendor_part_id = vendor_part_id , content = '' )
@@ -306,11 +306,11 @@ def _handle_start_of_response(
306306 )
307307 elif self ._could_be_tag_start (combined_content , start_tag ):
308308 # Might be start of start tag at position 0, buffer it
309- if vendor_part_id is not None :
309+ if vendor_part_id is not None : # pragma: lax no cover
310310 self ._tag_buffer [vendor_part_id ] = combined_content
311311 else :
312312 # Not a start tag, process as normal text
313- if vendor_part_id is not None :
313+ if vendor_part_id is not None : # pragma: lax no cover
314314 self ._tag_buffer .pop (vendor_part_id , None )
315315 yield from self ._handle_text_delta_simple (
316316 vendor_part_id = vendor_part_id ,
0 commit comments