🚚 chore: Move credential UserAuth.py to a new location

LlmKira · Feb 17, 2024 · 7e0ea99 · 7e0ea99
1 parent 3e92ea3
commit 7e0ea99
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 8 deletions.
diff --git a/src/novelai_python/credential/ApiToken.py b/src/novelai_python/credential/ApiToken.py
@@ -25,12 +25,12 @@ async def get_session(self, timeout: int = 180, update_headers: dict = None):
                 "Accept": "*/*",
                 "Accept-Language": "zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2",
                 "Accept-Encoding": "gzip, deflate, br",
-                "User-Agent": FAKE_UA.chrome,
+                "User-Agent": FAKE_UA.edge,
                 "Authorization": f"Bearer {self.api_token.get_secret_value()}",
                 "Content-Type": "application/json",
                 "Origin": "https://novelai.net",
                 "Referer": "https://novelai.net/",
-            }, impersonate="chrome110")
+            }, impersonate="edge101")
         self._session.headers.update(update_headers)
         return self._session
 

diff --git a/src/novelai_python/credential/JwtToken.py b/src/novelai_python/credential/JwtToken.py
@@ -24,13 +24,13 @@ async def get_session(self, timeout: int = 180, update_headers: dict = None):
             self._session = AsyncSession(timeout=timeout, headers={
                 "Accept": "*/*",
                 "Accept-Language": "zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2",
-                "User-Agent": FAKE_UA.chrome,
+                "User-Agent": FAKE_UA.edge,
                 "Accept-Encoding": "gzip, deflate, br",
                 "Authorization": f"Bearer {self.jwt_token.get_secret_value()}",
                 "Content-Type": "application/json",
                 "Origin": "https://novelai.net",
                 "Referer": "https://novelai.net/",
-            }, impersonate="chrome110")
+            }, impersonate="edge101")
         assert isinstance(update_headers, dict), "update_headers must be a dict"
         self._session.headers.update(update_headers)
         return self._session

diff --git a/src/novelai_python/credential/UserAuth.py b/src/novelai_python/credential/UserAuth.py
@@ -30,14 +30,14 @@ async def get_session(self, timeout: int = 180, update_headers: dict = None):
             resp = await Login.build(user_name=self.username, password=self.password.get_secret_value()).request()
             self._session = AsyncSession(timeout=timeout, headers={
                 "Accept": "*/*",
-                "User-Agent": FAKE_UA.chrome,
+                "User-Agent": FAKE_UA.edge,
                 "Accept-Language": "zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2",
                 "Accept-Encoding": "gzip, deflate, br",
                 "Authorization": f"Bearer {resp.accessToken}",
                 "Content-Type": "application/json",
                 "Origin": "https://novelai.net",
                 "Referer": "https://novelai.net/",
-            }, impersonate="chrome110")
+            }, impersonate="edge101")
             self._update_at = int(time.time())
         self._session.headers.update(update_headers)
         return self._session
diff --git a/src/novelai_python/credential/_base.py b/src/novelai_python/credential/_base.py
@@ -9,8 +9,6 @@
 
 FAKE_UA = UserAgent()
 
-print(FAKE_UA.chrome)
-
 
 class CredentialBase(BaseModel):
     _session: AsyncSession = None