diff --git a/scrape.py b/scrape.py index 714df1e..7abb226 100644 --- a/scrape.py +++ b/scrape.py @@ -143,13 +143,12 @@ class Tools: def __init__(self, req_limit: int): super().__init__(req_limit=req_limit) - class Valves(BaseModel): - request_limit: int = Field(default=5, description="Request Limit") + class UserValves(BaseModel): single_request: bool = Field(default=False, description="Single Request") pass - class UserValves(BaseModel): - single_request: bool = Field(default=False, description="Single Request") + class Valves(UserValves): + request_limit: int = Field(default=5, description="Request Limit") pass def __init__(self): @@ -168,7 +167,7 @@ class Tools: :params url: The URL of the web page to be scraped. """ - if self.user_valves.single_request: + if self.user_valves.single_request or self.valves.single_request: request_limit = 1 else: request_limit = self.valves.request_limit