@cephalization/phoenix-insight 0.4.0 → 1.0.2

package/dist/snapshot/utils.js

@@ -1,112 +0,0 @@
-/**
- * Snapshot discovery utilities
- *
- * Functions for listing and finding snapshots in the local filesystem.
- */
-import * as fs from "node:fs/promises";
-import * as path from "node:path";
-import * as os from "node:os";
-/**
- * Get the base snapshots directory path
- */
-export function getSnapshotsDir() {
-    return path.join(os.homedir(), ".phoenix-insight", "snapshots");
-}
-/**
- * Parse a snapshot directory name to extract timestamp
- *
- * Directory names are in format: `<timestamp>-<random>` where timestamp is Date.now()
- * Example: "1704067200000-abc123" -> Date(2024-01-01T00:00:00.000Z)
- *
- * @param dirName - The directory name to parse
- * @returns The parsed timestamp as Date, or null if invalid
- */
-function parseSnapshotDirName(dirName) {
-    // Format: <timestamp>-<random>
-    const match = dirName.match(/^(\d+)-[\w]+$/);
-    if (!match || !match[1]) {
-        return null;
-    }
-    const timestamp = parseInt(match[1], 10);
-    if (isNaN(timestamp) || timestamp <= 0) {
-        return null;
-    }
-    const date = new Date(timestamp);
-    // Validate the date is reasonable (between year 2000 and year 3000)
-    // Use UTC year to avoid timezone issues
-    const year = date.getUTCFullYear();
-    if (year < 2000 || year > 3000) {
-        return null;
-    }
-    return date;
-}
-/**
- * List all available snapshots
- *
- * Scans the snapshots directory and returns information about each valid snapshot.
- * Results are sorted by timestamp descending (most recent first).
- *
- * @returns Array of snapshot info objects, sorted by timestamp descending
- */
-export async function listSnapshots() {
-    const snapshotsDir = getSnapshotsDir();
-    // Check if snapshots directory exists
-    try {
-        await fs.access(snapshotsDir);
-    }
-    catch {
-        // Directory doesn't exist - return empty array
-        return [];
-    }
-    // Read directory contents
-    let entries;
-    try {
-        entries = await fs.readdir(snapshotsDir);
-    }
-    catch {
-        // Cannot read directory - return empty array
-        return [];
-    }
-    // Filter and parse valid snapshot directories
-    const snapshots = [];
-    for (const entry of entries) {
-        const timestamp = parseSnapshotDirName(entry);
-        if (!timestamp) {
-            // Invalid directory name format - skip
-            continue;
-        }
-        const snapshotPath = path.join(snapshotsDir, entry, "phoenix");
-        // Verify the phoenix subdirectory exists
-        try {
-            const stat = await fs.stat(snapshotPath);
-            if (!stat.isDirectory()) {
-                continue;
-            }
-        }
-        catch {
-            // Phoenix subdirectory doesn't exist or can't be accessed - skip
-            continue;
-        }
-        snapshots.push({
-            path: snapshotPath,
-            timestamp,
-            id: entry,
-        });
-    }
-    // Sort by timestamp descending (most recent first)
-    snapshots.sort((a, b) => b.timestamp.getTime() - a.timestamp.getTime());
-    return snapshots;
-}
-/**
- * Get the latest (most recent) snapshot
- *
- * @returns The most recent snapshot info, or null if no snapshots exist
- */
-export async function getLatestSnapshot() {
-    const snapshots = await listSnapshots();
-    if (snapshots.length === 0) {
-        return null;
-    }
-    // First element is the most recent due to descending sort
-    return snapshots[0] ?? null;
-}

package/dist/tsconfig.esm.tsbuildinfo

@@ -1 +0,0 @@
-{"root":["../src/cli.ts","../src/index.ts","../src/progress.ts","../src/agent/index.ts","../src/commands/index.ts","../src/commands/px-fetch-more-spans.ts","../src/commands/px-fetch-more-trace.ts","../src/config/index.ts","../src/config/loader.ts","../src/config/schema.ts","../src/modes/index.ts","../src/modes/local.ts","../src/modes/sandbox.ts","../src/modes/types.ts","../src/observability/index.ts","../src/prompts/index.ts","../src/prompts/system.ts","../src/snapshot/client.ts","../src/snapshot/context.ts","../src/snapshot/datasets.ts","../src/snapshot/experiments.ts","../src/snapshot/index.ts","../src/snapshot/projects.ts","../src/snapshot/prompts.ts","../src/snapshot/spans.ts","../src/snapshot/utils.ts"],"version":"5.9.3"}

package/src/agent/index.ts

@@ -1,323 +0,0 @@
-/**
- * Phoenix Insight AI agent setup using Vercel AI SDK
- */
-
-import {
-  generateText,
-  streamText,
-  tool,
-  stepCountIs,
-  type GenerateTextResult,
-  type StreamTextResult,
-} from "ai";
-import { anthropic } from "@ai-sdk/anthropic";
-import { z } from "zod";
-import type { ExecutionMode } from "../modes/types.js";
-import { INSIGHT_SYSTEM_PROMPT } from "../prompts/system.js";
-import {
-  fetchMoreSpans,
-  fetchMoreTrace,
-  type FetchMoreSpansOptions,
-  type FetchMoreTraceOptions,
-} from "../commands/index.js";
-import type { PhoenixClient } from "@arizeai/phoenix-client";
-import { PhoenixClientError } from "../snapshot/client.js";
-
-/**
- * Configuration for the Phoenix Insight agent
- */
-export interface PhoenixInsightAgentConfig {
-  /** The execution mode (sandbox or local) */
-  mode: ExecutionMode;
-  /** Phoenix client instance */
-  client: PhoenixClient;
-  /** Maximum number of agent steps before stopping (default: 25) */
-  maxSteps?: number;
-}
-
-/**
- * Phoenix Insight Agent
- */
-export class PhoenixInsightAgent {
-  private mode: ExecutionMode;
-  private client: PhoenixClient;
-  private maxSteps: number;
-  private tools: Record<string, any> | null = null;
-  private model = anthropic("claude-sonnet-4-5");
-
-  constructor(config: PhoenixInsightAgentConfig) {
-    this.mode = config.mode;
-    this.client = config.client;
-    this.maxSteps = config.maxSteps || 25;
-  }
-
-  /**
-   * Initialize the agent tools
-   */
-  private async initializeTools(): Promise<Record<string, any>> {
-    if (this.tools) return this.tools;
-
-    // Get the bash tool from the execution mode
-    const bashTool = await this.mode.getBashTool();
-
-    // Store references in closure for the custom tools
-    const client = this.client;
-    const mode = this.mode;
-
-    // Create custom px-fetch-more-spans tool using AI SDK's tool function
-    const pxFetchMoreSpans = tool({
-      description:
-        "Fetch additional spans from Phoenix. Use when you need more span data than what's in the snapshot. You must provide both project name and optionally a limit.",
-      inputSchema: z.object({
-        project: z.string().describe("The project name"),
-        limit: z
-          .number()
-          .optional()
-          .describe("Number of spans to fetch (default: 500)"),
-        startTime: z
-          .string()
-          .optional()
-          .describe("Start time filter in ISO format"),
-        endTime: z
-          .string()
-          .optional()
-          .describe("End time filter in ISO format"),
-      }),
-      execute: async (params) => {
-        try {
-          const options: FetchMoreSpansOptions = {
-            project: params.project,
-            limit: params.limit || 500,
-            startTime: params.startTime,
-            endTime: params.endTime,
-          };
-
-          await fetchMoreSpans(client, mode, options);
-
-          return {
-            success: true,
-            message: `Fetched additional spans for project ${params.project}. Data saved to /phoenix/projects/${params.project}/spans/`,
-          };
-        } catch (error) {
-          return {
-            success: false,
-            error: error instanceof Error ? error.message : String(error),
-          };
-        }
-      },
-    });
-
-    // Create custom px-fetch-more-trace tool using AI SDK's tool function
-    const pxFetchMoreTrace = tool({
-      description:
-        "Fetch a specific trace by ID from Phoenix. Use when you need to examine a particular trace in detail. You must provide both the trace ID and the project name.",
-      inputSchema: z.object({
-        traceId: z.string().describe("The trace ID to fetch"),
-        project: z.string().describe("The project name to search in"),
-      }),
-      execute: async (params) => {
-        try {
-          const options: FetchMoreTraceOptions = {
-            traceId: params.traceId,
-            project: params.project,
-          };
-
-          await fetchMoreTrace(client, mode, options);
-
-          return {
-            success: true,
-            message: `Fetched trace ${params.traceId}. Data saved to /phoenix/traces/${params.traceId}/`,
-          };
-        } catch (error) {
-          return {
-            success: false,
-            error: error instanceof Error ? error.message : String(error),
-          };
-        }
-      },
-    });
-
-    this.tools = {
-      bash: bashTool,
-      px_fetch_more_spans: pxFetchMoreSpans,
-      px_fetch_more_trace: pxFetchMoreTrace,
-    };
-
-    return this.tools;
-  }
-
-  /**
-   * Generate a response for a user query
-   */
-  async generate(
-    userQuery: string,
-    options?: {
-      onStepFinish?: (step: any) => void;
-    }
-  ): Promise<GenerateTextResult<any, any>> {
-    let tools;
-    try {
-      tools = await this.initializeTools();
-    } catch (error) {
-      throw new Error(
-        `Failed to initialize agent tools: ${error instanceof Error ? error.message : String(error)}`
-      );
-    }
-
-    try {
-      const result = await generateText({
-        model: this.model,
-        system: INSIGHT_SYSTEM_PROMPT,
-        prompt: userQuery,
-        tools,
-        stopWhen: stepCountIs(this.maxSteps),
-        onStepFinish: options?.onStepFinish,
-        experimental_telemetry: {
-          isEnabled: true,
-        },
-      });
-
-      return result;
-    } catch (error) {
-      // Check for specific AI SDK errors
-      if (error instanceof Error) {
-        if (error.message.includes("rate limit")) {
-          throw new Error(
-            "AI model rate limit exceeded. Please wait and try again."
-          );
-        }
-        if (error.message.includes("timeout")) {
-          throw new Error("AI model request timed out. Please try again.");
-        }
-        if (
-          error.message.includes("authentication") ||
-          error.message.includes("API key")
-        ) {
-          throw new Error(
-            "AI model authentication failed. Check your API key configuration."
-          );
-        }
-      }
-
-      throw new Error(
-        `AI generation failed: ${error instanceof Error ? error.message : String(error)}`
-      );
-    }
-  }
-
-  /**
-   * Stream a response for a user query
-   */
-  async stream(
-    userQuery: string,
-    options?: {
-      onStepFinish?: (step: any) => void;
-    }
-  ): Promise<StreamTextResult<any, any>> {
-    let tools;
-    try {
-      tools = await this.initializeTools();
-    } catch (error) {
-      throw new Error(
-        `Failed to initialize agent tools: ${error instanceof Error ? error.message : String(error)}`
-      );
-    }
-
-    try {
-      const result = streamText({
-        model: this.model,
-        system: INSIGHT_SYSTEM_PROMPT,
-        prompt: userQuery,
-        tools,
-        stopWhen: stepCountIs(this.maxSteps),
-        onStepFinish: options?.onStepFinish,
-        experimental_telemetry: {
-          isEnabled: true,
-        },
-      });
-
-      return result;
-    } catch (error) {
-      // Check for specific AI SDK errors
-      if (error instanceof Error) {
-        if (error.message.includes("rate limit")) {
-          throw new Error(
-            "AI model rate limit exceeded. Please wait and try again."
-          );
-        }
-        if (error.message.includes("timeout")) {
-          throw new Error("AI model request timed out. Please try again.");
-        }
-        if (
-          error.message.includes("authentication") ||
-          error.message.includes("API key")
-        ) {
-          throw new Error(
-            "AI model authentication failed. Check your API key configuration."
-          );
-        }
-      }
-
-      throw new Error(
-        `AI streaming failed: ${error instanceof Error ? error.message : String(error)}`
-      );
-    }
-  }
-
-  /**
-   * Clean up resources
-   */
-  async cleanup(): Promise<void> {
-    await this.mode.cleanup();
-  }
-}
-
-/**
- * Creates a Phoenix Insight agent
- */
-export async function createInsightAgent(
-  config: PhoenixInsightAgentConfig
-): Promise<PhoenixInsightAgent> {
-  return new PhoenixInsightAgent(config);
-}
-
-/**
- * Run a query with the Phoenix Insight agent
- */
-export async function runQuery(
-  agent: PhoenixInsightAgent,
-  userQuery: string,
-  options?: {
-    onStepFinish?: (step: any) => void;
-    stream?: boolean;
-  }
-): Promise<GenerateTextResult<any, any> | StreamTextResult<any, any>> {
-  const { stream = false, ...callbacks } = options || {};
-
-  if (stream) {
-    return await agent.stream(userQuery, callbacks);
-  } else {
-    return await agent.generate(userQuery, callbacks);
-  }
-}
-
-/**
- * Create and run a one-shot query
- */
-export async function runOneShotQuery(
-  config: PhoenixInsightAgentConfig,
-  userQuery: string,
-  options?: {
-    onStepFinish?: (step: any) => void;
-    stream?: boolean;
-  }
-): Promise<GenerateTextResult<any, any> | StreamTextResult<any, any>> {
-  const agent = await createInsightAgent(config);
-
-  try {
-    const result = await runQuery(agent, userQuery, options);
-    return result;
-  } finally {
-    await agent.cleanup();
-  }
-}