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