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 문서의)입니다.

메모

현재 다음 유형의 함수 매개변수를 지원합니다.

• 스칼라
  • 문자열 -> 문자열
  • int -> 정수
  • 부울 -> 부울
  • 부동 -> 숫자
• 목록
  • 문자열 -> 문자열
  • 정수 -> 정수
  • 부울 -> 부울
  • 부동 -> 숫자
• dict -> 객체

개발

 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