Text Generation - I

import boto3
import json


def generate_text_with_bedrock(prompt, model_id="amazon.titan-text-express-v1"):
    """
    使用 AWS Bedrock 的 Titan Express 模型生成文本。

    :param prompt: 用户输入的提示语。
    :param model_id: 使用的模型 ID，默认是 Titan Express。
    :return: 生成的文本。
    """
    # 初始化 Bedrock 客户端
    client = boto3.client("bedrock-runtime", region_name="us-west-2")  # 确保区域支持 Bedrock

    # 定义生成参数
    payload = {
        "inputText": prompt,
        "textGenerationConfig": {
            "maxTokenCount": 500,  # 最大生成的 token 数量
            "temperature": 0.7,  # 控制文本的随机性
            "topP": 0.9,  # 控制文本多样性
            "stopSequences": []  # 可定义停止生成的标记
        }
    }

    try:
        # 调用 Bedrock 服务生成文本
        response = client.invoke_model(
            modelId=model_id,
            body=json.dumps(payload),
            contentType="application/json"
        )

        # 解析返回的文本
        result = response["body"].read().decode("utf-8")
        return result
    except Exception as e:
        print("Error invoking Bedrock model:", str(e))
        return None


# 示例调用
prompt = "Write a short story about a brave knight and a dragon."
generated_text = generate_text_with_bedrock(prompt)

if generated_text:
    print("Generated Text:")
    print(generated_text)

{
  "inputTextTokenCount": 12,
  "results": [
    {
      "tokenCount": 494,
      "outputText": "\nA brave knight named Sir Egbert once fought a dragon. Sir Egbert was courageous and tenacious, and he was determined to slay the dragon and save the kingdom.\n\nSir Egbert battled the dragon for hours, but he was eventually successful. He wounded the dragon with his sword and then took the opportunity to strike. The dragon let out a roar of pain and agony as it collapsed to the ground.\n\nSir Egbert was victorious, and the kingdom was saved. He was congratulated by the people and hailed as a hero. Sir Egbert was proud of his accomplishment and knew that he had done his duty.\n\nHowever, Sir Egbert soon realized that the dragon had left a legacy behind. The kingdom was in ruins, and the people were in despair. Sir Egbert knew that he had to do something to help them rebuild their lives.\n\nSir Egbert gathered a group of brave knights and set out to find the dragon's lair. They traveled through dangerous terrain, fighting off all sorts of monsters and obstacles. Finally, they found the dragon's lair.\n\nThe dragon was still alive, but it was weak from its wounds. Sir Egbert and his knights attacked the dragon, and this time, they were successful. They killed the dragon and saved the kingdom.\n\nSir Egbert was hailed as a hero once again, and he was given a new title: King of the Dragons. He ruled the kingdom with wisdom and justice, and he ensured that the people were safe and happy.\n\nYears passed, and Sir Egbert grew old. He knew that his time was coming to an end, and he wanted to leave a legacy behind. He decided to build a castle for the people to live in, and he named it after himself.\n\nThe castle was magnificent, and it was filled with all sorts of treasures. Sir Egbert also built a school for the children of the kingdom, and he taught them the importance of honor, courage, and loyalty.\n\nSir Egbert died peacefully in his sleep, and the kingdom was in mourning. But Sir Egbert's legacy lived on. His castle and school were still standing, and the people of the kingdom were proud of him.\n\nSir Egbert's story was passed down through generations, and it inspired many others to be brave and courageous. Even today, the people of the kingdom celebrate Sir Egbert's memory and his accomplishments.",
      "completionReason": "FINISH"
    }
  ]
}