Fix select model type and custom prompts task.py

[cyun9601]

Fixed an issue where the LLM model and custom prompts could not distinguish between main and other tasks.