Add OpenAIChatGPTAdapter

Author: code-yeongyu

aishell/adapters/__init__.py

@@ -0,0 +1 @@
+from .openai_chatgpt_adapter import OpenAIChatGPTAdapter as OpenAIChatGPTAdapter

aishell/adapters/openai_chatgpt_adapter.py

@@ -0,0 +1,27 @@
+from http.cookiejar import Cookie
+from typing import Optional
+
+from yt_dlp.cookies import SUPPORTED_BROWSERS, extract_cookies_from_browser
+from yt_dlp.utils import YoutubeDLCookieJar
+
+
+class OpenAIChatGPTAdapter:
+
+    def __init__(self, browser_name: str):
+        if browser_name not in SUPPORTED_BROWSERS:  # type: ignore
+            raise ValueError(f'Browser {browser_name} is not supported. Supported browsers are: {SUPPORTED_BROWSERS}')
+        self.BROWSER_NAME = browser_name
+
+    def get_openai_cookies(self) -> dict[str, Optional[str]]:
+        all_cookies: YoutubeDLCookieJar = extract_cookies_from_browser(self.BROWSER_NAME)
+        try:
+            openai_cookies: dict[str, Cookie] = all_cookies.__dict__['_cookies']['chat.openai.com']['/']
+        except KeyError as error:
+            raise ValueError('Could not find OpenAI cookies. Make sure you are logged in to OpenAI.') from error
+
+        cookies: dict[str, Optional[str]] = {key: cookie.value for key, cookie in openai_cookies.items()}
+
+        return cookies
+
+    def get_openai_session_token(self) -> Optional[str]:
+        return self.get_openai_cookies()['__Secure-next-auth.session-token']

aishell/adapters/test/__init__.py

@@ -0,0 +1,17 @@
+import pytest
+from yt_dlp.cookies import SUPPORTED_BROWSERS
+
+from aishell.adapters import OpenAIChatGPTAdapter
+
+
+class TestChatGPTAccessTokenAdapterInit:
+
+    def test_fail_when_invalid_browser_name_is_given(self) -> None:
+        '''잘못된 브라우저 이름을 입력했을 때 ValueError가 발생한다.'''
+        with pytest.raises(ValueError):
+            OpenAIChatGPTAdapter('invalid_browser_name')
+
+    @pytest.mark.parametrize('browser_name', SUPPORTED_BROWSERS)
+    def test_success_when_valid_browser_name_is_given(self, browser_name: str) -> None:
+        '''올바른 브라우저 이름을 입력했을 때 ValueError가 발생하지 않는다'''
+        OpenAIChatGPTAdapter(browser_name)