bedrock tools

基岩工具

一个小型 Python 库，可简化 Amazon Bedrock Converse API 函数调用（即工具使用）。

该库减少了将本机 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 文档）：

笔记

目前支持的函数参数类型：

• 标量
  • 字符串 -> 字符串
  • int -> 整数
  • 布尔 -> 布尔值
  • 浮点数 -> 数字
• 列表
  • 字符串 -> 字符串
  • 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