claude-mpm 2.1.1__py3-none-any.whl → 3.0.0__py3-none-any.whl

claude_mpm/agents/base_agent_loader.py

@@ -21,6 +21,7 @@ Usage:
     full_prompt = prepend_base_instructions(get_documentation_agent_prompt())
 """
 
+import json
 import logging
 import os
 from pathlib import Path
@@ -39,27 +40,27 @@ def _get_base_agent_file() -> Path:
     """Get the base agent file path."""
     # Check if we're running from a wheel installation
     try:
-        import claude_pm
-        package_path = Path(claude_pm.__file__).parent
+        import claude_mpm
+        package_path = Path(claude_mpm.__file__).parent
         path_str = str(package_path.resolve())
         if 'site-packages' in path_str or 'dist-packages' in path_str:
             # For wheel installations, check data directory
-            data_base_agent = package_path / "data" / "agents" / "BASE_AGENT_TEMPLATE.md"
+            data_base_agent = package_path / "data" / "agents" / "base_agent.json"
             if data_base_agent.exists():
                 logger.debug(f"Using wheel installation base_agent: {data_base_agent}")
                 return data_base_agent
     except Exception:
         pass
     
-    # Use the BASE_AGENT_TEMPLATE.md in the agents directory
-    base_agent_path = Path(__file__).parent / "BASE_AGENT_TEMPLATE.md"
+    # Use the base_agent.json in the agents directory
+    base_agent_path = Path(__file__).parent / "base_agent.json"
     if base_agent_path.exists():
         logger.debug(f"Using base agent template: {base_agent_path}")
         return base_agent_path
     
     # Fallback error
     logger.error("Base agent template file not found")
-    raise FileNotFoundError("BASE_AGENT_TEMPLATE.md not found in agents directory")
+    raise FileNotFoundError("base_agent.json not found in agents directory")
 
 
 # Base agent file path (dynamically determined)
@@ -149,7 +150,7 @@ TEMPLATE_SECTIONS = {
 
 def load_base_agent_instructions(force_reload: bool = False) -> Optional[str]:
     """
-    Load base agent instructions from base_agent.md with caching.
+    Load base agent instructions from base_agent.json with caching.
     Conditionally includes test-mode instructions based on CLAUDE_PM_TEST_MODE.
     
     Args:
@@ -184,7 +185,21 @@ def load_base_agent_instructions(force_reload: bool = False) -> Optional[str]:
             return None
             
         logger.debug(f"Loading base agent instructions from: {base_agent_file}")
-        content = base_agent_file.read_text(encoding='utf-8')
+        
+        # Load JSON and extract instructions
+        with open(base_agent_file, 'r', encoding='utf-8') as f:
+            base_agent_data = json.load(f)
+        
+        # Extract instructions from the JSON structure
+        if 'narrative_fields' in base_agent_data and 'instructions' in base_agent_data['narrative_fields']:
+            content = base_agent_data['narrative_fields']['instructions']
+        else:
+            # Fallback for older format
+            content = base_agent_data.get('instructions', '')
+        
+        if not content:
+            logger.error("No instructions found in base agent JSON")
+            return None
         
         # If NOT in test mode, remove test-specific instructions to save context
         if not test_mode:

claude_mpm/agents/schema/agent_schema.json

@@ -1,156 +1,310 @@
 {
   "$schema": "http://json-schema.org/draft-07/schema#",
+  "version": "1.1.0",
   "title": "Claude MPM Agent Definition Schema",
-  "description": "Schema for defining Claude MPM agents matching the target YAML structure",
+  "description": "Schema for defining Claude MPM agent templates",
   "type": "object",
-  "required": ["version", "agent_type"],
+  "required": ["schema_version", "agent_id", "agent_version", "agent_type", "metadata", "capabilities", "instructions"],
   "properties": {
-    "version": {
-      "type": "integer",
-      "description": "Agent template version number",
-      "minimum": 1
+    "schema_version": {
+      "type": "string",
+      "description": "Schema version for the agent template format",
+      "pattern": "^\\d+\\.\\d+\\.\\d+$"
+    },
+    "agent_id": {
+      "type": "string",
+      "description": "Unique identifier for the agent",
+      "pattern": "^[a-z0-9_]+$"
+    },
+    "agent_version": {
+      "type": "string",
+      "description": "Version of the agent template",
+      "pattern": "^\\d+\\.\\d+\\.\\d+$"
     },
     "agent_type": {
       "type": "string",
-      "description": "Type of agent (base or specific agent name)",
+      "description": "Type of agent",
       "enum": ["base", "engineer", "qa", "documentation", "research", "security", "ops", "data_engineer", "version_control"]
     },
-    "narrative_fields": {
+    "metadata": {
       "type": "object",
-      "description": "Narrative content that gets combined between base and specific agents",
+      "description": "Agent metadata information",
+      "required": ["name", "description", "tags", "created_at", "updated_at"],
       "properties": {
-        "when_to_use": {
+        "name": {
+          "type": "string",
+          "description": "Human-readable name of the agent"
+        },
+        "description": {
+          "type": "string",
+          "description": "Brief description of the agent's purpose"
+        },
+        "category": {
+          "type": "string",
+          "description": "Category the agent belongs to"
+        },
+        "tags": {
           "type": "array",
           "items": {
             "type": "string"
           },
-          "description": "List of when to use this agent - maps to YAML when_to_use section"
+          "description": "Tags for categorizing and searching agents"
         },
-        "specialized_knowledge": {
-          "type": "array", 
-          "items": {
-            "type": "string"
-          },
-          "description": "List of specialized knowledge areas - maps to YAML rationale.specialized_knowledge"
+        "author": {
+          "type": "string",
+          "description": "Author or team that created the agent"
         },
-        "unique_capabilities": {
+        "created_at": {
+          "type": "string",
+          "description": "Creation timestamp in ISO 8601 format"
+        },
+        "updated_at": {
+          "type": "string",
+          "description": "Last update timestamp in ISO 8601 format"
+        },
+        "specializations": {
           "type": "array",
           "items": {
             "type": "string"
           },
-          "description": "List of unique capabilities - maps to YAML rationale.unique_capabilities"
-        },
-        "instructions": {
-          "type": "string",
-          "description": "Additional markdown instructions for the agent - appears after YAML in system prompt"
+          "description": "Agent specializations (used by some agents)"
         }
       },
       "additionalProperties": false
     },
-    "configuration_fields": {
+    "capabilities": {
       "type": "object",
-      "description": "Configuration values where specific agent overrides base agent",
+      "description": "Agent capabilities and configuration",
       "properties": {
-        "description": {
+        "model": {
           "type": "string",
-          "description": "Agent description - maps to YAML description field"
-        },
-        "tags": {
-          "type": "array",
-          "items": {
-            "type": "string"
-          },
-          "description": "Agent tags - maps to YAML tags field"
+          "description": "AI model to use for this agent"
         },
         "tools": {
           "type": "array",
           "items": {
             "type": "string"
           },
-          "description": "Available tools for the agent - maps to YAML tools field"
+          "description": "Available tools for the agent"
+        },
+        "resource_tier": {
+          "type": "string",
+          "description": "Resource tier for the agent",
+          "enum": ["basic", "standard", "intensive", "lightweight"]
+        },
+        "max_tokens": {
+          "type": "integer",
+          "minimum": 1,
+          "description": "Maximum tokens for responses"
         },
         "temperature": {
           "type": "number",
           "minimum": 0,
           "maximum": 1,
-          "description": "Model temperature setting - maps to YAML temperature field"
+          "description": "Model temperature setting"
         },
         "timeout": {
           "type": "integer",
           "minimum": 1,
-          "description": "Timeout in seconds - maps to YAML timeout and execution_timeout fields"
-        },
-        "max_tokens": {
-          "type": "integer",
-          "minimum": 1,
-          "description": "Maximum tokens for responses - maps to YAML max_tokens field"
+          "description": "Timeout in seconds"
         },
         "memory_limit": {
           "type": "integer",
           "minimum": 1,
-          "description": "Memory limit in MB - maps to YAML memory_limit field"
+          "description": "Memory limit in MB"
         },
         "cpu_limit": {
           "type": "integer",
           "minimum": 1,
           "maximum": 100,
-          "description": "CPU limit percentage - maps to YAML cpu_limit field"
+          "description": "CPU limit percentage"
         },
         "network_access": {
           "type": "boolean",
-          "description": "Whether agent has network access - maps to YAML network_access field"
+          "description": "Whether agent has network access"
         },
-        "primary_role": {
-          "type": "string",
-          "description": "Primary role of the agent - maps to YAML capabilities.primary_role field"
+        "file_access": {
+          "type": "object",
+          "description": "File access permissions",
+          "properties": {
+            "read_paths": {
+              "type": "array",
+              "items": {
+                "type": "string"
+              },
+              "description": "Paths the agent can read from"
+            },
+            "write_paths": {
+              "type": "array",
+              "items": {
+                "type": "string"
+              },
+              "description": "Paths the agent can write to"
+            }
+          }
         },
-        "specializations": {
+        "when_to_use": {
           "type": "array",
           "items": {
             "type": "string"
           },
-          "description": "Agent specializations - maps to YAML capabilities.specializations field"
+          "description": "List of scenarios when to use this agent"
         },
-        "authority": {
-          "type": "string",
-          "description": "Authority level of the agent - maps to YAML capabilities.authority field"
+        "specialized_knowledge": {
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "description": "List of specialized knowledge areas"
         },
-        "model": {
-          "type": "string",
-          "description": "Model to use - maps to YAML model field",
-          "default": "claude-3-5-sonnet-20241022"
+        "unique_capabilities": {
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "description": "List of unique capabilities"
+        }
+      },
+      "additionalProperties": true
+    },
+    "configuration": {
+      "type": "object",
+      "description": "Alternative configuration structure (used by some agents)",
+      "additionalProperties": true
+    },
+    "instructions": {
+      "type": "string",
+      "description": "Markdown instructions for the agent's behavior"
+    },
+    "knowledge": {
+      "type": "object",
+      "description": "Knowledge base for the agent",
+      "properties": {
+        "domain_expertise": {
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "description": "Areas of domain expertise"
         },
-        "file_access": {
-          "type": "string",
-          "description": "File access level - maps to YAML file_access field",
-          "enum": ["project", "limited", "none"],
-          "default": "project"
+        "best_practices": {
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "description": "Best practices the agent follows"
         },
-        "dangerous_tools": {
-          "type": "boolean",
-          "description": "Whether dangerous tools are allowed - maps to YAML dangerous_tools field",
-          "default": false
+        "constraints": {
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "description": "Constraints on the agent's behavior"
         },
-        "review_required": {
-          "type": "boolean",
-          "description": "Whether review is required - maps to YAML review_required field",
-          "default": false
+        "examples": {
+          "type": "array",
+          "items": {
+            "type": "object"
+          },
+          "description": "Example scenarios or code"
+        }
+      },
+      "additionalProperties": false
+    },
+    "interactions": {
+      "type": "object",
+      "description": "Agent interaction configurations",
+      "properties": {
+        "input_format": {
+          "type": "object",
+          "properties": {
+            "required_fields": {
+              "type": "array",
+              "items": {
+                "type": "string"
+              }
+            },
+            "optional_fields": {
+              "type": "array",
+              "items": {
+                "type": "string"
+              }
+            }
+          }
         },
-        "team": {
-          "type": "string",
-          "description": "Team name - maps to YAML team field",
-          "default": "mpm-framework"
+        "output_format": {
+          "type": "object",
+          "properties": {
+            "structure": {
+              "type": "string"
+            },
+            "includes": {
+              "type": "array",
+              "items": {
+                "type": "string"
+              }
+            }
+          }
         },
-        "project": {
-          "type": "string",
-          "description": "Project name - maps to YAML project field",
-          "default": "claude-mpm"
+        "handoff_agents": {
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "description": "Agents this agent can hand off to"
         },
-        "priority": {
-          "type": "string",
-          "description": "Priority level - maps to YAML priority field",
-          "enum": ["low", "medium", "high"],
-          "default": "high"
+        "triggers": {
+          "type": "array",
+          "items": {
+            "type": "object"
+          },
+          "description": "Triggers for agent activation"
+        }
+      },
+      "additionalProperties": false
+    },
+    "testing": {
+      "type": "object",
+      "description": "Testing configuration for the agent",
+      "properties": {
+        "test_cases": {
+          "type": "array",
+          "items": {
+            "type": "object",
+            "properties": {
+              "name": {
+                "type": "string"
+              },
+              "input": {
+                "type": "string"
+              },
+              "expected_behavior": {
+                "type": "string"
+              },
+              "validation_criteria": {
+                "type": "array",
+                "items": {
+                  "type": "string"
+                }
+              }
+            }
+          }
+        },
+        "performance_benchmarks": {
+          "type": "object",
+          "properties": {
+            "response_time": {
+              "type": "number"
+            },
+            "token_usage": {
+              "type": "integer"
+            },
+            "success_rate": {
+              "type": "number",
+              "minimum": 0,
+              "maximum": 1
+            }
+          }
         }
       },
       "additionalProperties": false

claude_mpm/agents/templates/data_engineer.json

@@ -1,6 +1,8 @@
 {
-  "id": "data_engineer",
-  "version": "1.0.0",
+  "schema_version": "1.2.0",
+  "agent_id": "data_engineer_agent",
+  "agent_version": "1.2.0",
+  "agent_type": "data_engineer",
   "metadata": {
     "name": "Data Engineer Agent",
     "description": "Data engineering and AI API integrations",
@@ -25,7 +27,8 @@
       "Grep",
       "Glob",
       "LS",
-      "WebSearch"
+      "WebSearch",
+      "TodoWrite"
     ],
     "resource_tier": "intensive",
     "max_tokens": 8192,

claude_mpm/agents/templates/documentation.json

@@ -1,6 +1,8 @@
 {
-  "id": "documentation",
-  "version": "1.0.0",
+  "schema_version": "1.2.0",
+  "agent_id": "documentation_agent",
+  "agent_version": "1.1.0",
+  "agent_type": "documentation",
   "metadata": {
     "name": "Documentation Agent",
     "description": "Documentation creation and maintenance",
@@ -25,7 +27,8 @@
       "Grep",
       "Glob",
       "LS",
-      "WebSearch"
+      "WebSearch",
+      "TodoWrite"
     ],
     "resource_tier": "lightweight",
     "max_tokens": 8192,

claude_mpm/agents/templates/engineer.json

@@ -1,6 +1,8 @@
 {
-  "id": "engineer",
-  "version": "1.0.0",
+  "schema_version": "1.2.0",
+  "agent_id": "engineer_agent",
+  "agent_version": "1.1.0",
+  "agent_type": "engineer",
   "metadata": {
     "name": "Engineer Agent",
     "description": "Research-guided code implementation with pattern adherence",
@@ -27,11 +29,12 @@
       "Grep",
       "Glob",
       "LS",
-      "WebSearch"
+      "WebSearch",
+      "TodoWrite"
     ],
     "resource_tier": "intensive",
     "max_tokens": 12288,
-    "temperature": 0.05,
+    "temperature": 0.2,
     "timeout": 1200,
     "memory_limit": 6144,
     "cpu_limit": 80,

claude_mpm/agents/templates/ops.json

@@ -1,6 +1,8 @@
 {
-  "id": "ops",
-  "version": "1.0.0",
+  "schema_version": "1.2.0",
+  "agent_id": "ops_agent",
+  "agent_version": "1.1.0",
+  "agent_type": "ops",
   "metadata": {
     "name": "Ops Agent",
     "description": "Operations, deployment, and infrastructure",
@@ -24,7 +26,8 @@
       "Bash",
       "Grep",
       "Glob",
-      "LS"
+      "LS",
+      "TodoWrite"
     ],
     "resource_tier": "standard",
     "max_tokens": 8192,

claude_mpm/agents/templates/qa.json

@@ -1,6 +1,8 @@
 {
-  "id": "qa",
-  "version": "1.0.0",
+  "schema_version": "1.2.0",
+  "agent_id": "qa_agent",
+  "agent_version": "2.3.0",
+  "agent_type": "qa",
   "metadata": {
     "name": "Qa Agent",
     "description": "Quality assurance and testing validation",
@@ -24,11 +26,12 @@
       "Bash",
       "Grep",
       "Glob",
-      "LS"
+      "LS",
+      "TodoWrite"
     ],
     "resource_tier": "standard",
     "max_tokens": 8192,
-    "temperature": 0.05,
+    "temperature": 0.0,
     "timeout": 600,
     "memory_limit": 3072,
     "cpu_limit": 50,
@@ -38,7 +41,9 @@
         "./"
       ],
       "write_paths": [
-        "./"
+        "./tests/",
+        "./test/",
+        "./scripts/"
       ]
     }
   },