Update app/core/service/playwright/models.py

app/core/service/playwright/models.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from typing import Literal
 
-from pydantic import BaseModel, Field, HttpUrl, ValidationInfo, field_validator
+from pydantic import BaseModel, Field, HttpUrl
 
 
 class ViewPortModel(BaseModel):
@@ -22,20 +22,8 @@ class GetContentModel(BaseModel):
     url: HttpUrl
     new_browser: bool | None = False
     query_selector: str | None = None
-    wait_selector: bool | None = False
     ms_delay: float = Field(default=0.0, gt=-1, lt=15_000.1)
 
-    @field_validator("wait_selector")
-    @classmethod
-    def check_query_selector(cls, v: bool | None, info: ValidationInfo) -> bool | None:
-        if v is True and info.data.get("query_selector") is None:
-            msg = "wait_selector cannot be set to True without specifying a query_selector"
-            raise ValueError(
-                msg,
-            )
-        return v
-
 
 class ScreenshotModel(GetContentModel):
     full_page: bool | None = False
-