AWS WebSocket API 运行在AWS Lambda函数中，需要调用 AWS API Gateway，以便连接到Websocket API。AWS Lambda函数使您能够将任何服务器端代码逻辑与 AWS WebSocket API 结合使用。您可以在 Lambda 函数中定义一个处理程序函数，该函数将在使用 Websocket API 时被触发。

以下是使用 AWS WebSocket API 调用 Lambda 函数的Python示例代码：

import json
import boto3

def lambda_handler(event, context):
    print("Running Lambda Function")
    print(event['requestContext']['connectionId'])

    client = boto3.client('apigatewaymanagementapi',
                          endpoint_url='https://' + event['requestContext']['domainName'] + '/' + event['requestContext']['stage'])
    client.post_to_connection(ConnectionId=event['requestContext']['connectionId'],
                              Data=json.dumps({"message": "Hello World"}).encode('utf-8'))

在上面的示例中，lambda_handler() 是 Lambda 函数的处理程序函数。它将打印 ConnectionId 和 "Hello World" 消息，然后通过 API Gateway 使用 post_to_connection() 方法向连接的客户端发送消息。

使用 WebSocket API 调用Lambda函数的步骤如下：

1. 创建一个 Lambda 函数，编写处理程序函数。

2. 在 AWS API Gateway 中创建 WebSocket API 并与Lambda函数关联。

3. 在 WebSocket API 与Lambda函数之间创建Integration和Route，向 Lambda 函数发送数据。

4. 接收来自 WebSocket API 的请求并通过 API Gateway 调用 Lambda 函数。

5. 从Lambda函数使用post_to_connection()方法向WebSocket API发送数据。

以上是一个概括的步骤。在您的实际应用中，可能需要做更多的操作及其他配置调整来实现您的需求。