Merge pull request #2 from 0Shark/copilot/sub-pr-1

Fix LLMObs span double-exit issue with exit state tracking

Author: 0Shark

client/joinly_client/agent.py

@@ -234,6 +234,7 @@ async def _call_llm(self, messages: list[ModelMessage]) -> ModelResponse:
 
         # Try to use LLMObs for proper LLM Observability
         llmobs_span = None
+        llmobs_span_exited = False
         try:
             from ddtrace.llmobs import LLMObs
 
@@ -272,6 +273,7 @@ async def _call_llm(self, messages: list[ModelMessage]) -> ModelResponse:
         except Exception as e:
             if llmobs_span:
                 self._annotate_llmobs(llmobs_span, llmobs_input, error=e)
+                llmobs_span_exited = True
             raise
 
         logger.debug(
@@ -281,7 +283,7 @@ async def _call_llm(self, messages: list[ModelMessage]) -> ModelResponse:
             response.usage.response_tokens or 0,
         )
 
-        if llmobs_span:
+        if llmobs_span and not llmobs_span_exited:
             self._annotate_llmobs(llmobs_span, llmobs_input, response=response)
 
         self._usage.add(