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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: selenium-python-ai-agent
-Version: 0.2.0
+Version: 0.2.1
 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.1}/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.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "selenium-python-ai-agent"
-version = "0.2.0"
+version = "0.2.1"
 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.1}/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.1"
 __author__ = "Ankit Tripathi"
 __all__ = ["SeleniumAgent"]

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.1}/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.1")
 
     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.1}/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.1}/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.1/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.1
 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.1}/tests/test_agent_enhancements.py

@@ -333,3 +333,28 @@ 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

{selenium_python_ai_agent-0.2.0 → selenium_python_ai_agent-0.2.1}/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.1"
     assert selenium_agent.SeleniumAgent.__name__ == "Orchestrator"