From cadc507fad2fee59b23d1d8e73c472c077f468fc Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Sat, 18 Nov 2023 04:38:31 +0100 Subject: Add AiChatOnline, ChatgptDemoAi, ChatgptNext Providers Add OnlineGpt and TalkAi Provider Add get_random_string and get_random_hex helper Remove funcaptcha package --- g4f/Provider/Koala.py | 13 +++---------- 1 file changed, 3 insertions(+), 10 deletions(-) (limited to 'g4f/Provider/Koala.py') diff --git a/g4f/Provider/Koala.py b/g4f/Provider/Koala.py index 70a40882..0e3ba13d 100644 --- a/g4f/Provider/Koala.py +++ b/g4f/Provider/Koala.py @@ -1,12 +1,11 @@ from __future__ import annotations -import random -import string import json from aiohttp import ClientSession from ..typing import AsyncResult, Messages from .base_provider import AsyncGeneratorProvider +from .helper import get_random_string class Koala(AsyncGeneratorProvider): url = "https://koala.sh" @@ -32,7 +31,7 @@ class Koala(AsyncGeneratorProvider): "Referer": f"{cls.url}/chat", "Content-Type": "application/json", "Flag-Real-Time-Data": "false", - "Visitor-ID": random_string(), + "Visitor-ID": get_random_string(20), "Origin": cls.url, "Alt-Used": "koala.sh", "Connection": "keep-alive", @@ -62,10 +61,4 @@ class Koala(AsyncGeneratorProvider): response.raise_for_status() async for chunk in response.content: if chunk.startswith(b"data: "): - yield json.loads(chunk[6:]) - - -def random_string(length: int = 20): - return ''.join(random.choice( - string.ascii_letters + string.digits - ) for _ in range(length)) \ No newline at end of file + yield json.loads(chunk[6:]) \ No newline at end of file -- cgit v1.2.3