GPT Action 库（中间件）：AWS Lambda

引言

这个特定的 GPT Action 提供了关于如何构建一个 AWS Lambda 函数的概述。此文档帮助用户设置一个受 OAuth 保护的 AWS 函数，以连接到 GPT Action 和一个示例应用程序。本示例使用 AWS SAM（无服务器应用程序模型）来设置 AWS 堆栈。

价值 + 示例业务用例

价值：用户现在可以利用 ChatGPT 的功能连接到 AWS 函数。这使您能够连接到 AWS 中的任何服务并在其上运行代码/应用程序。这可以通过几种方式实现：

访问第三方服务，如 AWS Redshift、AWS DynamoDB、AWS S3 等！
允许预处理来自 API 的文本响应（克服上下文限制、添加上下文或元数据作为示例）。
能够返回文件而不是从第三方 API 检索文本。这对于展示用于数据分析的 CSV 文件，或者带回 PDF 文件并让 ChatGPT 将其视为上传非常有用。

示例用例：

用户需要查询 Redshift 中的数据，但需要一个介于 ChatGPT 和 Redshift 之间的中间件应用程序来返回文件（数据分析数据的精确性以及大量数据）。
用户已在 AWS 函数中构建了多个步骤，并且需要能够使用 ChatGPT 启动该过程。

应用程序信息和先决条件

我们将利用 AWS Lambda 服务来创建一个中间件函数。您可以通过访问以下链接来熟悉此堆栈：

Lambda 网站：https://aws.amazon.com/lambda/
Lambda 文档：https://docs.aws.amazon.com/lambda/
AWS SAM 文档：https://docs.aws.amazon.com/serverless-application-model/

先决条件

在开始之前，请确保您拥有一个 AWS 控制台，并具有创建以下资源的访问权限：Lambda 函数、S3 存储桶、应用程序堆栈、Cognito 用户池、Cognito 用户池应用程序客户端、API Gateway、Lambda 角色、CloudFormation 堆栈（这似乎很多，但创建这些服务是自动化的！）。

创建 AWS Lambda 函数

要创建 AWS 函数，您可以使用 AWS SAM。SAM 模板的示例可以在此处 [0] 找到。

此模板包括：

用户池和用户池客户端，用于 OAuth
确保函数只能由经过身份验证的用户调用才能调用的 Cognito 授权方
将 Lambda 函数映射到现有的 VPC（有助于连接到其他 AWS 服务）
具有可以动态设置的参数（例如：凭据/变量）
一个将 HTTP 路由映射到函数的 API Gateway

此代码纯粹是信息性的，旨在帮助您入门，不需要预先存在的 AWS 资源。我们建议映射现有的用户池（如果您有的话），而不是创建新的用户池，并且将您的 Lambda 设置在可以访问其他 AWS 资源的 VPC 中（如果您需要利用这些资源）。您可以在RedShift 食谱中看到类似设置的示例。

Cognito 授权方是确保您的函数只能由经过身份验证的用户调用/访问的关键，因此请确保正确设置您的环境。

[0]

AWSTemplateFormatVersion: '2010-09-09'
Transform: AWS::Serverless-2016-10-31
Description: >
  aws-middleware

  AWS middleware function

Parameters:
  CognitoUserPoolName:
    Type: String
    Default: MyCognitoUserPool
  CognitoUserPoolClientName:
    Type: String
    Default: MyCognitoUserPoolClient

Resources:
  MyCognitoUserPool:
    Type: AWS::Cognito::UserPool
    Properties:
      UserPoolName: !Ref CognitoUserPoolName
      Policies:
        PasswordPolicy:
          MinimumLength: 8
      UsernameAttributes:

        - email
      Schema:

        - AttributeDataType: String
          Name: email
          Required: false

  MyCognitoUserPoolClient:
    Type: AWS::Cognito::UserPoolClient
    Properties:
      UserPoolId: !Ref MyCognitoUserPool
      ClientName: !Ref CognitoUserPoolClientName
      GenerateSecret: true

  MiddlewareApi:
    Type: AWS::Serverless::Api
    Properties:
      StageName: Prod
      Cors: "'*'"
      Auth:
        DefaultAuthorizer: MyCognitoAuthorizer
        Authorizers:
          MyCognitoAuthorizer:
            AuthorizationScopes:

              - openid
              - email
              - profile
            UserPoolArn: !GetAtt MyCognitoUserPool.Arn

  MiddlewareFunction:
    Type: AWS::Serverless::Function
    Properties:
      CodeUri: aws-middleware/
      Handler: app.lambda_handler
      Runtime: python3.11
      Timeout: 45
      Architectures:

        - x86_64
      Events:
        SqlStatement:
          Type: Api
          Properties:
            Path: /my_route
            Method: post
            RestApiId: !Ref MiddlewareApi

Outputs:
  MiddlewareApi:
    Description: "API Gateway endpoint URL for Prod stage for SQL Statement function"
    Value: !Sub "https://${MiddlewareApi}.execute-api.${AWS::Region}.amazonaws.com/Prod/my_route"
  MiddlewareFunction:
    Description: "SQL Statement Lambda Function ARN"
    Value: !GetAtt MiddlewareFunction.Arn
  MiddlewareFunctionIamRole:
    Description: "Implicit IAM Role created for SQL Statement function"
    Value: !GetAtt MiddlewareFunctionRole.Arn
  CognitoUserPoolArn:
    Description: "ARN of the Cognito User Pool"
    Value: !GetAtt MyCognitoUserPool.Arn

您可以克隆 openai-cookbook 存储库并从 lambda-middleware 目录中获取示例 Python 代码和 SAM 模板：

git clone https://github.com/pap-openai/lambda-middleware
cd lambda-middleware

要构建和部署您的函数，请从此目录运行以下命令：

sam build
sam deploy --template-file template.yaml --stack-name aws-middleware --capabilities CAPABILITY_IAM

部署后，您可以检查 AWS Lambda 上的应用程序：

../../../images/aws_lambda_1.png

您可以通过运行不带任何身份验证的 curl 命令来确认该函数在未经验证的情况下无法访问：

curl -d {} <deploy_command_from_middleware_api_output>

这将返回 {"message":"Unauthorized"}。

在 AWS Cognito 中设置身份验证

可选：仅当您创建了用户池且未使用现有用户池时才执行这些步骤

让我们在新创建的用户池中创建一个用户。为此，请获取部署命令中的 CognitoUserPoolArn 输出，并获取斜杠后的值，该值应为格式：your-region_xxxxx。

aws cognito-idp admin-create-user \
    --user-pool-id "your-region_xxxxx" \
    --username johndoe@example.com \
    --user-attributes Name=email,Value=johndoe@example.com \
    --temporary-password "TempPassword123"

现在，让我们确保创建一个可以登录的网页/域。转到 AWS Cognito，选择新创建的用户池，然后转到“应用程序集成”选项卡：

../../../images/aws_lambda_3.png

通过单击“域”然后“创建 Cognito 域”来创建 Cognito 域。

../../../images/aws_lambda_8.png

在用户池的“应用程序集成”页面的“应用程序客户端列表”下滚动：

../../../images/aws_lambda_9.png

选择您的应用程序客户端并编辑托管 UI：

../../../images/aws_lambda_10.png

并添加回调 URL、授权方案和 OAuth 范围：

../../../images/aws_lambda_11.png

请注意，当 ChatGPT 生成操作的身份验证回调 URL 时，您将返回到此步骤。Postman URL 仅应用于开发目的。

您可以在 Postman 中尝试此连接，在“授权”下为您的 <api_url>，将值复制粘贴到 client_id、client_secret 和您为 auth 域设置的 URL，请确保在范围中添加 openid 以获取有效的 access_token：

../../../images/aws_lambda_12.png

../../../images/aws_lambda_13.png

如果您现在在 Postman 中进行请求，使用您刚刚检索到的 access_token，您将收到一个成功的 JSON 返回：

../../../images/aws_lambda_14.png

在 ChatGPT 中创建操作

现在让我们将其集成到 ChatGPT 中。

创建一个操作并复制粘贴以下规范：

openapi: 3.1.0
info:
  title: Success API
  description: API that returns a success message.
  version: 1.0.0
servers:

  - url: https://3ho5n15aef.execute-api.us-east-1.amazonaws.com/Prod
    description: Main production server
paths:
  /my_route:
    post:
      operationId: postSuccess
      summary: Returns a success message.
      description: Endpoint to check the success status.
      responses:
        '200':
          description: A JSON object indicating success.
          content:
            application/json:
              schema:
                type: object
                properties:
                  success:
                    type: boolean
                    example: true

如果您尝试测试该操作（您可以单击“测试”按钮），您会看到一个 401 错误，因为您未经验证。

现在让我们在操作中添加身份验证。

单击“身份验证”>“OAuth”。我们现在需要获取 AWS Cognito 的变量。让我们转到您的用户池 > 用户池应用程序客户端。从那里您可以检索您的客户端 ID 和客户端密钥。

../../../images/aws_lambda_15.png