bedrock

type Adapter struct {
	// contains filtered or unexported fields
}

type BedrockError struct {
	// Code is the AWS error code (e.g., "ThrottlingException", "AccessDeniedException").
	Code string

	// Message is a human-readable error description.
	Message string

	// RequestID is the AWS request ID for debugging with AWS support.
	RequestID string

	// Region is the AWS region where the error occurred.
	Region string

	// Retryable indicates if this error is transient and safe to retry.
	// True for: ThrottlingException, ModelTimeoutException, InternalServerException
	// False for: AccessDeniedException, ValidationException, ModelNotReadyException
	Retryable bool

	// OriginalError is the underlying AWS SDK error.
	OriginalError error
}

type ClaudeSchemaTranslator struct{}

type Config struct {
	// Region is the AWS region for Bedrock endpoint (e.g., "us-east-1").
	// Required. Must be a valid AWS region where Bedrock is available.
	Region string

	// ModelID is the Bedrock model identifier (e.g., "anthropic.claude-3-5-sonnet-20241022-v2:0").
	// Required. Format: "provider.model-name[:version]"
	ModelID string

	// CredentialsProvider explicitly sets AWS credentials.
	// Optional. If nil, uses AWS SDK default credential chain:
	// 1. Environment variables (AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY)
	// 2. Shared credentials file (~/.aws/credentials)
	// 3. IAM role (for EC2/ECS/Lambda)
	CredentialsProvider aws.CredentialsProvider

	// EndpointURL overrides the default Bedrock endpoint.
	// Optional. Used for VPC endpoints or testing with localstack.
	EndpointURL string

	// FallbackRegions provides ordered list of backup regions for automatic failover.
	// Optional. On throttling or service errors, adapter retries in fallback regions.
	FallbackRegions []string

	// MaxRetries sets maximum retry attempts for transient errors.
	// Optional. Default: 3. Range: 0-10.
	// Retries use exponential backoff with jitter.
	MaxRetries int

	// Temperature controls randomness in model output.
	// Optional. Range: 0.0 (deterministic) to 1.0 (very random).
	// Default: model-specific (typically 1.0 for Claude, 0.5 for Llama).
	Temperature float64

	// MaxTokens limits the maximum tokens to generate.
	// Optional. Range: 1 to model maximum (4096 for Claude, 8192 for Llama).
	// Default: model-specific.
	MaxTokens int

	// TopP controls nucleus sampling diversity.
	// Optional. Range: 0.0 to 1.0.
	// Default: model-specific (typically 0.9).
	TopP float64

	// StopSequences defines sequences that stop generation.
	// Optional. Maximum 4 sequences (Bedrock API limit).
	StopSequences []string

	// StreamingEnabled enables streaming responses token-by-token.
	// Optional. Default: false.
	// Requires model support (Claude supports, Llama 4 Instruct does not).
	StreamingEnabled bool
}

type ModelFamily int

type NovaSchemaTranslator struct{}

type SchemaTranslator interface {
	// TranslateRequest converts LangGraph messages to Bedrock request format.
	//
	// Parameters:
	// - messages: LangGraph conversation history
	// - tools: Optional tool specifications
	// - config: Bedrock configuration for generation parameters
	//
	// Returns:
	// - JSON-encoded request body for Bedrock InvokeModel API
	// - Error if translation fails (e.g., unsupported features)
	TranslateRequest(messages []model.Message, tools []model.ToolSpec, config *Config) (json.RawMessage, error)

	// TranslateResponse converts Bedrock response to LangGraph ChatOut format.
	//
	// Parameters:
	// - response: JSON response from Bedrock InvokeModel API
	//
	// Returns:
	// - ChatOut with extracted text and tool calls
	// - Error if parsing fails
	TranslateResponse(response json.RawMessage) (model.ChatOut, error)

	// TranslateStreamEvent converts a Bedrock streaming event to StreamChunk.
	//
	// Only applicable for models that support streaming (e.g., Claude).
	//
	// Parameters:
	// - event: JSON streaming event from Bedrock InvokeModelWithResponseStream
	//
	// Returns:
	// - StreamChunk with incremental content
	// - Error if parsing fails
	TranslateStreamEvent(event json.RawMessage) (StreamChunk, error)

	// SupportsStreaming returns true if this model family supports streaming responses.
	SupportsStreaming() bool

	// SupportsTools returns true if this model family supports tool/function calling.
	SupportsTools() bool
}

type StreamCallback func(chunk StreamChunk) error

type StreamChunk struct {
	// Delta contains incremental text content.
	// Empty for non-text events (e.g., metadata-only events).
	Delta string

	// ToolCallDelta contains incremental tool call data (if applicable).
	// Only used for models that support tools and streaming tool calls.
	ToolCallDelta *ToolCallDelta

	// FinishReason indicates why generation stopped (if this is the final chunk).
	// Values: "end_turn", "max_tokens", "stop_sequence", "tool_use"
	// Empty for intermediate chunks.
	FinishReason string

	// Metadata contains model-specific metadata (token counts, request IDs, etc.).
	Metadata map[string]interface{}
}

type ToolCallDelta struct {
	// Index identifies which tool call this delta belongs to.
	// Multiple tool calls may be streamed in parallel.
	Index int

	// Name is the tool name (only present in first delta for this tool call).
	Name string

	// PartialJSON contains a partial JSON fragment for tool input.
	// Must be accumulated across deltas and parsed when complete.
	PartialJSON string
}