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
This commit is contained in:
hlohaus 2025-02-03 20:23:55 +01:00
parent 0638cbc175
commit 9c8aedbeb1
5 changed files with 6 additions and 6 deletions

View file

@ -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

View file

@ -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

View file

@ -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"

View file

@ -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,

View file

@ -6,7 +6,7 @@ import asyncio
from .. import debug
from ..typing import CreateResult, Messages
from .types import BaseProvider, ProviderType
from ..image import ImageResponse
from ..providers.response import ImageResponse
system_message = """
You can generate images, pictures, photos or img with the DALL-E 3 image generator.