AWS .NET SDK 中 AmazonBedrockAgentRuntimeClient.RetrieveAndGenerateAsync 方法使用指南

问题背景

在使用 AWS .NET SDK 调用 AmazonBedrockAgentRuntimeClient 的 RetrieveAndGenerateAsync 方法时，开发者可能会遇到 Internal Server Error 错误。这个问题通常与请求参数配置不当有关，特别是 RetrieveAndGenerateConfiguration 对象的 Type 属性设置。

关键问题分析

RetrieveAndGenerateAsync 方法是 AWS Bedrock Agent Runtime 服务提供的一个重要接口，用于从知识库中检索信息并生成响应。当开发者直接调用此方法时，可能会遇到以下常见问题：

1. Type 属性缺失：RetrieveAndGenerateConfiguration 对象必须明确指定 Type 属性，当前唯一有效值为 "KNOWLEDGE_BASE"

2. SessionId 使用不当：首次调用时不应提供 SessionId，只有在后续调用中才需要使用服务返回的 SessionId 来维持会话状态

3. 模型ARN格式：ModelArn 参数需要正确指定基础模型的ARN路径

正确使用方法

以下是正确使用 RetrieveAndGenerateAsync 方法的示例代码：

public static async Task<RetrieveAndGenerateResponse> InvokeKnowledgeBaseAsync(
    string knowledgeBaseId, 
    string prompt, 
    string sessionId = "")
{
    var client = new AmazonBedrockAgentRuntimeClient(RegionEndpoint.USEast1);
    
    var request = new RetrieveAndGenerateRequest
    {
        Input = new RetrieveAndGenerateInput { Text = prompt },
        RetrieveAndGenerateConfiguration = new RetrieveAndGenerateConfiguration
        {
            Type = "KNOWLEDGE_BASE", // 必须明确指定
            KnowledgeBaseConfiguration = new KnowledgeBaseRetrieveAndGenerateConfiguration
            {
                KnowledgeBaseId = knowledgeBaseId,
                ModelArn = "arn:aws:bedrock:us-east-1::foundation-model/anthropic.claude-v2"
            }
        }
    };

    if (!string.IsNullOrWhiteSpace(sessionId))
    {
        request.SessionId = sessionId;
    }

    return await client.RetrieveAndGenerateAsync(request);
}

最佳实践建议

1. 会话管理：首次调用时不提供 SessionId，后续调用使用服务返回的 SessionId 来维持对话上下文

2. 错误处理：实现完善的错误处理机制，捕获 AmazonBedrockAgentRuntimeException 异常

3. 日志记录：建议启用详细日志记录以帮助调试：

   AWSConfigs.LoggingConfig.LogResponses = ResponseLoggingOption.Always;
   AWSConfigs.LoggingConfig.LogTo = LoggingOptions.Console;
   

4. 参数验证：在调用前验证 knowledgeBaseId 和 prompt 参数的有效性

常见问题排查

如果仍然遇到问题，可以检查以下方面：

1. AWS凭证：确保使用的IAM角色有访问Bedrock服务的权限

2. 区域设置：确认客户端配置的区域与知识库所在的区域一致

3. 知识库状态：验证知识库是否已正确配置并处于活动状态

4. 模型兼容性：检查指定的基础模型是否支持知识库检索功能

通过遵循这些指导原则，开发者可以有效地使用 AmazonBedrockAgentRuntimeClient 的 RetrieveAndGenerateAsync 方法来实现知识库检索和响应生成功能。