因此,我为脚本创建了一个 lambda 函数,该函数本质上允许用户将查询传递给 Amazon bedrock 上的 amazon titan LLM。这是我的部署包中 main.py 文件的内容。
from langchain.llms.bedrock import Bedrock
import boto3
from langchain.retrievers import AmazonKendraRetriever
from langchain.chains import RetrievalQA
from langchain.prompts import PromptTemplate
import json
from botocore.exceptions import ClientError
def get_secret():
secret_name = "kendraRagApp"
# Create a Secrets Manager client
session = boto3.session.Session()
client = session.client(
service_name='secretsmanager',
)
try:
get_secret_value_response = client.get_secret_value(
SecretId=secret_name
)
except ClientError as e:
raise e
# Decrypts secret using the associated KMS key.
secret = get_secret_value_response['SecretString']
return secret
def …Run Code Online (Sandbox Code Playgroud) amazon-ec2 aws-lambda aws-lambda-layers aws-iam-authenticator aws-iam-policy