From 9c8aedbeb1f33c6b6f87f59892f76cfff94306d8 Mon Sep 17 00:00:00 2001
From: hlohaus <983577+hlohaus@users.noreply.github.com>
Date: Mon, 3 Feb 2025 20:23:55 +0100
Subject: Improve select custom model in UI Updates for the response of the
 BackendApi Update of the demo model list Improve web search tool Moved
 copy_images to /image

---
 g4f/Provider/not_working/Airforce.py      | 3 +--
 g4f/Provider/not_working/AmigoChat.py     | 2 +-
 g4f/Provider/not_working/ReplicateHome.py | 2 +-
 g4f/Provider/template/BackendApi.py       | 3 ++-
 4 files changed, 5 insertions(+), 5 deletions(-)

(limited to 'g4f/Provider')

diff --git a/g4f/Provider/not_working/Airforce.py b/g4f/Provider/not_working/Airforce.py
index 862e59d9..003b7fa2 100644
--- a/g4f/Provider/not_working/Airforce.py
+++ b/g4f/Provider/not_working/Airforce.py
@@ -6,8 +6,7 @@ from aiohttp import ClientSession
 from typing import List
 
 from ...typing import AsyncResult, Messages
-from ...image import ImageResponse
-from ...providers.response import FinishReason, Usage
+from ...providers.response import ImageResponse, FinishReason, Usage
 from ...requests.raise_for_status import raise_for_status
 from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin
 
diff --git a/g4f/Provider/not_working/AmigoChat.py b/g4f/Provider/not_working/AmigoChat.py
index 31d1b10b..77190188 100644
--- a/g4f/Provider/not_working/AmigoChat.py
+++ b/g4f/Provider/not_working/AmigoChat.py
@@ -5,7 +5,7 @@ import uuid
 
 from ...typing import AsyncResult, Messages
 from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin
-from ...image import ImageResponse
+from ...providers.response import ImageResponse
 from ...requests import StreamSession, raise_for_status
 from ...errors import ResponseStatusError
 
diff --git a/g4f/Provider/not_working/ReplicateHome.py b/g4f/Provider/not_working/ReplicateHome.py
index e8a99e83..cafc4ce0 100644
--- a/g4f/Provider/not_working/ReplicateHome.py
+++ b/g4f/Provider/not_working/ReplicateHome.py
@@ -9,7 +9,7 @@ from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin
 from ...requests.aiohttp import get_connector
 from ...requests.raise_for_status import raise_for_status
 from ..helper import format_prompt
-from ...image import ImageResponse
+from ...providers.response import ImageResponse
 
 class ReplicateHome(AsyncGeneratorProvider, ProviderModelMixin):
     url = "https://replicate.com"
diff --git a/g4f/Provider/template/BackendApi.py b/g4f/Provider/template/BackendApi.py
index 0bca9a22..91067006 100644
--- a/g4f/Provider/template/BackendApi.py
+++ b/g4f/Provider/template/BackendApi.py
@@ -10,6 +10,7 @@ from ... import debug
 
 class BackendApi(AsyncGeneratorProvider, ProviderModelMixin):
     ssl = None
+    headers = {}
 
     @classmethod
     async def create_async_generator(
@@ -21,7 +22,7 @@ class BackendApi(AsyncGeneratorProvider, ProviderModelMixin):
     ) -> AsyncResult:
         debug.log(f"{cls.__name__}: {api_key}")
         async with StreamSession(
-            headers={"Accept": "text/event-stream"},
+            headers={"Accept": "text/event-stream", **cls.headers},
         ) as session:
             async with session.post(f"{cls.url}/backend-api/v2/conversation", json={
                 "model": model,
-- 
cgit v1.2.3