elasticdash-sdk 0.2.0

package/src/http.ts

@@ -0,0 +1,13 @@
+/**
+ * Lightweight entry point for use inside framework apps (Next.js, Remix, etc.).
+ * Only exports what is needed for HTTP workflow mode instrumentation.
+ * Import from 'elasticdash-sdk/http' to avoid bundling server-only deps.
+ */
+export { setHttpRunContext, initHttpRunContext, getHttpRunContext } from './interceptors/telemetry-push.js'
+export { wrapTool } from './interceptors/tool.js'
+export { wrapAI } from './interceptors/workflow-ai.js'
+export { wrapDB, wrapPgClient, wrapKnex, wrapMongoCollection, wrapRedisClient } from './interceptors/db.js'
+export { interceptFetch, restoreFetch, readVercelAIStream } from './interceptors/http.js'
+export type { VercelAIStreamResult } from './interceptors/http.js'
+export { initObservability, shutdownObservability, startTrace, endTrace, wrapWorkflow } from './observability.js'
+export { recordToolCall } from './tracing.js'

package/src/index.ts

@@ -0,0 +1,138 @@
+// Public API surface for programmatic use
+
+// Test framework
+export { aiTest, beforeAll, afterAll, beforeEach, afterEach, getRegistry, clearRegistry } from './core/registry.js'
+export { runFiles } from './runner.js'
+export { reportResults } from './reporter.js'
+export { registerMatchers } from './matchers/index.js'
+export { expect } from './matchers/index.js'
+export type { TestResult, FileResult, RunnerOptions } from './runner.js'
+
+// Trace adapter
+export { createTraceHandle, startTraceSession, setCurrentTrace, getCurrentTrace } from './trace-adapter/context.js'
+export type { AITestContext, TraceHandle, LLMStep, ToolCall, CustomStep, TraceStep, RunnerHooks } from './trace-adapter/context.js'
+
+// Workflow capture & replay
+export { TraceRecorder, setCaptureContext, getCaptureContext } from './capture/recorder.js'
+export { ReplayController } from './capture/replay.js'
+export type { WorkflowEvent, WorkflowEventType, WorkflowTrace } from './capture/event.js'
+export type { CaptureContext } from './capture/recorder.js'
+
+// Tool & AI interceptors
+export { wrapTool } from './interceptors/tool.js'
+export { wrapAI } from './interceptors/workflow-ai.js'
+
+// Tool registry (rerunnable tools defined anywhere in the project)
+export { edTool, defineTool, getRegisteredTool, getRegisteredTools, clearToolRegistry } from './tool-registry.js'
+export type { RegisteredTool } from './tool-registry.js'
+
+// HTTP run context (ALS + global fallback for streaming frameworks)
+export {
+  setHttpRunContext,
+  initHttpRunContext,
+  getHttpRunContext,
+  getHttpFrozenEvent,
+  getHttpPromptMock,
+  pushTelemetryEvent,
+  tryAutoInitHttpContext,
+  runInHttpContext,
+  runWithInitializedHttpContext,
+  setGlobalHttpContext,
+  clearGlobalHttpContext,
+} from './interceptors/telemetry-push.js'
+export type { HttpRunContext } from './interceptors/telemetry-push.js'
+
+// Mock resolution (standalone utilities — work without ALS)
+export {
+  resolveMock,
+  resolveAIMock,
+  resolvePromptMock,
+  resolveUserPromptMock,
+  lookupMockEntry,
+  applyUserPromptMocks,
+  applySystemPromptMocks,
+} from './internals/mock-resolver.js'
+export type {
+  AIMockEntry,
+  AIMockConfig,
+  UserPromptMockEntry,
+  UserPromptMockConfig,
+  SystemPromptMockEntry,
+  SystemPromptMockConfig,
+} from './internals/mock-resolver.js'
+
+// Database interceptors
+export { wrapDB, wrapPgClient, wrapKnex, wrapMongoCollection, wrapRedisClient } from './interceptors/db.js'
+export { installDBAutoInterceptor, uninstallDBAutoInterceptor } from './interceptors/db-auto.js'
+
+// HTTP & fetch interceptors
+export { interceptFetch, restoreFetch, readVercelAIStream } from './interceptors/http.js'
+export type { VercelAIStreamResult } from './interceptors/http.js'
+
+// Side-effect interceptors
+export { interceptRandom, restoreRandom, interceptDateNow, restoreDateNow } from './interceptors/side-effects.js'
+
+// AI interceptor (monkey-patch based)
+export { installAIInterceptor, uninstallAIInterceptor } from './interceptors/ai-interceptor.js'
+
+// Observability
+export { initObservability, shutdownObservability, startTrace, endTrace, wrapWorkflow } from './observability.js'
+export type { ObservabilityOptions, ObservabilityHandle } from './observability.js'
+export { TelemetryBatcher } from './telemetry-batcher.js'
+export type { TelemetryBatcherOptions, TriggerSignal, TriggerStep } from './telemetry-batcher.js'
+export type { ObservabilityContext } from './interceptors/telemetry-push.js'
+export { checkToolAvailability, checkAIAvailability } from './portal-executor.js'
+export type { AvailabilityResult } from './portal-executor.js'
+
+// Socket connector (real-time backend communication)
+export { connectToBackend, disconnectFromBackend } from './socket-connector.js'
+export type { SocketConnectorOptions } from './socket-connector.js'
+
+// LLM proxy
+export { startLLMProxy, fetchCapturedTrace } from './proxy/llm-capture.js'
+
+// Tracing
+export { recordToolCall } from './tracing.js'
+export { safeRecordToolCall, isWorker } from './internals/conditional-recorder.js'
+
+// Workflow runner
+export { runWorkflow } from './workflow-runner.js'
+export type { RunWorkflowOptions, WorkflowRunResult } from './workflow-runner.js'
+
+// Agent mid-trace replay
+export { serializeAgentState, deserializeAgentState, extractTaskOutputs, resolveTaskInput } from './core/agent-state.js'
+export type { AgentTask, AgentPlan, AgentState, AgentTaskStatus, AgentPlanStatus } from './types/agent.js'
+
+// Portal (remote rerun queue)
+export { startPortalServer } from './portal-server.js'
+export { executePortalTask } from './portal-executor.js'
+export type { PortalTask, PortalTaskResult, PortalServerOptions, PortalServerHandle, PortalStatus } from './types/portal.js'
+
+// Trace capture & test file recognition (Phase 1)
+export { discoverTestFiles } from './ci/test-discovery.js'
+export { defineTest } from './ci/test-registry.js'
+export { loadTests } from './ci/test-loader.js'
+export type { TestDefinition, TestTarget, TestBenchmarks } from './ci/test-registry.js'
+export type { ValidatedTest, ValidationError } from './ci/test-loader.js'
+export type { DiskTrace, DiskTraceStep } from './ci/trace-schema.js'
+
+// Test runner & upload (Phase 3)
+export { runEdTests } from './ci/ed-runner.js'
+export type { EdTestRunOptions, EdTestResult, EdTestRunResult } from './ci/ed-runner.js'
+export { uploadResults, buildUploadPayload } from './ci/upload-client.js'
+export type { UploadPayload, UploadTestResult } from './ci/upload-client.js'
+
+// CI runner (programmatic API)
+export { runCI } from './ci/runner.js'
+export { fetchTestGroups, submitTestRun, createBatch, fetchEvaluatorConfig } from './ci/api-client.js'
+export { detectGitInfo } from './ci/git-info.js'
+export type { CIRunConfig, CIRunSummary, CITestResult, CISingleRunResult, CIExpectationResult } from './ci/types.js'
+export type { GitInfo } from './ci/git-info.js'
+
+// ─── Eager auto-init ────────────────────────────────────────
+// When ELASTICDASH_API_KEY is set, automatically initialise observability mode
+// at import time so the socket connection is established immediately (e.g. for
+// onboarding detection). tryAutoInitHttpContext is idempotent and deduplicates
+// concurrent calls, so this is safe even if interceptors trigger it again later.
+import { tryAutoInitHttpContext } from './interceptors/telemetry-push.js'
+tryAutoInitHttpContext().catch(() => {})