def get_query_ids_mapping()

in databao/executors/lighthouse/graph.py [0:0]

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

def get_query_ids_mapping(messages: list[BaseMessage]) -> dict[str, ToolMessage]:
    query_ids = {}
    for message in messages:
        if isinstance(message, ToolMessage) and isinstance(message.artifact, dict) and "query_id" in message.artifact:
            query_ids[message.artifact["query_id"]] = message
    return query_ids