summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/unfinished/Komo.py
diff options
context:
space:
mode:
authorTekky <98614666+xtekky@users.noreply.github.com>2023-10-05 20:06:21 +0200
committerGitHub <noreply@github.com>2023-10-05 20:06:21 +0200
commita59dc2fb36b42be735d52db24016f29f09cefff6 (patch)
tree76240685a9dcc320535cf0ffb3a5b931b2a14b72 /g4f/Provider/unfinished/Komo.py
parent~ | Merge pull request #984 from HexyeDEV/patch-2 (diff)
parentMerge branch 'main' into bom (diff)
downloadgpt4free-a59dc2fb36b42be735d52db24016f29f09cefff6.tar
gpt4free-a59dc2fb36b42be735d52db24016f29f09cefff6.tar.gz
gpt4free-a59dc2fb36b42be735d52db24016f29f09cefff6.tar.bz2
gpt4free-a59dc2fb36b42be735d52db24016f29f09cefff6.tar.lz
gpt4free-a59dc2fb36b42be735d52db24016f29f09cefff6.tar.xz
gpt4free-a59dc2fb36b42be735d52db24016f29f09cefff6.tar.zst
gpt4free-a59dc2fb36b42be735d52db24016f29f09cefff6.zip
Diffstat (limited to 'g4f/Provider/unfinished/Komo.py')
-rw-r--r--g4f/Provider/unfinished/Komo.py44
1 files changed, 44 insertions, 0 deletions
diff --git a/g4f/Provider/unfinished/Komo.py b/g4f/Provider/unfinished/Komo.py
new file mode 100644
index 00000000..84d8d634
--- /dev/null
+++ b/g4f/Provider/unfinished/Komo.py
@@ -0,0 +1,44 @@
+from __future__ import annotations
+
+import json
+
+from ...requests import StreamSession
+from ...typing import AsyncGenerator
+from ..base_provider import AsyncGeneratorProvider, format_prompt
+
+class Komo(AsyncGeneratorProvider):
+ url = "https://komo.ai/api/ask"
+ supports_gpt_35_turbo = True
+
+ @classmethod
+ async def create_async_generator(
+ cls,
+ model: str,
+ messages: list[dict[str, str]],
+ **kwargs
+ ) -> AsyncGenerator:
+ async with StreamSession(impersonate="chrome107") as session:
+ prompt = format_prompt(messages)
+ data = {
+ "query": prompt,
+ "FLAG_URLEXTRACT": "false",
+ "token": "",
+ "FLAG_MODELA": "1",
+ }
+ headers = {
+ 'authority': 'komo.ai',
+ 'accept': 'text/event-stream',
+ 'cache-control': 'no-cache',
+ 'referer': 'https://komo.ai/',
+ }
+
+ async with session.get(cls.url, params=data, headers=headers) as response:
+ response.raise_for_status()
+ next = False
+ async for line in response.iter_lines():
+ if line == b"event: line":
+ next = True
+ elif next and line.startswith(b"data: "):
+ yield json.loads(line[6:])
+ next = False
+