@apo-ai/sdk 0.2.0

package/dist/types-C-nrDEVf.d.ts

@@ -0,0 +1,140 @@
+type ParameterOverrides = Record<string, unknown>;
+interface ClientOptions {
+    parameterOverrides?: ParameterOverrides;
+    runEvals?: boolean;
+}
+interface IngestionEvent {
+    id: string;
+    timestamp: Date;
+    type: "run-create" | "call-create" | "call-update" | "score-create";
+    body: Record<string, unknown>;
+}
+interface CreateTraceParams {
+    id?: string;
+    project: string;
+    task_id?: string;
+    flow_name?: string;
+    version?: string;
+    user_id?: string;
+    session_id?: string;
+    environment?: string;
+    external_id?: string;
+    tags?: string[];
+    run_metadata?: Record<string, unknown>;
+}
+interface CreateSpanParams {
+    id?: string;
+    project: string;
+    task_id: string;
+    run_id?: string;
+    parent_call_id?: string;
+    flow_name?: string;
+    step_name?: string;
+    step_index?: number;
+    version?: string;
+    model?: string;
+    input?: Record<string, unknown>;
+    messages?: Record<string, unknown>[];
+    output?: Record<string, unknown>;
+    observation_type?: "GENERATION" | "SPAN" | "TOOL" | "CHAIN" | "RETRIEVER" | "EVALUATOR" | "EMBEDDING" | "GUARDRAIL" | "AGENT";
+    level?: "DEBUG" | "DEFAULT" | "WARNING" | "ERROR";
+    user_id?: string;
+    metadata?: Record<string, unknown>;
+}
+interface EndSpanParams {
+    id: string;
+    output?: Record<string, unknown>;
+    latency_ms?: number;
+    prompt_tokens?: number;
+    completion_tokens?: number;
+    status_message?: string;
+    level?: "DEBUG" | "DEFAULT" | "WARNING" | "ERROR";
+    metadata?: Record<string, unknown>;
+}
+interface TraceRunRootSpanOptions extends Omit<CreateSpanParams, "project" | "run_id" | "parent_call_id"> {
+}
+interface TraceRunOptions extends Omit<CreateTraceParams, "project"> {
+    project?: string;
+    rootSpan?: TraceRunRootSpanOptions;
+}
+interface TraceStepOptions extends Omit<CreateSpanParams, "project" | "task_id" | "run_id" | "output"> {
+    task_id?: string;
+    summarize?: (result: unknown) => Record<string, unknown> | undefined;
+}
+interface TraceEventOptions extends Omit<CreateSpanParams, "project" | "task_id" | "run_id"> {
+    task_id?: string;
+    latency_ms?: number;
+}
+interface TraceRunContext {
+    runId: string;
+    rootSpanId: string;
+    step<T>(options: TraceStepOptions, fn: (spanId: string) => Promise<T>): Promise<T>;
+    recordEvent(options: TraceEventOptions): string;
+    endRoot(params?: Omit<EndSpanParams, "id">): void;
+    traceTool<T>(name: string, params: Record<string, unknown>, fn: () => Promise<T>): Promise<T>;
+    traceRetriever<T>(query: string, fn: () => Promise<T>): Promise<T>;
+    traceChain<T>(name: string, fn: () => Promise<T>): Promise<T>;
+    traceAgent<T>(name: string, fn: () => Promise<T>): Promise<T>;
+    traceGuardrail<T>(name: string, fn: () => Promise<T>): Promise<T>;
+    traceEmbedding<T>(model: string, input: unknown, fn: () => Promise<T>): Promise<T>;
+    score(params: CreateScoreParams): Promise<void>;
+}
+interface CreateScoreParams {
+    name: string;
+    value: number | string | boolean;
+    dataType?: "NUMERIC" | "CATEGORICAL" | "BOOLEAN";
+    source?: "API" | "EVAL" | "ANNOTATION";
+    configId?: number;
+    comment?: string;
+    observationId?: string;
+}
+interface ObserveOptions {
+    /** Unique ID for this observation (trace or span) */
+    id?: string;
+    /** Project name for grouping traces */
+    project?: string;
+    /** Flow name (alternative to project) */
+    flowName?: string;
+    /** Step name for spans */
+    stepName?: string;
+    /** Observation name (defaults to method name if not provided) */
+    name?: string;
+    /** User ID for tracking */
+    userId?: string;
+    /** Session ID for session tracking */
+    sessionId?: string;
+    /** Environment (e.g., "production", "development") */
+    environment?: string;
+    /** External ID for linking to external systems */
+    externalId?: string;
+    /** Tags for filtering and grouping */
+    tags?: string[];
+    /** Metadata for additional context */
+    metadata?: Record<string, unknown>;
+    /** Model name (for spans) */
+    model?: string;
+    /** Explicit input data (overrides automatic extraction) */
+    input?: Record<string, unknown>;
+    /** Messages array (for chat completions) */
+    messages?: Record<string, unknown>[];
+    /** Observation type (Langfuse-style) */
+    observationType?: "GENERATION" | "SPAN" | "TOOL" | "CHAIN" | "RETRIEVER" | "EVALUATOR" | "EMBEDDING" | "GUARDRAIL" | "AGENT";
+    /** Shorthand for observation type */
+    as?: "tool" | "retriever" | "chain" | "agent" | "guardrail" | "embedding";
+    /** Observation level */
+    level?: "DEBUG" | "DEFAULT" | "WARNING" | "ERROR";
+    /** Whether to capture output (default: true) */
+    captureOutput?: boolean;
+}
+interface ObservationContext {
+    /** Trace ID (run ID) */
+    traceId: string;
+    /** Current span ID */
+    spanId: string;
+    /** Flow name */
+    flowName: string;
+    /** Project name */
+    projectName: string;
+}
+
+export type { CreateSpanParams as C, EndSpanParams as E, IngestionEvent as I, ObservationContext as O, ParameterOverrides as P, TraceRunContext as T, TraceRunOptions as a, ClientOptions as b, CreateScoreParams as c, CreateTraceParams as d, ObserveOptions as e, TraceEventOptions as f, TraceStepOptions as g };

package/package.json

@@ -0,0 +1,92 @@
+{
+  "name": "@apo-ai/sdk",
+  "version": "0.2.0",
+  "description": "TypeScript SDK for testing and observing AI agents with Apo",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist",
+    "LICENSE",
+    "README.md"
+  ],
+  "engines": {
+    "node": ">=20"
+  },
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/samikuikka/apo.git",
+    "directory": "packages/sdk"
+  },
+  "homepage": "https://github.com/samikuikka/apo#readme",
+  "bugs": {
+    "url": "https://github.com/samikuikka/apo/issues"
+  },
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "default": "./dist/index.js"
+    },
+    "./otel": {
+      "types": "./dist/otel/index.d.ts",
+      "import": "./dist/otel/index.js",
+      "default": "./dist/otel/index.js"
+    },
+    "./agent-task": {
+      "types": "./dist/agent-task/public.d.ts",
+      "import": "./dist/agent-task/public.js",
+      "default": "./dist/agent-task/public.js"
+    },
+    "./agent-task/integrations/ai-sdk": {
+      "types": "./dist/agent-task/integrations/ai-sdk.d.ts",
+      "import": "./dist/agent-task/integrations/ai-sdk.js",
+      "default": "./dist/agent-task/integrations/ai-sdk.js"
+    },
+    "./agent-task/integrations/openai": {
+      "types": "./dist/agent-task/integrations/openai.d.ts",
+      "import": "./dist/agent-task/integrations/openai.js",
+      "default": "./dist/agent-task/integrations/openai.js"
+    },
+    "./agent-task/integrations/anthropic": {
+      "types": "./dist/agent-task/integrations/anthropic.d.ts",
+      "import": "./dist/agent-task/integrations/anthropic.js",
+      "default": "./dist/agent-task/integrations/anthropic.js"
+    }
+  },
+  "publishConfig": {
+    "access": "public",
+    "registry": "https://registry.npmjs.org/"
+  },
+  "devDependencies": {
+    "@types/node": "^20.0.0",
+    "@vitest/coverage-v8": "^4.1.9",
+    "esbuild": "^0.21.5",
+    "publint": "^0.3.22",
+    "tsup": "^8.5.1",
+    "typescript": "^5.0.0",
+    "vitest": "^4.1.9",
+    "zod": "^3.22.0"
+  },
+  "dependencies": {
+    "@opentelemetry/api": "^1.9.1",
+    "@opentelemetry/context-async-hooks": "^2.9.0",
+    "@opentelemetry/core": "^2.9.0",
+    "@opentelemetry/exporter-trace-otlp-http": "^0.220.0",
+    "@opentelemetry/resources": "^2.9.0",
+    "@opentelemetry/sdk-trace-base": "^2.9.0",
+    "@opentelemetry/semantic-conventions": "^1.43.0",
+    "effect": "^3.19.12"
+  },
+  "scripts": {
+    "typecheck": "tsc --noEmit && tsc --noEmit -p tsconfig.type-tests.json",
+    "build": "tsup",
+    "build:watch": "tsup --watch",
+    "build:agent-task-runtime": "node scripts/build-agent-task-runtime.mjs",
+    "package:check": "node scripts/verify-package.mjs",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "test:coverage": "vitest run --coverage"
+  }
+}