summaryrefslogtreecommitdiffstats
path: root/g4f
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--g4f/__init__.py28
-rw-r--r--g4f/gui/run.py4
-rw-r--r--g4f/interference/run.py4
-rw-r--r--g4f/typing.py2
4 files changed, 24 insertions, 14 deletions
diff --git a/g4f/__init__.py b/g4f/__init__.py
index bd4bfde0..4b1e4b80 100644
--- a/g4f/__init__.py
+++ b/g4f/__init__.py
@@ -5,7 +5,8 @@ from .Provider import BaseProvider
from .typing import CreateResult, Union
from .debug import logging
-version = '0.1.5.5'
+version = '0.1.5.6'
+version_check = True
def check_pypi_version() -> None:
try:
@@ -18,9 +19,10 @@ def check_pypi_version() -> None:
except Exception as e:
print(f'Failed to check g4f pypi version: {e}')
-check_pypi_version()
-
-def get_model_and_provider(model: Union[Model, str], provider: Union[type[BaseProvider], None], stream: bool) -> tuple[Model, type[BaseProvider]]:
+def get_model_and_provider(model : Union[Model, str],
+ provider : Union[type[BaseProvider], None],
+ stream : bool) -> tuple[Model, type[BaseProvider]]:
+
if isinstance(model, str):
if model in ModelUtils.convert:
model = ModelUtils.convert[model]
@@ -46,14 +48,11 @@ def get_model_and_provider(model: Union[Model, str], provider: Union[type[BasePr
class ChatCompletion:
@staticmethod
- def create(
- model: Union[Model, str],
- messages: list[dict[str, str]],
- provider: Union[type[BaseProvider], None] = None,
- stream: bool = False,
- auth: Union[str, None] = None,
- **kwargs
- ) -> Union[CreateResult, str]:
+ def create(model: Union[Model, str],
+ messages : list[dict[str, str]],
+ provider : Union[type[BaseProvider], None] = None,
+ stream : bool = False,
+ auth : Union[str, None] = None, **kwargs) -> Union[CreateResult, str]:
model, provider = get_model_and_provider(model, provider, stream)
@@ -104,4 +103,7 @@ class Completion:
result = provider.create_completion(model.name, [{"role": "user", "content": prompt}], stream, **kwargs)
- return result if stream else ''.join(result) \ No newline at end of file
+ return result if stream else ''.join(result)
+
+if version_check:
+ check_pypi_version() \ No newline at end of file
diff --git a/g4f/gui/run.py b/g4f/gui/run.py
new file mode 100644
index 00000000..da672d59
--- /dev/null
+++ b/g4f/gui/run.py
@@ -0,0 +1,4 @@
+from g4f.gui import run_gui
+
+if __name__ == '__main__':
+ run_gui() \ No newline at end of file
diff --git a/g4f/interference/run.py b/g4f/interference/run.py
new file mode 100644
index 00000000..e527ce11
--- /dev/null
+++ b/g4f/interference/run.py
@@ -0,0 +1,4 @@
+from g4f.interference import run_interference
+
+if __name__ == '__main__':
+ run_interference() \ No newline at end of file
diff --git a/g4f/typing.py b/g4f/typing.py
index 840e4624..cfddf4a8 100644
--- a/g4f/typing.py
+++ b/g4f/typing.py
@@ -8,7 +8,7 @@ else:
SHA256 = NewType('sha_256_hash', str)
CreateResult = Generator[str, None, None]
-AsyncResult = AsyncGenerator[str]
+AsyncResult = AsyncGenerator[str, None]
Messages = List[Dict[str, str]]
__all__ = [