import { PrivacyConfig, normalizeToolDefinitions } from "./core/privacy.js"; import { AIConfig, AIConfigOptions, ContentMode } from "./config.js"; import { EvidenceQuote, MessageLabel, RubricScore, SessionEnrichments, TopicClassification } from "./core/enrichments.js"; import { AmplitudeAILike, AmplitudeClientLike, AmplitudeEvent, AmplitudeLike, AmplitudeOrAI, AnthropicParams, AnthropicResponse, Attachment, BedrockConverseParams, BedrockConverseResponse, ChatCompletionParams, ChatCompletionResponse, ChatMessage, ContentBlock, MistralChatParams, MistralChatResponse, ToolCallShape, TrackCallOptions, TrackFn, resolveAmplitude } from "./types.js"; import { ReviewRequest } from "./review.js"; import { TenantHandle } from "./tenant.js"; import { AmplitudeAI, FlushOptions } from "./client.js"; import { Session, SessionOptions } from "./session.js"; import { AgentOptions, AiMessageOpts, BoundAgent, EmbeddingOpts, ScoreOpts, SessionEndOpts, SessionEnrichmentOpts, SpanOpts, ToolCallOpts, UserMessageOpts } from "./bound-agent.js"; import { SessionContext, getActiveContext, isTrackerManaged, pushContext, runWithContext, runWithContextAsync } from "./context.js"; import { EVENT_AI_RESPONSE, EVENT_EMBEDDING, EVENT_SCORE, EVENT_SESSION_END, EVENT_SESSION_ENRICHMENT, EVENT_SPAN, EVENT_TOOL_CALL, EVENT_USER_MESSAGE, MAX_SERIALIZED_LENGTH, PROP_ABANDONMENT_TURN, PROP_AGENT_CHAIN, PROP_AGENT_DESCRIPTION, PROP_AGENT_ID, PROP_AGENT_VERSION, PROP_ATTACHMENTS, PROP_ATTACHMENT_COUNT, PROP_ATTACHMENT_TYPES, PROP_CACHE_CREATION_TOKENS, PROP_CACHE_READ_TOKENS, PROP_COMMENT, PROP_COMPONENT_TYPE, PROP_CONTENT_MODE, PROP_CONTEXT, PROP_COST_USD, PROP_CUSTOMER_ORG_ID, PROP_EDITED_MESSAGE_ID, PROP_EMBEDDING_DIMENSIONS, PROP_ENRICHMENTS, PROP_ENV, PROP_ERROR_MESSAGE, PROP_ERROR_TYPE, PROP_EVALUATION_SOURCE, PROP_FINISH_REASON, PROP_HAS_ATTACHMENTS, PROP_HAS_REASONING, PROP_IDLE_TIMEOUT_MINUTES, PROP_INPUT_STATE, PROP_INPUT_TOKENS, PROP_INVOCATION_ID, PROP_IS_EDIT, PROP_IS_ERROR, PROP_IS_REGENERATION, PROP_IS_STREAMING, PROP_LATENCY_MS, PROP_LOCALE, PROP_MAX_OUTPUT_TOKENS, PROP_MESSAGE_ID, PROP_MESSAGE_LABELS, PROP_MESSAGE_LABEL_MAP, PROP_MESSAGE_SOURCE, PROP_MODEL_NAME, PROP_MODEL_TIER, PROP_OUTPUT_STATE, PROP_OUTPUT_TOKENS, PROP_PARENT_AGENT_ID, PROP_PARENT_MESSAGE_ID, PROP_PARENT_SPAN_ID, PROP_PROMPT_ID, PROP_PROVIDER, PROP_PROVIDER_REQUEST_ID, PROP_QUALITY_SCORE, PROP_REASONING_CONTENT, PROP_REASONING_TOKENS, PROP_REQUEST_COMPLEXITY, PROP_ROOT_AGENT_NAME, PROP_RUNTIME, PROP_SCORE_NAME, PROP_SCORE_VALUE, PROP_SDK_VERSION, PROP_SENTIMENT_SCORE, PROP_SESSION_ID, PROP_SESSION_REPLAY_ID, PROP_SPAN_ID, PROP_SPAN_KIND, PROP_SPAN_NAME, PROP_SYSTEM_PROMPT, PROP_SYSTEM_PROMPT_LENGTH, PROP_TARGET_ID, PROP_TARGET_TYPE, PROP_TASK_FAILURE_REASON, PROP_TASK_FAILURE_TYPE, PROP_TEMPERATURE, PROP_TOOL_CALLS, PROP_TOOL_DEFINITIONS, PROP_TOOL_DEFINITIONS_COUNT, PROP_TOOL_DEFINITIONS_HASH, PROP_TOOL_INPUT, PROP_TOOL_NAME, PROP_TOOL_OUTPUT, PROP_TOOL_OWNER, PROP_TOOL_SUCCESS, PROP_TOOL_TYPE, PROP_TOP_P, PROP_TOTAL_ATTACHMENT_SIZE, PROP_TOTAL_TOKENS, PROP_TRACE_ID, PROP_TTFB_MS, PROP_TURN_ID, PROP_WAS_CACHED, PROP_WAS_COPIED, SDK_RUNTIME, SDK_VERSION } from "./core/constants.js"; import { serializeToJsonString, trackAiMessage, trackConversation, trackEmbedding, trackScore, trackSessionEnd, trackSessionEnrichment, trackSpan, trackToolCall, trackUserMessage } from "./core/tracking.js"; import { ObserveSpanType, ObserveWrapped, ToolCallTracker, ToolWrapped, observe, tool } from "./decorators.js"; import { AmplitudeAIError, ConfigurationError, CostCalculationError, ProviderError, TrackingError, ValidationError } from "./exceptions.js"; import { isServerless } from "./serverless.js"; import { MockAmplitudeAI } from "./testing.js"; import { GENERATED_FILES, MCP_PROMPTS, MCP_RESOURCES, MCP_SERVER_NAME, MCP_TOOLS } from "./mcp/contract.js"; import { generateVerifyTest } from "./mcp/generate-verify-test.js"; import { InstrumentFileOptions, instrumentFile } from "./mcp/instrument-file.js"; import { extractContext, injectContext } from "./propagation.js"; import { MiddlewareOptions, createAmplitudeAIMiddleware } from "./middleware.js"; import { StreamingAccumulator } from "./utils/streaming.js"; import { ANTHROPIC_AVAILABLE, Anthropic, extractAnthropicContent, extractAnthropicSystemPrompt } from "./providers/anthropic.js"; import { resolveOpenAIProvider } from "./utils/openai-provider.js"; import { OPENAI_AVAILABLE, OpenAI, WrappedResponses, extractSystemPrompt } from "./providers/openai.js"; import { AzureOpenAI } from "./providers/azure-openai.js"; import { BEDROCK_AVAILABLE, Bedrock, extractBedrockResponse } from "./providers/bedrock.js"; import { GEMINI_AVAILABLE, Gemini, extractGeminiResponse } from "./providers/gemini.js"; import { GOOGLE_GENAI_AVAILABLE, GoogleGenAI, GoogleGenAIOptions, extractGoogleGenAIResponse } from "./providers/google-genai.js"; import { MISTRAL_AVAILABLE, Mistral } from "./providers/mistral.js"; import { AmplitudeAIWrapError, wrap } from "./wrappers.js"; import { patch, patchAnthropic, patchAzureOpenAI, patchBedrock, patchGemini, patchMistral, patchOpenAI, patchedProviders, unpatch, unpatchAnthropic, unpatchAzureOpenAI, unpatchBedrock, unpatchGemini, unpatchMistral, unpatchOpenAI } from "./patching.js"; import { AmplitudeCallbackHandler, createAmplitudeCallback } from "./integrations/langchain.js"; import { AmplitudeAgentExporter, AmplitudeGenAIExporter, AmplitudeSpanExporter } from "./integrations/opentelemetry.js"; import { AmplitudeLlamaIndexHandler, createAmplitudeLlamaIndexHandler } from "./integrations/llamaindex.js"; import { AmplitudeTracingProcessor } from "./integrations/openai-agents.js"; import { AmplitudeToolLoop } from "./integrations/anthropic-tools.js"; import { AmplitudeCrewAIHooks } from "./integrations/crewai.js"; import { inferProviderFromModel } from "./utils/providers.js"; import { calculateCost, enableLivePriceUpdates, fireworksExplicitPriceCost, fireworksFastTierCost, getGenaiPriceLookupCandidates, inferProvider, stripProviderPrefix } from "./utils/costs.js"; import { countMessageTokens, countTokens, estimateTokens } from "./utils/tokens.js"; import { TIER_FAST, TIER_REASONING, TIER_STANDARD, inferModelTier } from "./utils/model-tiers.js"; import { OfflineEvalServerZone, OfflineEvalUploadError, OfflineEvalUploadResult, OfflineEvalValidationError, ReportOfflineEvalOptions, reportOfflineEval } from "./offline-eval.js"; import { BraintrustExperiment, BraintrustExperimentRow, braintrustExperimentsToDocument } from "./integrations/braintrust-experiment.js"; import { GitMetadata, _resetCache, getGitMetadata } from "./utils/git_metadata.js"; export { AIConfig, type AIConfigOptions, ANTHROPIC_AVAILABLE, OPENAI_AVAILABLE as AZURE_OPENAI_AVAILABLE, type AgentOptions, type AiMessageOpts, AmplitudeAI, AmplitudeAIError, type AmplitudeAILike, AmplitudeAIWrapError, AmplitudeAgentExporter, AmplitudeCallbackHandler, type AmplitudeClientLike, AmplitudeCrewAIHooks, type AmplitudeEvent, AmplitudeGenAIExporter, type AmplitudeLike, AmplitudeLlamaIndexHandler, type AmplitudeOrAI, AmplitudeSpanExporter, AmplitudeToolLoop, AmplitudeTracingProcessor, Anthropic, type AnthropicParams, type AnthropicResponse, type Attachment, AzureOpenAI, BEDROCK_AVAILABLE, Bedrock, type BedrockConverseParams, type BedrockConverseResponse, BoundAgent, type BraintrustExperiment, type BraintrustExperimentRow, type ChatCompletionParams, type ChatCompletionResponse, type ChatMessage, ConfigurationError, type ContentBlock, ContentMode, CostCalculationError, EVENT_AI_RESPONSE, EVENT_EMBEDDING, EVENT_SCORE, EVENT_SESSION_END, EVENT_SESSION_ENRICHMENT, EVENT_SPAN, EVENT_TOOL_CALL, EVENT_USER_MESSAGE, type EmbeddingOpts, EvidenceQuote, type FlushOptions, GEMINI_AVAILABLE, GENERATED_FILES, GOOGLE_GENAI_AVAILABLE, Gemini, type GitMetadata, GoogleGenAI, type GoogleGenAIOptions, type InstrumentFileOptions, MAX_SERIALIZED_LENGTH, MCP_PROMPTS, MCP_RESOURCES, MCP_SERVER_NAME, MCP_TOOLS, MISTRAL_AVAILABLE, MessageLabel, type MiddlewareOptions, Mistral, type MistralChatParams, type MistralChatResponse, MockAmplitudeAI, OPENAI_AVAILABLE, type ObserveSpanType, type ObserveWrapped, type OfflineEvalServerZone, OfflineEvalUploadError, type OfflineEvalUploadResult, type OfflineEvalValidationError, OpenAI, PROP_ABANDONMENT_TURN, PROP_AGENT_CHAIN, PROP_AGENT_DESCRIPTION, PROP_AGENT_ID, PROP_AGENT_VERSION, PROP_ATTACHMENTS, PROP_ATTACHMENT_COUNT, PROP_ATTACHMENT_TYPES, PROP_CACHE_CREATION_TOKENS, PROP_CACHE_READ_TOKENS, PROP_COMMENT, PROP_COMPONENT_TYPE, PROP_CONTENT_MODE, PROP_CONTEXT, PROP_COST_USD, PROP_CUSTOMER_ORG_ID, PROP_EDITED_MESSAGE_ID, PROP_EMBEDDING_DIMENSIONS, PROP_ENRICHMENTS, PROP_ENV, PROP_ERROR_MESSAGE, PROP_ERROR_TYPE, PROP_EVALUATION_SOURCE, PROP_FINISH_REASON, PROP_HAS_ATTACHMENTS, PROP_HAS_REASONING, PROP_IDLE_TIMEOUT_MINUTES, PROP_INPUT_STATE, PROP_INPUT_TOKENS, PROP_INVOCATION_ID, PROP_IS_EDIT, PROP_IS_ERROR, PROP_IS_REGENERATION, PROP_IS_STREAMING, PROP_LATENCY_MS, PROP_LOCALE, PROP_MAX_OUTPUT_TOKENS, PROP_MESSAGE_ID, PROP_MESSAGE_LABELS, PROP_MESSAGE_LABEL_MAP, PROP_MESSAGE_SOURCE, PROP_MODEL_NAME, PROP_MODEL_TIER, PROP_OUTPUT_STATE, PROP_OUTPUT_TOKENS, PROP_PARENT_AGENT_ID, PROP_PARENT_MESSAGE_ID, PROP_PARENT_SPAN_ID, PROP_PROMPT_ID, PROP_PROVIDER, PROP_PROVIDER_REQUEST_ID, PROP_QUALITY_SCORE, PROP_REASONING_CONTENT, PROP_REASONING_TOKENS, PROP_REQUEST_COMPLEXITY, PROP_ROOT_AGENT_NAME, PROP_RUNTIME, PROP_SCORE_NAME, PROP_SCORE_VALUE, PROP_SDK_VERSION, PROP_SENTIMENT_SCORE, PROP_SESSION_ID, PROP_SESSION_REPLAY_ID, PROP_SPAN_ID, PROP_SPAN_KIND, PROP_SPAN_NAME, PROP_SYSTEM_PROMPT, PROP_SYSTEM_PROMPT_LENGTH, PROP_TARGET_ID, PROP_TARGET_TYPE, PROP_TASK_FAILURE_REASON, PROP_TASK_FAILURE_TYPE, PROP_TEMPERATURE, PROP_TOOL_CALLS, PROP_TOOL_DEFINITIONS, PROP_TOOL_DEFINITIONS_COUNT, PROP_TOOL_DEFINITIONS_HASH, PROP_TOOL_INPUT, PROP_TOOL_NAME, PROP_TOOL_OUTPUT, PROP_TOOL_OWNER, PROP_TOOL_SUCCESS, PROP_TOOL_TYPE, PROP_TOP_P, PROP_TOTAL_ATTACHMENT_SIZE, PROP_TOTAL_TOKENS, PROP_TRACE_ID, PROP_TTFB_MS, PROP_TURN_ID, PROP_WAS_CACHED, PROP_WAS_COPIED, PrivacyConfig, ProviderError, type ReportOfflineEvalOptions, type ReviewRequest, RubricScore, SDK_RUNTIME, SDK_VERSION, type ScoreOpts, Session, SessionContext, type SessionEndOpts, type SessionEnrichmentOpts, SessionEnrichments, type SessionOptions, type SpanOpts, StreamingAccumulator, TIER_FAST, TIER_REASONING, TIER_STANDARD, TenantHandle, type ToolCallOpts, type ToolCallShape, ToolCallTracker, type ToolWrapped, TopicClassification, type TrackCallOptions, type TrackFn, TrackingError, type UserMessageOpts, ValidationError, WrappedResponses, _resetCache as _resetGitCache, braintrustExperimentsToDocument, calculateCost, countMessageTokens, countTokens, createAmplitudeAIMiddleware, createAmplitudeCallback, createAmplitudeLlamaIndexHandler, enableLivePriceUpdates, estimateTokens, extractAnthropicContent, extractAnthropicSystemPrompt, extractBedrockResponse, extractContext, extractGeminiResponse, extractGoogleGenAIResponse, extractSystemPrompt, fireworksExplicitPriceCost, fireworksFastTierCost, generateVerifyTest, getActiveContext, getGenaiPriceLookupCandidates, getGitMetadata, inferModelTier, inferProvider, inferProviderFromModel, injectContext, instrumentFile, isServerless, isTrackerManaged, normalizeToolDefinitions, observe, patch, patchAnthropic, patchAzureOpenAI, patchBedrock, patchGemini, patchMistral, patchOpenAI, patchedProviders, pushContext, reportOfflineEval, resolveAmplitude, resolveOpenAIProvider, runWithContext, runWithContextAsync, serializeToJsonString, stripProviderPrefix, tool, trackAiMessage, trackConversation, trackEmbedding, trackScore, trackSessionEnd, trackSessionEnrichment, trackSpan, trackToolCall, trackUserMessage, unpatch, unpatchAnthropic, unpatchAzureOpenAI, unpatchBedrock, unpatchGemini, unpatchMistral, unpatchOpenAI, wrap };