VibesAI-api 1.3.2__tar.gz → 1.4.2__tar.gz

{vibesai_api-1.3.2 → vibesai_api-1.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: VibesAI-api
-Version: 1.3.2
+Version: 1.4.2
 Summary: Unofficial Python API client for vibes.ai
 Author-email: Ashiq Hussain Mir <imsrenpsycho@gmail.com>
 License: MIT
@@ -29,7 +29,7 @@ This package was built by reverse-engineering the vibes.ai Next.js bundles and o
 
 ```bash
 # 1) Install
-pip install -e /home/z/my-project/download/vibes-api
+pip install VibesAI-api
 
 # 2) Set your cookie (grab from DevTools → Application → Cookies → vibes.ai)
 export VIBES_META_SESSION="e60e910a-242a-...-K54E"
@@ -555,7 +555,7 @@ python /home/z/my-project/download/vibes-api/examples/01_generate_video.py
 
 ```bash
 # Install in editable mode
-pip install -e /home/z/my-project/download/vibes-api
+pip install VibesAI-api
 
 # Run tests (requires VIBES_META_SESSION env var)
 pytest

{vibesai_api-1.3.2 → vibesai_api-1.4.2}/README.md

@@ -10,7 +10,7 @@ This package was built by reverse-engineering the vibes.ai Next.js bundles and o
 
 ```bash
 # 1) Install
-pip install -e /home/z/my-project/download/vibes-api
+pip install VibesAI-api
 
 # 2) Set your cookie (grab from DevTools → Application → Cookies → vibes.ai)
 export VIBES_META_SESSION="e60e910a-242a-...-K54E"
@@ -536,7 +536,7 @@ python /home/z/my-project/download/vibes-api/examples/01_generate_video.py
 
 ```bash
 # Install in editable mode
-pip install -e /home/z/my-project/download/vibes-api
+pip install VibesAI-api
 
 # Run tests (requires VIBES_META_SESSION env var)
 pytest

{vibesai_api-1.3.2 → vibesai_api-1.4.2}/VibesAI_api.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: VibesAI-api
-Version: 1.3.2
+Version: 1.4.2
 Summary: Unofficial Python API client for vibes.ai
 Author-email: Ashiq Hussain Mir <imsrenpsycho@gmail.com>
 License: MIT
@@ -29,7 +29,7 @@ This package was built by reverse-engineering the vibes.ai Next.js bundles and o
 
 ```bash
 # 1) Install
-pip install -e /home/z/my-project/download/vibes-api
+pip install VibesAI-api
 
 # 2) Set your cookie (grab from DevTools → Application → Cookies → vibes.ai)
 export VIBES_META_SESSION="e60e910a-242a-...-K54E"
@@ -555,7 +555,7 @@ python /home/z/my-project/download/vibes-api/examples/01_generate_video.py
 
 ```bash
 # Install in editable mode
-pip install -e /home/z/my-project/download/vibes-api
+pip install VibesAI-api
 
 # Run tests (requires VIBES_META_SESSION env var)
 pytest

{vibesai_api-1.3.2 → vibesai_api-1.4.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "VibesAI-api"
-version = "1.3.2"
+version = "1.4.2"
 description = "Unofficial Python API client for vibes.ai"
 readme = "README.md"
 license = { text = "MIT" }

{vibesai_api-1.3.2 → vibesai_api-1.4.2}/tests/test_all.py

@@ -42,9 +42,9 @@ class TestImports(unittest.TestCase):
         for name in vibes_api.__all__:
             self.assertTrue(hasattr(vibes_api, name), f"Missing export: {name}")
 
-    def test_version_is_1_3_0(self):
+    def test_version_is_1_4_2(self):
         import vibes_api
-        self.assertEqual(vibes_api.__version__, "1.3.0")
+        self.assertEqual(vibes_api.__version__, "1.4.2")
 
     def test_no_syntax_errors(self):
         import vibes_api.client

{vibesai_api-1.3.2 → vibesai_api-1.4.2}/vibes_api/__init__.py

@@ -94,7 +94,7 @@ from .ingredients import (
 )
 from .composition import Composition
 
-__version__ = "1.3.0"
+__version__ = "1.4.2"
 __all__ = [
     # Client
     "VibesClient",

{vibesai_api-1.3.2 → vibesai_api-1.4.2}/vibes_api/client.py

@@ -182,6 +182,7 @@ class VibesClient:
         on_cookie_refresh: Optional[callable] = None,
         on_session_expired: Optional[callable] = None,
         auto_relogin: bool = False,
+        auth_meta_cookies: Optional[str] = None,
     ):
         self.base_url = base_url.rstrip("/")
         self.timeout = timeout
@@ -191,6 +192,7 @@ class VibesClient:
         self.on_cookie_refresh = on_cookie_refresh
         self.on_session_expired = on_session_expired
         self.auto_relogin = auto_relogin
+        self.auth_meta_cookies = auth_meta_cookies
         self._background_thread = None
         self._background_stop = None
         self._session_expired = False
@@ -213,11 +215,7 @@ class VibesClient:
         self._session_expired = False
 
     def update_cookie(self, meta_session: str) -> None:
-        """Update the session cookie at runtime.
-
-        Use this when the ``on_session_expired`` callback fires and
-        you've obtained a fresh cookie (e.g., from the browser).
-        """
+        """Update the session cookie at runtime."""
         self._set_cookie(meta_session)
         if self.on_cookie_refresh:
             try:
@@ -225,16 +223,24 @@ class VibesClient:
             except Exception:
                 pass
 
+    def set_auth_meta_cookies(self, cookies: str) -> None:
+        """Set the auth.meta.com cookies for automatic re-login.
+
+        Parameters
+        ----------
+        cookies : str
+            Cookie string from auth.meta.com (e.g.,
+            ``"ps_n=1;datr=...;fs=...;locale=en_GB;ps_l=1"``).
+            Get these from DevTools → Application → Cookies → auth.meta.com.
+        """
+        self.auth_meta_cookies = cookies
+        self.auto_relogin = True  # Enable auto re-login
+
     def get_current_cookie(self) -> str:
         return self._current_meta_session
 
     def _maybe_refresh_cookie(self, resp: requests.Response) -> None:
-        """Intercept Set-Cookie headers and update the session cookie.
-
-        Note: vibes.ai does NOT send Set-Cookie on regular API responses.
-        This is kept for completeness — if the server ever adds sliding
-        session behavior, this will automatically pick it up.
-        """
+        """Intercept Set-Cookie headers and update the session cookie."""
         if not self.auto_refresh:
             return
         set_cookie = resp.headers.get("Set-Cookie") or resp.headers.get("set-cookie")
@@ -258,19 +264,22 @@ class VibesClient:
 
         Returns True if the session was successfully restored.
         """
+        # Don't retry re-login if we're already in the middle of one
+        if self._session_expired:
+            return False
+
         self._session_expired = True
 
-        # Call the session expired callback
         if self.on_session_expired:
             try:
                 self.on_session_expired(self._current_meta_session)
             except Exception:
                 pass
 
-        # Attempt auto re-login via headless browser
-        if self.auto_relogin:
+        if self.auto_relogin and self.auth_meta_cookies:
             new_cookie = self._attempt_oidc_relogin()
             if new_cookie:
+                # _set_cookie resets _session_expired to False
                 self._set_cookie(new_cookie)
                 if self.on_cookie_refresh:
                     try:
@@ -279,60 +288,117 @@ class VibesClient:
                         pass
                 return True
 
+        # Reset the flag so future 401s can attempt re-login again
+        self._session_expired = False
         return False
 
     def _attempt_oidc_relogin(self) -> Optional[str]:
-        """Attempt to re-login via the OIDC flow using a headless browser.
+        """Attempt to re-login via the OIDC flow using auth.meta.com cookies.
 
-        This navigates to ``/api/meta-oidc/start`` which redirects to
-        Meta's OIDC login. If the browser has an active Meta session
-        (from facebook.com/meta.com cookies), the flow completes
-        automatically and a new ``meta_session`` cookie is set.
+        Follows the full OIDC redirect chain using requests.get() (not Session)
+        to avoid cookie/header interference between steps.
 
-        Returns the new cookie value, or None if re-login failed.
+        Returns the new meta_session cookie value, or None if re-login failed.
         """
-        import subprocess, json as _json
+        if not self.auth_meta_cookies:
+            return None
 
-        try:
-            # Use agent-browser CLI to navigate to the OIDC start URL
-            # The browser must have Meta cookies for this to work
-            result = subprocess.run(
-                ["agent-browser", "open", f"{self.base_url}/api/meta-oidc/start"],
-                capture_output=True, text=True, timeout=30,
-            )
-            if result.returncode != 0:
-                return None
+        ua = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36"
 
-            # Wait for redirects to complete
-            import time as _time
-            _time.sleep(3)
+        for attempt in range(3):
+            try:
+                # Step 1: Start OIDC flow at vibes.ai
+                r1 = requests.get(
+                    f"{self.base_url}/api/meta-oidc/start",
+                    headers={"User-Agent": ua, "Cookie": "cookie_ack=true"},
+                    allow_redirects=False, timeout=15,
+                )
+                if r1.status_code not in (302, 307):
+                    if attempt < 2:
+                        time.sleep(5)
+                        continue
+                    return None
+                redirect_url = r1.headers.get("Location", "")
+                if not redirect_url:
+                    return None
+
+                # Extract oauth_csrf_token from Set-Cookie
+                csrf_token = None
+                for part in r1.headers.get("Set-Cookie", "").split(";"):
+                    if part.strip().startswith("oauth_csrf_token="):
+                        csrf_token = part.strip()[len("oauth_csrf_token="):]
+                        break
+
+                # Step 2: auth.meta.com (with auth.meta.com cookies only)
+                r2 = requests.get(
+                    redirect_url,
+                    headers={"User-Agent": ua, "Cookie": self.auth_meta_cookies},
+                    allow_redirects=False, timeout=15,
+                )
+                if r2.status_code not in (302, 307):
+                    # 400 = auth.meta.com rate limited or cookies expired
+                    # Don't retry — makes rate limiting worse
+                    return None
+                redirect2 = r2.headers.get("Location", "")
+                if not redirect2 or "code=" not in redirect2:
+                    return None
+
+                # Step 3: auth.meta.ai/ecto (code exchange)
+                r3 = requests.get(
+                    redirect2,
+                    headers={"User-Agent": ua},
+                    allow_redirects=False, timeout=15,
+                )
+                if r3.status_code not in (302, 307):
+                    return None
+                redirect3 = r3.headers.get("Location", "")
+                if not redirect3 or "vibes.ai" not in redirect3:
+                    return None
+
+                # Step 4: vibes.ai callback (CSRF token only, no old session)
+                callback_cookie = "cookie_ack=true"
+                if csrf_token:
+                    callback_cookie += f";oauth_csrf_token={csrf_token}"
+                r4 = requests.get(
+                    redirect3,
+                    headers={"User-Agent": ua, "Cookie": callback_cookie},
+                    allow_redirects=False, timeout=15,
+                )
 
-            # Get the current URL (should be back on vibes.ai)
-            result = subprocess.run(
-                ["agent-browser", "get", "url"],
-                capture_output=True, text=True, timeout=10,
-            )
-            current_url = result.stdout.strip()
+                # Check Set-Cookie for new meta_session
+                set_cookie = r4.headers.get("Set-Cookie", "").replace(",", ";")
+                for part in set_cookie.split(";"):
+                    part = part.strip()
+                    if part.startswith("meta_session="):
+                        val = part[len("meta_session="):]
+                        if val and val != "deleted" and val != self._current_meta_session:
+                            return val
+
+                # Step 5: Follow final redirect if no cookie in step 4
+                final_redirect = r4.headers.get("Location", "")
+                if final_redirect and "error=" not in final_redirect:
+                    if not final_redirect.startswith("http"):
+                        final_redirect = f"{self.base_url}{final_redirect}"
+                    r5 = requests.get(
+                        final_redirect,
+                        headers={"User-Agent": ua, "Cookie": "cookie_ack=true"},
+                        allow_redirects=True, timeout=15,
+                    )
+                    set_cookie_5 = r5.headers.get("Set-Cookie", "").replace(",", ";")
+                    for part in set_cookie_5.split(";"):
+                        part = part.strip()
+                        if part.startswith("meta_session="):
+                            val = part[len("meta_session="):]
+                            if val and val != "deleted" and val != self._current_meta_session:
+                                return val
 
-            # Check if we're back on vibes.ai (OIDC callback completed)
-            if "vibes.ai" not in current_url:
                 return None
 
-            # Extract the new meta_session cookie
-            result = subprocess.run(
-                ["agent-browser", "cookies"],
-                capture_output=True, text=True, timeout=10,
-            )
-            cookies_text = result.stdout.strip()
-            for line in cookies_text.split("\n"):
-                line = line.strip()
-                if line.startswith("meta_session="):
-                    new_cookie = line[len("meta_session="):]
-                    if new_cookie and new_cookie != "deleted":
-                        return new_cookie
-
-        except (subprocess.TimeoutExpired, FileNotFoundError, Exception):
-            pass
+            except Exception:
+                if attempt < 2:
+                    time.sleep(5)
+                    continue
+                pass
 
         return None
 
@@ -449,9 +515,10 @@ class VibesClient:
         except VibesAPIError as e:
             if e.status == 401 and not self._session_expired:
                 if self._handle_401():
-                    # Session was restored — retry the request
+                    # Session was restored — retry the request ONCE
                     resp = self.session.get(self._url(path), params=params, timeout=self.timeout, **kw)
                     return self._check(resp)
+            # Re-login failed or not enabled — raise the original error
             raise
 
     def _post(self, path: str, json_body: Optional[dict] = None, **kw) -> dict: