chat_with_manual_tool_implementation.py

import re
import requests
from dotenv import load_dotenv
from anthropic import Anthropic

load_dotenv()
client = Anthropic()


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


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


def main():
    print("Chat with Claude (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")

                if content.text.startswith("ping_address("):
                    ip_address = re.search(r"ping_address\((.+)\)", content.text)[1]
                    ping_response = ping_address(ip_address)

                    print(f"\u001b[93mClaude\u001b[0m: Tool usage requested, pinging {ip_address}")
                    print(f"\u001b[93mClaude\u001b[0m: {ping_response}")


if __name__ == "__main__":
    main()