From 4a9d7714adab375beda9db2a16bb34d139efef90 Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <heiner@lohaus.eu>
Date: Sun, 8 Oct 2023 11:39:19 +0200
Subject: Add Cromicle to provider list

---
 g4f/Provider/Cromicle.py | 30 +++++++++++++++---------------
 1 file changed, 15 insertions(+), 15 deletions(-)

(limited to 'g4f/Provider/Cromicle.py')

diff --git a/g4f/Provider/Cromicle.py b/g4f/Provider/Cromicle.py
index bd29544d..5f521b3e 100644
--- a/g4f/Provider/Cromicle.py
+++ b/g4f/Provider/Cromicle.py
@@ -2,32 +2,32 @@ from __future__ import annotations
 
 from aiohttp import ClientSession
 from hashlib import sha256
+from typing import AsyncGenerator, Dict, List
 
-from ..typing import AsyncGenerator
 from .base_provider import AsyncGeneratorProvider
+from .helper import format_prompt
 
 
 class Cromicle(AsyncGeneratorProvider):
-    url                   = 'https://cromicle.top'
-    working               = True
-    supports_gpt_35_turbo = True
+    url: str = 'https://cromicle.top'
+    working: bool = True
+    supports_gpt_35_turbo: bool = True
 
     @classmethod
     async def create_async_generator(
         cls,
         model: str,
-        messages: list[dict[str, str]],
+        messages: List[Dict[str, str]],
         proxy: str = None,
         **kwargs
-    ) -> AsyncGenerator:
-        message = messages[-1]["content"]
+    ) -> AsyncGenerator[str, None]:
         async with ClientSession(
             headers=_create_header()
         ) as session:
             async with session.post(
-                cls.url + '/chat',
+                f'{cls.url}/chat',
                 proxy=proxy,
-                json=_create_payload(message, **kwargs)
+                json=_create_payload(format_prompt(messages))
             ) as response:
                 response.raise_for_status()
                 async for stream in response.content.iter_any():
@@ -35,16 +35,16 @@ class Cromicle(AsyncGeneratorProvider):
                         yield stream.decode()
 
 
-def _create_header():
+def _create_header() -> Dict[str, str]:
     return {
         'accept': '*/*',
         'content-type': 'application/json',
     }
 
 
-def _create_payload(message: str):
+def _create_payload(message: str) -> Dict[str, str]:
     return {
-        'message'    : message,
-        'token' : 'abc',
-        'hash'    : sha256('abc'.encode() + message.encode()).hexdigest()
-    }
+        'message': message,
+        'token': 'abc',
+        'hash': sha256('abc'.encode() + message.encode()).hexdigest()
+    }
\ No newline at end of file
-- 
cgit v1.2.3