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/GptGo.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'g4f/Provider/GptGo.py') diff --git a/g4f/Provider/GptGo.py b/g4f/Provider/GptGo.py index 5f6cc362..f9b94a5c 100644 --- a/g4f/Provider/GptGo.py +++ b/g4f/Provider/GptGo.py @@ -3,7 +3,7 @@ from __future__ import annotations from aiohttp import ClientSession import json -from ..typing import AsyncGenerator +from ..typing import AsyncResult, Messages from .base_provider import AsyncGeneratorProvider, format_prompt @@ -16,10 +16,10 @@ class GptGo(AsyncGeneratorProvider): async def create_async_generator( cls, model: str, - messages: list[dict[str, str]], + messages: Messages, proxy: str = None, **kwargs - ) -> AsyncGenerator: + ) -> AsyncResult: headers = { "User-Agent" : "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/116.0.0.0 Safari/537.36", "Accept" : "*/*", -- cgit v1.2.3