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/Myshell.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'g4f/Provider/Myshell.py') diff --git a/g4f/Provider/Myshell.py b/g4f/Provider/Myshell.py index 6ed4fd7a..847bac2f 100644 --- a/g4f/Provider/Myshell.py +++ b/g4f/Provider/Myshell.py @@ -6,7 +6,7 @@ from aiohttp import ClientSession from aiohttp.http import WSMsgType import asyncio -from ..typing import AsyncGenerator +from ..typing import AsyncResult, Messages from .base_provider import AsyncGeneratorProvider, format_prompt @@ -27,11 +27,11 @@ class Myshell(AsyncGeneratorProvider): async def create_async_generator( cls, model: str, - messages: list[dict[str, str]], + messages: Messages, proxy: str = None, timeout: int = 90, **kwargs - ) -> AsyncGenerator: + ) -> AsyncResult: if not model: bot_id = models["samantha"] elif model in models: -- cgit v1.2.3