added gpt4v helpers and browser context

Author: mishushakov

browserbase/__init__.py

@@ -19,7 +19,8 @@ def load(self, url: str):
             browser = p.chromium.connect_over_cdp(
                 "wss://api.browserbase.com?apiKey=" + self.api_key
             )
-            page = browser.new_page()
+            default_context = browser.contexts[0]
+            page = default_context.pages[0]
             page.goto(url)
             html = page.content()
             browser.close()
@@ -36,8 +37,10 @@ def load_urls(self, urls: List[str]):
                 "wss://api.browserbase.com?apiKey=" + self.api_key
             )
 
+            default_context = browser.contexts[0]
+            page = default_context.pages[0]
+
             for url in urls:
-                page = browser.new_page()
                 page.goto(url)
                 yield page.content()
 
@@ -52,6 +55,7 @@ def screenshot(self, url: str, full_page: bool = False):
             browser = p.chromium.connect_over_cdp(
                 "wss://api.browserbase.com?apiKey=" + self.api_key
             )
+
             page = browser.new_page()
             page.goto(url)
             screenshot = page.screenshot(full_page=full_page)

browserbase/helpers/gpt4v.py

@@ -0,0 +1,15 @@
+from base64 import b64encode
+
+
+def GPT4Image(img: bytes, detail: str):
+    if not img:
+        raise ValueError("Image was not provided")
+
+    img_encoded = b64encode(img).decode()
+    return {
+        "type": "image_url",
+        "image_url": {
+            "url": "data:image/jpeg;base64," + img_encoded,
+            "detail": detail,
+        },
+    }