返回提交历史
Modified
g4f/Provider/Bing.py
+3
-3
Modified
g4f/Provider/Raycast.py
+1
-1
Modified
g4f/__init__.py
+6
-6
Modified
g4f/models.py
+1
-1
Modified
g4f/typing.py
+1
-1
XFEstudio/gpt4free
~ | major refractoring | v0.0.2.0
6d96a1ef
代码差异
5 个文件
+12
-12
@@ -9,7 +9,7 @@ import aiohttp
9
9
import certifi
10
10
import requests
11
11
12
from ..typing import Any, AsyncGenerator, CreateResult, Tuple
12
from ..typing import Any, AsyncGenerator, CreateResult, Tuple, Union
13
13
from .base_provider import BaseProvider
14
14
15
15
@@ -79,7 +79,7 @@ def _format(msg: dict[str, Any]) -> str:
79
79
async def stream_generate(
80
80
prompt: str,
81
81
mode: dict[str, list[str]] = jailbreak,
82
context: bool | str = False,
82
context: Union[bool, str] = False,
83
83
):
84
84
timeout = aiohttp.ClientTimeout(total=900)
85
85
session = aiohttp.ClientSession(timeout=timeout)
@@ -350,7 +350,7 @@ class Defaults:
350
350
}
351
351
352
352
353
def run(generator: AsyncGenerator[Any | str, Any]):
353
def run(generator: AsyncGenerator[Union[Any, str], Any]):
354
354
loop = asyncio.get_event_loop()
355
355
gen = generator.__aiter__()
356
356
@@ -2,7 +2,7 @@ import json
2
2
3
3
import requests
4
4
5
from ..provider.base_provider import BaseProvider
5
from ..Provider.base_provider import BaseProvider
6
6
from ..typing import Any, CreateResult
7
7
8
8
@@ -1,6 +1,6 @@
1
1
from . import models
2
from .provider import BaseProvider
3
from .typing import Any, CreateResult
2
from .Provider import BaseProvider
3
from .typing import Any, CreateResult, Union
4
4
5
5
logging = False
6
6
@@ -8,13 +8,13 @@ logging = False
8
8
class ChatCompletion:
9
9
@staticmethod
10
10
def create(
11
model: models.Model | str,
11
model: Union[models.Model, str],
12
12
messages: list[dict[str, str]],
13
provider: type[BaseProvider] | None = None,
13
provider: Union[type[BaseProvider], None] = None,
14
14
stream: bool = False,
15
auth: str | None = None,
15
auth: Union[str, None] = None,
16
16
**kwargs: Any,
17
) -> CreateResult | str:
17
) -> Union[CreateResult, str]:
18
18
if isinstance(model, str):
19
19
try:
20
20
model = models.ModelUtils.convert[model]
@@ -1,6 +1,6 @@
1
1
from dataclasses import dataclass
2
2
3
from .provider import Bard, BaseProvider, GetGpt, H2o, Liaobots, Vercel
3
from .Provider import Bard, BaseProvider, GetGpt, H2o, Liaobots, Vercel
4
4
5
5
6
6
@dataclass
@@ -1,4 +1,4 @@
1
from typing import Any, AsyncGenerator, Generator, NewType, Tuple, TypedDict
1
from typing import Any, AsyncGenerator, Generator, NewType, Tuple, TypedDict, Union
2
2
3
3
SHA256 = NewType("sha_256_hash", str)
4
4
CreateResult = Generator[str, None, None]