selenium-python-ai-agent 0.2.1__tar.gz → 0.2.3__tar.gz

{selenium_python_ai_agent-0.2.1/selenium_python_ai_agent.egg-info → selenium_python_ai_agent-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: selenium-python-ai-agent
-Version: 0.2.1
+Version: 0.2.3
 Summary: AI-powered multi-agent framework to plan, generate and heal Selenium Python tests (pytest + pytest-bdd)
 Author: Ankit Tripathi
 License: MIT
@@ -35,7 +35,7 @@ Dynamic: license-file
 
 [![PyPI version](https://badge.fury.io/py/selenium-python-ai-agent.svg)](https://pypi.org/project/selenium-python-ai-agent/)
 [![Python](https://img.shields.io/badge/python-3.9+-blue.svg)](https://python.org)
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://github.com/AnkitDeepBlue/selenium-python-ai-agent/blob/main/LICENSE)
 
 ---
 
@@ -405,7 +405,7 @@ Subcommands: `selenium-agent config …`, `selenium-agent init-agents …`, `sel
 
 ## 🤝 Contributing
 
-PRs welcome! See [CONTRIBUTING.md](CONTRIBUTING.md)
+PRs welcome! See [CONTRIBUTING.md](https://github.com/AnkitDeepBlue/selenium-python-ai-agent/blob/main/CONTRIBUTING.md)
 
 ---
 

{selenium_python_ai_agent-0.2.1 → selenium_python_ai_agent-0.2.3}/README.md

@@ -5,7 +5,7 @@
 
 [![PyPI version](https://badge.fury.io/py/selenium-python-ai-agent.svg)](https://pypi.org/project/selenium-python-ai-agent/)
 [![Python](https://img.shields.io/badge/python-3.9+-blue.svg)](https://python.org)
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://github.com/AnkitDeepBlue/selenium-python-ai-agent/blob/main/LICENSE)
 
 ---
 
@@ -375,7 +375,7 @@ Subcommands: `selenium-agent config …`, `selenium-agent init-agents …`, `sel
 
 ## 🤝 Contributing
 
-PRs welcome! See [CONTRIBUTING.md](CONTRIBUTING.md)
+PRs welcome! See [CONTRIBUTING.md](https://github.com/AnkitDeepBlue/selenium-python-ai-agent/blob/main/CONTRIBUTING.md)
 
 ---
 

{selenium_python_ai_agent-0.2.1 → selenium_python_ai_agent-0.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "selenium-python-ai-agent"
-version = "0.2.1"
+version = "0.2.3"
 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.1 → selenium_python_ai_agent-0.2.3}/selenium_agent/__init__.py

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

{selenium_python_ai_agent-0.2.1 → selenium_python_ai_agent-0.2.3}/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.1 → selenium_python_ai_agent-0.2.3}/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.1 → selenium_python_ai_agent-0.2.3}/selenium_agent/cli.py

@@ -201,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.1")
+    parser.add_argument("--version",      action="version", version="selenium-agent 0.2.3")
 
     args = parser.parse_args()
 

{selenium_python_ai_agent-0.2.1 → selenium_python_ai_agent-0.2.3}/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.1 → selenium_python_ai_agent-0.2.3}/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.1 → selenium_python_ai_agent-0.2.3/selenium_python_ai_agent.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: selenium-python-ai-agent
-Version: 0.2.1
+Version: 0.2.3
 Summary: AI-powered multi-agent framework to plan, generate and heal Selenium Python tests (pytest + pytest-bdd)
 Author: Ankit Tripathi
 License: MIT
@@ -35,7 +35,7 @@ Dynamic: license-file
 
 [![PyPI version](https://badge.fury.io/py/selenium-python-ai-agent.svg)](https://pypi.org/project/selenium-python-ai-agent/)
 [![Python](https://img.shields.io/badge/python-3.9+-blue.svg)](https://python.org)
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://github.com/AnkitDeepBlue/selenium-python-ai-agent/blob/main/LICENSE)
 
 ---
 
@@ -405,7 +405,7 @@ Subcommands: `selenium-agent config …`, `selenium-agent init-agents …`, `sel
 
 ## 🤝 Contributing
 
-PRs welcome! See [CONTRIBUTING.md](CONTRIBUTING.md)
+PRs welcome! See [CONTRIBUTING.md](https://github.com/AnkitDeepBlue/selenium-python-ai-agent/blob/main/CONTRIBUTING.md)
 
 ---
 

{selenium_python_ai_agent-0.2.1 → selenium_python_ai_agent-0.2.3}/tests/test_agent_enhancements.py

@@ -358,3 +358,65 @@ def test_config_persists_project(tmp_path: Path, monkeypatch: pytest.MonkeyPatch
     # 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.1 → selenium_python_ai_agent-0.2.3}/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.1"
+    assert selenium_agent.__version__ == "0.2.3"
     assert selenium_agent.SeleniumAgent.__name__ == "Orchestrator"