mirror of
https://github.com/j93es/browser-use-oauth.git
synced 2026-06-04 06:21:52 +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: 해당하는 프롬프트 문자열 또는 (프롬프트, 모델) 튜플
|
||||
"""
|
||||
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
|
||||
|
|
|
|||
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