XFE Git
XFE Studio Git
Git 首页 全局搜索
XFE 主站 文档 NuGet
公开
关注 0 Fork 0 Star 1
返回提交历史

XFEstudio/gpt4free

Update GeminiPro.py

95e61ad7
H Lohaus <hlohaus@users.noreply.github.com>
提交于

代码差异

1 个文件 +2 -2
Modified g4f/Provider/GeminiPro.py +2 -2
@@ -76,7 +76,7 @@ class GeminiPro(AsyncGeneratorProvider, ProviderModelMixin):
76 76 if not response.ok:
77 77 data = await response.json()
78 78 data = data[0] if isinstance(data, list) else data
79 raise RuntimeError(data["error"]["message"])
79 raise RuntimeError(f"Response {response.status}: {data["error"]["message"]}")
80 80 if stream:
81 81 lines = []
82 82 async for chunk in response.content:
@@ -88,7 +88,7 @@ class GeminiPro(AsyncGeneratorProvider, ProviderModelMixin):
88 88 data = json.loads(data)
89 89 yield data["candidates"][0]["content"]["parts"][0]["text"]
90 90 except:
91 data = data.decode() if isinstance(data, bytes) else data
91 data = data.decode(errors="ignore") if isinstance(data, bytes) else data
92 92 raise RuntimeError(f"Read chunk failed: {data}")
93 93 lines = []
94 94 else: