mirror of
https://github.com/j93es/browser-use-oauth.git
synced 2026-10-12 07:21:21 +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"]:
|
||||||
|
|
@ -19,7 +19,19 @@ def get_prompt(type: str) -> tuple[str, Type[BaseModel]] | str:
|
||||||
elif type.lower() in ["microsoft", "microsoftonline"]:
|
elif type.lower() in ["microsoft", "microsoftonline"]:
|
||||||
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
|
||||||
6
src/lib/llm/prompt/_fallback/model.py
Normal file
6
src/lib/llm/prompt/_fallback/model.py
Normal file
|
|
@ -0,0 +1,6 @@
|
||||||
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
class model(BaseModel):
|
||||||
|
msg: str | None = None
|
||||||
|
status: str | None = None # "success", "mfa_required", "blocked", "sso_not_found", "login_page_not_found", "invalid_credentials"
|
||||||
|
final_url: str | None = None
|
||||||
122
src/lib/llm/prompt/_fallback/prompt.py
Normal file
122
src/lib/llm/prompt/_fallback/prompt.py
Normal file
|
|
@ -0,0 +1,122 @@
|
||||||
|
# Extended planner prompt
|
||||||
|
prompt = f"""
|
||||||
|
🎯 목적: 웹 자동화를 위한 **SSO 로그인 리디렉션 URL 수집**
|
||||||
|
|
||||||
|
📌 주의사항 (전제 조건)
|
||||||
|
- ❌ **검색 엔진(Google, Bing 등) 사용 금지**
|
||||||
|
- ✅ **초기 제공된 URL 내에서만 탐색**
|
||||||
|
- ❌ 직접 이동하거나 추측한 링크 클릭 금지
|
||||||
|
- ⛔ 추측한 URL은 대답하거나 클릭하지 마세요
|
||||||
|
- OAuth가 아닌 일반 로그인은 무시
|
||||||
|
- OAuth가 없다면 **즉시 중단**하고 빈 배열 반환
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🧩 Step 0: 페이지 차단(Block) 여부 확인
|
||||||
|
|
||||||
|
초기 URL의 로그인 페이지에 접근하여 다음 사항을 점검합니다:
|
||||||
|
|
||||||
|
- 🚫 페이지 차단됨 (Firewall, Access Denied 등) → 즉시 중단
|
||||||
|
- 🔒 CAPTCHA는 통과 가능 (해결하고 계속 진행)
|
||||||
|
- ❗ 로그인 UI가 정상적으로 로드되지 않으면 중단
|
||||||
|
|
||||||
|
📤 차단 시 즉시 종료
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🔍 Step 1: 로그인 페이지 탐색
|
||||||
|
|
||||||
|
* 초기 URL에 접속하여 **클라이언트용 로그인 페이지**로 진입합니다.
|
||||||
|
* 쿠키 동의, 개인정보 안내 등 팝업은 무시하거나 닫고 계속 진행하세요.
|
||||||
|
* 페이지가 정상 로드되었다고 가정합니다.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 👀 Step 2: SSO 로그인 버튼 식별
|
||||||
|
|
||||||
|
아래 **OAuth SSO 버튼들만** 유효합니다:
|
||||||
|
|
||||||
|
* ✅ Google, GitHub, Facebook, LinkedIn, Microsoft, Naver
|
||||||
|
|
||||||
|
**유효한 버튼 기준**:
|
||||||
|
|
||||||
|
* OAuth 인증 흐름을 실제로 트리거
|
||||||
|
* `window.location` 또는 `<a href=...>` 또는 JS로 redirect가 발생
|
||||||
|
|
||||||
|
**제외 버튼들 (클릭 금지)**:
|
||||||
|
|
||||||
|
* ❌ 일반 로그인, 패스키, 이메일/전화번호, 인증서 기반, 비밀번호 입력
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## ✅ Step 3: 모든 SSO 버튼 클릭 및 로그인 시도
|
||||||
|
|
||||||
|
> 각 SSO 로그인 버튼을 클릭한 뒤 반드시 아래 절차를 **완전히 수행**해야 합니다.
|
||||||
|
|
||||||
|
각 SSO 버튼에 대해 다음을 수행:
|
||||||
|
|
||||||
|
1. 버튼 클릭
|
||||||
|
2. ✅ 로그인 진행:
|
||||||
|
- 로그인 페이지에서 OAuth 인증을 완료합니다.
|
||||||
|
- sign in with your username(email) x_username and password is x_password
|
||||||
|
- 버튼같은게 안눌리면 새로고침을 해봐
|
||||||
|
- **로그인 완료 후 authorize 등 버튼이 있으면 클릭**
|
||||||
|
- GitHub같은 경우 Authorize 버튼이 뜨는데 오래걸릴 수 있음, 기다려야 할 수도 있음
|
||||||
|
- 만약 버튼을 눌러도 반응이 없을 경우 새로고침을 한번 해주세요.
|
||||||
|
- **OAuth Flow가 완료되면 (callback URL 도달 또는 인증 완료) 즉시 작업 종료**
|
||||||
|
4. 로그인이 성공하면 모두 쿠키를 삭제하고 다음 SSO 버튼을 클릭합니다.
|
||||||
|
5. 다음 SSO 버튼으로 반복 진행
|
||||||
|
|
||||||
|
쿠키 삭제 방법:
|
||||||
|
chrome://settings/clearBrowserData에 들어가서 삭제해주세요.
|
||||||
|
|
||||||
|
🛑 절대 아래와 같이 해석하지 말 것:
|
||||||
|
- ❌ 버튼 클릭 후 페이지 로딩만 기다리고 돌아가기
|
||||||
|
- ❌ URL 저장 없이 go_back() 호출
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### ✨ 추가 안전 장치: "뒤로가기(go_back) 호출 조건" 제한
|
||||||
|
|
||||||
|
```text
|
||||||
|
🛑 뒤로가기(go_back)은 다음 조건이 모두 충족될 때만 사용 => 다만 로그인 실패 시, 뒤로가기 수행:
|
||||||
|
- ✅ 로그인 흐름이 완료됨 (예: redirect back to app, or callback URL)
|
||||||
|
- ✅ 현재 리디렉션 URL이 수집됨
|
||||||
|
- ✅ 결과에 저장 후 다음 버튼 탐색을 위해 복귀 필요할 때
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🚫 Step 4: 버튼 없음 또는 예외 발생 시
|
||||||
|
|
||||||
|
* 유효한 SSO 버튼이 **전혀 없을 경우**
|
||||||
|
* 예외, 오류 등 발생 시
|
||||||
|
|
||||||
|
-> 즉시 중단
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
|
||||||
|
최종 반환:
|
||||||
|
Return the result in the following format only:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{{
|
||||||
|
"msg": "Google login completed",
|
||||||
|
"status": "success" | "already_logged_in" | "mfa_required" | "window_blocked" | "idpw_required" | "blocked" | "sso_not_found" | "login_page_not_found",
|
||||||
|
"final_url": "<url_after_login_redirect or empty string>"
|
||||||
|
}}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 📎 중요 규칙 요약
|
||||||
|
|
||||||
|
* ✅ **모든 SSO 로그인은 반드시 실행** (가능한 버튼은 모두 클릭)
|
||||||
|
* 🔁 단계는 반드시 순서대로 진행
|
||||||
|
* 🔐 로그인은 쿠키/세션으로 유지된 상태에서 수행
|
||||||
|
* 👀 직접 OAuth Providor ID/PW를 입력하여도 됨 가지고 있다면
|
||||||
|
* ⛔ 추측한 URL은 접속하지 않음
|
||||||
|
|
||||||
|
---
|
||||||
|
"""
|
||||||
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
|
||||||
6
src/lib/llm/prompt/_get_oauth/model.py
Normal file
6
src/lib/llm/prompt/_get_oauth/model.py
Normal file
|
|
@ -0,0 +1,6 @@
|
||||||
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
class model(BaseModel):
|
||||||
|
msg: str | None = None
|
||||||
|
url: str | None = None
|
||||||
|
sso_list: list[str] = [] # List of SSO providers found on the login page
|
||||||
61
src/lib/llm/prompt/_get_oauth/prompt.py
Normal file
61
src/lib/llm/prompt/_get_oauth/prompt.py
Normal file
|
|
@ -0,0 +1,61 @@
|
||||||
|
prompt = """
|
||||||
|
You are an expert in finding login pages.
|
||||||
|
|
||||||
|
Your task is to navigate to the login page of the given URL. Follow the steps below strictly and return results only in the specified format.
|
||||||
|
|
||||||
|
※ You are NOT allowed to navigate to URLs that are not directly discoverable within the initial domain. Do NOT use search engines or guess external login URLs.
|
||||||
|
|
||||||
|
0. INITIAL BLOCK CHECK
|
||||||
|
- If the browser is blocked when trying to access the page — due to firewall, CAPTCHA, regional restrictions, or other access denials — immediately terminate the process and return the following JSON:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"msg": "Blocked",
|
||||||
|
"url": "",
|
||||||
|
"sso_list": []
|
||||||
|
}
|
||||||
|
```
|
||||||
|
- Do NOT proceed to further steps in this case.
|
||||||
|
|
||||||
|
1. LOGIN PAGE NAVIGATION
|
||||||
|
- Navigate only to a **client-side (non-enterprise)** login page within the provided domain.
|
||||||
|
- Do NOT rely on external tools, search engines, or links not directly found on the site.
|
||||||
|
- If a consent popup (e.g. for privacy/cookies) appears, you MUST dismiss or close it before proceeding.
|
||||||
|
- Since step 0 confirmed access, assume the page now loads properly.
|
||||||
|
|
||||||
|
2. SSO BUTTON IDENTIFICATION
|
||||||
|
- On the login page, look for the following social login (SSO) buttons:
|
||||||
|
- Google, GitHub, Facebook, LinkedIn, Microsoft, Naver, Slack, Etc.
|
||||||
|
- ✅ Proceed only if it is clearly an **actual SSO button**.
|
||||||
|
- ❌ Exclude the following:
|
||||||
|
- Passkey-related buttons
|
||||||
|
- Username/password fields
|
||||||
|
- Email-based login
|
||||||
|
- Non-OAuth methods such as certificate or phone verification
|
||||||
|
|
||||||
|
3. RETURN FORMAT
|
||||||
|
- If the login page is successfully found, return:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"msg": "Login page found",
|
||||||
|
"url": "https://example.com/login",
|
||||||
|
"sso_list": ["Google", "GitHub"]
|
||||||
|
}
|
||||||
|
```
|
||||||
|
- If the login page cannot be found, return:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"msg": "Login page not found",
|
||||||
|
"url": "",
|
||||||
|
"sso_list": []
|
||||||
|
}
|
||||||
|
```
|
||||||
|
- If blocked (as in step 0), return:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"msg": "Blocked",
|
||||||
|
"url": "",
|
||||||
|
"sso_list": []
|
||||||
|
}
|
||||||
|
```
|
||||||
|
- Return ONLY the JSON object. Do NOT include any explanation, logging, or extra output.
|
||||||
|
"""
|
||||||
|
|
@ -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…
Reference in a new issue