From 53192b86b129380660f7454170fa987faf2da3c5 Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Tue, 10 Oct 2023 09:49:29 +0200 Subject: Some small fixes --- g4f/Provider/H2o.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'g4f/Provider/H2o.py') diff --git a/g4f/Provider/H2o.py b/g4f/Provider/H2o.py index d92bd6d1..65429a28 100644 --- a/g4f/Provider/H2o.py +++ b/g4f/Provider/H2o.py @@ -5,7 +5,7 @@ import uuid from aiohttp import ClientSession -from ..typing import AsyncGenerator +from ..typing import AsyncResult, Messages from .base_provider import AsyncGeneratorProvider, format_prompt @@ -18,10 +18,10 @@ class H2o(AsyncGeneratorProvider): async def create_async_generator( cls, model: str, - messages: list[dict[str, str]], + messages: Messages, proxy: str = None, **kwargs - ) -> AsyncGenerator: + ) -> AsyncResult: model = model if model else cls.model headers = {"Referer": cls.url + "/"} -- cgit v1.2.3