返回提交历史
Modified
g4f/client/async.py
+4
-18
XFEstudio/gpt4free
Update async.py
393be6be
代码差异
1 个文件
+4
-18
@@ -6,6 +6,7 @@ import time
6
6
import random
7
7
import string
8
8
9
from .types import Client as BaseClient
9
10
from .types import BaseProvider, ProviderType, FinishReason
10
11
from .stubs import ChatCompletion, ChatCompletionChunk, Image, ImagesResponse
11
12
from ..typing import Union, Iterator, Messages, ImageType, AsyncIerator
@@ -61,30 +62,15 @@ async def iter_append_model_and_provider(response: AsyncIterResponse) -> IterRes
61
62
chunk.provider = last_provider.get("name")
62
63
yield chunk
63
64
64
class Client():
65
class Client(BaseClient):
65
66
def __init__(
66
67
self,
67
api_key: str = None,
68
proxies: Proxies = None,
69
provider: ProviderType = None,
70
image_provider: ImageProvider = None,
71
68
**kwargs
72
) -> None:
73
self.api_key: str = api_key
74
self.proxies: Proxies = proxies
69
):
70
super().__init__(**kwargs)
75
71
self.chat: Chat = Chat(self, provider)
76
72
self.images: Images = Images(self, image_provider)
77
73
78
def get_proxy(self) -> Union[str, None]:
79
if isinstance(self.proxies, str):
80
return self.proxies
81
elif self.proxies is None:
82
return os.environ.get("G4F_PROXY")
83
elif "all" in self.proxies:
84
return self.proxies["all"]
85
elif "https" in self.proxies:
86
return self.proxies["https"]
87
88
74
def filter_none(**kwargs):
89
75
for key in list(kwargs.keys()):
90
76
if kwargs[key] is None: