zishuo/gpt4free
1
0
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-10-17 05:41:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add public gen api

Browse Source
This commit is contained in:
hlohaus
2025-07-13 15:31:09 +02:00
parent 426130f3d0
commit a31cb50d6a
3 changed files with 22 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
g4f/api/__init__.py
View File

@@ -82,7 +82,7 @@ from .stubs import (
from g4f import debug
try:
from g4f.gui.server.crypto import create_or_read_keys, decrypt_data
from g4f.gui.server.crypto import create_or_read_keys, decrypt_data, get_session_key
has_crypto = True
except ImportError:
has_crypto = False
@@ -217,6 +217,7 @@ class Api:
print(f"Register authentication key: {''.join(['*' for _ in range(len(AppConfig.g4f_api_key))])}")
if has_crypto:
private_key, _ = create_or_read_keys()
session_key = get_session_key()
@self.app.middleware("http")
async def authorization(request: Request, call_next):
if AppConfig.g4f_api_key is not None or AppConfig.demo:
@@ -231,10 +232,15 @@ class Api:
if has_crypto and user_g4f_api_key:
try:
expires, user = decrypt_data(private_key, user_g4f_api_key).split(":", 1)
except:
try:
data = json.loads(decrypt_data(session_key, user_g4f_api_key))
expires = int(decrypt_data(private_key, data["data"])) + 86400
user = data.get("user", None)
except:
return ErrorResponse.from_message(f"Invalid G4F API key", HTTP_401_UNAUTHORIZED)
expires = int(expires) - int(time.time())
debug.log(f"User: '{user}' G4F API key expires in {expires} seconds")
except Exception as e:
return ErrorResponse.from_message(f"Invalid G4F API key: {e}", HTTP_401_UNAUTHORIZED)
if expires < 0:
return ErrorResponse.from_message("G4F API key expired", HTTP_401_UNAUTHORIZED)
else:

4
g4f/gui/server/backend_api.py
View File

@@ -29,7 +29,7 @@ try:
except ImportError as e:
has_markitdown = False
try:
from .crypto import rsa, serialization, create_or_read_keys, decrypt_data, encrypt_data
from .crypto import rsa, serialization, create_or_read_keys, decrypt_data, encrypt_data, get_session_key
has_crypto = True
except ImportError:
has_crypto = False
@@ -79,7 +79,7 @@ class Backend_Api(Api):
self.chat_cache = {}
if has_crypto:
private_key_obj = rsa.generate_private_key(public_exponent=65537, key_size=4096)
private_key_obj = get_session_key()
public_key_obj = private_key_obj.public_key()
public_key_pem = public_key_obj.public_bytes(
encoding=serialization.Encoding.PEM,

9
g4f/gui/server/crypto.py
View File

@@ -9,6 +9,15 @@ from cryptography.hazmat.primitives.asymmetric.rsa import RSAPublicKey, RSAPriva
from ...cookies import get_cookies_dir
SESSION_KEY: RSAPrivateKey = None
def get_session_key() -> RSAPrivateKey:
global SESSION_KEY
if SESSION_KEY is not None:
return SESSION_KEY
SESSION_KEY = rsa.generate_private_key(public_exponent=65537, key_size=4096)
return SESSION_KEY
def create_or_read_keys() -> tuple[RSAPrivateKey, RSAPublicKey]:
private_key_file = os.path.join(get_cookies_dir(), "private_key.pem")
public_key_file = os.path.join(get_cookies_dir(), "public_key.pem")