mirror of
https://github.com/xtekky/gpt4free.git
synced 2025-12-15 06:41:34 -08:00
bdc356c4c2
- Replaced imports of `STATIC_URL` from `..constants` to `..config` in:
- `g4f/Provider/PollinationsAI.py`
- `g4f/Provider/PollinationsImage.py`
- Updated `client.py` to import `CONFIG_DIR` and `COOKIES_DIR` from `g4f.config` instead of defining platform-specific directories.
- Changed the handling of conversation history in `ConversationManager`:
- Updated `self.history` to retrieve data from `data.get("items", [])` instead of `data.get("history", [])`.
- Modified the `stream_response` function to use `media` instead of `image` for handling media content.
- Updated the `save_content` function to accept `media_content` of type `Optional[MediaResponse]` instead of `content`.
- Adjusted the `run_client_args` function to handle media URLs and files more effectively, appending valid media to a list.
- Removed the `constants.py` file and added a new `config.py` file to centralize configuration settings.
- Updated the `CookiesConfig` class to set `cookies_dir` based on the existence of `CUSTOM_COOKIES_DIR`.
- Adjusted the `render` function in `website.py` to correctly handle file paths and requests for HTML files.
- Updated various references to use the new `config` module instead of the removed `constants` module.
123 lines
3.9 KiB
Python
123 lines
3.9 KiB
Python
from __future__ import annotations
|
|
|
|
from os import environ
|
|
import requests
|
|
from functools import cached_property
|
|
from importlib.metadata import version as get_package_version, PackageNotFoundError
|
|
from subprocess import check_output, CalledProcessError, PIPE
|
|
from .errors import VersionNotFoundError
|
|
from .config import PACKAGE_NAME, GITHUB_REPOSITORY
|
|
from . import debug
|
|
|
|
def get_pypi_version(package_name: str) -> str:
|
|
"""
|
|
Retrieves the latest version of a package from PyPI.
|
|
|
|
Args:
|
|
package_name (str): The name of the package for which to retrieve the version.
|
|
|
|
Returns:
|
|
str: The latest version of the specified package from PyPI.
|
|
|
|
Raises:
|
|
VersionNotFoundError: If there is an error in fetching the version from PyPI.
|
|
"""
|
|
try:
|
|
response = requests.get(f"https://pypi.org/pypi/{package_name}/json").json()
|
|
return response["info"]["version"]
|
|
except requests.RequestException as e:
|
|
raise VersionNotFoundError(f"Failed to get PyPI version: {e}")
|
|
|
|
def get_github_version(repo: str) -> str:
|
|
"""
|
|
Retrieves the latest release version from a GitHub repository.
|
|
|
|
Args:
|
|
repo (str): The name of the GitHub repository.
|
|
|
|
Returns:
|
|
str: The latest release version from the specified GitHub repository.
|
|
|
|
Raises:
|
|
VersionNotFoundError: If there is an error in fetching the version from GitHub.
|
|
"""
|
|
try:
|
|
response = requests.get(f"https://api.github.com/repos/{repo}/releases/latest")
|
|
response.raise_for_status()
|
|
return response.json()["tag_name"]
|
|
except requests.RequestException as e:
|
|
raise VersionNotFoundError(f"Failed to get GitHub release version: {e}")
|
|
|
|
def get_git_version() -> str:
|
|
# Read from git repository
|
|
try:
|
|
command = ["git", "describe", "--tags", "--abbrev=0"]
|
|
return check_output(command, text=True, stderr=PIPE).strip()
|
|
except CalledProcessError:
|
|
return None
|
|
|
|
class VersionUtils:
|
|
"""
|
|
Utility class for managing and comparing package versions of 'g4f'.
|
|
"""
|
|
@cached_property
|
|
def current_version(self) -> str:
|
|
"""
|
|
Retrieves the current version of the 'g4f' package.
|
|
|
|
Returns:
|
|
str: The current version of 'g4f'.
|
|
|
|
Raises:
|
|
VersionNotFoundError: If the version cannot be determined from the package manager,
|
|
Docker environment, or git repository.
|
|
"""
|
|
if debug.version:
|
|
return debug.version
|
|
|
|
# Read from package manager
|
|
try:
|
|
return get_package_version(PACKAGE_NAME)
|
|
except PackageNotFoundError:
|
|
pass
|
|
|
|
# Read from docker environment
|
|
version = environ.get("G4F_VERSION")
|
|
if version:
|
|
return version
|
|
|
|
return get_git_version()
|
|
|
|
@property
|
|
def latest_version(self) -> str:
|
|
"""
|
|
Retrieves the latest version of the 'g4f' package.
|
|
|
|
Returns:
|
|
str: The latest version of 'g4f'.
|
|
"""
|
|
# Is installed via package manager?
|
|
try:
|
|
get_package_version(PACKAGE_NAME)
|
|
except PackageNotFoundError:
|
|
return get_github_version(GITHUB_REPOSITORY)
|
|
return get_pypi_version(PACKAGE_NAME)
|
|
|
|
@cached_property
|
|
def latest_version_cached(self) -> str:
|
|
return self.latest_version
|
|
|
|
def check_version(self) -> None:
|
|
"""
|
|
Checks if the current version of 'g4f' is up to date with the latest version.
|
|
|
|
Note:
|
|
If a newer version is available, it prints a message with the new version and update instructions.
|
|
"""
|
|
try:
|
|
if self.current_version != self.latest_version:
|
|
print(f'New g4f version: {self.latest_version} (current: {self.current_version}) | pip install -U g4f')
|
|
except Exception as e:
|
|
print(f'Failed to check g4f version: {e}')
|
|
|
|
utils = VersionUtils()
|