summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/Koala.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--g4f/Provider/Koala.py13
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