Type constraints

We can ask completions to be restricted to valid python types:

from outlines import models, generate

model = models.transformers("mistralai/Mistral-7B-v0.1")
generator = generate.format(model, int)
answer = generator("When I was 6 my sister was half my age. Now I’m 70 how old is my sister?")
print(answer)
# 67

The following types are currently available:

• int
• float
• bool
• datetime.date
• datetime.time
• datetime.datetime
• We also provide custom types