bedrock tools

ferramentas de base

Uma pequena biblioteca Python que simplifica a chamada de função da API Amazon Bedrock Converse (ou seja, o uso de ferramentas).

Essa biblioteca reduz o código padrão necessário para integrar funções nativas do Python com a API Amazon Bedrock Converse, facilitando a criação de aplicativos de IA conversacionais poderosos e aprimorados por ferramentas.

Exemplo de uso

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 }

Aqui está um exemplo (da documentação da API Bedrock Converse) com e sem a biblioteca:

Notas

Atualmente suporta parâmetros de função do tipo:

• escalar
  • string -> string
  • int -> inteiro
  • bool -> booleano
  • flutuante -> número
• lista
  • string -> string
  • int -> inteiro
  • bool -> booleano
  • flutuante -> número
• ditado -> objeto

Desenvolvimento

 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