@traceroot-ai/traceroot 0.1.6 → 0.1.8

package/dist/pi.d.ts

@@ -0,0 +1,156 @@
+import { SpanStatusCode } from '@opentelemetry/api';
+import type { Context, Span, Tracer } from '@opentelemetry/api';
+/**
+ * Hand-transcribed local mirror of @earendil-works/pi-coding-agent / pi-agent-core, verified against published .d.ts at 0.80.6.
+ */
+export interface Usage {
+    input: number;
+    output: number;
+    cacheRead: number;
+    cacheWrite: number;
+    reasoning?: number;
+    totalTokens: number;
+    cost: {
+        input: number;
+        output: number;
+        cacheRead: number;
+        cacheWrite: number;
+        total: number;
+    };
+}
+export type StopReason = 'stop' | 'length' | 'toolUse' | 'error' | 'aborted';
+export interface AssistantMessage {
+    role: 'assistant';
+    content: unknown[];
+    api: string;
+    provider: string;
+    model: string;
+    responseModel?: string;
+    responseId?: string;
+    usage: Usage;
+    stopReason: StopReason;
+    errorMessage?: string;
+    timestamp: number;
+}
+export interface UserMessage {
+    role: 'user';
+    content: unknown;
+    timestamp: number;
+}
+export interface ToolResultMessage {
+    role: 'toolResult';
+    toolCallId: string;
+    toolName: string;
+    content: unknown[];
+    details?: unknown;
+    isError: boolean;
+    timestamp: number;
+}
+export interface OtherAgentMessage {
+    role: 'bashExecution' | 'custom' | 'branchSummary' | 'compactionSummary';
+    timestamp?: number;
+}
+export type AgentMessage = AssistantMessage | UserMessage | ToolResultMessage | OtherAgentMessage;
+export type AgentEvent = {
+    type: 'agent_start';
+} | {
+    type: 'agent_end';
+    messages: AgentMessage[];
+    willRetry?: boolean;
+} | {
+    type: 'turn_start';
+} | {
+    type: 'turn_end';
+    message: AgentMessage;
+    toolResults: ToolResultMessage[];
+} | {
+    type: 'message_start';
+    message: AgentMessage;
+} | {
+    type: 'message_update';
+    message: AgentMessage;
+} | {
+    type: 'message_end';
+    message: AgentMessage;
+} | {
+    type: 'tool_execution_start';
+    toolCallId: string;
+    toolName: string;
+    args: unknown;
+} | {
+    type: 'tool_execution_update';
+    toolCallId: string;
+    toolName: string;
+    args: unknown;
+    partialResult: unknown;
+} | {
+    type: 'tool_execution_end';
+    toolCallId: string;
+    toolName: string;
+    result: unknown;
+    isError: boolean;
+};
+export interface PromptOptions {
+    images?: unknown[];
+    streamingBehavior?: 'steer' | 'followUp';
+}
+export interface AgentSessionInstance {
+    readonly sessionId?: string;
+    /** True while a run is actively executing; prompt() routes through steer()/followUp() instead of starting fresh (see isQueueOnlySteer). */
+    readonly isStreaming?: boolean;
+    prompt(text: string, options?: PromptOptions): Promise<void>;
+    /** Queue a steering message mid-run. Optional so a partial double doesn't disable instrumentation over one missing method. */
+    steer?(text: string, images?: unknown[]): Promise<void>;
+    /** Queue a follow-up message, processed once the agent has no more tool calls or steering messages left. */
+    followUp?(text: string, images?: unknown[]): Promise<void>;
+    subscribe(listener: (event: AgentEvent) => void): () => void;
+    /** dispose() reassigns the private _eventListeners array to a fresh empty array, so no listener (including this file's own) fires again after dispose(). */
+    dispose(): void;
+}
+export interface AgentSessionConstructor {
+    prototype: AgentSessionInstance;
+}
+/** Structural shape of the imported `@earendil-works/pi-coding-agent` module namespace. */
+export interface PiCodingAgentModule {
+    AgentSession?: AgentSessionConstructor;
+    [key: string]: unknown;
+}
+export declare const TRACER_NAME = "@traceroot-ai/pi-coding-agent";
+export interface PiInstrumentationConfig {
+    /** Capture prompt/response text as input.value/output.value on AGENT and LLM spans. Default true. */
+    captureContent?: boolean;
+    /** Capture tool call args/results as input.value/output.value on TOOL spans. Default true. */
+    captureToolIo?: boolean;
+}
+export interface ResolvedPiInstrumentationConfig {
+    captureContent: boolean;
+    captureToolIo: boolean;
+}
+export declare function resolveConfig(config?: PiInstrumentationConfig): ResolvedPiInstrumentationConfig;
+export declare function openRootSpan(tracer: Pick<Tracer, 'startSpan'>, parentCtx: Context, input: {
+    text: string | undefined;
+    sessionId: string | undefined;
+    captureContent: boolean;
+}): Span;
+export declare function stampRootOutput(span: Span, finalMessages: AgentMessage[], captureContent: boolean): void;
+export declare function finalizeRootSpan(span: Span, retryCount: number, status: {
+    code: SpanStatusCode;
+    message?: string;
+}, error?: unknown): void;
+export declare function openLlmSpan(tracer: Pick<Tracer, 'startSpan'>, parentCtx: Context, message: AssistantMessage): Span;
+export declare function closeLlmSpan(span: Span, message: AssistantMessage, captureContent: boolean): void;
+export declare function describeToolCallSpan(toolName: string, args: unknown): string;
+export declare function openToolSpan(tracer: Pick<Tracer, 'startSpan'>, parentCtx: Context, toolCallId: string, toolName: string, args: unknown, captureToolIo: boolean): Span;
+export declare function closeToolSpan(span: Span, result: unknown, isError: boolean, captureToolIo: boolean): void;
+export declare function closeDanglingSpan(span: Span | undefined): void;
+/**
+ * Dispatcher entry point for pi, called once by wireInstrumentations() in
+ * instrumentation.ts. Kept here rather than inline in that shared file so the
+ * dispatcher stays symmetric with claude/openai (one import + one call). Unwraps
+ * the optional { module, config } form; pi carries its own captureContent/
+ * captureToolIo config, passed through unmerged (there is no apiKey/baseUrl to
+ * thread). A throw from instrumentPiCodingAgent() surfaces, same as claude's.
+ */
+export declare function wirePiCodingAgentInstrumentation(entry: unknown): void;
+export declare function instrumentPiCodingAgent(sdk: unknown, config?: PiInstrumentationConfig): unknown;
+//# sourceMappingURL=pi.d.ts.map

package/dist/pi.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pi.d.ts","sourceRoot":"","sources":["../src/pi.ts"],"names":[],"mappings":"AACA,OAAO,EAAyC,cAAc,EAAS,MAAM,oBAAoB,CAAC;AAClG,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAiBhE;;GAEG;AAEH,MAAM,WAAW,KAAK;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,KAAK,EAAE,MAAM,CAAC;KACf,CAAC;CACH;AAED,MAAM,MAAM,UAAU,GAAG,MAAM,GAAG,QAAQ,GAAG,SAAS,GAAG,OAAO,GAAG,SAAS,CAAC;AAE7E,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,WAAW,CAAC;IAClB,OAAO,EAAE,OAAO,EAAE,CAAC;IACnB,GAAG,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,KAAK,CAAC;IACb,UAAU,EAAE,UAAU,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,YAAY,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,OAAO,EAAE,CAAC;IACnB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;CACnB;AAGD,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,eAAe,GAAG,QAAQ,GAAG,eAAe,GAAG,mBAAmB,CAAC;IACzE,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,MAAM,YAAY,GAAG,gBAAgB,GAAG,WAAW,GAAG,iBAAiB,GAAG,iBAAiB,CAAC;AAGlG,MAAM,MAAM,UAAU,GAClB;IAAE,IAAI,EAAE,aAAa,CAAA;CAAE,GACvB;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,QAAQ,EAAE,YAAY,EAAE,CAAC;IAAC,SAAS,CAAC,EAAE,OAAO,CAAA;CAAE,GACpE;IAAE,IAAI,EAAE,YAAY,CAAA;CAAE,GACtB;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,OAAO,EAAE,YAAY,CAAC;IAAC,WAAW,EAAE,iBAAiB,EAAE,CAAA;CAAE,GAC7E;IAAE,IAAI,EAAE,eAAe,CAAC;IAAC,OAAO,EAAE,YAAY,CAAA;CAAE,GAChD;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,OAAO,EAAE,YAAY,CAAA;CAAE,GACjD;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,OAAO,EAAE,YAAY,CAAA;CAAE,GAC9C;IAAE,IAAI,EAAE,sBAAsB,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,OAAO,CAAA;CAAE,GACrF;IACE,IAAI,EAAE,uBAAuB,CAAC;IAC9B,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,OAAO,CAAC;IACd,aAAa,EAAE,OAAO,CAAC;CACxB,GACD;IACE,IAAI,EAAE,oBAAoB,CAAC;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,OAAO,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;CAClB,CAAC;AAEN,MAAM,WAAW,aAAa;IAC5B,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC;IACnB,iBAAiB,CAAC,EAAE,OAAO,GAAG,UAAU,CAAC;CAC1C;AAED,MAAM,WAAW,oBAAoB;IACnC,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IAC5B,2IAA2I;IAC3I,QAAQ,CAAC,WAAW,CAAC,EAAE,OAAO,CAAC;IAC/B,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7D,8HAA8H;IAC9H,KAAK,CAAC,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACxD,4GAA4G;IAC5G,QAAQ,CAAC,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC3D,SAAS,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,UAAU,KAAK,IAAI,GAAG,MAAM,IAAI,CAAC;IAC7D,4JAA4J;IAC5J,OAAO,IAAI,IAAI,CAAC;CACjB;AAED,MAAM,WAAW,uBAAuB;IACtC,SAAS,EAAE,oBAAoB,CAAC;CACjC;AAED,2FAA2F;AAC3F,MAAM,WAAW,mBAAmB;IAClC,YAAY,CAAC,EAAE,uBAAuB,CAAC;IACvC,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,eAAO,MAAM,WAAW,kCAAkC,CAAC;AAE3D,MAAM,WAAW,uBAAuB;IACtC,qGAAqG;IACrG,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,8FAA8F;IAC9F,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB;AAED,MAAM,WAAW,+BAA+B;IAC9C,cAAc,EAAE,OAAO,CAAC;IACxB,aAAa,EAAE,OAAO,CAAC;CACxB;AAED,wBAAgB,aAAa,CAAC,MAAM,CAAC,EAAE,uBAAuB,GAAG,+BAA+B,CAO/F;AA+JD,wBAAgB,YAAY,CAC1B,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,EACjC,SAAS,EAAE,OAAO,EAClB,KAAK,EAAE;IAAE,IAAI,EAAE,MAAM,GAAG,SAAS,CAAC;IAAC,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;IAAC,cAAc,EAAE,OAAO,CAAA;CAAE,GAC1F,IAAI,CAMN;AAGD,wBAAgB,eAAe,CAC7B,IAAI,EAAE,IAAI,EACV,aAAa,EAAE,YAAY,EAAE,EAC7B,cAAc,EAAE,OAAO,GACtB,IAAI,CAUN;AAGD,wBAAgB,gBAAgB,CAC9B,IAAI,EAAE,IAAI,EACV,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,OAAO,CAAC,EAAE,MAAM,CAAA;CAAE,EAClD,KAAK,CAAC,EAAE,OAAO,GACd,IAAI,CAYN;AAED,wBAAgB,WAAW,CACzB,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,EACjC,SAAS,EAAE,OAAO,EAClB,OAAO,EAAE,gBAAgB,GACxB,IAAI,CASN;AAED,wBAAgB,YAAY,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,gBAAgB,EAAE,cAAc,EAAE,OAAO,GAAG,IAAI,CA0BjG;AA6BD,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,MAAM,CAkB5E;AAED,wBAAgB,YAAY,CAC1B,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,EACjC,SAAS,EAAE,OAAO,EAClB,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,OAAO,EACb,aAAa,EAAE,OAAO,GACrB,IAAI,CA0BN;AAED,wBAAgB,aAAa,CAC3B,IAAI,EAAE,IAAI,EACV,MAAM,EAAE,OAAO,EACf,OAAO,EAAE,OAAO,EAChB,aAAa,EAAE,OAAO,GACrB,IAAI,CAeN;AAGD,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,IAAI,GAAG,SAAS,GAAG,IAAI,CAS9D;AA+DD;;;;;;;GAOG;AACH,wBAAgB,gCAAgC,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI,CAQrE;AAED,wBAAgB,uBAAuB,CAAC,GAAG,EAAE,OAAO,EAAE,MAAM,CAAC,EAAE,uBAAuB,GAAG,OAAO,CA4J/F"}