Modify Agent Prompts and Configuration

Agent Architecture

review-item-processor/
├── agent.py               # Prompt generators + agent execution
├── model_config.py        # Model registry with capabilities
├── tools/
│   ├── factory.py         # Dynamic tool creation
│   ├── knowledge_base.py  # Bedrock KB integration
│   ├── code_interpreter.py
│   └── mcp_tool.py        # MCP tool support
└── index.py               # Lambda handler entry point

Common Modifications

1. Modifying Document Review Prompts

File: agent.py

Key functions:

• get_document_review_prompt() - Main entry point
• _get_document_review_prompt_with_citations() - With citations
• _get_document_review_prompt_legacy() - Without citations

JSON Schema:

{
  "result": "pass" | "fail",
  "confidence": <number 0-1>,
  "explanation": "<detailed reasoning>",
  "shortExplanation": "<max 80 chars>",
  "pageNumber": <integer from 1>,
  "citations": ["<quoted text>", ...]
}

2. Modifying Image Review Prompts

File: agent.py, function get_image_review_prompt()

JSON Schema (Nova models add boundingBoxes):

{
  "result": "pass" | "fail",
  "confidence": <number 0-1>,
  "explanation": "<detailed reasoning>",
  "shortExplanation": "<max 80 chars>",
  "usedImageIndexes": [<list of indexes>]
}

3. Model Configuration

Environment Variables (set in CDK):

• DOCUMENT_PROCESSING_MODEL_ID, IMAGE_REVIEW_MODEL_ID, BEDROCK_REGION
• ENABLE_CITATIONS, ENABLE_CODE_INTERPRETER

Change models via CDK:

cdk deploy -c rapid.documentProcessingModelId="global.anthropic.claude-opus-4-6-v1"

Add new model to model_config.py in _get_model_configs():

"model-id": ModelConfig(
    model_id="model-id",
    supports_document_block=True,
    supports_citation=True,
    supports_caching=True,
    input_per_1k=0.XXX,
    output_per_1k=0.XXX,
)

4. Adding New Environment Variables

1. Define in cdk/lib/parameter-schema.ts
2. Pass to Lambda in cdk/lib/constructs/agent.ts
3. Read in agent.py with os.environ.get()

5. Adding New Tools

For detailed tool creation guide, see references/TOOL-CREATION.md.

Quick Reference

Troubleshooting

After Modification

1. Run /build-and-format if Python code changed
2. Run /deploy-cdk-stack if CDK changes made
3. Test with sample documents and monitor CloudWatch logs