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

app/core/service/playwright/playwright_context.py

@@ -39,23 +39,25 @@ class AsyncPlaywrightContext:
             self.default_context = await self.browser.new_context()
         return self
 
-   async def new_context_page(
-       self,
-       screenshot_model,
-       browser, page_model) -> Page:
-    params = {
-        'color_scheme': page_model.color_scheme,
-        'java_script_enabled': page_model.java_script_enabled,
-        'no_viewport': page_model.no_viewport,
-        'proxy': page_model.proxy.model_dump() if page_model.proxy else None,
-        'viewport': page_model.viewport.model_dump() if page_model.viewport else None,
-    }
-
-    if not screenshot_model.new_browser:
-        return await self.browser.new_page(**params)
-
-    new_context = await browser.new_context(**params)
-    return await new_context.new_page()
+    async def new_context_page(
+        self,
+        screenshot_model: GetContentModel,
+        page_model: PageModel,
+    ) -> Page:
+            
+        params = {
+            'color_scheme': page_model.color_scheme,
+            'java_script_enabled': page_model.java_script_enabled,
+            'no_viewport': page_model.no_viewport,
+            'proxy': page_model.proxy.model_dump() if page_model.proxy else None,
+            'viewport': page_model.viewport.model_dump() if page_model.viewport else None,
+        }
+        
+        if not screenshot_model.new_browser:
+            return await self.browser.new_page(**params)
+        
+        new_context = await browser.new_context(**params)
+        return await new_context.new_page()
 
 
     async def screenshot(
@@ -130,4 +132,4 @@ class AsyncPlaywrightContext:
         # if self.browser:
         #     await self.browser.close()
         # if self.playwright:
-        #     await self.playwright.stop()
+        #     await self.playwright.stop()