Techioz Blog

Ruby を使用して AWS Bedrock でモデルを呼び出すにはどうすればよいですか?

概要

AWS Ruby SDK を使用して AWS Bedrock でモデルを呼び出す方法をドキュメントで理解するのは難しいことがわかりました。

解決策

Ruby-amazon-bedrock gem https://github.com/AAlvAAro/ruby-amazon-bedrock を作成しました。これは、aws-sdk-bedrockruntime gem の活用を容易にするように設計されています。これは、より直観的なインターフェイスを提供し、さまざまなペイロード構造の処理に伴う複雑さの一部を抽象化します。

これを使用する方法の例をいくつか示します。

client = RubyAmazonBedrock::Client.new(
  region: "aws_region",
  access_key_id: "access_key_id",
  access_token: "access_token"
)

# Claude Instant 1.2
client.invoke_model(
  id: "anthropic.claude-instant-v1",
  input: "What is a neural network?"
)

# Meta: Llama 2 Chat 70B
client.invoke_model(
  id: "meta.llama2-70b-chat-v1",
  input: "Generate a Facebook add to promote a new website that is selling Ruby on Rails and AI courses"
)

# Stability AI: SDXL 1.0
client.invoke_model(
  id: "stability.stable-diffusion-xl-v1",
  input: "Generate an image of a white gold ring with a ruby on it",
  options: { file_path: "ruby-ring.jpg" }
)

ドキュメントをできるだけ明確にするよう努めましたが、ドキュメントを改善するために常にフィードバックを求めています。この宝石により、AWS Bedrock での作業が少しでもスムーズになることを願っています。役立つかどうか、またはさらに質問がある場合はお知らせください。