Add default values for rdns

pull/1633/head
Heiner Lohaus 7 months ago
parent 02b84fb519
commit 6bbe60572e

@ -70,7 +70,7 @@ class Bing(AsyncGeneratorProvider):
gpt4_turbo = True if model.startswith("gpt-4-turbo") else False gpt4_turbo = True if model.startswith("gpt-4-turbo") else False
return stream_generate(prompt, tone, image, context, cookies, get_connector(connector, proxy), web_search, gpt4_turbo, timeout) return stream_generate(prompt, tone, image, context, cookies, get_connector(connector, proxy, True), web_search, gpt4_turbo, timeout)
def create_context(messages: Messages) -> str: def create_context(messages: Messages) -> str:
""" """

@ -105,7 +105,7 @@ class Liaobots(AsyncGeneratorProvider, ProviderModelMixin):
async with ClientSession( async with ClientSession(
headers=headers, headers=headers,
cookie_jar=cls._cookie_jar, cookie_jar=cls._cookie_jar,
connector=get_connector(connector, proxy) connector=get_connector(connector, proxy, True)
) as session: ) as session:
cls._auth_code = auth if isinstance(auth, str) else cls._auth_code cls._auth_code = auth if isinstance(auth, str) else cls._auth_code
if not cls._auth_code: if not cls._auth_code:

@ -51,11 +51,10 @@ def get_random_hex() -> str:
""" """
return secrets.token_hex(16).zfill(32) return secrets.token_hex(16).zfill(32)
def get_connector(connector: BaseConnector = None, proxy: str = None) -> Optional[BaseConnector]: def get_connector(connector: BaseConnector = None, proxy: str = None, rdns: bool = False) -> Optional[BaseConnector]:
if proxy and not connector: if proxy and not connector:
try: try:
from aiohttp_socks import ProxyConnector from aiohttp_socks import ProxyConnector
rdns = False
if proxy.startswith("socks5h://"): if proxy.startswith("socks5h://"):
proxy = proxy.replace("socks5h://", "socks5://") proxy = proxy.replace("socks5h://", "socks5://")
rdns = True rdns = True

Loading…
Cancel
Save