Working with Anthropic in Pixeltable

%pip install -qU pixeltable anthropic

import getpass
import os

if 'ANTHROPIC_API_KEY' not in os.environ:
    os.environ['ANTHROPIC_API_KEY'] = getpass.getpass(
        'Anthropic API Key:'
    )

import pixeltable as pxt

# Remove the 'anthropic_demo' directory and its contents, if it exists
pxt.drop_dir('anthropic_demo', force=True)
pxt.create_dir('anthropic_demo')

from pixeltable.functions import anthropic

# Create a table in Pixeltable and add a computed column that calls Anthropic

t = pxt.create_table('anthropic_demo/chat', {'input': pxt.String})

msgs = [{'role': 'user', 'content': t.input}]
t.add_computed_column(
    output=anthropic.messages(
        messages=msgs,
        model='claude-haiku-4-5-20251001',
        max_tokens=300,
        model_kwargs={
            # Optional dict with parameters for the Anthropic API
            'system': 'Respond to the prompt with detailed historical information.',
            'temperature': 0.7,
        },
    )
)

# Parse the response into a new column
t.add_computed_column(response=t.output.content[0].text)

# Start a conversation
t.insert(
    [
        {
            'input': 'What was the outcome of the 1904 US Presidential election?'
        }
    ]
)
t.select(t.input, t.response).show()