simple_chat.py

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 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")


if __name__ == "__main__":
    main()