Abstract: Implementations relate to receiving natural language (NL) input associated with a client device; and generating response(s) to the NL input. Generating the response(s) includes: determining, based on the NL input, a base prompt; processing, using a first generative model (GM), first GM input to generate corresponding first GM output, the first GM input including the base prompt; determining, based on the first GM output, a plurality of extended prompts; obtaining, based on filtering the plurality of extended prompts, a subset of extended prompts; for each extended prompt of the subset: processing, using the first GM or a second GM, second GM input to generate corresponding second GM output, the second GM input including the respective extended prompt, and determining, based on the second GM output, a respective candidate response corresponding to the respective extended prompt; and obtaining, based on filtering the candidate response(s), the response(s) to the NL input.