// This file is auto-generated by @hey-api/openapi-ts

import * as z from 'zod'

/**
 * BedrockRetrievalSetting
 *
 * Retrieval settings for Amazon Bedrock knowledge base queries.
 */
export const zBedrockRetrievalSetting = z.object({
  score_threshold: z.number().optional().default(0),
  top_k: z.int().nullish(),
})

/**
 * BedrockRetrievalPayload
 */
export const zBedrockRetrievalPayload = z.object({
  knowledge_id: z.string(),
  query: z.string(),
  retrieval_setting: zBedrockRetrievalSetting,
})

export const zPostTestRetrievalBody = zBedrockRetrievalPayload

/**
 * Bedrock retrieval test completed
 */
export const zPostTestRetrievalResponse = z.record(z.string(), z.unknown())