[Update]

lib/agents/find_login_page.py

@@ -6,46 +6,87 @@ from browser_use import (
 )
 from lib.agents.run_agent import run_agent
 from lib.utils.logger import logger
-from lib.browser_use_utils.clean_resources import clean_agent_resources
 from lib.browser_use_utils.create_google_ai import create_google_ai
 from lib.config import GOOGLE_MODEL, GOOGLE_PLANNER_MODEL
 
 NOT_FOUND_LOGIN_PAGE = 0
 FOUND_LOGIN_PAGE = 1
 
-class IsFound(BaseModel):
-    status: int
+class FindLoginPageResponse(BaseModel):
+    status: int = NOT_FOUND_LOGIN_PAGE  # 0 if not found, 1 if found
+    msg: str | None = None
+    url: str | None = None
 
-async def find_login_page(target_url, session):
+async def find_login_page(target_url, session) -> tuple[bool, str | None]:
     initial_actions = [{"open_tab": {"url": target_url}}]
-    task = "Navigate to the login page, and stop"
-    extend_planner_system_message = "You are an expert in finding login pages. Your task is to navigate to the login page of the given URL and stop there."
+    task = """
+    You are an expert in finding login pages.
 
-    controller = Controller(output_model=IsFound, exclude_actions=['search_google'])
+    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
+        {
+            "status": 0,
+            "msg": "Blocked",
+            "url": "" 
+        }
+        ```
+    - 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. RETURN FORMAT
+    - Once the login page is reached, return a JSON object matching the following schema:
+        ```json
+        {
+            "status": 1,  // 1 if login page is found, 0 otherwise
+            "msg": "Login page found",  // Optional message 
+            "url": "https://example.com/login"  // Full URL of the login page if found
+        }
+        ```
+    - If the login page cannot be found, return:
+        ```json
+        {
+            "status": 0,
+            "msg": "Login page not found",
+            "url": ""
+        }
+        ```
+    - Return ONLY the JSON object. Do NOT include any explanation, logging, or extra output.
+    """
+
+
+    controller = Controller(output_model=FindLoginPageResponse, exclude_actions=['search_google'])
     agent = Agent(
         browser_session=session,
         initial_actions=initial_actions,
         task=task,
         llm=create_google_ai(GOOGLE_MODEL),
-        planner_llm=create_google_ai(GOOGLE_PLANNER_MODEL),
         controller=controller,
-        extend_planner_system_message=extend_planner_system_message,
     )
 
-    status, final_result = await run_agent(agent)
-    if status:
+    is_failed, final_result = await run_agent(agent)
+    if is_failed:
         logger(f"⚠️ 스캔 실패: {target_url} | {final_result}")
         print(f"⚠️ 스캔 실패: {target_url} | {final_result}")
         return False, None; 
     
     data = json.loads(final_result)
     try:
-        is_found = IsFound(**data)
-        if is_found.status == NOT_FOUND_LOGIN_PAGE:
-            return False, "로그인 페이지를 찾을 수 없습니다."
+        resp = FindLoginPageResponse(**data)
+        if resp.status == FOUND_LOGIN_PAGE and len(resp.url) > 0:
+            return True, resp.url 
         else:
-            return True, "로그인 페이지를 찾았습니다."
+            return False, resp.msg
     except Exception as e:
-        logger(f"⚠️ 결과 파싱 실패: {target_url} | {e}\n원본 결과: {final_result}")
-        print(f"⚠️ 결과 파싱 실패: {target_url} | {e}\n원본 결과: {final_result}")
-        return False, "결과 파싱 실패"
+        logger(f"⚠️ 결과 파싱 실패: {target_url} | {e}\n원본 결과: {data.msg}")
+        print(f"⚠️ 결과 파싱 실패: {target_url} | {e}\n원본 결과: {data.msg}")
+        return False, data.msg