summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/needs_auth/HuggingChat.py
diff options
context:
space:
mode:
authornullstreak <139914347+nullstreak@users.noreply.github.com>2023-12-15 23:58:13 +0100
committerGitHub <noreply@github.com>2023-12-15 23:58:13 +0100
commitf554018da26e32cd10c752162fbe2db0c210aadc (patch)
treeacea5998f0cfd7d247d5e968cf4c4211250d4c78 /g4f/Provider/needs_auth/HuggingChat.py
parentMerge pull request #1350 from devAdityaa/main (diff)
downloadgpt4free-f554018da26e32cd10c752162fbe2db0c210aadc.tar
gpt4free-f554018da26e32cd10c752162fbe2db0c210aadc.tar.gz
gpt4free-f554018da26e32cd10c752162fbe2db0c210aadc.tar.bz2
gpt4free-f554018da26e32cd10c752162fbe2db0c210aadc.tar.lz
gpt4free-f554018da26e32cd10c752162fbe2db0c210aadc.tar.xz
gpt4free-f554018da26e32cd10c752162fbe2db0c210aadc.tar.zst
gpt4free-f554018da26e32cd10c752162fbe2db0c210aadc.zip
Diffstat (limited to 'g4f/Provider/needs_auth/HuggingChat.py')
-rw-r--r--g4f/Provider/needs_auth/HuggingChat.py9
1 files changed, 7 insertions, 2 deletions
diff --git a/g4f/Provider/needs_auth/HuggingChat.py b/g4f/Provider/needs_auth/HuggingChat.py
index 530069c0..41c938b4 100644
--- a/g4f/Provider/needs_auth/HuggingChat.py
+++ b/g4f/Provider/needs_auth/HuggingChat.py
@@ -47,14 +47,19 @@ class HuggingChat(AsyncGeneratorProvider):
"web_search": web_search
}
async with session.post(f"{cls.url}/conversation/{conversation_id}", json=send, proxy=proxy) as response:
+ first_token = True
async for line in response.content:
line = json.loads(line[:-1])
if "type" not in line:
raise RuntimeError(f"Response: {line}")
elif line["type"] == "stream":
- yield line["token"]
+ token = line["token"]
+ if first_token:
+ token = token.lstrip()
+ first_token = False
+ yield token
elif line["type"] == "finalAnswer":
break
async with session.delete(f"{cls.url}/conversation/{conversation_id}", proxy=proxy) as response:
- response.raise_for_status() \ No newline at end of file
+ response.raise_for_status()