awsbedrock

type AnyToolChoice struct{}

type AutoToolChoice struct{}

type BedrockException struct {
	// Status code of the aws error response.
	Code string `json:"code,omitempty"`
	// Error type of the aws response.
	Type string `json:"type,omitempty"`
	// Error message of the aws response.
	Message string `json:"message"`
}

type ContentBlock struct {
	// A tool result that is a document.
	Document *DocumentBlock `json:"document,omitempty"`

	// A tool result that is an image.
	//
	// This field is only supported by Anthropic Claude 3 models.
	Image *ImageBlock `json:"image,omitempty"`
	// Text to include in the message.
	Text *string `json:"text,omitempty"`

	// The result for a tool request that a model makes.
	ToolResult *ToolResultBlock `json:"toolResult,omitempty"`

	// Information about a tool use request from a model.
	ToolUse *ToolUseBlock `json:"toolUse,omitempty"`
}

type ContentBlockStart struct {
	ToolUse *ToolUseBlockStart `json:"toolUse,omitempty"`
}

type ConverseInput struct {
	// Additional model parameters field paths to return in the response. Converse
	// returns the requested fields as a JSON Pointer object in the additionalModelResponseFields
	// field. The following is example JSON for additionalModelResponseFieldPaths.
	//
	// [ "/stop_sequence" ]
	//
	// For information about the JSON Pointer syntax, see the Internet Engineering
	// Task Force (IETF) (https://datatracker.ietf.org/doc/html/rfc6901) documentation.
	//
	// Converse rejects an empty JSON Pointer or incorrectly structured JSON Pointer
	// with a 400 error code. if the JSON Pointer is valid, but the requested field
	// is not in the model response, it is ignored by Converse.
	AdditionalModelResponseFieldPaths []*string `json:"additionalModelResponseFieldPaths,omitempty"`

	// Configuration information for a guardrail that you want to use in the request.
	GuardrailConfig *GuardrailConfiguration `json:"guardrailConfig,omitempty"`

	// Inference parameters to pass to the model. Converse supports a base set of
	// inference parameters. If you need to pass additional parameters that the
	// model supports, use the additionalModelRequestFields request field.
	InferenceConfig *InferenceConfiguration `json:"inferenceConfig,omitempty"`

	// The messages that you want to send to the model.
	//
	// Messages is a required field.
	Messages []*Message `json:"messages"`

	// A system prompt to pass to the model.
	System []*SystemContentBlock `json:"system,omitempty"`

	// Configuration information for the tools that the model can use when generating
	// a response.
	//
	// This field is only supported by Anthropic Claude 3, Cohere Command R, Cohere
	// Command R+, and Mistral Large models.
	ToolConfig *ToolConfiguration `json:"toolConfig,omitempty"`
}

type ConverseMetrics struct {
	// The latency of the call to Converse, in milliseconds.
	//
	// LatencyMs is a required field.
	LatencyMs *int64 `json:"latencyMs"`
}

type ConverseOutput struct {
	Message Message `json:"message,omitempty"`
}

type ConverseResponse struct {
	// Metrics for the call to Converse.
	//
	// Metrics is a required field.
	Metrics *ConverseMetrics `json:"metrics"`

	// The result from the call to Converse.
	//
	// Output is a required field.
	Output *ConverseOutput `json:"output"`

	// The reason why the model stopped generating output.
	//
	// StopReason is a required field.
	//
	// Valid Values: end_turn | tool_use | max_tokens | stop_sequence | guardrail_intervened | content_filtered.
	StopReason *string `json:"stopReason"`

	// The total number of tokens used in the call to Converse. The total includes
	// the tokens input to the model and the tokens generated by the model.
	//
	// Usage is a required field.
	Usage *TokenUsage `json:"usage"`
}

type ConverseStreamEvent struct {
	ContentBlockIndex int                                   `json:"contentBlockIndex,omitempty"`
	Delta             *ConverseStreamEventContentBlockDelta `json:"delta,omitempty"`
	Role              *string                               `json:"role,omitempty"`
	StopReason        *string                               `json:"stopReason,omitempty"`
	Usage             *TokenUsage                           `json:"usage,omitempty"`
	Start             *ContentBlockStart                    `json:"start,omitempty"`
}

type ConverseStreamEventContentBlockDelta struct {
	Text    *string            `json:"text,omitempty"`
	ToolUse *ToolUseBlockDelta `json:"toolUse,omitempty"`
}

type DocumentBlock struct {
	// The format of a document, or its extension.
	//
	// Format is a required field.
	Format string `json:"format"`

	// A name for the document. The name can only contain the following characters:
	//
	//    * Alphanumeric characters
	//
	//    * Whitespace characters (no more than one in a row)
	//
	//    * Hyphens
	//
	//    * Parentheses
	//
	//    * Square brackets
	//
	// This field is vulnerable to prompt injections, because the model might inadvertently
	// interpret it as instructions. Therefore, we recommend that you specify a
	// neutral name.
	//
	// Name is a required field.
	Name string `json:"name"`

	// Contains the content of the document.
	//
	// Source is a required field.
	Source DocumentSource `json:"source"`
}

type DocumentSource struct {
	// The raw bytes for the document. If you use an Amazon Web Services SDK, you
	// don't need to encode the bytes in base64.
	// Bytes are automatically base64 encoded/decoded by the SDK.
	Bytes []byte `json:"bytes"`
}

type GuardrailConfiguration struct {
	// The identifier for the guardrail.
	//
	// GuardrailIdentifier is a required field.
	GuardrailIdentifier *string `json:"guardrailIdentifier"`

	// The version of the guardrail.
	//
	// GuardrailVersion is a required field.
	GuardrailVersion *string `json:"guardrailVersion"`

	// The trace behavior for the guardrail.
	Trace *string `json:"trace,omitempty"`
}

type GuardrailConverseContentBlock struct {
	// The text to guard.
	Text *GuardrailConverseTextBlock `json:"text"`
}

type GuardrailConverseTextBlock struct {
	// The qualifier details for the guardrails contextual grounding filter.
	Qualifiers []*string `json:"qualifiers,omitempty"`

	// The text that you want to guard.
	//
	// Text is a required field.
	Text *string `json:"text"`
}

type ImageBlock struct {
	// The format of the image.
	//
	// Format is a required field.
	Format string `json:"format"`

	// The source for the image.
	//
	// Source is a required field.
	Source ImageSource `json:"source"`
}

type ImageSource struct {
	// The raw image bytes for the image. If you use an AWS SDK, you don't need
	// to encode the image bytes in base64.
	// Bytes are automatically base64 encoded/decoded by the SDK.
	Bytes []byte `json:"bytes"`
}

type InferenceConfiguration struct {
	// The maximum number of tokens to allow in the generated response. The default
	// value is the maximum allowed value for the model that you are using. For
	// more information, see Inference parameters for foundation models (https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters.html).
	MaxTokens *int64 `json:"maxTokens,omitempty"`

	// A list of stop sequences. A stop sequence is a sequence of characters that
	// causes the model to stop generating the response.
	StopSequences []*string `json:"stopSequences,omitempty"`

	// The likelihood of the model selecting higher-probability options while generating
	// a response. A lower value makes the model more likely to choose higher-probability
	// options, while a higher value makes the model more likely to choose lower-probability
	// options.
	//
	// The default value is the default value for the model that you are using.
	// For more information, see Inference parameters for foundation models (https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters.html).
	Temperature *float64 `json:"temperature,omitempty"`

	// The percentage of most-likely candidates that the model considers for the
	// next token. For example, if you choose a value of 0.8 for topP, the model
	// selects from the top 80% of the probability distribution of tokens that could
	// be next in the sequence.
	//
	// The default value is the default value for the model that you are using.
	// For more information, see Inference parameters for foundation models (https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters.html).
	TopP *float64 `json:"topP,omitempty"`
}

type Message struct {
	// The message content. Note the following restrictions:
	//
	//    * You can include up to 20 images. Each image's size, height, and width
	//    must be no more than 3.75 MB, 8000 px, and 8000 px, respectively.
	//
	//    * You can include up to five documents. Each document's size must be no
	//    more than 4.5 MB.
	//
	//    * If you include a ContentBlock with a document field in the array, you
	//    must also include a ContentBlock with a text field.
	//
	//    * You can only include images and documents if the role is user.
	//
	// Content is a required field.
	Content []*ContentBlock `json:"content"`

	// The role that the message plays in the message.
	//
	// Role is a required field.
	Role string `json:"role"`
}

type SpecificToolChoice struct {
	// The name of the tool that the model must request.
	//
	// Name is a required field.
	Name *string `json:"name"`
}

type SystemContentBlock struct {
	// A content block to assess with the guardrail. Use with the Converse (https://docs.aws.amazon.com/bedrock/latest/APIReference/API_runtime_Converse.html)
	// or ConverseStream (https://docs.aws.amazon.com/bedrock/latest/APIReference/API_runtime_ConverseStream.html)
	// API operations.
	//
	// For more information, see Use a guardrail with the Converse API in the Amazon
	// Bedrock User Guide.
	GuardContent *GuardrailConverseContentBlock `json:"guardContent,omitempty"`

	// A system prompt for the model.
	Text string `json:"text"`
}

type TokenUsage struct {
	InputTokens  int `json:"inputTokens"`
	OutputTokens int `json:"outputTokens"`
	TotalTokens  int `json:"totalTokens"`
}

type Tool struct {
	// The specification for the tool.
	ToolSpec *ToolSpecification `json:"toolSpec"`
}

type ToolChoice struct {
	// The model must request at least one tool (no text is generated).
	Any *AnyToolChoice `json:"any,omitempty"`

	// (Default). The Model automatically decides if a tool should be called or
	// whether to generate text instead.
	Auto *AutoToolChoice `json:"auto,omitempty"`

	// The Model must request the specified tool. Only supported by Anthropic Claude
	// 3 models.
	Tool *SpecificToolChoice `json:"tool,omitempty"`
}

type ToolConfiguration struct {
	// If supported by model, forces the model to request a tool.
	ToolChoice *ToolChoice `json:"toolChoice,omitempty"`

	// An array of tools that you want to pass to a model.
	//
	// Tools is a required field.
	Tools []*Tool `json:"tools"`
}

type ToolInputSchema struct {
	JSON any `json:"json"`
}

type ToolResultBlock struct {
	// The content for tool result content block.
	//
	// Content is a required field.
	Content []*ToolResultContentBlock `json:"content"`

	// The status for the tool result content block.
	//
	// This field is only supported Anthropic Claude 3 models.
	Status *string `json:"status"`

	// The ID of the tool request that this is the result for.
	//
	// ToolUseId is a required field.
	ToolUseID *string `json:"toolUseId"`
}

type ToolResultContentBlock struct {
	// A tool result that is a document.
	Document *DocumentBlock `json:"document,omitempty"`

	// A tool result that is an image.
	//
	// This field is only supported by Anthropic Claude 3 models.
	Image *ImageBlock `json:"image,omitempty"`

	// A tool result that is text.
	Text *string `json:"text,omitempty"`

	// A tool result that is JSON format data.
	JSON *string `json:"json,omitempty"`
}

type ToolSpecification struct {
	// The description for the tool.
	Description *string `json:"description,omitempty"`

	// The schema for the tool in JSON format.
	//
	// InputSchema is a required field.
	InputSchema *ToolInputSchema `json:"inputSchema"`

	// The name for the tool.
	//
	// Name is a required field.
	Name *string `json:"name"`
}

type ToolUseBlock struct {
	// Name is the name the tool that the model wants to use.
	Name string `json:"name"`
	// Input is to pass to the tool in JSON format.
	Input map[string]interface{} `json:"input"`
	// ToolUseID is the ID for the tool request, pattern is ^[a-zA-Z0-9_-]+$.
	ToolUseID string `json:"toolUseId"`
}

type ToolUseBlockDelta struct {
	Input string `json:"input"`
}

type ToolUseBlockStart struct {
	Name      string `json:"name"`
	ToolUseID string `json:"toolUseId"`
}