mirror of
https://github.com/j93es/browser-use-oauth.git
synced 2026-06-04 22:51:27 +09:00
refect: OAuth 프롬프트 및 모델 구조 개선
This commit is contained in:
parent
65c865b620
commit
146c187b05
9 changed files with 18 additions and 6 deletions
|
|
@ -9,7 +9,7 @@ def get_prompt(type: str) -> tuple[str, Type[BaseModel]] | str:
|
||||||
:return: 해당하는 프롬프트 문자열 또는 (프롬프트, 모델) 튜플
|
:return: 해당하는 프롬프트 문자열 또는 (프롬프트, 모델) 튜플
|
||||||
"""
|
"""
|
||||||
if type.lower() == "auth":
|
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
|
return prompt, model
|
||||||
|
|
||||||
elif type.lower() in ["google", "google account"]:
|
elif type.lower() in ["google", "google account"]:
|
||||||
|
|
@ -20,6 +20,18 @@ def get_prompt(type: str) -> tuple[str, Type[BaseModel]] | str:
|
||||||
from lib.llm.prompt.microsoft import prompt, model
|
from lib.llm.prompt.microsoft import prompt, model
|
||||||
return 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:
|
else:
|
||||||
from lib.llm.prompt.fallback import model, prompt
|
from lib.llm.prompt._fallback import model, prompt
|
||||||
return prompt, model
|
return prompt, model
|
||||||
|
|
|
||||||
2
src/lib/llm/prompt/_fallback/__init__.py
Normal file
2
src/lib/llm/prompt/_fallback/__init__.py
Normal file
|
|
@ -0,0 +1,2 @@
|
||||||
|
from lib.llm.prompt._fallback.prompt import prompt
|
||||||
|
from lib.llm.prompt._fallback.model import model
|
||||||
2
src/lib/llm/prompt/_get_oauth/__init__.py
Normal file
2
src/lib/llm/prompt/_get_oauth/__init__.py
Normal file
|
|
@ -0,0 +1,2 @@
|
||||||
|
from lib.llm.prompt._get_oauth.prompt import prompt
|
||||||
|
from lib.llm.prompt._get_oauth.model import model
|
||||||
|
|
@ -1,2 +0,0 @@
|
||||||
from lib.llm.prompt.fallback.prompt import prompt
|
|
||||||
from lib.llm.prompt.fallback.model import model
|
|
||||||
|
|
@ -1,2 +0,0 @@
|
||||||
from lib.llm.prompt.get_oauth.prompt import prompt
|
|
||||||
from lib.llm.prompt.get_oauth.model import model
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue