Praison AI

PromptExpanderAgent Module

Expand short prompts into detailed, comprehensive prompts for better task execution

PromptExpanderAgent Module

The PromptExpanderAgent class expands short prompts into detailed, comprehensive prompts for better task execution.

Key Difference from QueryRewriterAgent

  • QueryRewriterAgent: Optimizes queries for search/retrieval (RAG)
  • PromptExpanderAgent: Expands prompts for detailed task execution

Supported Strategies

  • BASIC: Simple expansion with clarity improvements
  • DETAILED: Rich expansion with context, constraints, and examples
  • STRUCTURED: Expansion with clear structure (task, format, requirements)
  • CREATIVE: Expansion with creative flair and vivid language
  • AUTO: Automatically selects the best strategy based on prompt analysis

Import

python
from praisonaiagents import PromptExpanderAgent, ExpandStrategy

Quick Example

python
from praisonaiagents import PromptExpanderAgent, ExpandStrategy

agent = PromptExpanderAgent()

# Basic expansion
result = agent.expand("write a movie script in 3 lines")
print(result.expanded_prompt)

# Detailed expansion
result = agent.expand("blog about AI", strategy=ExpandStrategy.DETAILED)
print(result.expanded_prompt)

Constructor

PromptExpanderAgent()

Creates a new PromptExpanderAgent instance.

Parameters:

ParameterTypeDefaultDescription
namestr"Prompt Expander"Agent name
modelstr"gpt-4o-mini"LLM model to use
verboseboolFalseEnable verbose logging
temperaturefloat0.7Temperature for generation (higher = more creative)
toolslist[]Optional tools for context gathering

Methods

expand(prompt, strategy)

Expands a prompt using the specified strategy.

Parameters:

  • prompt (str): The original short prompt
  • strategy (ExpandStrategy): Expansion strategy to use (default: AUTO)

Returns: ExpandResult – Contains expanded prompt and metadata

ExpandStrategy Enum

python
class ExpandStrategy(Enum):
    BASIC = "basic"
    DETAILED = "detailed"
    STRUCTURED = "structured"
    CREATIVE = "creative"
    AUTO = "auto"

ExpandResult

python
@dataclass
class ExpandResult:
    original_prompt: str
    expanded_prompt: str
    strategy_used: ExpandStrategy
    metadata: Dict[str, Any]

Examples

Creative Expansion

python
from praisonaiagents import PromptExpanderAgent, ExpandStrategy

agent = PromptExpanderAgent(model="gpt-4o-mini", )
result = agent.expand("write a poem", strategy=ExpandStrategy.CREATIVE)
print(result.expanded_prompt)

With Tools for Context

python
from praisonaiagents import PromptExpanderAgent
from praisonaiagents import internet_search

agent = PromptExpanderAgent(tools=[internet_search], )
result = agent.expand("latest AI developments")
# Gathers context first, then expands

Structured Expansion

python
from praisonaiagents import PromptExpanderAgent, ExpandStrategy

agent = PromptExpanderAgent()
result = agent.expand(
    "create a REST API",
    strategy=ExpandStrategy.STRUCTURED
)
# Returns structured prompt with task, format, requirements

Related