Generation from partially done JSON/List

[isamu-isozaki]

What I want is to complete a list of BaseTasks given couple BaseTasks are already defined like [BaseTask1, BaseTask2
and I want the model to complete the list. The schema is below

class CompletionStatus(str, Enum):
    todo = 'todo'
    done = 'done'
    inprogress = "inprogress"

class BaseTask(BaseModel):
    status: CompletionStatus
    task_description: constr(max_length=50)

[rlouf]

If I understand correctly you can already do this, and it could be handled more elegantly once #667 is implemented. For instance:

gen_choice = generate.choice(model, [", ", "]"])
gen_task = generate.json(model, BaseTask)

prompt = "[BaseTask1, BaseTask"
while True:
    result = gen_choice(prompt)
    if result == "]":
        return prompt + "]"
    else:
        prompt += ", "
        result = gen_task(prompt)
        prompt += result