Import Traces

import trajectory_sdk as tj
from trajectory import Client
from trajectory.lib import upload_trajectories

# 1. Initialize
tj.init(
    provider="langsmith",
    api_key="lsv2_pt_...",
    project_id="your-project-id",
)

# 2. List conversations
conversations = tj.list_conversations(limit=5)

# 3. Extract IDs and import as Trajectories
ids = [c.conversation_id for c in conversations]
trajectories = tj.import_conversations(ids)

# 4. Upload to the Trajectory platform (reads TRAJECTORY_API_KEY by default)
result = upload_trajectories(Client(), trajectories, dataset="my-dataset-v1")

tj.save(trajectories, "./exports")

trajectories = tj.import_conversations([
    "cc_abc123",
    "cc_def456",
])