From 24345bc07bf77bdfa4bb799de62233ffdb1c5d73 Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Mon, 8 Apr 2024 23:27:54 +0200 Subject: Add DuckDuckGo Provider, Add SpeechRecognition to gui --- g4f/Provider/DuckDuckGo.py | 64 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 64 insertions(+) create mode 100644 g4f/Provider/DuckDuckGo.py (limited to 'g4f/Provider') diff --git a/g4f/Provider/DuckDuckGo.py b/g4f/Provider/DuckDuckGo.py new file mode 100644 index 00000000..5269ced3 --- /dev/null +++ b/g4f/Provider/DuckDuckGo.py @@ -0,0 +1,64 @@ +from __future__ import annotations + +import json +import aiohttp + +from .base_provider import AsyncGeneratorProvider, ProviderModelMixin +from ..typing import AsyncResult, Messages +from ..requests.raise_for_status import raise_for_status + +class DuckDuckGo(AsyncGeneratorProvider, ProviderModelMixin): + url = "https://duckduckgo.com/duckchat" + working = True + supports_gpt_35_turbo = True + supports_message_history = True + + default_model = "gpt-3.5-turbo-0125" + models = ["gpt-3.5-turbo-0125", "claude-instant-1.2"] + model_aliases = {"gpt-3.5-turbo": "gpt-3.5-turbo-0125"} + + status_url = "https://duckduckgo.com/duckchat/v1/status" + chat_url = "https://duckduckgo.com/duckchat/v1/chat" + user_agent = 'Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:123.0) Gecko/20100101 Firefox/123.0' + headers = { + 'User-Agent': user_agent, + 'Accept': 'text/event-stream', + 'Accept-Language': 'de,en-US;q=0.7,en;q=0.3', + 'Accept-Encoding': 'gzip, deflate, br', + 'Referer': 'https://duckduckgo.com/', + 'Content-Type': 'application/json', + 'Origin': 'https://duckduckgo.com', + 'Connection': 'keep-alive', + 'Cookie': 'dcm=1', + 'Sec-Fetch-Dest': 'empty', + 'Sec-Fetch-Mode': 'cors', + 'Sec-Fetch-Site': 'same-origin', + 'Pragma': 'no-cache', + 'TE': 'trailers' + } + + @classmethod + async def create_async_generator( + cls, + model: str, + messages: Messages, + **kwargs + ) -> AsyncResult: + async with aiohttp.ClientSession(headers=cls.headers) as session: + async with session.get(cls.status_url, headers={"x-vqd-accept": "1"}) as response: + await raise_for_status(response) + vqd_4 = response.headers.get("x-vqd-4") + payload = { + 'model': cls.get_model(model), + 'messages': messages + } + async with session.post(cls.chat_url, json=payload, headers={"x-vqd-4": vqd_4}) as response: + await raise_for_status(response) + async for line in response.content: + if line.startswith(b"data: "): + chunk = line[6:] + if chunk.startswith(b"[DONE]"): + break + data = json.loads(chunk) + if "message" in data: + yield data["message"] \ No newline at end of file -- cgit v1.2.3