From 3982f39424ea037aca1086d45c6f657b4bfc457c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E2=B2=98r=E1=B9=A8h=E0=B8=AA=E2=88=82ow?=
 <71973368+MrShadowDev@users.noreply.github.com>
Date: Mon, 23 Oct 2023 09:46:25 +0200
Subject: 'Refactored by Sourcery' (#1125)

Co-authored-by: Sourcery AI <>
---
 g4f/Provider/deprecated/Wuguokai.py | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

(limited to 'g4f/Provider/deprecated/Wuguokai.py')

diff --git a/g4f/Provider/deprecated/Wuguokai.py b/g4f/Provider/deprecated/Wuguokai.py
index 311131cf..079f0541 100644
--- a/g4f/Provider/deprecated/Wuguokai.py
+++ b/g4f/Provider/deprecated/Wuguokai.py
@@ -41,15 +41,20 @@ class Wuguokai(BaseProvider):
             "userId": f"#/chat/{random.randint(1,99999999)}",
             "usingContext": True
         }
-        response = requests.post("https://ai-api20.wuguokai.xyz/api/chat-process", headers=headers, timeout=3, json=data, proxies=kwargs['proxy'] if 'proxy' in kwargs else {})
+        response = requests.post(
+            "https://ai-api20.wuguokai.xyz/api/chat-process",
+            headers=headers,
+            timeout=3,
+            json=data,
+            proxies=kwargs.get('proxy', {}),
+        )
         _split = response.text.split("> 若回答失败请重试或多刷新几次界面后重试")
-        if response.status_code == 200:
-            if len(_split) > 1:
-                yield _split[1].strip()
-            else:
-                yield _split[0].strip()
-        else:
+        if response.status_code != 200:
             raise Exception(f"Error: {response.status_code} {response.reason}")
+        if len(_split) > 1:
+            yield _split[1].strip()
+        else:
+            yield _split[0].strip()
 
     @classmethod
     @property
-- 
cgit v1.2.3