Function get_sessions_as_traces

app/api/agentops/exporter/processor.py:249–258

Get sessions and convert them to traces with parallel processing

(offset: int, limit: int)

async def get_sessions_as_traces(offset: int, limit: int) -> AsyncGenerator[Trace, None]:
    """Get sessions and convert them to traces with parallel processing"""
    sessions = await get_sessions(offset, limit)
    for i in range(0, len(sessions), PARALLEL_READS):
        batch = sessions[i : i + PARALLEL_READS]
        tasks = [asyncio.create_task(get_session_as_trace(session)) for session in batch]
        for task in asyncio.as_completed(tasks):
            trace = await task
            print(trace.id)
            yield trace


# Cache for session_id to project_id mapping