Add nodriver to Gemini provider,

Add slim docker image with google-chrome usage,
Add the new docker images to publish worklow,
Update requirements.txt and pip requirements

Author: hlohaus

.github/workflows/publish-workflow.yaml

@@ -48,3 +48,15 @@ jobs:
           labels: ${{ steps.metadata.outputs.labels }}
           build-args: |
             G4F_VERSION=${{ github.ref_name }}
+      - name: Build and push slim image
+        uses: docker/build-push-action@v5
+        with:
+          context: .
+          file: docker/Dockerfile-slim
+          push: true
+          tags: |
+            hlohaus789/g4f=slim
+            hlohaus789/g4f=${{ github.ref_name }}-slim
+          labels: ${{ steps.metadata.outputs.labels }}
+          build-args: |
+            G4F_VERSION=${{ github.ref_name }}

docker-compose-slim.yml

@@ -0,0 +1,25 @@
+version: '3'
+
+services:
+  g4f-gui:
+    container_name: g4f-gui
+    image: hlohaus789/g4f:slim
+    build:
+      context: .
+      dockerfile: docker/Dockerfile-slim
+    command: python -m g4f.cli gui -debug
+    volumes:
+      - .:/app
+    ports:
+      - '8080:8080'
+  g4f-api:
+    container_name: g4f-api
+    image: hlohaus789/g4f:slim
+    build:
+      context: .
+      dockerfile: docker/Dockerfile-slim
+    command: python -m g4f.cli api
+    volumes:
+      - .:/app
+    ports:
+      - '1337:1337'

docker/Dockerfile

@@ -40,6 +40,7 @@ RUN apt-get -qqy update \
 
 # Update entrypoint
 COPY docker/supervisor.conf /etc/supervisor/conf.d/selenium.conf
+COPY docker/supervisor-api.conf /etc/supervisor/conf.d/api.conf
 COPY docker/supervisor-gui.conf /etc/supervisor/conf.d/gui.conf
 
 # If no gui

docker/Dockerfile-slim

@@ -0,0 +1,68 @@
+FROM python:bookworm
+
+ARG G4F_VERSION
+ARG G4F_USER=g4f
+ARG G4F_USER_ID=1000
+ARG PYDANTIC_VERSION=1.8.1
+
+ENV G4F_VERSION $G4F_VERSION
+ENV G4F_USER $G4F_USER
+ENV G4F_USER_ID $G4F_USER_ID
+ENV G4F_DIR /app
+
+RUN apt-get update && apt-get upgrade -y \
+  && apt-get install -y git \
+  && apt-get install --quiet --yes --no-install-recommends \
+      build-essential \
+# Add user and user group
+  && groupadd -g $G4F_USER_ID $G4F_USER \
+  && useradd -rm -G sudo -u $G4F_USER_ID -g $G4F_USER_ID $G4F_USER \
+  && mkdir -p /var/log/supervisor \
+  && chown "${G4F_USER_ID}:${G4F_USER_ID}" /var/log/supervisor \
+  && echo "${G4F_USER}:${G4F_USER}" | chpasswd
+
+USER $G4F_USER_ID
+WORKDIR $G4F_DIR
+
+ENV HOME /home/$G4F_USER
+ENV PATH "${HOME}/.local/bin:${HOME}/.cargo/bin:${PATH}"
+
+# Create app dir and copy the project's requirements file into it
+RUN mkdir -p $G4F_DIR
+COPY requirements-slim.txt $G4F_DIR
+
+# Install rust toolchain
+RUN curl https://sh.rustup.rs -sSf | bash -s -- -y
+
+# Upgrade pip for the latest features and install the project's Python dependencies.
+RUN python -m pip install --upgrade pip \
+  && pip install --no-cache-dir \
+    Cython==0.29.22 \
+    setuptools \
+  # Install PyDantic
+  && pip install \
+    -vvv \
+    --no-cache-dir \
+    --no-binary pydantic \
+    --global-option=build_ext \
+    --global-option=-j8 \
+    pydantic==${PYDANTIC_VERSION} \
+  && pip install --no-cache-dir -r requirements-slim.txt \
+  # Remove build packages
+  && pip uninstall --yes \
+      Cython \
+      setuptools
+
+USER root
+
+# Clean up build deps
+RUN rustup self uninstall -y \
+  && apt-get purge --auto-remove --yes \
+    build-essential \
+  && apt-get clean \
+  && rm --recursive --force /var/lib/apt/lists/* /tmp/* /var/tmp/*
+
+USER $G4F_USER_ID
+
+# Copy the entire package into the container.
+ADD --chown=$G4F_USER:$G4F_USER g4f $G4F_DIR/g4f

docker/supervisor-api.conf

@@ -0,0 +1,12 @@
+[program:g4f-api]
+priority=15
+command=python -m g4f.cli api
+directory=/app
+stopasgroup=true
+autostart=true
+autorestart=true
+
+;Logs (all Hub activity redirected to stdout so it can be seen through "docker logs"
+redirect_stderr=true
+stdout_logfile=/dev/stdout
+stdout_logfile_maxbytes=0

docker/supervisor-gui.conf

@@ -1,6 +1,6 @@
 [program:g4f-gui]
 priority=15
-command=python -m g4f.cli gui
+command=python -m g4f.cli gui -debug
 directory=/app
 stopasgroup=true
 autostart=true

docker/supervisor.conf

@@ -47,17 +47,4 @@ stderr_logfile_maxbytes=50MB
 stdout_logfile_backups=5
 stderr_logfile_backups=5
 stdout_capture_maxbytes=50MB
-stderr_capture_maxbytes=50MB
-
-[program:g4f-api]
-priority=15
-command=python -m g4f.cli api
-directory=/app
-stopasgroup=true
-autostart=true
-autorestart=true
-
-;Logs (all Hub activity redirected to stdout so it can be seen through "docker logs"
-redirect_stderr=true
-stdout_logfile=/dev/stdout
-stdout_logfile_maxbytes=0
+stderr_capture_maxbytes=50MB

docs/docker.md

@@ -28,12 +28,22 @@
    ```
 
 2. **Build and Run with Docker Compose**
+
+   Pull the latest image and run a container with Google Chrome support:
+   ```bash
+      docker pull hlohaus789/g4f
+      docker-compose up -d
+   ```
+   Or run the small docker images without Google Chrome:
    ```bash
-   docker-compose up --build
+      docker-compose -f docker-compose-slim.yml up -d
    ```
 
-3. **Access the API**
-   The server will be accessible at `http://localhost:1337`
+3. **Access the API or the GUI**
+
+   The api server will be accessible at `http://localhost:1337`
+
+   And the gui at this url: `http://localhost:8080`
 
 ### Non-Docker Method
 If you encounter issues with Docker, you can run the project directly using Python:
@@ -54,8 +64,12 @@ If you encounter issues with Docker, you can run the project directly using Pyth
    python -m g4f.api.run
    ```
 
-4. **Access the API**
-   The server will be accessible at `http://localhost:1337`
+4. **Access the API or the GUI**
+
+   The api server will be accessible at `http://localhost:1337`
+
+   And the gui at this url: `http://localhost:8080`
+
 
 ## Testing the API
 **You can test the API using curl or by creating a simple Python script:**

g4f/Provider/Cloudflare.py

@@ -7,6 +7,7 @@
 from ..typing import AsyncResult, Messages, Cookies
 from .base_provider import AsyncGeneratorProvider, ProviderModelMixin, get_running_loop
 from ..requests import Session, StreamSession, get_args_from_nodriver, raise_for_status, merge_cookies
+from ..errors import ResponseStatusError
 
 class Cloudflare(AsyncGeneratorProvider, ProviderModelMixin):
     label = "Cloudflare AI"
@@ -42,10 +43,14 @@ def get_models(cls) -> str:
                 cls._args = asyncio.run(args)
             with Session(**cls._args) as session:
                 response = session.get(cls.models_url)
-                raise_for_status(response)
+                cls._args["cookies"] = merge_cookies(cls._args["cookies"] , response)
+                try:
+                    raise_for_status(response)
+                except ResponseStatusError as e:
+                    cls._args = None
+                    raise e
                 json_data = response.json()
                 cls.models = [model.get("name") for model in json_data.get("models")]
-                cls._args["cookies"] = merge_cookies(cls._args["cookies"] , response)
         return cls.models
 
     @classmethod
@@ -74,8 +79,12 @@ async def create_async_generator(
                 cls.api_endpoint,
                 json=data,
             ) as response:
-                await raise_for_status(response)
                 cls._args["cookies"] = merge_cookies(cls._args["cookies"] , response)
+                try:
+                    await raise_for_status(response)
+                except ResponseStatusError as e:
+                    cls._args = None
+                    raise e
                 async for line in response.iter_lines():
                     if line.startswith(b'data: '):
                         if line == b'data: [DONE]':

g4f/Provider/HuggingChat.py

@@ -4,12 +4,13 @@
 import requests
 
 try:
-    from curl_cffi import requests as cf_reqs
+    from curl_cffi import Session
     has_curl_cffi = True
 except ImportError:
     has_curl_cffi = False
 from ..typing import CreateResult, Messages
 from ..errors import MissingRequirementsError
+from ..requests.raise_for_status import raise_for_status
 from .base_provider import ProviderModelMixin, AbstractProvider
 from .helper import format_prompt
 
@@ -18,7 +19,7 @@ class HuggingChat(AbstractProvider, ProviderModelMixin):
     working = True
     supports_stream = True
     default_model = "meta-llama/Meta-Llama-3.1-70B-Instruct"
-    
+
     models = [
         'meta-llama/Meta-Llama-3.1-70B-Instruct',
         'CohereForAI/c4ai-command-r-plus-08-2024',
@@ -30,7 +31,7 @@ class HuggingChat(AbstractProvider, ProviderModelMixin):
         'mistralai/Mistral-Nemo-Instruct-2407',
         'microsoft/Phi-3.5-mini-instruct',
     ]
-    
+
     model_aliases = {
         "llama-3.1-70b": "meta-llama/Meta-Llama-3.1-70B-Instruct",
         "command-r-plus": "CohereForAI/c4ai-command-r-plus-08-2024",
@@ -43,15 +44,6 @@ class HuggingChat(AbstractProvider, ProviderModelMixin):
         "phi-3.5-mini": "microsoft/Phi-3.5-mini-instruct",
     }
 
-    @classmethod
-    def get_model(cls, model: str) -> str:
-        if model in cls.models:
-            return model
-        elif model in cls.model_aliases:
-            return cls.model_aliases[model]
-        else:
-            return cls.default_model
-            
     @classmethod
     def create_completion(
         cls,
@@ -65,7 +57,7 @@ def create_completion(
         model = cls.get_model(model)
 
         if model in cls.models:
-            session = cf_reqs.Session()
+            session = Session()
             session.headers = {
                 'accept': '*/*',
                 'accept-language': 'en',
@@ -82,20 +74,18 @@ def create_completion(
                 'sec-fetch-site': 'same-origin',
                 'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/127.0.0.0 Safari/537.36',
             }
-
             json_data = {
                 'model': model,
             }
-
             response = session.post('https://huggingface.co/chat/conversation', json=json_data)
-            if response.status_code != 200:
-                raise RuntimeError(f"Request failed with status code: {response.status_code}, response: {response.text}")
+            raise_for_status(response)
 
             conversationId = response.json().get('conversationId')
 
             # Get the data response and parse it properly
             response = session.get(f'https://huggingface.co/chat/conversation/{conversationId}/__data.json?x-sveltekit-invalidated=11')
-            
+            raise_for_status(response)
+
             # Split the response content by newlines and parse each line as JSON
             try:
                 json_data = None
@@ -156,6 +146,7 @@ def create_completion(
                 headers=headers,
                 files=files,
             )
+            raise_for_status(response)
 
             full_response = ""
             for line in response.iter_lines():
@@ -182,9 +173,4 @@ def create_completion(
             full_response = full_response.replace('<|im_end|', '').replace('\u0000', '').strip()
 
             if not stream:
-                yield full_response
-
-    @classmethod
-    def supports_model(cls, model: str) -> bool:
-        """Check if the model is supported by the provider."""
-        return model in cls.models or model in cls.model_aliases
+                yield full_response