Michael Ingvarsson
15 lines
428 B
Python
15 lines
428 B
Python
import logging
|
|
from core.agent import BaseAgent
|
|
from core.prompts import PLANNING_AGENT_SYSTEM_PROMPT
|
|
|
|
logger: logging.Logger = logging.getLogger("agent-planning")
|
|
|
|
|
|
class PlanningAgent(BaseAgent):
|
|
"""AI agent that analyzes a PR/issue and builds an implementation plan."""
|
|
|
|
def __init__(self, model_name: str) -> None:
|
|
super().__init__(model_name)
|
|
self.system_prompt: str = PLANNING_AGENT_SYSTEM_PROMPT
|
|
|