selenium-python-ai-agent 0.2.0__tar.gz → 0.2.2__tar.gz

{selenium_python_ai_agent-0.2.0/selenium_python_ai_agent.egg-info → selenium_python_ai_agent-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: selenium-python-ai-agent
-Version: 0.2.0
+Version: 0.2.2
 Summary: AI-powered multi-agent framework to plan, generate and heal Selenium Python tests (pytest + pytest-bdd)
 Author: Ankit Tripathi
 License: MIT
@@ -135,11 +135,17 @@ selenium-agent --heal-only generated_tests/tests/test_checkout.py
 Most teams don't start from scratch — they already have a Selenium project. Point the agent at it with `--project` and it **adapts to your structure instead of imposing its own**:
 
 ```bash
-# First, see what the scanner detects about your project (read-only)
+# First, see what the scanner detects about your project (read-only).
+# The path is SAVED to config — you never need to repeat it.
 selenium-agent --scan /path/to/your/project
 
-# Then generate tests that FIT INTO it
-selenium-agent "test the invoice creation flow" --project /path/to/your/project
+# From now on every run fits into it automatically:
+selenium-agent "test the invoice creation flow"
+selenium-agent "test the user permissions page"
+
+# Manage the saved project:
+selenium-agent config --project /other/project   # switch
+selenium-agent config --project none             # clear
 ```
 
 The built-in **ProjectScanner** detects and follows:
@@ -359,7 +365,7 @@ generated_tests/
 | `--headless` | Headless browser | from config (`false`) |
 | `--explore N` | Explore up to N extra same-origin pages while planning (relevance-ranked, multi-hop) | `0` |
 | `--output-dir` | Where to save generated files | `generated_tests` |
-| `--project` | Fit into existing project path | — |
+| `--project` | Existing project path (saved to config — set once) | from config |
 | `--max-retries` | Healer fix attempts | `5` |
 | `--no-heal` | Skip healing after codegen | `false` |
 | `--plan-only` | Save + show plan, no code | `false` |

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.2}/README.md

@@ -105,11 +105,17 @@ selenium-agent --heal-only generated_tests/tests/test_checkout.py
 Most teams don't start from scratch — they already have a Selenium project. Point the agent at it with `--project` and it **adapts to your structure instead of imposing its own**:
 
 ```bash
-# First, see what the scanner detects about your project (read-only)
+# First, see what the scanner detects about your project (read-only).
+# The path is SAVED to config — you never need to repeat it.
 selenium-agent --scan /path/to/your/project
 
-# Then generate tests that FIT INTO it
-selenium-agent "test the invoice creation flow" --project /path/to/your/project
+# From now on every run fits into it automatically:
+selenium-agent "test the invoice creation flow"
+selenium-agent "test the user permissions page"
+
+# Manage the saved project:
+selenium-agent config --project /other/project   # switch
+selenium-agent config --project none             # clear
 ```
 
 The built-in **ProjectScanner** detects and follows:
@@ -329,7 +335,7 @@ generated_tests/
 | `--headless` | Headless browser | from config (`false`) |
 | `--explore N` | Explore up to N extra same-origin pages while planning (relevance-ranked, multi-hop) | `0` |
 | `--output-dir` | Where to save generated files | `generated_tests` |
-| `--project` | Fit into existing project path | — |
+| `--project` | Existing project path (saved to config — set once) | from config |
 | `--max-retries` | Healer fix attempts | `5` |
 | `--no-heal` | Skip healing after codegen | `false` |
 | `--plan-only` | Save + show plan, no code | `false` |

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "selenium-python-ai-agent"
-version = "0.2.0"
+version = "0.2.2"
 description = "AI-powered multi-agent framework to plan, generate and heal Selenium Python tests (pytest + pytest-bdd)"
 readme = "README.md"
 license = { text = "MIT" }

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.2}/selenium_agent/__init__.py

@@ -15,6 +15,6 @@ Usage (CLI):
 
 from selenium_agent.core.orchestrator import Orchestrator as SeleniumAgent
 
-__version__ = "0.2.0"
+__version__ = "0.2.2"
 __author__ = "Ankit Tripathi"
 __all__ = ["SeleniumAgent"]

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.2}/selenium_agent/agents/coder.py

@@ -245,6 +245,51 @@ Output valid JSON only, no markdown:
 ]}}
 """
 
+PROJECT_NATIVE_SYSTEM_PROMPT = """
+You are an expert Selenium Python engineer extending an EXISTING test
+framework. The project profile below — especially its sample code — is the
+ONLY architecture reference. You are a guest in this codebase: mirror its
+patterns exactly, never import your own.
+
+{project_context}
+
+══ HARD RULES ══
+
+1. FILE LOCATIONS — write ONLY into the project's own folders:
+     page objects → {pages_dir}/
+     test files   → {tests_dir}/
+   FORBIDDEN: creating new top-level folders (generated_tests/, output/, ...).
+
+2. ARCHITECTURE — copy the sample page object's style exactly: same base
+   class (or none), same helper/service/logger imports, same constructor
+   signature, same method style. Do NOT import selenium_agent modules
+   unless the samples themselves do.
+
+3. DRIVER — the project's conftest.py already provides the fixture named
+   '{fixture_name}'. Test functions accept it as a parameter:
+       def test_something({fixture_name}):
+   NEVER define fixtures, NEVER create drivers, NEVER assume the fixture
+   is called 'driver' if the project calls it something else.
+
+4. IMPORTS — follow the project's import style shown in the samples
+   (e.g. 'from pages.login_page import LoginPage').
+
+══ QUALITY RULES (always apply) ══
+  - Use ONLY locators from the plan / DOM scan — never invent
+  - Unique runtime test data for entity-creating flows:
+      import uuid; unique = uuid.uuid4().hex[:8]
+      email = "qa." + unique + "@example.com"; password = "Xk9#" + unique + "!Qz"
+  - Fill EVERY form field the scan shows, not just the ones named
+  - Assert POST-action outcomes (what changes/appears), never pre-action text
+  - No time.sleep(), no pytest.skip() placeholders, no TODO steps
+
+Output valid JSON only, no markdown:
+{{"files": [
+  {{"filename": "{pages_dir}/<name>_page.py", "content": "..."}},
+  {{"filename": "{tests_dir}/test_<name>.py",  "content": "..."}}
+]}}
+"""
+
 REPAIR_PROMPT = """
 The code you generated has validation errors. Fix them and return the
 COMPLETE corrected set of files in the same JSON format.
@@ -270,7 +315,7 @@ class CoderAgent:
         mode = plan.get("mode", "pytest")
         logger.info(f"💻 Generating [{mode.upper()}] code (headless={plan.get('headless', False)})...")
 
-        system_prompt = CODER_SYSTEM_PROMPT_BDD if mode == "bdd" else CODER_SYSTEM_PROMPT_PYTEST
+        system_prompt = self._system_prompt_for(mode, project_profile)
         user_prompt = self._build_user_prompt(plan, project_profile, mode)
         max_tokens = self._token_budget(plan)
 
@@ -375,6 +420,24 @@ class CoderAgent:
 
     # ── Prompt building ────────────────────────────────────────────────
 
+    @staticmethod
+    def _system_prompt_for(mode: str, project_profile=None) -> str:
+        """
+        Standalone mode teaches the framework's own BasePage architecture.
+        Project mode REPLACES that entirely: the user's existing framework
+        (its samples, folders, fixture name) is the only reference —
+        supplementing a strong default architecture with a small 'fit in'
+        note demonstrably loses to the baked-in style.
+        """
+        if project_profile is not None and mode != "bdd":
+            return PROJECT_NATIVE_SYSTEM_PROMPT.format(
+                project_context=project_profile.to_llm_context(),
+                pages_dir=project_profile.pages_dir.rstrip("/"),
+                tests_dir=project_profile.tests_dir.rstrip("/"),
+                fixture_name=project_profile.driver_fixture_name,
+            )
+        return CODER_SYSTEM_PROMPT_BDD if mode == "bdd" else CODER_SYSTEM_PROMPT_PYTEST
+
     def _build_user_prompt(self, plan: dict, project_profile, mode: str) -> str:
         project_context = ""
         if project_profile:

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.2}/selenium_agent/agents/healer.py

@@ -483,7 +483,8 @@ class HealerAgent:
 
     # ── Main heal loop ─────────────────────────────────────────────────
 
-    def heal(self, saved_files: list[str], test_filter: str | None = None) -> dict:
+    def heal(self, saved_files: list[str], test_filter: str | None = None,
+             project_profile=None) -> dict:
         resolved = self._resolve_paths(saved_files)
 
         extra = self._auto_discover_related_files(resolved)
@@ -519,7 +520,8 @@ class HealerAgent:
 
             logger.warning(f"❌ Failed on attempt {attempt}")
             try:
-                applied = self._fix_once(resolved, output, test_filter, scan_cache)
+                applied = self._fix_once(resolved, output, test_filter, scan_cache,
+                                         project_profile=project_profile)
             except LLMJSONError as exc:
                 logger.error(f"💥 Healer LLM returned unusable JSON: {exc}")
                 applied = False
@@ -537,7 +539,8 @@ class HealerAgent:
         return {"status": "failed", "attempts": self.max_retries, "output": output}
 
     def _fix_once(self, resolved: dict[str, Path], output: str,
-                  test_filter: str | None, scan_cache: dict[str, str]) -> bool:
+                  test_filter: str | None, scan_cache: dict[str, str],
+                  project_profile=None) -> bool:
         """One LLM fix round. Returns True if at least one file was updated."""
         known_fix = SeleniumErrorMap.get_fix_summary(output)
 
@@ -582,6 +585,18 @@ class HealerAgent:
             system_prompt = HEALER_SYSTEM_PROMPT
             target_instruction = ""
 
+        project_instruction = ""
+        if project_profile is not None:
+            project_instruction = (
+                f"\n\n🏗️ EXISTING-PROJECT MODE — this codebase has its OWN "
+                f"architecture. The profile below OVERRIDES the default "
+                f"architecture/import rules above: mirror the project's "
+                f"patterns exactly, never convert files to the "
+                f"selenium_agent BasePage style, and tests use the fixture "
+                f"named '{project_profile.driver_fixture_name}'.\n\n"
+                f"{project_profile.to_llm_context()}\n"
+            )
+
         raw = self.client.generate_text(
             system_prompt=system_prompt,
             user_prompt=(
@@ -589,7 +604,8 @@ class HealerAgent:
                 f"SELENIUM ERROR ANALYSIS:\n{known_fix}\n\n"
                 f"{locator_context}\n"
                 f"PYTEST OUTPUT:\n{self._trim_output(output)}\n"
-                f"{target_instruction}\n"
+                f"{target_instruction}"
+                f"{project_instruction}\n"
                 f"CURRENT CODE:\n{files_text}"
             ),
             max_tokens=8000,

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.2}/selenium_agent/cli.py

@@ -27,6 +27,8 @@ def _handle_config(argv):
     p.add_argument("--headless",    dest="headless", action="store_true",  default=None)
     p.add_argument("--no-headless", dest="headless", action="store_false", default=None)
     p.add_argument("--mode",     choices=["pytest", "bdd"])
+    p.add_argument("--project",  help="Existing project to fit tests into "
+                                      "(use 'none' to clear)")
     p.add_argument("--show",     action="store_true", help="Print current config")
     args = p.parse_args(argv)
 
@@ -35,6 +37,9 @@ def _handle_config(argv):
         print(json.dumps(config_manager.load(), indent=2))
         return
 
+    if args.project and args.project.lower() in ("none", "clear", "off"):
+        args.project = ""  # empty string clears it (consumers treat "" as unset)
+
     updates = {k: v for k, v in vars(args).items() if v is not None and k != "show"}
     if not updates:
         p.print_help()
@@ -115,8 +120,8 @@ def _handle_help():
   --headless    headless browser
   --explore N   scan N extra same-origin pages while planning
   --output-dir  where to save files         (default: generated_tests)
-  --project     path to existing project
-  --max-retries healer retries              (default: 3)
+  --project     existing project path      (auto-saved to config)
+  --max-retries healer retries              (default: 5)
   --no-heal     skip healing
   --plan-only   preview plan only (saved to specs/)
   --from-plan   generate from saved plan JSON
@@ -196,7 +201,7 @@ Other examples:
     parser.add_argument("--test",          default=None, metavar="TEST_NAME",
                         help="Specific test function to run/heal. "
                              "e.g. --test test_login_locked_out_user")
-    parser.add_argument("--version",      action="version", version="selenium-agent 0.2.0")
+    parser.add_argument("--version",      action="version", version="selenium-agent 0.2.2")
 
     args = parser.parse_args()
 
@@ -210,6 +215,11 @@ Other examples:
         except Exception as e:
             print(f"❌ Scan failed: {e}")
             sys.exit(1)
+        # Remember the project — future runs fit into it automatically,
+        # no need to repeat --project every time.
+        config_manager.save({"project": args.scan})
+        print(f"\n💾 Project saved to config — future runs will fit into it.")
+        print(f"   (change: selenium-agent config --project <path> | clear: --project none)")
         sys.exit(0)
 
     # ── Merge config + CLI args ───────────────────────────────────────
@@ -220,6 +230,16 @@ Other examples:
         "mode":     args.mode,
     })
 
+    # `--model gpt-5` without `--provider` must not hit the default
+    # (Anthropic) API — the model name itself tells us the provider.
+    # An explicit --provider flag always wins.
+    if args.provider is None and cfg.get("model"):
+        from selenium_agent.utils.llm import infer_provider_for_model
+        inferred = infer_provider_for_model(cfg["model"])
+        if inferred and inferred != cfg.get("provider"):
+            logger.info(f"🧭 Model '{cfg['model']}' implies provider '{inferred}'")
+            cfg["provider"] = inferred
+
     # URL priority: --url flag > extracted from instruction > saved base_url
     from selenium_agent.utils.url_extractor import extract_url
     override_url = args.url
@@ -235,6 +255,14 @@ Other examples:
         config_manager.save({"base_url": args.url})
         logger.info(f"💾 base_url saved: {args.url}")
 
+    # Project priority: --project flag > saved config. Persist like --url.
+    project_root = args.project or (cfg.get("project") or None)
+    if args.project and args.project != cfg.get("project"):
+        config_manager.save({"project": args.project})
+        logger.info(f"💾 project saved: {args.project}")
+    elif not args.project and project_root:
+        logger.info(f"🏗️  Using saved project: {project_root}")
+
     # ── Build orchestrator ────────────────────────────────────────────
     try:
         agent = Orchestrator(
@@ -245,7 +273,7 @@ Other examples:
             max_heal_retries=args.max_retries or 5,
             auto_heal=not args.no_heal,
             mode=cfg["mode"],
-            project_root=args.project,
+            project_root=project_root,
             headless=cfg["headless"],
             explore_pages=args.explore,
         )

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.2}/selenium_agent/core/orchestrator.py

@@ -148,7 +148,8 @@ class Orchestrator:
         return self.coder.code(plan, project_profile=self.project_profile)
 
     def heal_only(self, file_paths: list, test_filter: str | None = None) -> dict:
-        return self.healer.heal(file_paths, test_filter=test_filter)
+        return self.healer.heal(file_paths, test_filter=test_filter,
+                                project_profile=self.project_profile)
 
     def scan_only(self, project_root: str) -> str:
         from selenium_agent.scanner.project_scanner import ProjectScanner
@@ -183,7 +184,8 @@ class Orchestrator:
         heal_result = None
         if self.auto_heal:
             logger.info("\n🩺 STEP 3: HEALER")
-            heal_result = self.healer.heal(saved_files)
+            heal_result = self.healer.heal(saved_files,
+                                           project_profile=self.project_profile)
         else:
             logger.info("\n⏭️  Auto-heal skipped")
         return saved_files, heal_result

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.2}/selenium_agent/scanner/project_scanner.py

@@ -54,6 +54,7 @@ class ProjectProfile:
     # Driver setup
     driver_setup: str = "fixture"               # fixture | setUp | class-level
     driver_scope: str = "function"              # function | class | module
+    driver_fixture_name: str = "driver"         # actual fixture name (e.g. 'browser')
     browser: str = "chrome"
     headless: bool = False
 
@@ -91,7 +92,8 @@ class ProjectProfile:
             f"BDD detected      : {self.has_bdd}",
             f"Base Page class   : {self.base_page_class}",
             f"Base Page import  : {self.base_page_import or 'not detected'}",
-            f"Driver setup      : {self.driver_setup} (scope={self.driver_scope})",
+            f"Driver setup      : {self.driver_setup} (scope={self.driver_scope}, "
+            f"fixture name='{self.driver_fixture_name}')",
             f"Browser           : {self.browser} (headless={self.headless})",
             f"Test file pattern : {self.test_file_prefix}<name>.py",
             f"Page file pattern : <name>{self.page_file_suffix}.py",
@@ -169,12 +171,18 @@ class ProjectScanner:
     # ── Detection methods ─────────────────────────────────────────────
 
     def _detect_folders(self):
-        """Find pages, tests, features, steps directories."""
-        all_dirs = [
-            d for d in self.root.rglob("*")
-            if d.is_dir() and not self._is_ignored(d)
-        ]
-        dir_names = {d.name.lower(): d for d in all_dirs}
+        """Find pages, tests, features, steps directories.
+
+        Shallowest directory wins: a project's real `pages/` at the root
+        must beat any deeper `something/pages/` twin."""
+        all_dirs = sorted(
+            (d for d in self.root.rglob("*")
+             if d.is_dir() and not self._is_ignored(d)),
+            key=lambda d: len(d.parts),
+        )
+        dir_names: dict = {}
+        for d in all_dirs:
+            dir_names.setdefault(d.name.lower(), d)
 
         for name in self.PAGES_DIRS:
             if name in dir_names:
@@ -301,11 +309,19 @@ class ProjectScanner:
                 continue
 
             # Fixture pattern
-            if "@pytest.fixture" in content and "driver" in content:
+            if "@pytest.fixture" in content and "driver" in content.lower():
                 self.profile.driver_setup = "fixture"
                 scope_match = re.search(r'@pytest\.fixture\s*\(\s*scope=["\'](\w+)["\']', content)
                 if scope_match:
                     self.profile.driver_scope = scope_match.group(1)
+                # The fixture NAME matters — generated tests must request the
+                # project's actual fixture ('browser', 'web_driver', ...),
+                # not assume it is called 'driver'.
+                name_match = re.search(
+                    r"@pytest\.fixture(?:\([^)]*\))?\s*\ndef\s+(\w+)\s*\(", content
+                )
+                if name_match and f.name == "conftest.py":
+                    self.profile.driver_fixture_name = name_match.group(1)
 
             # unittest setUp pattern
             elif "def setUp(self)" in content:
@@ -445,8 +461,14 @@ class ProjectScanner:
                 continue
 
     def _is_ignored(self, path: Path) -> bool:
-        """Skip venv, cache, git, node_modules etc."""
+        """Skip venv, cache, git, node_modules etc.
+
+        Also skips the agent's OWN output folders (generated_tests/, specs/):
+        scanning them would make the profile describe the agent's previous
+        output instead of the user's real framework — circular pollution.
+        """
         ignored = {".venv", "venv", "__pycache__", ".git", ".idea",
                    "node_modules", ".pytest_cache", "dist", "build",
-                   ".eggs", "*.egg-info", ".tox", ".mypy_cache"}
+                   ".eggs", "*.egg-info", ".tox", ".mypy_cache",
+                   "generated_tests", "specs"}
         return any(part in ignored for part in path.parts)

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.2}/selenium_agent/utils/config_manager.py

@@ -23,6 +23,7 @@ DEFAULTS = {
     "headless":  False,
     "mode":      "pytest",
     "base_url":  None,   # persisted across runs — no need to pass --url repeatedly
+    "project":   None,   # persisted by --scan/--project — fit into this project automatically
 }
 
 

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.2}/selenium_agent/utils/llm.py

@@ -37,6 +37,26 @@ _NON_RETRYABLE = (
 )
 
 
+def infer_provider_for_model(model: str | None) -> str | None:
+    """
+    Infer the provider from a model name — 'gpt-5' implies OpenAI,
+    'claude-fable-5' implies Anthropic.
+
+    Fresh installs have no saved config, so `--model gpt-5` without
+    `--provider openai` would otherwise send an OpenAI model to the
+    default (Anthropic) API and fail with a confusing auth/model error.
+    Returns None when the name matches no known family.
+    """
+    if not model:
+        return None
+    m = model.strip().lower()
+    if m.startswith(("gpt-", "o1", "o3", "o4", "chatgpt")):
+        return "openai"
+    if m.startswith("claude"):
+        return "anthropic"
+    return None
+
+
 def normalize_provider(provider: str | None) -> str:
     normalized = (provider or DEFAULT_PROVIDER).strip().lower()
     if normalized not in API_KEY_ENV_VARS:

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.2/selenium_python_ai_agent.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: selenium-python-ai-agent
-Version: 0.2.0
+Version: 0.2.2
 Summary: AI-powered multi-agent framework to plan, generate and heal Selenium Python tests (pytest + pytest-bdd)
 Author: Ankit Tripathi
 License: MIT
@@ -135,11 +135,17 @@ selenium-agent --heal-only generated_tests/tests/test_checkout.py
 Most teams don't start from scratch — they already have a Selenium project. Point the agent at it with `--project` and it **adapts to your structure instead of imposing its own**:
 
 ```bash
-# First, see what the scanner detects about your project (read-only)
+# First, see what the scanner detects about your project (read-only).
+# The path is SAVED to config — you never need to repeat it.
 selenium-agent --scan /path/to/your/project
 
-# Then generate tests that FIT INTO it
-selenium-agent "test the invoice creation flow" --project /path/to/your/project
+# From now on every run fits into it automatically:
+selenium-agent "test the invoice creation flow"
+selenium-agent "test the user permissions page"
+
+# Manage the saved project:
+selenium-agent config --project /other/project   # switch
+selenium-agent config --project none             # clear
 ```
 
 The built-in **ProjectScanner** detects and follows:
@@ -359,7 +365,7 @@ generated_tests/
 | `--headless` | Headless browser | from config (`false`) |
 | `--explore N` | Explore up to N extra same-origin pages while planning (relevance-ranked, multi-hop) | `0` |
 | `--output-dir` | Where to save generated files | `generated_tests` |
-| `--project` | Fit into existing project path | — |
+| `--project` | Existing project path (saved to config — set once) | from config |
 | `--max-retries` | Healer fix attempts | `5` |
 | `--no-heal` | Skip healing after codegen | `false` |
 | `--plan-only` | Save + show plan, no code | `false` |

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.2}/tests/test_agent_enhancements.py

@@ -333,3 +333,90 @@ def test_token_budget_uses_structure_not_keywords():
         "locators": [],
     }
     assert coder._token_budget(big) == 12000
+
+
+def test_infer_provider_from_model_name():
+    from selenium_agent.utils.llm import infer_provider_for_model
+
+    assert infer_provider_for_model("gpt-5") == "openai"
+    assert infer_provider_for_model("gpt-4o-mini") == "openai"
+    assert infer_provider_for_model("o3-mini") == "openai"
+    assert infer_provider_for_model("claude-fable-5") == "anthropic"
+    assert infer_provider_for_model("claude-sonnet-5") == "anthropic"
+    assert infer_provider_for_model("some-unknown-model") is None
+    assert infer_provider_for_model(None) is None
+
+
+def test_config_persists_project(tmp_path: Path, monkeypatch: pytest.MonkeyPatch):
+    from selenium_agent.utils import config_manager
+
+    monkeypatch.chdir(tmp_path)
+    config_manager.save({"project": "/path/to/my/framework"})
+    cfg = config_manager.load()
+    assert cfg["project"] == "/path/to/my/framework"
+
+    # Clearing: empty string is treated as unset by consumers
+    config_manager.save({"project": ""})
+    assert (config_manager.load()["project"] or None) is None
+
+
+# ── project-native mode ────────────────────────────────────────────────
+
+
+def _make_fake_project(tmp_path: Path):
+    (tmp_path / "pages").mkdir()
+    (tmp_path / "tests").mkdir()
+    (tmp_path / "generated_tests" / "pages").mkdir(parents=True)
+    (tmp_path / "pages" / "home_page.py").write_text(
+        "from utils.helpers import HelperBot\n\nclass HomePage:\n    def __init__(self, driver):\n        self.bot = HelperBot(driver)\n"
+    )
+    (tmp_path / "generated_tests" / "pages" / "old_page.py").write_text(
+        "from selenium_agent.selenium.base_page import BasePage\nclass OldPage(BasePage):\n    pass\n"
+    )
+    (tmp_path / "tests" / "conftest.py").write_text(
+        "import pytest\nfrom selenium import webdriver\n\n"
+        "@pytest.fixture(scope=\"session\")\ndef browser():\n"
+        "    d = webdriver.Chrome()\n    yield d\n    d.quit()\n"
+    )
+    (tmp_path / "tests" / "test_home.py").write_text(
+        "def test_home(browser):\n    assert browser\n"
+    )
+
+
+def test_scanner_ignores_own_output_and_prefers_shallow_dirs(tmp_path: Path):
+    from selenium_agent.scanner.project_scanner import ProjectScanner
+
+    _make_fake_project(tmp_path)
+    profile = ProjectScanner(str(tmp_path)).scan()
+
+    assert profile.pages_dir == "pages"                # not generated_tests/pages
+    assert "selenium_agent" not in profile.sample_page_code  # own output excluded
+    assert "HelperBot" in profile.sample_page_code     # the REAL project style
+
+
+def test_scanner_detects_fixture_name(tmp_path: Path):
+    from selenium_agent.scanner.project_scanner import ProjectScanner
+
+    _make_fake_project(tmp_path)
+    profile = ProjectScanner(str(tmp_path)).scan()
+
+    assert profile.driver_fixture_name == "browser"
+    assert "fixture name='browser'" in profile.to_llm_context()
+
+
+def test_project_native_prompt_replaces_default_architecture(tmp_path: Path):
+    from selenium_agent.agents.coder import CoderAgent
+    from selenium_agent.scanner.project_scanner import ProjectScanner
+
+    _make_fake_project(tmp_path)
+    profile = ProjectScanner(str(tmp_path)).scan()
+    prompt = CoderAgent._system_prompt_for("pytest", profile)
+
+    assert "def test_something(browser):" in prompt
+    assert "generated_tests/" in prompt          # named as FORBIDDEN
+    assert "HelperBot" in prompt                 # project's sample code included
+    assert "selenium_agent.selenium.base_page" not in prompt  # our BasePage gone
+
+    # Standalone mode is unchanged
+    standalone = CoderAgent._system_prompt_for("pytest", None)
+    assert "selenium_agent.selenium.base_page" in standalone

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.2}/tests/test_path_handling.py

@@ -25,7 +25,7 @@ def make_healer(output_dir: Path) -> HealerAgent:
 
 
 def test_package_import_exposes_public_api():
-    assert selenium_agent.__version__ == "0.2.0"
+    assert selenium_agent.__version__ == "0.2.2"
     assert selenium_agent.SeleniumAgent.__name__ == "Orchestrator"