summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/Ylokh.py
diff options
context:
space:
mode:
authorHeiner Lohaus <heiner@lohaus.eu>2023-09-18 07:15:43 +0200
committerHeiner Lohaus <heiner@lohaus.eu>2023-09-18 07:15:43 +0200
commit3b8dfff974618499b177c5a724638919b93b702e (patch)
treee48ce2533f3cff7b1c45a3e789f63da01e541ac7 /g4f/Provider/Ylokh.py
parentAdd GptGo Provider, Fix AItianhu Provider (diff)
downloadgpt4free-3b8dfff974618499b177c5a724638919b93b702e.tar
gpt4free-3b8dfff974618499b177c5a724638919b93b702e.tar.gz
gpt4free-3b8dfff974618499b177c5a724638919b93b702e.tar.bz2
gpt4free-3b8dfff974618499b177c5a724638919b93b702e.tar.lz
gpt4free-3b8dfff974618499b177c5a724638919b93b702e.tar.xz
gpt4free-3b8dfff974618499b177c5a724638919b93b702e.tar.zst
gpt4free-3b8dfff974618499b177c5a724638919b93b702e.zip
Diffstat (limited to 'g4f/Provider/Ylokh.py')
-rw-r--r--g4f/Provider/Ylokh.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/g4f/Provider/Ylokh.py b/g4f/Provider/Ylokh.py
index 1986b6d3..c7b92089 100644
--- a/g4f/Provider/Ylokh.py
+++ b/g4f/Provider/Ylokh.py
@@ -51,7 +51,9 @@ class Ylokh(AsyncGeneratorProvider):
if stream:
async for line in response.content:
line = line.decode()
- if line.startswith("data: ") and not line.startswith("data: [DONE]"):
+ if line.startswith("data: "):
+ if line.startswith("data: [DONE]"):
+ break
line = json.loads(line[6:-1])
content = line["choices"][0]["delta"].get("content")
if content:
@@ -71,6 +73,7 @@ class Ylokh(AsyncGeneratorProvider):
("stream", "bool"),
("proxy", "str"),
("temperature", "float"),
+ ("top_p", "float"),
]
param = ", ".join([": ".join(p) for p in params])
return f"g4f.provider.{cls.__name__} supports: ({param})" \ No newline at end of file