如何建立一個帶有知識庫的 Amazon Bedrock 代理並將其整合到 Lambda 函數中？

使用 AWS Bedrock 的程式碼範例

這段程式碼是用來呼叫 AWS Bedrock 的一個代理服務。它使用 Python 語言和一些特定的庫來完成這個任務。

程式碼的主要部分

首先，我們需要導入一些必要的庫：

import json

import boto3

import re

接下來，我們需要設定 AWS 的存取金鑰和區域：

AWS_ACCESS_KEY = 

AWS_SECRET_KEY = 

REGION_NAME = 

呼叫 Bedrock 的函數

這個函數叫做 bedrock_call，它會接受一個使用者的提示文字：

def bedrock_call(userPrompt):

在這裡，我們會建立一個 Bedrock 的客戶端，然後嘗試呼叫代理服務：

bedrock = boto3.client(service_name="bedrock-agent-runtime", region_name=REGION_NAME, aws_access_key_id=AWS_ACCESS_KEY, aws_secret_access_key=AWS_SECRET_KEY)

處理回應

當我們呼叫代理後，會得到一個回應。我們需要從這個回應中解析出最終的答案：

response = bedrock.invoke_agent(...)

接著，我們會檢查回應中的資料，並將它轉換成可讀的格式：

final_answer = data.decode('utf8')

錯誤處理

如果在過程中發生錯誤，我們會捕捉到這些錯誤並顯示相關的訊息：

except Exception as e:

    print("unexpected event.", e)

Lambda 函數

最後，我們有一個 Lambda 函數，這個函數會處理事件並呼叫 bedrock_call 函數：

def lambda_handler(event, context):

在這裡，我們會取得使用者的輸入，然後將其傳遞給 bedrock_call 函數，並返回結果：

return { 'statusCode': 200, 'body': json.dumps(response) }