index.ts

// Main class
export { LLMock } from "./llmock.js";

// Server
export { createServer, type ServerInstance } from "./server.js";

// Fixture loading
export {
  loadFixtureFile,
  loadFixturesFromDir,
  validateFixtures,
  normalizeResponse,
} from "./fixture-loader.js";
export type { ValidationResult } from "./fixture-loader.js";

// Logger
export { Logger } from "./logger.js";
export type { LogLevel } from "./logger.js";

// Journal
export { Journal, DEFAULT_TEST_ID } from "./journal.js";

// Router
export { matchFixture, matchFixtureDiagnostic, getTextContent } from "./router.js";
export type { MatchFixtureDiagnostic } from "./router.js";

// Provider handlers
export {
  handleResponses,
  buildTextStreamEvents,
  buildToolCallStreamEvents,
  buildContentWithToolCallsStreamEvents,
} from "./responses.js";
export type { ResponsesSSEEvent } from "./responses.js";
export { handleMessages } from "./messages.js";
export { handleGemini } from "./gemini.js";
export { handleEmbeddings } from "./embeddings.js";
export { handleBedrock, bedrockToCompletionRequest, handleBedrockStream } from "./bedrock.js";

// Bedrock Converse
export {
  handleConverse,
  handleConverseStream,
  converseToCompletionRequest,
} from "./bedrock-converse.js";

// Gemini Interactions
export {
  handleGeminiInteractions,
  geminiInteractionsToCompletionRequest,
} from "./gemini-interactions.js";

// AWS Event Stream
export {
  encodeEventStreamFrame,
  encodeEventStreamMessage,
  writeEventStream,
} from "./aws-event-stream.js";

// Metrics
export { createMetricsRegistry, normalizePathLabel } from "./metrics.js";
export type { MetricsRegistry } from "./metrics.js";

// NDJSON
export { writeNDJSONStream } from "./ndjson-writer.js";
export type { NDJSONStreamOptions } from "./ndjson-writer.js";

// Ollama
export { handleOllama, handleOllamaGenerate, ollamaToCompletionRequest } from "./ollama.js";

// Cohere
export { handleCohere, cohereToCompletionRequest } from "./cohere.js";

// Service mocks
export { handleSearch } from "./search.js";
export type { SearchResult, SearchFixture } from "./search.js";
export { handleRerank } from "./rerank.js";
export type { RerankResult, RerankFixture } from "./rerank.js";
export { handleModeration } from "./moderation.js";
export type { ModerationResult, ModerationFixture } from "./moderation.js";
export type { ServiceFixtures } from "./server.js";

// WebSocket
export { WebSocketConnection, upgradeToWebSocket, computeAcceptKey } from "./ws-framing.js";
export { handleWebSocketResponses } from "./ws-responses.js";
export { handleWebSocketRealtime } from "./ws-realtime.js";
export { handleWebSocketGeminiLive } from "./ws-gemini-live.js";

// Multimedia handlers
export { handleImages } from "./images.js";
export { handleSpeech } from "./speech.js";
export { handleTranscription } from "./transcription.js";
export { handleVideoCreate, handleVideoStatus, VideoStateMap } from "./video.js";
export {
  handleOpenRouterVideoCreate,
  handleOpenRouterVideoStatus,
  handleOpenRouterVideoContent,
  handleOpenRouterVideoModels,
  OpenRouterVideoJobMap,
  OPENROUTER_VIDEO_MAX_ENTRIES,
  OPENROUTER_VIDEO_DEFAULT_MAX_CONTENT_BYTES,
} from "./openrouter-video.js";
export {
  handleVeoVideoCreate,
  handleVeoVideoStatus,
  VeoVideoJobMap,
  VEO_VIDEO_MAX_ENTRIES,
} from "./veo-video.js";
export {
  handleGrokVideoCreate,
  handleGrokVideoStatus,
  GrokVideoJobMap,
  GROK_VIDEO_MAX_ENTRIES,
} from "./grok-video.js";
export { handleElevenLabsAudio } from "./elevenlabs-audio.js";
export { handleFalQueue } from "./fal-audio.js";
export { handleFal, FalQueueStateMap } from "./fal.js";

// Helpers
export {
  flattenHeaders,
  generateId,
  generateToolCallId,
  generateMessageId,
  generateToolUseId,
  buildTextChunks,
  buildToolCallChunks,
  buildContentWithToolCallsChunks,
  buildTextCompletion,
  buildToolCallCompletion,
  buildContentWithToolCallsCompletion,
  extractOverrides,
  isTextResponse,
  isToolCallResponse,
  isContentWithToolCallsResponse,
  resolveFixtureBlocks,
  isErrorResponse,
  isEmbeddingResponse,
  isImageResponse,
  isAudioResponse,
  isTranscriptionResponse,
  isVideoResponse,
  generateDeterministicEmbedding,
  buildEmbeddingResponse,
  FORMAT_TO_CONTENT_TYPE,
  formatToMime,
  strictNoMatchMessage,
  strictNoMatchLogLine,
} from "./helpers.js";
export type { EmbeddingAPIResponse } from "./helpers.js";

// Interruption
export { createInterruptionSignal } from "./interruption.js";
export type { InterruptionControl } from "./interruption.js";

// SSE
export { writeSSEStream, writeErrorResponse, delay, calculateDelay } from "./sse-writer.js";
export type { StreamOptions, ErrorResponseOptions } from "./sse-writer.js";

// Chaos
export { evaluateChaos, applyChaos } from "./chaos.js";
export type { ChaosAction } from "./types.js";

// Recorder
export { proxyAndRecord } from "./recorder.js";

// URL
export { resolveUpstreamUrl } from "./url.js";

// Stream Collapse
export {
  collapseOpenAISSE,
  collapseAnthropicSSE,
  collapseGeminiSSE,
  collapseGeminiInteractionsSSE,
  collapseOllamaNDJSON,
  collapseCohereSSE,
  collapseBedrockEventStream,
  collapseStreamingResponse,
  MAX_COLLAPSE_STRING_LENGTH,
  setCollapseStringLimitForTests,
} from "./stream-collapse.js";
export type { CollapseResult } from "./stream-collapse.js";

// Mountable
export type { Mountable } from "./types.js";

// MCP
export { MCPMock } from "./mcp-mock.js";
export type {
  MCPMockOptions,
  MCPToolDefinition,
  MCPResourceDefinition,
  MCPPromptDefinition,
  MCPContent,
  MCPResourceContent,
  MCPPromptResult,
  MCPSession,
} from "./mcp-types.js";

// Vector
export { VectorMock } from "./vector-mock.js";
export type {
  VectorMockOptions,
  VectorCollection,
  VectorEntry,
  QueryResult,
  VectorQuery,
  QueryHandler,
} from "./vector-types.js";

// A2A
export { A2AMock } from "./a2a-mock.js";
export type {
  A2AMockOptions,
  A2AAgentDefinition,
  A2APart,
  A2AArtifact,
  A2ATaskResponse,
  A2AStreamEvent,
  A2ATask,
  A2AMessage,
  A2ARole,
  A2ATaskState,
} from "./a2a-types.js";

// AG-UI
export { AGUIMock } from "./agui-mock.js";
export {
  proxyAndRecordAGUI,
  DEFAULT_AGUI_RECORD_BUFFER_BYTES,
  AGUI_RECORD_BUFFER_HARD_CEILING,
  setAGUIRecordBufferCeilingForTests,
} from "./agui-recorder.js";
export type {
  AGUIMockOptions,
  AGUIRunAgentInput,
  AGUIMessage,
  AGUIMessageContentPart,
  AGUIMessageRole,
  AGUIToolDefinition,
  AGUIToolCall,
  AGUIEvent,
  AGUIEventType,
  AGUIFixture,
  AGUIFixtureMatch,
  AGUIRecordConfig,
  // Base event
  AGUIBaseEvent,
  // Lifecycle events
  AGUIRunStartedEvent,
  AGUIRunFinishedEvent,
  AGUIRunErrorEvent,
  AGUIStepStartedEvent,
  AGUIStepFinishedEvent,
  // Text message events
  AGUITextMessageStartEvent,
  AGUITextMessageContentEvent,
  AGUITextMessageEndEvent,
  AGUITextMessageChunkEvent,
  AGUITextMessageRole,
  // Tool call events
  AGUIToolCallStartEvent,
  AGUIToolCallArgsEvent,
  AGUIToolCallEndEvent,
  AGUIToolCallChunkEvent,
  AGUIToolCallResultEvent,
  // State events
  AGUIStateSnapshotEvent,
  AGUIStateDeltaEvent,
  AGUIMessagesSnapshotEvent,
  // Activity events
  AGUIActivitySnapshotEvent,
  AGUIActivityDeltaEvent,
  // Reasoning events
  AGUIReasoningStartEvent,
  AGUIReasoningMessageStartEvent,
  AGUIReasoningMessageContentEvent,
  AGUIReasoningMessageEndEvent,
  AGUIReasoningMessageChunkEvent,
  AGUIReasoningEndEvent,
  AGUIReasoningEncryptedValueEvent,
  AGUIReasoningEncryptedValueSubtype,
  // Special events
  AGUIRawEvent,
  AGUICustomEvent,
  // Deprecated thinking events
  AGUIThinkingStartEvent,
  AGUIThinkingEndEvent,
  AGUIThinkingTextMessageStartEvent,
  AGUIThinkingTextMessageContentEvent,
  AGUIThinkingTextMessageEndEvent,
  // Interrupt/Resume
  AGUIInterrupt,
  AGUIResumeEntry,
  AGUIRunFinishedOutcome,
} from "./agui-types.js";
export type { AGUIBuildOpts } from "./agui-handler.js";
export { matchesFixture as matchesAGUIFixture } from "./agui-handler.js";
export {
  buildTextResponse as buildAGUITextResponse,
  buildTextChunkResponse as buildAGUITextChunkResponse,
  buildToolCallResponse as buildAGUIToolCallResponse,
  buildStateUpdate as buildAGUIStateUpdate,
  buildStateDelta as buildAGUIStateDelta,
  buildMessagesSnapshot as buildAGUIMessagesSnapshot,
  buildReasoningResponse as buildAGUIReasoningResponse,
  buildActivityResponse as buildAGUIActivityResponse,
  buildErrorResponse as buildAGUIErrorResponse,
  buildStepWithText as buildAGUIStepWithText,
  buildCompositeResponse as buildAGUICompositeResponse,
  buildActivityDelta as buildAGUIActivityDelta,
  buildToolCallChunk as buildAGUIToolCallChunk,
  buildRawEvent as buildAGUIRawEvent,
  buildCustomEvent as buildAGUICustomEvent,
  buildReasoningChunk as buildAGUIReasoningChunk,
  buildReasoningEncryptedValue as buildAGUIReasoningEncryptedValue,
  extractLastUserMessage as extractAGUILastUserMessage,
  getLastMessageIfToolResult as getAGUILastMessageIfToolResult,
  findFixture as findAGUIFixture,
  writeAGUIEventStream,
} from "./agui-handler.js";

// JSON-RPC
export { createJsonRpcDispatcher } from "./jsonrpc.js";
export type { JsonRpcResponse, MethodHandler, JsonRpcDispatcherOptions } from "./jsonrpc.js";

// Config loader
export { loadConfig, startFromConfig } from "./config-loader.js";
export type { AimockConfig } from "./config-loader.js";

// Suite
export { createMockSuite } from "./suite.js";
export type { MockSuite, MockSuiteOptions } from "./suite.js";

// Types
export type {
  ChatMessage,
  ChatCompletionRequest,
  ContentPart,
  ToolDefinition,
  FixtureMatch,
  TextResponse,
  ToolCall,
  FixtureBlock,
  ToolCallResponse,
  ErrorResponse,
  EmbeddingResponse,
  FixtureResponse,
  Fixture,
  FixtureFile,
  FixtureFileEntry,
  JournalEntry,
  SSEChunk,
  SSEChoice,
  SSEDelta,
  SSEToolCallDelta,
  ChaosConfig,
  MockServerOptions,
  StreamingProfile,
  FixtureOpts,
  EmbeddingFixtureOpts,
  ToolCallMessage,
  RecordConfig,
  RecordProviderKey,
  ChatCompletion,
  ChatCompletionChoice,
  ChatCompletionMessage,
  ImageItem,
  ImageResponse,
  AudioResponse,
  TranscriptionResponse,
  VideoResponse,
  RawJSONResponse,
  ResponseOverrides,
  ContentWithToolCallsResponse,
  FixtureFileResponse,
  FixtureFileToolCall,
  FixtureFileTextResponse,
  FixtureFileToolCallResponse,
  FixtureFileContentWithToolCallsResponse,
} from "./types.js";