diff --git a/src/lib/llm/prompt/__init__.py b/src/lib/llm/prompt/__init__.py index 1f9e97b..8f09c98 100644 --- a/src/lib/llm/prompt/__init__.py +++ b/src/lib/llm/prompt/__init__.py @@ -9,7 +9,7 @@ def get_prompt(type: str) -> tuple[str, Type[BaseModel]] | str: :return: 해당하는 프롬프트 문자열 또는 (프롬프트, 모델) 튜플 """ if type.lower() == "auth": - from lib.llm.prompt.get_oauth import prompt, model + from lib.llm.prompt._get_oauth import prompt, model return prompt, model elif type.lower() in ["google", "google account"]: @@ -19,7 +19,19 @@ def get_prompt(type: str) -> tuple[str, Type[BaseModel]] | str: elif type.lower() in ["microsoft", "microsoftonline"]: from lib.llm.prompt.microsoft import prompt, model return prompt, model + + elif type.lower() in ["meta", "facebook"]: + from lib.llm.prompt.facebook import prompt, model + return prompt, model + + elif type.lower() in ["apple"]: + from lib.llm.prompt.apple import prompt, model + return prompt, model + + elif type.lower() in ["github"]: + from lib.llm.prompt.github import prompt, model + return prompt, model else: - from lib.llm.prompt.fallback import model, prompt + from lib.llm.prompt._fallback import model, prompt return prompt, model diff --git a/src/lib/llm/prompt/_fallback/__init__.py b/src/lib/llm/prompt/_fallback/__init__.py new file mode 100644 index 0000000..eebae1b --- /dev/null +++ b/src/lib/llm/prompt/_fallback/__init__.py @@ -0,0 +1,2 @@ +from lib.llm.prompt._fallback.prompt import prompt +from lib.llm.prompt._fallback.model import model diff --git a/src/lib/llm/prompt/fallback/model.py b/src/lib/llm/prompt/_fallback/model.py similarity index 100% rename from src/lib/llm/prompt/fallback/model.py rename to src/lib/llm/prompt/_fallback/model.py diff --git a/src/lib/llm/prompt/fallback/prompt.py b/src/lib/llm/prompt/_fallback/prompt.py similarity index 100% rename from src/lib/llm/prompt/fallback/prompt.py rename to src/lib/llm/prompt/_fallback/prompt.py diff --git a/src/lib/llm/prompt/_get_oauth/__init__.py b/src/lib/llm/prompt/_get_oauth/__init__.py new file mode 100644 index 0000000..0ca3d4f --- /dev/null +++ b/src/lib/llm/prompt/_get_oauth/__init__.py @@ -0,0 +1,2 @@ +from lib.llm.prompt._get_oauth.prompt import prompt +from lib.llm.prompt._get_oauth.model import model diff --git a/src/lib/llm/prompt/get_oauth/model.py b/src/lib/llm/prompt/_get_oauth/model.py similarity index 100% rename from src/lib/llm/prompt/get_oauth/model.py rename to src/lib/llm/prompt/_get_oauth/model.py diff --git a/src/lib/llm/prompt/get_oauth/prompt.py b/src/lib/llm/prompt/_get_oauth/prompt.py similarity index 100% rename from src/lib/llm/prompt/get_oauth/prompt.py rename to src/lib/llm/prompt/_get_oauth/prompt.py diff --git a/src/lib/llm/prompt/fallback/__init__.py b/src/lib/llm/prompt/fallback/__init__.py deleted file mode 100644 index c9a486e..0000000 --- a/src/lib/llm/prompt/fallback/__init__.py +++ /dev/null @@ -1,2 +0,0 @@ -from lib.llm.prompt.fallback.prompt import prompt -from lib.llm.prompt.fallback.model import model diff --git a/src/lib/llm/prompt/get_oauth/__init__.py b/src/lib/llm/prompt/get_oauth/__init__.py deleted file mode 100644 index 1759f89..0000000 --- a/src/lib/llm/prompt/get_oauth/__init__.py +++ /dev/null @@ -1,2 +0,0 @@ -from lib.llm.prompt.get_oauth.prompt import prompt -from lib.llm.prompt.get_oauth.model import model