?HAMA

Navigation

Breadcrumbs

/ / opt / hc_python / lib / python3.12 / site-packages / sentry_sdk / integrations

Create

Create File | Create Folder | Upload Files | Run Command

Edit File Size: 11.56 KB

//opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/langgraph.py

Text Base64

from functools import wraps
from typing import Any, Callable, List, Optional

import sentry_sdk
from sentry_sdk.ai.utils import (
    set_data_normalized,
    normalize_message_roles,
    truncate_and_annotate_messages,
)
from sentry_sdk.consts import OP, SPANDATA
from sentry_sdk.integrations import DidNotEnable, Integration
from sentry_sdk.scope import should_send_default_pii
from sentry_sdk.utils import safe_serialize

try:
    from langgraph.graph import StateGraph
    from langgraph.pregel import Pregel
except ImportError:
    raise DidNotEnable("langgraph not installed")

class LanggraphIntegration(Integration):
    identifier = "langgraph"
    origin = f"auto.ai.{identifier}"

def __init__(self, include_prompts=True):
        # type: (LanggraphIntegration, bool) -> None
        self.include_prompts = include_prompts

@staticmethod
    def setup_once():
        # type: () -> None
        # LangGraph lets users create agents using a StateGraph or the Functional API.
        # StateGraphs are then compiled to a CompiledStateGraph. Both CompiledStateGraph and
        # the functional API execute on a Pregel instance. Pregel is the runtime for the graph
        # and the invocation happens on Pregel, so patching the invoke methods takes care of both.
        # The streaming methods are not patched, because due to some internal reasons, LangGraph
        # will automatically patch the streaming methods to run through invoke, and by doing this
        # we prevent duplicate spans for invocations.
        StateGraph.compile = _wrap_state_graph_compile(StateGraph.compile)
        if hasattr(Pregel, "invoke"):
            Pregel.invoke = _wrap_pregel_invoke(Pregel.invoke)
        if hasattr(Pregel, "ainvoke"):
            Pregel.ainvoke = _wrap_pregel_ainvoke(Pregel.ainvoke)

def _get_graph_name(graph_obj):
    # type: (Any) -> Optional[str]
    for attr in ["name", "graph_name", "__name__", "_name"]:
        if hasattr(graph_obj, attr):
            name = getattr(graph_obj, attr)
            if name and isinstance(name, str):
                return name
    return None

def _normalize_langgraph_message(message):
    # type: (Any) -> Any
    if not hasattr(message, "content"):
        return None

parsed = {"role": getattr(message, "type", None), "content": message.content}

for attr in ["name", "tool_calls", "function_call", "tool_call_id"]:
        if hasattr(message, attr):
            value = getattr(message, attr)
            if value is not None:
                parsed[attr] = value

return parsed

def _parse_langgraph_messages(state):
    # type: (Any) -> Optional[List[Any]]
    if not state:
        return None

messages = None

if isinstance(state, dict):
        messages = state.get("messages")
    elif hasattr(state, "messages"):
        messages = state.messages
    elif hasattr(state, "get") and callable(state.get):
        try:
            messages = state.get("messages")
        except Exception:
            pass

if not messages or not isinstance(messages, (list, tuple)):
        return None

normalized_messages = []
    for message in messages:
        try:
            normalized = _normalize_langgraph_message(message)
            if normalized:
                normalized_messages.append(normalized)
        except Exception:
            continue

return normalized_messages if normalized_messages else None

def _wrap_state_graph_compile(f):
    # type: (Callable[..., Any]) -> Callable[..., Any]
    @wraps(f)
    def new_compile(self, *args, **kwargs):
        # type: (Any, Any, Any) -> Any
        integration = sentry_sdk.get_client().get_integration(LanggraphIntegration)
        if integration is None:
            return f(self, *args, **kwargs)
        with sentry_sdk.start_span(
            op=OP.GEN_AI_CREATE_AGENT,
            origin=LanggraphIntegration.origin,
        ) as span:
            compiled_graph = f(self, *args, **kwargs)

compiled_graph_name = getattr(compiled_graph, "name", None)
            span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "create_agent")
            span.set_data(SPANDATA.GEN_AI_AGENT_NAME, compiled_graph_name)

if compiled_graph_name:
                span.description = f"create_agent {compiled_graph_name}"
            else:
                span.description = "create_agent"

if kwargs.get("model", None) is not None:
                span.set_data(SPANDATA.GEN_AI_REQUEST_MODEL, kwargs.get("model"))

tools = None
            get_graph = getattr(compiled_graph, "get_graph", None)
            if get_graph and callable(get_graph):
                graph_obj = compiled_graph.get_graph()
                nodes = getattr(graph_obj, "nodes", None)
                if nodes and isinstance(nodes, dict):
                    tools_node = nodes.get("tools")
                    if tools_node:
                        data = getattr(tools_node, "data", None)
                        if data and hasattr(data, "tools_by_name"):
                            tools = list(data.tools_by_name.keys())

if tools is not None:
                span.set_data(SPANDATA.GEN_AI_REQUEST_AVAILABLE_TOOLS, tools)

return compiled_graph

return new_compile

def _wrap_pregel_invoke(f):
    # type: (Callable[..., Any]) -> Callable[..., Any]

@wraps(f)
    def new_invoke(self, *args, **kwargs):
        # type: (Any, Any, Any) -> Any
        integration = sentry_sdk.get_client().get_integration(LanggraphIntegration)
        if integration is None:
            return f(self, *args, **kwargs)

graph_name = _get_graph_name(self)
        span_name = (
            f"invoke_agent {graph_name}".strip() if graph_name else "invoke_agent"
        )

with sentry_sdk.start_span(
            op=OP.GEN_AI_INVOKE_AGENT,
            name=span_name,
            origin=LanggraphIntegration.origin,
        ) as span:
            if graph_name:
                span.set_data(SPANDATA.GEN_AI_PIPELINE_NAME, graph_name)
                span.set_data(SPANDATA.GEN_AI_AGENT_NAME, graph_name)

span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "invoke_agent")

# Store input messages to later compare with output
            input_messages = None
            if (
                len(args) > 0
                and should_send_default_pii()
                and integration.include_prompts
            ):
                input_messages = _parse_langgraph_messages(args[0])
                if input_messages:
                    normalized_input_messages = normalize_message_roles(input_messages)
                    scope = sentry_sdk.get_current_scope()
                    messages_data = truncate_and_annotate_messages(
                        normalized_input_messages, span, scope
                    )
                    if messages_data is not None:
                        set_data_normalized(
                            span,
                            SPANDATA.GEN_AI_REQUEST_MESSAGES,
                            messages_data,
                            unpack=False,
                        )

result = f(self, *args, **kwargs)

_set_response_attributes(span, input_messages, result, integration)

return result

return new_invoke

def _wrap_pregel_ainvoke(f):
    # type: (Callable[..., Any]) -> Callable[..., Any]

@wraps(f)
    async def new_ainvoke(self, *args, **kwargs):
        # type: (Any, Any, Any) -> Any
        integration = sentry_sdk.get_client().get_integration(LanggraphIntegration)
        if integration is None:
            return await f(self, *args, **kwargs)

graph_name = _get_graph_name(self)
        span_name = (
            f"invoke_agent {graph_name}".strip() if graph_name else "invoke_agent"
        )

span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "invoke_agent")

input_messages = None
            if (
                len(args) > 0
                and should_send_default_pii()
                and integration.include_prompts
            ):
                input_messages = _parse_langgraph_messages(args[0])
                if input_messages:
                    normalized_input_messages = normalize_message_roles(input_messages)
                    scope = sentry_sdk.get_current_scope()
                    messages_data = truncate_and_annotate_messages(
                        normalized_input_messages, span, scope
                    )
                    if messages_data is not None:
                        set_data_normalized(
                            span,
                            SPANDATA.GEN_AI_REQUEST_MESSAGES,
                            messages_data,
                            unpack=False,
                        )

result = await f(self, *args, **kwargs)

_set_response_attributes(span, input_messages, result, integration)

return result

return new_ainvoke

def _get_new_messages(input_messages, output_messages):
    # type: (Optional[List[Any]], Optional[List[Any]]) -> Optional[List[Any]]
    """Extract only the new messages added during this invocation."""
    if not output_messages:
        return None

if not input_messages:
        return output_messages

# only return the new messages, aka the output messages that are not in the input messages
    input_count = len(input_messages)
    new_messages = (
        output_messages[input_count:] if len(output_messages) > input_count else []
    )

return new_messages if new_messages else None

def _extract_llm_response_text(messages):
    # type: (Optional[List[Any]]) -> Optional[str]
    if not messages:
        return None

for message in reversed(messages):
        if isinstance(message, dict):
            role = message.get("role")
            if role in ["assistant", "ai"]:
                content = message.get("content")
                if content and isinstance(content, str):
                    return content

return None

def _extract_tool_calls(messages):
    # type: (Optional[List[Any]]) -> Optional[List[Any]]
    if not messages:
        return None

tool_calls = []
    for message in messages:
        if isinstance(message, dict):
            msg_tool_calls = message.get("tool_calls")
            if msg_tool_calls and isinstance(msg_tool_calls, list):
                tool_calls.extend(msg_tool_calls)

return tool_calls if tool_calls else None

def _set_response_attributes(span, input_messages, result, integration):
    # type: (Any, Optional[List[Any]], Any, LanggraphIntegration) -> None
    if not (should_send_default_pii() and integration.include_prompts):
        return

parsed_response_messages = _parse_langgraph_messages(result)
    new_messages = _get_new_messages(input_messages, parsed_response_messages)

llm_response_text = _extract_llm_response_text(new_messages)
    if llm_response_text:
        set_data_normalized(span, SPANDATA.GEN_AI_RESPONSE_TEXT, llm_response_text)
    elif new_messages:
        set_data_normalized(span, SPANDATA.GEN_AI_RESPONSE_TEXT, new_messages)
    else:
        set_data_normalized(span, SPANDATA.GEN_AI_RESPONSE_TEXT, result)

tool_calls = _extract_tool_calls(new_messages)
    if tool_calls:
        set_data_normalized(
            span,
            SPANDATA.GEN_AI_RESPONSE_TOOL_CALLS,
            safe_serialize(tool_calls),
            unpack=False,
        )

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Directory Contents

Dirs: 10 × Files: 70

Sort:

Enable drag-to-move

Name	Size	Perms	Modified	Actions
celery DIR	-	drwxr-xr-x	2025-12-03 08:02:53	Edit Download recursive recursive
django DIR	-	drwxr-xr-x	2025-12-03 08:02:53	Edit Download recursive recursive
google_genai DIR	-	drwxr-xr-x	2025-12-03 08:02:53	Edit Download recursive recursive
grpc DIR	-	drwxr-xr-x	2025-12-03 08:02:53	Edit Download recursive recursive
openai_agents DIR	-	drwxr-xr-x	2025-12-03 08:02:53	Edit Download recursive recursive
opentelemetry DIR	-	drwxr-xr-x	2025-12-03 08:02:53	Edit Download recursive recursive
pydantic_ai DIR	-	drwxr-xr-x	2025-12-03 08:02:53	Edit Download recursive recursive
redis DIR	-	drwxr-xr-x	2025-12-03 08:02:53	Edit Download recursive recursive
spark DIR	-	drwxr-xr-x	2025-12-03 08:02:53	Edit Download recursive recursive
__pycache__ DIR	-	drwxr-xr-x	2025-12-03 08:02:53	Edit Download recursive recursive
aiohttp.py	12.70 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
anthropic.py	14.17 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
argv.py	911 B	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
ariadne.py	5.70 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
arq.py	7.70 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
asgi.py	12.50 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
asyncio.py	4.50 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
asyncpg.py	6.37 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
atexit.py	1.61 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
aws_lambda.py	17.65 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
beam.py	5.06 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
boto3.py	4.31 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
bottle.py	6.46 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
chalice.py	4.59 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
clickhouse_driver.py	5.96 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
cloud_resource_context.py	7.60 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
cohere.py	9.18 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
dedupe.py	1.93 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
dramatiq.py	7.28 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
excepthook.py	2.35 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
executing.py	1.95 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
falcon.py	9.28 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
fastapi.py	4.48 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
flask.py	8.54 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
gcp.py	8.26 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
gnu_backtrace.py	2.75 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
gql.py	4.69 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
graphene.py	4.92 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
httpx.py	5.94 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
huey.py	5.32 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
huggingface_hub.py	14.60 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
langchain.py	39.02 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
langgraph.py	11.56 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
launchdarkly.py	1.89 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
litellm.py	9.99 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
litestar.py	11.55 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
logging.py	13.57 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
loguru.py	6.38 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
mcp.py	19.58 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
modules.py	820 B	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
openai.py	24.54 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
openfeature.py	1.10 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
otlp.py	3.06 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
pure_eval.py	4.50 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
pymongo.py	6.23 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
pyramid.py	7.19 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
quart.py	7.24 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
ray.py	5.20 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
rq.py	5.18 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
rust_tracing.py	8.87 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
sanic.py	12.66 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
serverless.py	1.76 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
socket.py	3.09 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
sqlalchemy.py	4.24 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
starlette.py	25.62 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
starlite.py	10.31 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
statsig.py	1.20 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
stdlib.py	8.76 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
strawberry.py	13.82 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
sys_exit.py	2.43 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
threading.py	6.94 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
tornado.py	7.04 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
trytond.py	1.61 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
typer.py	1.77 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
unleash.py	1.03 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
unraisablehook.py	1.71 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
wsgi.py	10.56 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
_asgi_common.py	3.11 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
_wsgi_common.py	7.38 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
__init__.py	12.44 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive