chat_with_tools.py

import requests
from dotenv import load_dotenv
from anthropic import Anthropic

load_dotenv()
client = Anthropic()

TOOLS = [
    {
        "name": "ping_address",
        "description": "Check the availability of an IP address, by pinging the service and parsing the response.",
        "input_schema": {
            "type": "object",
            "properties": {
                "ip_address": {
                    "type": "string",
                    "description": "The IP address of the service to check."
                },
            },
            "required": ["ip_address"]
        }
    }
]


def ping_address(ip_address):
    ip_address = ip_address.replace('"', '')
    try:
        response = requests.get("https://" + ip_address, timeout=5)
        message = f"{ip_address} is reachable, status code {response.status_code}"
    except requests.ConnectionError:
        message = f"Failed to reach {ip_address}"
    return message


def run_inference(messages):
    response = client.messages.create(
        max_tokens=1024,
        tools=TOOLS,
        messages=messages,
        model="claude-3-7-sonnet-latest",
        stream=False,
    )
    return response


def main():
    print("Chat with Claude 3.7 (use 'Ctrl-c to quit)")

    conversation = []
    while True:
        message = input("\u001b[94mYou\u001b[0m: ")
        print("")
        conversation.append({"role": "user", "content": message})
        response = run_inference(conversation)

        for content in response.content:
            if content.type == "text":
                conversation.append({"role": "assistant", "content": content.text})
                print(f"\u001b[93mClaude\u001b[0m: {content.text}\n")
            elif content.type == "tool_use":
                if content.name == "ping_address":
                    ping_response = ping_address(**content.input)
                    print(f"\u001b[93mClaude\u001b[0m: {ping_response}")
                    conversation.append({"role": "assistant", "content": ping_response})


if __name__ == "__main__":
    main()