diff --git a/lib/llm/prompt/__init__.py b/lib/llm/prompt/__init__.py index 314b2dd..1417b03 100644 --- a/lib/llm/prompt/__init__.py +++ b/lib/llm/prompt/__init__.py @@ -10,7 +10,7 @@ def get_prompt(type:str) -> str: :param type: 'extend_planner' 또는 'oauth_login' :return: 해당하는 프롬프트 문자열 """ - if type == "auth": + if type.lower() == "auth": from lib.llm.prompt.auth_list import extract_oauth_list_prompt return extract_oauth_list_prompt else: