@@ -647,16 +647,13 @@ def _sentry_patched_create_sync(f: "Any", *args: "Any", **kwargs: "Any") -> "Any
647647 "sentry.origin" : AnthropicIntegration .origin ,
648648 },
649649 )
650-
651- set_on_span = span .set_attribute
652650 else :
653651 span = get_start_span_function ()(
654652 op = OP .GEN_AI_CHAT ,
655653 name = f"chat { model } " .strip (),
656654 origin = AnthropicIntegration .origin ,
657655 )
658656
659- set_on_span = span .set_data
660657 span .__enter__ ()
661658
662659 _set_create_input_data (span , kwargs , integration )
@@ -712,10 +709,8 @@ def _sentry_patched_create_sync(f: "Any", *args: "Any", **kwargs: "Any") -> "Any
712709 response_id = getattr (result , "id" , None ),
713710 finish_reason = getattr (result , "stop_reason" , None ),
714711 )
715- span .__exit__ (None , None , None )
716- else :
717- set_on_span ("unknown_response" , True )
718- span .__exit__ (None , None , None )
712+
713+ span .__exit__ (None , None , None )
719714
720715 return result
721716
@@ -749,16 +744,13 @@ async def _sentry_patched_create_async(
749744 "sentry.origin" : AnthropicIntegration .origin ,
750745 },
751746 )
752-
753- set_on_span = span .set_attribute
754747 else :
755748 span = get_start_span_function ()(
756749 op = OP .GEN_AI_CHAT ,
757750 name = f"chat { model } " .strip (),
758751 origin = AnthropicIntegration .origin ,
759752 )
760753
761- set_on_span = span .set_data
762754 span .__enter__ ()
763755
764756 _set_create_input_data (span , kwargs , integration )
@@ -814,10 +806,8 @@ async def _sentry_patched_create_async(
814806 response_id = getattr (result , "id" , None ),
815807 finish_reason = getattr (result , "stop_reason" , None ),
816808 )
817- span .__exit__ (None , None , None )
818- else :
819- set_on_span ("unknown_response" , True )
820- span .__exit__ (None , None , None )
809+
810+ span .__exit__ (None , None , None )
821811
822812 return result
823813
0 commit comments