diff options
Diffstat (limited to 'g4f/Provider/Cromicle.py')
-rw-r--r-- | g4f/Provider/Cromicle.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/g4f/Provider/Cromicle.py b/g4f/Provider/Cromicle.py index 5f521b3e..93ef8035 100644 --- a/g4f/Provider/Cromicle.py +++ b/g4f/Provider/Cromicle.py @@ -2,7 +2,7 @@ from __future__ import annotations from aiohttp import ClientSession from hashlib import sha256 -from typing import AsyncGenerator, Dict, List +from ..typing import AsyncResult, Messages, Dict from .base_provider import AsyncGeneratorProvider from .helper import format_prompt @@ -17,10 +17,10 @@ class Cromicle(AsyncGeneratorProvider): async def create_async_generator( cls, model: str, - messages: List[Dict[str, str]], + messages: Messages, proxy: str = None, **kwargs - ) -> AsyncGenerator[str, None]: + ) -> AsyncResult: async with ClientSession( headers=_create_header() ) as session: |