LangChain callback handler: "benign" cross-context detach is logged at ERROR by OpenTelemetry (the `except ValueError` guard never fires)

已關閉
#1,062 0 則留言 0 個 reaction 已指派 0 人 在 GitHub 檢視

維護者通常 1 天內回覆

已經有一個關聯 PR 被合併了。

  • #1064 來自 @sushan5140 —— 已關閉,未合併
  • #1065 來自 @pcbeingused333 —— 已關閉,未合併
  • #1071 來自 @AbhiPrasad —— 已合併

評估

難度
3/5
預估耗時
1-2 天
新手友好度
55/100
Issue 類型
缺陷
描述清晰度
基本清楚
活躍度
活躍
技術堆疊
python
領域
observability

研究方向

Start in braintrust/otel/context.py at ContextManager.unset_current_span, then read _end_span in braintrust/integrations/langchain/callbacks.py, where the dead except ValueError sits. Run the reproduction script in the issue and confirm the opentelemetry.context ERROR appears. Done when that log no longer appears, span parenting is unchanged, and the existing tests pass. Choose one of the three suggested fixes with a maintainer before writing code.

由索引模型根據 Issue 內容生成。

描述

Summary

With BRAINTRUST_OTEL_COMPAT=true, BraintrustCallbackHandler sets each run's span as the current OTel context when the run starts and unsets it when the run ends. In LangGraph's async runtime, a run can start in one asyncio task and end in another. The OTel token then belongs to a different contextvars.Context, and every such run logs this at ERROR with a traceback:

ERROR:opentelemetry.context:Failed to detach context
Traceback (most recent call last):
  File ".../opentelemetry/context/__init__.py", line 155, in detach
    _RUNTIME_CONTEXT.detach(token)
  File ".../opentelemetry/context/contextvars_context.py", line 53, in detach
    self._current_context.reset(token)
ValueError: <Token var=<ContextVar name='current_context' ...> at 0x...> was created in a different Context

The handler already anticipates this. In braintrust/integrations/langchain/callbacks.py, _end_span:

# Async callbacks may unset from a different context; span state is
# tracked in self.spans, so this ValueError is benign.
try:
    span.unset_current()
except ValueError as e:
    if "was created in a different Context" in str(e):
        pass
    else:
        raise

But that except can never catch the error. span.unset_current() reaches opentelemetry.context.detach(token) (via braintrust/otel/context.py, ContextManager.unset_current_span). OTel's detach catches the ValueError itself and logs it with logger.exception("Failed to detach context"). Nothing is raised to Braintrust, so the guard is dead code and the "benign" error reaches the logs at ERROR.

We agree it's benign. Span parenting in our traces is correct. But in production it was 100% of the ERROR lines on our agent service for a week. That breaks "no ERRORs after deploy" checks and buries real errors.

Reproduction (no network)
import contextvars
import logging
import os
import uuid

os.environ["BRAINTRUST_OTEL_COMPAT"] = "true"
logging.basicConfig(level=logging.INFO)

from braintrust import logger as bt
from braintrust.test_helpers import init_test_logger, simulate_login
from braintrust.integrations.langchain import BraintrustCallbackHandler

# In-memory logger, no network.
simulate_login()
with bt._internal_with_memory_background_logger():
    handler = BraintrustCallbackHandler(logger=init_test_logger("repro"))
    run_id = uuid.uuid4()
    # LangGraph's async runtime can fire start and end for one run in different tasks.
    contextvars.copy_context().run(handler.on_chain_start, {"name": "node"}, {}, run_id=run_id)
    contextvars.copy_context().run(handler.on_chain_end, {}, run_id=run_id)
    # -> ERROR:opentelemetry.context:Failed to detach context
    #    ValueError: <Token ...> was created in a different Context

A real async LangGraph run shows the same thing with no manual context juggling. A two-node StateGraph of async nodes, run with await graph.ainvoke(..., config={"callbacks": [handler]}), logs 3 of these errors per invocation.

Versions
  • braintrust 0.31.0. The relevant code (_end_span in integrations/langchain/callbacks.py and ContextManager.unset_current_span in otel/context.py) is unchanged in 0.49.2.
  • opentelemetry-api 1.37.0
  • langchain-core 1.5.3, langgraph 1.2.10
  • Python 3.12
Suggested fix

Handle the mismatch where the token is actually reset, rather than relying on OTel to raise. Some options:

  1. In ContextManager.unset_current_span, skip OTel's public context.detach (which always logs) and reset the token without logging, catching ValueError there. This would use OTel's runtime context directly.
  2. Record which contextvars.Context a span was attached in (for example, compare contextvars.copy_context() identity or a marker ContextVar set alongside the attach). Only detach when ending in that same context; otherwise drop the token.
  3. At minimum, remove the dead except ValueError in _end_span and document the ERROR log, so users know to filter opentelemetry.context themselves.
Workaround we use

A logging.Filter on the opentelemetry.context logger. It drops a record only when all three hold: the message is Failed to detach context, the exception is this ValueError, and braintrust/integrations/langchain/callbacks.py:_end_span is on the live call stack. Any other detach failure still logs. It works, but every Braintrust + LangGraph + OTel user will hit this, so it would be better fixed in the SDK.

Related

#7 was the same ValueError in the Claude Agent SDK wrapper, where it was raised (fixed in 0.9.0). This one is in the LangChain integration, and OpenTelemetry logs it rather than raising it, so a try/except around unset_current() cannot catch it.

主要語言
Python
星號
22
分支
30
平均合併
9 小時
30 天內合併 PR
128

環境準備

  • 沒有 Dockerfile 或 Docker Compose 檔案
  • 沒有 Pull Request 範本
  • 閱讀貢獻指南

從這裡開始

  1. 先讀完整個 Issue,再讀專案的貢獻指南。
  2. 在 Issue 下留言說明你要接手 —— 這能避免兩個人做同樣的事。
  3. Fork 儲存庫,在一個分支上完成修改。
  4. 送出 Pull Request,並在描述裡引用這個 Issue 編號。

braintrustdata/braintrust-sdk-python 的其他 Issue

查看 braintrustdata/braintrust-sdk-python 的全部 Issue

相似的 Issue

更多 Python Issue

把新 issue 寄到你的電子郵件信箱

精選適合新手參與的 GitHub issue 摘要。