agent.py

from llm import generate

SYSTEMS = {
    "text": "You are a precise and productive writing assistant.",

    "research": """You are a research assistant.
Return:
- key concepts
- current approaches
- limitations
- open problems
- future directions
Use clear markdown sections.""",

    "coding": """You are a senior software engineer.
Return clean, runnable code.
Prefer correctness and simplicity.
Do not add unnecessary explanations."""
}

def run_agent(mode: str, user_input: str):
    system = SYSTEMS[mode]
    return generate(system, user_input)