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

XFEstudio/gpt4free

Update (g4f/Provider/needs_auth/GeminiPro.py)

795e42a2
kqlio67 <kqlio67@users.noreply.github.com>
提交于

代码差异

1 个文件 +5 -1
Modified g4f/Provider/needs_auth/GeminiPro.py +5 -1
@@ -104,4 +104,8 @@ class GeminiPro(AsyncGeneratorProvider, ProviderModelMixin):
104 104 lines.append(chunk)
105 105 else:
106 106 data = await response.json()
107 yield data["candidates"][0]["content"]["parts"][0]["text"]
107 candidate = data["candidates"][0]
108 if candidate["finishReason"] == "STOP":
109 yield candidate["content"]["parts"][0]["text"]
110 else:
111 yield candidate["finishReason"] + ' ' + candidate["safetyRatings"]