Pi 확장

npm install weave @earendil-works/pi-coding-agent
npm install --save-dev @types/node tsx typescript

import {init, createOtelExtension} from 'weave';

import {
  createAgentSession,
  DefaultResourceLoader,
  SessionManager,
  getAgentDir,
} from '@earendil-works/pi-coding-agent';

async function main() {
  // 1. Weave를 초기화합니다. Weave 프로젝트를 가리키는 OTEL TracerProvider를
  //    설정합니다. createOtelExtension()이 생성하는 모든 span은
  //    이곳으로 자동으로 내보내집니다.
  await init('[YOUR-TEAM]/[YOUR-PROJECT]');

  // 2. 리소스 로더를 생성하고 Weave OTEL 확장을 주입합니다.
  //    리소스 로더는 에이전트 활동을 트레이싱하는 데 사용되는 Pi 런타임 환경과
  //    확장 라이프사이클을 제공합니다.
  const resourceLoader = new DefaultResourceLoader({
    cwd: process.cwd(),
    agentDir: getAgentDir(),
    extensionFactories: [createOtelExtension({})],
  });

  await resourceLoader.reload();

  // 3. 에이전트 세션을 시작합니다
  const {session} = await createAgentSession({
    resourceLoader,
    sessionManager: SessionManager.inMemory(),
  });

  // 4. 확장을 바인딩합니다. session_start 이벤트가 트리거되어 OTEL 어댑터가
  //    루트 대화 span을 생성하고 대화 ID를 캡처합니다.
  await session.bindExtensions({});

  // 5. assistant 출력을 stdout으로 스트리밍합니다
  session.subscribe(event => {
    if (
      event.type === 'message_update' &&
      event.assistantMessageEvent.type === 'text_delta'
    ) {
      process.stdout.write(event.assistantMessageEvent.delta);
    }
  });

  // 6. 프롬프트를 전송하고 전체 응답을 기다립니다
  await session.prompt('What files are in the current directory?');
  console.log();
}

main();