pi SDK

import * as pi from '@earendil-works/pi-coding-agent';
import { TraceRoot } from '@traceroot-ai/traceroot';

TraceRoot.initialize({
  instrumentModules: { piCodingAgent: pi },
});

import { createAgentSession, ModelRegistry, ModelRuntime } from '@earendil-works/pi-coding-agent';

// pi 0.80's model + credential surface. Created async (it reads the on-disk
// model catalog), then handed to createAgentSession. Injecting OPENAI_API_KEY
// as a runtime credential authenticates from env alone — no `pi auth login`.
const modelRuntime = await ModelRuntime.create();
await modelRuntime.setRuntimeApiKey('openai', process.env.OPENAI_API_KEY);

const modelRegistry = new ModelRegistry(modelRuntime);
const model = modelRegistry.find('openai', 'gpt-4o-mini');

const { session } = await createAgentSession({ model, modelRuntime });
try {
  await session.prompt('Write add.js exporting add(a, b), then write and run a test for it.');
} finally {
  session.dispose();
}

AgentSession.prompt              # AGENT span — one per prompt() call
└── gpt-4o-mini                  # LLM span — provider, tokens
    ├── write: add.js            # TOOL span — arguments + result
    ├── write: test-add.js
    └── bash: node test-add.js

TraceRoot.initialize({
  instrumentModules: {
    piCodingAgent: { module: pi, config: { captureContent: false, captureToolIo: false } },
  },
});