diff options
Diffstat (limited to 'g4f/Provider/AItianhuSpace.py')
| -rw-r--r-- | g4f/Provider/AItianhuSpace.py | 23 |
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") |