mirror of https://github.com/xtekky/gpt4free.git
Update GeminiPro.py
This commit is contained in:
parent
8894160f42
commit
95e61ad79b
|
@ -76,7 +76,7 @@ class GeminiPro(AsyncGeneratorProvider, ProviderModelMixin):
|
||||||
if not response.ok:
|
if not response.ok:
|
||||||
data = await response.json()
|
data = await response.json()
|
||||||
data = data[0] if isinstance(data, list) else data
|
data = data[0] if isinstance(data, list) else data
|
||||||
raise RuntimeError(data["error"]["message"])
|
raise RuntimeError(f"Response {response.status}: {data["error"]["message"]}")
|
||||||
if stream:
|
if stream:
|
||||||
lines = []
|
lines = []
|
||||||
async for chunk in response.content:
|
async for chunk in response.content:
|
||||||
|
@ -88,7 +88,7 @@ class GeminiPro(AsyncGeneratorProvider, ProviderModelMixin):
|
||||||
data = json.loads(data)
|
data = json.loads(data)
|
||||||
yield data["candidates"][0]["content"]["parts"][0]["text"]
|
yield data["candidates"][0]["content"]["parts"][0]["text"]
|
||||||
except:
|
except:
|
||||||
data = data.decode() if isinstance(data, bytes) else data
|
data = data.decode(errors="ignore") if isinstance(data, bytes) else data
|
||||||
raise RuntimeError(f"Read chunk failed: {data}")
|
raise RuntimeError(f"Read chunk failed: {data}")
|
||||||
lines = []
|
lines = []
|
||||||
else:
|
else:
|
||||||
|
|
Loading…
Reference in New Issue