maxtext.experimental.agent.ckpt_conversion_agent.prompt_chain module

maxtext.experimental.agent.ckpt_conversion_agent.prompt_chain module#

This is a baseline agent, using prompt-chain + validator/executer architecture

class maxtext.experimental.agent.ckpt_conversion_agent.prompt_chain.prompt_chaining_agent(api_key, target_model='gemma3', max_retries=3, dir_path='context')[source]#

Bases: BaseAgent

Demonstrates a multi-step prompt chain to generate a model conversion script, with verification that every parameter is actually mapped.

run_chain(max_retries=3)[source]#

Run chain