import daft

# import prompt from the daft.functions library
from daft.functions import prompt

# configure the provider to "openai" with OpenAI client options like base_url
daft.set_provider("openai", base_url="...")

df = daft.from_pydict({
    "input": [
        "What sound does a cat make?",
        "What sound does a dog make?",
        "What sound does a cow make?",
        "What does the fox say?",
    ]
})

# generate responses using a chat model
df = df.with_column("response", prompt(df["input"], model="gpt-5-mini"))