tool_use_print_utils.py

# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0


def header():
    """
    Logs the welcome message and usage guide for the tool use demo.
    """
    print("\033[0m")
    separator("=")
    print("Welcome to the Amazon Bedrock Tool Use demo!")
    separator("=")
    print(
        "This assistant provides multi-tool use capability."
    )
    print(
        "You can talk to the bot to invoke tools."
    )
    print("")
    print("Example queries:")
    print("- What's the weather like in New York?")
    print("- Is it warmer in Rome or Barcelona today?")
    print("")
    print("To exit the program, simply type 'x' and press Enter.")
    print("")
    print("Have fun and experiment with the app!")


def footer():
    """
    Logs the footer information for the tool use demo.
    """
    print("\033[0m")
    separator("=")
    print("Thank you for using Skerava's Bedrock Demo!The new feature will be added soon.")
    print("learned something new, or got some inspiration for your own apps today!")
    print("")
    print(
        "For more Bedrock examples in different programming languages, have a look at:"
    )
    print(
        "https://docs.aws.amazon.com/bedrock/latest/userguide/service_code_examples.html"
    )
    separator("=")


def call_to_bedrock(conversation):
    """
    Logs information about the call to Amazon Bedrock.

    :param conversation: The conversation history.
    """
    if "toolResult" in conversation[-1]["content"][0]:
        print("\033[0;90mReturning the tool response(s) to the model...\033[0m")
    else:
        print("\033[0;90mSending the query to the model...\033[0m")


def tool_use(tool_name, input_data):
    """
    Logs information about the tool use.

    :param tool_name: The name of the tool being used.
    :param input_data: The input data for the tool.
    """
    print(f"\033[0;90mExecuting tool: {tool_name} with input: {input_data}...\033[0m")


def model_response(message):
    """
    Logs the model's response.

    :param message: The model's response message.
    """
    print("\033[0;90mThe model's response:\033[0m")
    print(message)


def separator(char="-"):
    """
    Logs a separator line.

    :param char: The character to use for the separator line.
    """
    print(char * 80)