diff options
Diffstat (limited to 'g4f/Provider/Koala.py')
| -rw-r--r-- | g4f/Provider/Koala.py | 13 |
1 files changed, 3 insertions, 10 deletions
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 |