Routing

def classify_input(user_input, llm):
    """Classify user input into a category."""
    prompt = f"""
    Classify this input into one category:
    - technical: Programming, debugging, system design
    - creative: Writing, brainstorming, storytelling
    - analytical: Data analysis, research, comparison
    - general: Greetings, simple questions

    Input: {user_input}
    Respond with just the category name.
    """
    return llm.generate(prompt).content.strip().lower()

def route_to_handler(category, user_input, llm):
    """Route to the appropriate specialist handler."""
    handlers = {
        "technical": "You are a senior software engineer...",
        "creative": "You are a creative writing expert...",
        "analytical": "You are a data analyst...",
        "general": "You are a friendly assistant...",
    }
    system_prompt = handlers.get(category, handlers["general"])
    prompt = f"{system_prompt}\n\nUser request: {user_input}"
    return llm.generate(prompt).content

# Usage
category = classify_input("Help me debug this Python error", llm)
# category = "technical"
response = route_to_handler(category, user_input, llm)