maxtext.experimental.agent.ckpt_conversion_agent.base module

maxtext.experimental.agent.ckpt_conversion_agent.base module#

A base class for other agents

class maxtext.experimental.agent.ckpt_conversion_agent.base.BaseAgent(api_key, model_id='gemini-2.5-pro')[source]#

Bases: object

A base class for agents that provides text generation capabilities.

generate_text(prompt, tools=None)[source]#

Generates text using the provided prompt and tools.

Parameters:
  • prompt (str) – The input prompt for the model.

  • tools – Optional tools for function calling.

Returns:

The generated text as a string.