Skip to content

instrument_remote_reply

autogen.opentelemetry.instrumentators.agent_instrumentators.remote.instrument_remote_reply #

instrument_remote_reply(agent, *, tracer)
Source code in autogen/opentelemetry/instrumentators/agent_instrumentators/remote.py 
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
def instrument_remote_reply(agent: Agent, *, tracer: Tracer) -> Agent:
    # Instrument `a_generate_remote_reply` as a remote invoke_agent span
    if hasattr(agent, "a_generate_remote_reply"):
        # Find the original reply func in _reply_func_list
        original_reply_func = None
        original_reply_func_index = None
        for i, reply_func_tuple in enumerate(agent._reply_func_list):
            if getattr(reply_func_tuple["reply_func"], "__name__", None) == "a_generate_remote_reply":
                original_reply_func = reply_func_tuple["reply_func"]
                original_reply_func_index = i
                break

        if original_reply_func and not hasattr(original_reply_func, "__otel_wrapped__"):
            old_httpx_client_factory = agent._httpx_client_factory

            def httpx_client_factory_traced():
                httpx_client = old_httpx_client_factory()
                TRACE_PROPAGATOR.inject(httpx_client.headers)
                return httpx_client

            agent._httpx_client_factory = httpx_client_factory_traced

            # Create traced wrapper that accepts self as first arg (like unbound method)
            async def a_generate_remote_reply_traced(
                self_agent: Any,
                *args: Any,
                **kwargs: Any,
            ) -> Any:
                with tracer.start_as_current_span(f"invoke_agent {self_agent.name}") as span:
                    span.set_attribute("ag2.span.type", SpanType.AGENT.value)
                    span.set_attribute("gen_ai.operation.name", "invoke_agent")
                    span.set_attribute("gen_ai.agent.name", self_agent.name)
                    span.set_attribute("gen_ai.agent.remote", True)
                    if hasattr(self_agent, "url") and self_agent.url:
                        span.set_attribute("server.address", self_agent.url)
                    return await original_reply_func(self_agent, *args, **kwargs)

            a_generate_remote_reply_traced.__otel_wrapped__ = True
            # Update _reply_func_list to use the traced function
            agent._reply_func_list[original_reply_func_index]["reply_func"] = a_generate_remote_reply_traced

    return agent