bedrock tools

岩盤ツール

Amazon Bedrock Converse API 関数の呼び出し (つまり、ツールの使用) を簡素化する小さな Python ライブラリ。

このライブラリは、ネイティブ Python 関数を Amazon Bedrock Converse API と統合するために必要な定型コードを削減し、ツールで強化された強力な会話型 AI アプリケーションの作成を容易にします。

使用例

pip install bedrock-tools

 from bedrock_tools import BedrockTools

# define native functions as tools (using type annotations)

def add_numbers ( a : int , b : int ) -> int :
    """Add two numbers together."""
    return a + b


def greet ( name : str ) -> str :
    """Greet a person by name."""
    return f"Hello, { name } !"


def get_weather ( city : str , state : str ) -> dict :
    """Get the weather for a location."""
    return {
        "city" : city ,
        "state" : state ,
        "temperature" : "75°F" ,
        "condition" : "Partly Cloudy" ,
    }

# setup
tools = BedrockTools ()
tools . add_function ( add_numbers )
tools . add_function ( greet )
tools . add_function ( get_weather )

# Use the config in your Bedrock Converse API call
response = bedrock . converse (
    modelId = model_id ,
    toolConfig = tools . get_tool_config ()
    messages = messages ,
)

# When you receive a toolUse from the API, invoke the tool
if "toolUse" in content_block :
    tool_results . append ( tools . invoke ( content_block [ "toolUse" ]))

message = { "role" : "user" , "content" : tool_results }

ライブラリを使用した場合と使用しない場合の例 (Bedrock Converse API ドキュメントから) を次に示します。

注意事項

現在、次のタイプの関数パラメータをサポートしています。

• スカラー
  • str -> 文字列
  • int -> 整数
  • ブール -> ブール値
  • 浮動小数点 -> 数値
• リスト
  • str -> 文字列
  • int -> 整数
  • ブール -> ブール値
  • 浮動小数点 -> 数値
• 辞書 -> オブジェクト

発達

 Choose a make command to run

  init      run this once to initialize a new python project
  install   install project dependencies
  start     run local project
  test      run unit tests