Import ATIF trajectories

import json
from phoenix.client import Client
from phoenix.client.helpers.atif import upload_atif_trajectories_as_spans

# Load one or more ATIF trajectory dicts
with open("trajectory.json") as f:
    trajectory = json.load(f)

client = Client()
result = upload_atif_trajectories_as_spans(
    client, [trajectory], project_name="my-agent-eval"
)
print(result)  # Counts of spans received and queued

assistant                  AGENT
  iteration 1              CHAIN
    gpt-4                  LLM
    search                 TOOL
  iteration 2              CHAIN
    gpt-4                  LLM

with open("parent_trajectory.json") as f:
    parent = json.load(f)
with open("child_trajectory.json") as f:
    child = json.load(f)

# Upload together so cross-references resolve
upload_atif_trajectories_as_spans(
    client, [parent, child], project_name="my-agent-eval"
)

orchestrator               AGENT
  iteration 1              CHAIN
    gpt-4                  LLM
    delegate_task          TOOL
      researcher           AGENT
        iteration 1        CHAIN
          gpt-4            LLM