1 files changed, 12 insertions, 11 deletions

diff --git a/g4f/Provider/AItianhuSpace.py b/g4f/Provider/AItianhuSpace.py
index 46856060..d316fc6f 100644
--- a/g4f/Provider/AItianhuSpace.py
+++ b/g4f/Provider/AItianhuSpace.py
@@ -27,26 +27,26 @@ class AItianhuSpace(AsyncGeneratorProvider):
         
         if not model:
             model = "gpt-3.5-turbo"
-        
-        elif not model in domains:
+
+        elif model not in domains:
             raise ValueError(f"Model are not supported: {model}")
-        
+
         if not domain:
             chars = 'abcdefghijklmnopqrstuvwxyz0123456789'
             rand = ''.join(random.choice(chars) for _ in range(6))
             domain = f"{rand}.{domains[model]}"
-        
+
         if debug.logging:
             print(f"AItianhuSpace | using domain: {domain}")
-        
+
         if not cookies:
             cookies = get_cookies('.aitianhu.space')
-            if not cookies:
-                raise RuntimeError(f"g4f.provider.{cls.__name__} requires cookies [refresh https://{domain} on chrome]")
-        
+        if not cookies:
+            raise RuntimeError(f"g4f.provider.{cls.__name__} requires cookies [refresh https://{domain} on chrome]")
+
         url = f'https://{domain}'
         async with StreamSession(proxies={"https": proxy},
-            cookies=cookies, timeout=timeout, impersonate="chrome110", verify=False) as session:
+                cookies=cookies, timeout=timeout, impersonate="chrome110", verify=False) as session:
             
             data = {
                 "prompt": format_prompt(messages),
@@ -71,8 +71,9 @@ class AItianhuSpace(AsyncGeneratorProvider):
                         raise RuntimeError("Platform's Risk Control")
                     line = json.loads(line)
                     if "detail" in line:
-                        content = line["detail"]["choices"][0]["delta"].get("content")
-                        if content:
+                        if content := line["detail"]["choices"][0]["delta"].get(
+                            "content"
+                        ):
                             yield content
                     elif "message" in line and "AI-4接口非常昂贵" in line["message"]:
                         raise RuntimeError("Rate limit for GPT 4 reached")