xtekky/gpt4free
1
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-11-26 21:21:21 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update DeepInfra.py

Browse Source
This commit is contained in:
H Lohaus 2024-04-06 23:47:42 +02:00 committed by GitHub
parent 68089ff0a9
commit 7f16b7bb21
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
g4f/Provider/DeepInfra.py
View File

@ -2,9 +2,9 @@ from __future__ import annotations
import json
import requests
from ..typing import AsyncResult, Messages
from ..typing import AsyncResult, Messages
from .base_provider import AsyncGeneratorProvider, ProviderModelMixin
from ..requests import StreamSession
from ..requests import StreamSession, raise_for_status
class DeepInfra(AsyncGeneratorProvider, ProviderModelMixin):
url = "https://deepinfra.com"
@ -67,7 +67,7 @@ class DeepInfra(AsyncGeneratorProvider, ProviderModelMixin):
}
async with session.post('https://api.deepinfra.com/v1/openai/chat/completions',
json=json_data) as response:
response.raise_for_status()
await raise_for_status(response)
first = True
async for line in response.iter_lines():
if not line.startswith(b"data: "):