selenium-python-ai-agent 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

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.2"
 __author__ = "Ankit Tripathi"
 __all__ = ["SeleniumAgent"]

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_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_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.2")
 
     args = parser.parse_args()
 

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_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.dist-info → selenium_python_ai_agent-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: selenium-python-ai-agent
-Version: 0.2.1
+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

{selenium_python_ai_agent-0.2.1.dist-info → selenium_python_ai_agent-0.2.2.dist-info}/RECORD

@@ -1,17 +1,17 @@
-selenium_agent/__init__.py,sha256=3t43wB1vdxMTtwnGRdfNGRwIB2cbqAhl9hUq5rjNTsg,603
-selenium_agent/cli.py,sha256=-tV39MBz5MXr1r83iRxfvbqK34JFaj4VWoaF_wbNXoU,15435
+selenium_agent/__init__.py,sha256=Tz7CXR4KoDc5_KZv7geXV7zZ9UKAyEgE2uTbLPmcfic,603
+selenium_agent/cli.py,sha256=l0MU95DYB_4zyihe5QhdrcvwsVDFXfXCj-TYLHBpsEA,15435
 selenium_agent/agents/__init__.py,sha256=7QaYnHxDYyTrmXVdiMXy12FpDhtWpCOnj2UkR22-k1Y,216
-selenium_agent/agents/coder.py,sha256=7wIhkVig20xtXUProiUf1-xh8TWexsk15P6rNCh_Qc4,21449
+selenium_agent/agents/coder.py,sha256=-5WR9RSsAWBxB2R97JcUdiAuQqV_aZnEbTKUOgoNdc8,24313
 selenium_agent/agents/definitions.py,sha256=RodYbSq6ZsUgwxmse3nPZRyAPJ5SW2kQONaZjaZL_pU,5366
-selenium_agent/agents/healer.py,sha256=ctP50_PZAS963k7H2D03ZQaWnhajeZweXhwb0Les7Sc,28337
+selenium_agent/agents/healer.py,sha256=834aAEXprm6Rmdkgv9AcwzzkvndHEHqEUN0BUtbtfXQ,29151
 selenium_agent/agents/planner.py,sha256=Qs3L0-5eagGAMOcicONjkVhayyvRMxX6F7G9OHo1FyI,10433
 selenium_agent/bdd/__init__.py,sha256=0fsSGdNXcNy9Vjmt-KpG8Dx_IHE07UpmNT3abBEuwNk,358
 selenium_agent/bdd/gherkin_advisor.py,sha256=sRnYOMWejCZ17xgpYNgh5nFnJ4hZ5LgGZii7zRKufVI,6873
 selenium_agent/bdd/templates.py,sha256=yQ6ssK0cRpfOXz5mIuNNUY7yrBQtnQmGw7F8Kj8Vlu0,2105
 selenium_agent/core/__init__.py,sha256=hdKa4Oxd99FGeHpeYsUoEkRgqdvyN6JxVX_i4SOVcVQ,86
-selenium_agent/core/orchestrator.py,sha256=N1uWsTJVNU8lFlLFnYt-M0oQd87NPJKUOrfMpsLdGUI,8211
+selenium_agent/core/orchestrator.py,sha256=bBGtkQMKUSFHcOgJiZKoBBTt2a1GvRlvXlO9vSeFNzQ,8362
 selenium_agent/scanner/__init__.py,sha256=RIsY5NnOEMC-2FBOTTeueA53ei2tIgiKb3oCP4LmSlw,130
-selenium_agent/scanner/project_scanner.py,sha256=oaXngF4uz8i9by20-HP1xdmSyJn1wBt0yLLsbHHbDso,17632
+selenium_agent/scanner/project_scanner.py,sha256=oUAMbYBIoOedGfUFmVSnyQ8KzOtzIowOuQ8y9Zbz7kw,18783
 selenium_agent/selenium/__init__.py,sha256=ENE_t5NwiLbD1Hzry3f1DWnkMkoRGa0eop-UHmJMfos,329
 selenium_agent/selenium/base_page.py,sha256=ChrTtXM35QjsQ74seyudHOWYaOeliXdOjCmODM54bwA,18105
 selenium_agent/selenium/driver_factory.py,sha256=o6b1NgAbpPZHbmfFqy_Gk4DHGX3bcW2MepdxcfiMOPI,8927
@@ -27,9 +27,9 @@ selenium_agent/utils/logger.py,sha256=wIttutxhMXkADarPgad3PjyzD3hXwN7GzY3phsa07W
 selenium_agent/utils/paths.py,sha256=UWz30G7H86gD5DHPfwaHky3iuw5kfD4nJj-PilU98so,1864
 selenium_agent/utils/spec_writer.py,sha256=HT86mRKoe063lM9kR7xUb1R4GaBZEqE0VWi4L0uBc1o,4552
 selenium_agent/utils/url_extractor.py,sha256=ibRUeQdmA8tMV89340WH9G8SVPppsOzX-l1QinMnV0I,1684
-selenium_python_ai_agent-0.2.1.dist-info/licenses/LICENSE,sha256=MjRANRVeUVoTsw3zJoK-NiHLwTKjMHNl9QrG1l8u4ds,1071
-selenium_python_ai_agent-0.2.1.dist-info/METADATA,sha256=3CHptknH0AOqsshTycigiegx6lS5AqHI11sbyXxxE_I,20136
-selenium_python_ai_agent-0.2.1.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
-selenium_python_ai_agent-0.2.1.dist-info/entry_points.txt,sha256=fZIhpEO_RKRV-t-rahZgQ1XXzlQN2i9zas5CmJKNF5k,59
-selenium_python_ai_agent-0.2.1.dist-info/top_level.txt,sha256=D7yM-RyAopuwik-nPbvgT6hXagm0aX0U5FQkFDW3yA8,15
-selenium_python_ai_agent-0.2.1.dist-info/RECORD,,
+selenium_python_ai_agent-0.2.2.dist-info/licenses/LICENSE,sha256=MjRANRVeUVoTsw3zJoK-NiHLwTKjMHNl9QrG1l8u4ds,1071
+selenium_python_ai_agent-0.2.2.dist-info/METADATA,sha256=B5FPPDEPtUEDN8LupKPLhuicdEBn2rRImxbyrWvdBws,20136
+selenium_python_ai_agent-0.2.2.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+selenium_python_ai_agent-0.2.2.dist-info/entry_points.txt,sha256=fZIhpEO_RKRV-t-rahZgQ1XXzlQN2i9zas5CmJKNF5k,59
+selenium_python_ai_agent-0.2.2.dist-info/top_level.txt,sha256=D7yM-RyAopuwik-nPbvgT6hXagm0aX0U5FQkFDW3yA8,15
+selenium_python_ai_agent-0.2.2.dist-info/RECORD,,