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 List
from ...typing import AsyncResult, Messages from ...typing import AsyncResult, Messages
from ...image import ImageResponse from ...providers.response import ImageResponse, FinishReason, Usage
from ...providers.response import FinishReason, Usage
from ...requests.raise_for_status import raise_for_status from ...requests.raise_for_status import raise_for_status
from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin

View file

@ -5,7 +5,7 @@ import uuid
from ...typing import AsyncResult, Messages from ...typing import AsyncResult, Messages
from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin
from ...image import ImageResponse from ...providers.response import ImageResponse
from ...requests import StreamSession, raise_for_status from ...requests import StreamSession, raise_for_status
from ...errors import ResponseStatusError from ...errors import ResponseStatusError

View file

@ -9,7 +9,7 @@ from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin
from ...requests.aiohttp import get_connector from ...requests.aiohttp import get_connector
from ...requests.raise_for_status import raise_for_status from ...requests.raise_for_status import raise_for_status
from ..helper import format_prompt from ..helper import format_prompt
from ...image import ImageResponse from ...providers.response import ImageResponse
class ReplicateHome(AsyncGeneratorProvider, ProviderModelMixin): class ReplicateHome(AsyncGeneratorProvider, ProviderModelMixin):
url = "https://replicate.com" url = "https://replicate.com"

View file

@ -10,6 +10,7 @@ from ... import debug
class BackendApi(AsyncGeneratorProvider, ProviderModelMixin): class BackendApi(AsyncGeneratorProvider, ProviderModelMixin):
ssl = None ssl = None
headers = {}
@classmethod @classmethod
async def create_async_generator( async def create_async_generator(
@ -21,7 +22,7 @@ class BackendApi(AsyncGeneratorProvider, ProviderModelMixin):
) -> AsyncResult: ) -> AsyncResult:
debug.log(f"{cls.__name__}: {api_key}") debug.log(f"{cls.__name__}: {api_key}")
async with StreamSession( async with StreamSession(
headers={"Accept": "text/event-stream"}, headers={"Accept": "text/event-stream", **cls.headers},
) as session: ) as session:
async with session.post(f"{cls.url}/backend-api/v2/conversation", json={ async with session.post(f"{cls.url}/backend-api/v2/conversation", json={
"model": model, "model": model,

View file

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