zishuo/gpt4free
1
0
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-12-24 13:07:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use StreamSession for proper line-by-line NDJSON parsing

Browse Source 
Co-authored-by: hlohaus <983577+hlohaus@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2025-11-29 04:24:09 +00:00
parent f0ea4c5b95
commit da4d7d118d
1 changed files with 5 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
g4f/Provider/GradientNetwork.py
View File

@@ -2,10 +2,9 @@ from __future__ import annotations
import json
from aiohttp import ClientSession
from ..typing import AsyncResult, Messages
from ..providers.response import Reasoning
from ..requests import StreamSession
from .base_provider import AsyncGeneratorProvider, ProviderModelMixin
@@ -58,7 +57,7 @@ class GradientNetwork(AsyncGeneratorProvider, ProviderModelMixin):
proxy: Optional proxy URL
temperature: Optional temperature parameter
max_tokens: Optional max tokens parameter
enable_thinking: Enable the thinking/analysis channel
enable_thinking: Enable the thinking/analysis channel (maps to enableThinking in API)
**kwargs: Additional arguments
Yields:
@@ -87,19 +86,14 @@ class GradientNetwork(AsyncGeneratorProvider, ProviderModelMixin):
if enable_thinking:
payload["enableThinking"] = True
async with ClientSession(headers=headers) as session:
async with StreamSession(headers=headers, proxy=proxy) as session:
async with session.post(
cls.api_endpoint,
json=payload,
proxy=proxy
) as response:
response.raise_for_status()
async for line_bytes in response.content:
if not line_bytes:
continue
line = line_bytes.decode("utf-8").strip()
async for line in response.iter_lines():
if not line:
continue
@@ -127,5 +121,5 @@ class GradientNetwork(AsyncGeneratorProvider, ProviderModelMixin):
# as they are for GPU cluster visualization only
except json.JSONDecodeError:
# Skip non-JSON lines
# Skip non-JSON lines (may be partial data or empty)
continue