From 7f16b7bb214cad17f4da2da3601dafd9658639c5 Mon Sep 17 00:00:00 2001 From: H Lohaus Date: Sat, 6 Apr 2024 23:47:42 +0200 Subject: [PATCH] Update DeepInfra.py --- g4f/Provider/DeepInfra.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/g4f/Provider/DeepInfra.py b/g4f/Provider/DeepInfra.py index f44679ff..6cf52694 100644 --- a/g4f/Provider/DeepInfra.py +++ b/g4f/Provider/DeepInfra.py @@ -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: "):