Update async.py

pull/1794/head
H Lohaus 2 months ago committed by GitHub
parent c9a83e53b7
commit 2beade3c24
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -13,7 +13,7 @@ from ..typing import Union, Iterator, Messages, ImageType, AsyncIerator
from ..image import ImageResponse as ImageProviderResponse
from ..errors import NoImageResponseError, RateLimitError, MissingAuthError
from .. import get_model_and_provider, get_last_provider
from .helper import read_json
from .helper import read_json, find_stop
from .Provider.BingCreateImages import BingCreateImages
from .Provider.needs_auth import Gemini, OpenaiChat
@ -30,7 +30,7 @@ async def iter_response(
finish_reason = None
completion_id = ''.join(random.choices(string.ascii_letters + string.digits, k=28))
count: int = 0
async for idx, chunk in response:
async for chunk in response:
if isinstance(chunk, FinishReason):
finish_reason = chunk.reason
break
@ -71,12 +71,6 @@ class Client(BaseClient):
self.chat: Chat = Chat(self, provider)
self.images: Images = Images(self, image_provider)
def filter_none(**kwargs):
for key in list(kwargs.keys()):
if kwargs[key] is None:
del kwargs[key]
return kwargs
class Completions():
def __init__(self, client: Client, provider: ProviderType = None):
self.client: Client = client

Loading…
Cancel
Save