def retrieve_trace()

in llm_demo/orchestrator/langgraph/langgraph_orchestrator.py [0:0]

• 10 lines of code
• 5 McCabe index (conditional complexity)

    def retrieve_trace(self, messages: Sequence[BaseMessage]):
        trace = []
        for m in messages:
            if isinstance(m, ToolMessage):
                trace_info = {"tool_call_id": m.name, "results": m.content}
                add_kwargs = m.additional_kwargs
                if add_kwargs and add_kwargs.get("sql"):
                    trace_info["sql"] = add_kwargs.get("sql")
                trace.append(trace_info)
        return trace