From 0b4c6247d051a1dd704f519062686716e8d9f201 Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Sun, 21 Jan 2024 09:43:46 +0100 Subject: Fix load json in DeepInfra --- g4f/Provider/DeepInfra.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) (limited to 'g4f') diff --git a/g4f/Provider/DeepInfra.py b/g4f/Provider/DeepInfra.py index 12bd49c7..acde1200 100644 --- a/g4f/Provider/DeepInfra.py +++ b/g4f/Provider/DeepInfra.py @@ -60,17 +60,13 @@ class DeepInfra(AsyncGeneratorProvider): async for line in response.iter_lines(): if not line.startswith(b"data: "): continue - try: - decoded_line = line.decode().lstrip("data: ") - json_line = json.loads(decoded_line) - + json_line = json.loads(line[6:]) choices = json_line.get("choices", [{}]) - finish_reason = choices[0].get("finish_reason", "") + finish_reason = choices[0].get("finish_reason") if finish_reason: break - token = choices[0].get("delta", {}).get("content", "") - + token = choices[0].get("delta", {}).get("content") if token: if first: token = token.lstrip() -- cgit v1.2.3