from openai_helper import OpenAIHandler
from claude import ClaudeHandler

def get_handler(model_name, temperature):
    if "claude" in model_name:
        return ClaudeHandler(model_name, temperature)
    else:
        return OpenAIHandler(model_name, temperature)