mirror of
https://github.com/xtekky/gpt4free.git
synced 2025-12-06 02:30:41 -08:00
1edd0fff17
- Removed empty string mapping from model_map in AnyModelProviderMixin. - Updated clean_name function to exclude 'chat' from version patterns. - Added stream_timeout parameter to AsyncGeneratorProvider for more flexible timeout handling. - Enhanced chunk yielding in AsyncAuthedProvider to support stream_timeout, allowing for better control over asynchronous responses.
51 lines
No EOL
1.6 KiB
Python
51 lines
No EOL
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
|
|
try:
|
|
from duckai import DuckAI
|
|
has_requirements = True
|
|
except ImportError:
|
|
has_requirements = False
|
|
|
|
from ...typing import CreateResult, Messages
|
|
from ..base_provider import AbstractProvider, ProviderModelMixin
|
|
from ..helper import get_last_user_message
|
|
|
|
class DuckDuckGo(AbstractProvider, ProviderModelMixin):
|
|
label = "Duck.ai (duckduckgo_search)"
|
|
url = "https://duckduckgo.com/aichat"
|
|
api_base = "https://duckduckgo.com/duckchat/v1/"
|
|
|
|
working = False
|
|
supports_stream = True
|
|
supports_system_message = True
|
|
supports_message_history = True
|
|
|
|
default_model = "gpt-4o-mini"
|
|
models = [default_model, "meta-llama/Llama-3.3-70B-Instruct-Turbo", "claude-3-haiku-20240307", "o3-mini", "mistralai/Mistral-Small-24B-Instruct-2501"]
|
|
|
|
duck_ai: DuckAI = None
|
|
|
|
model_aliases = {
|
|
"gpt-4": "gpt-4o-mini",
|
|
"llama-3.3-70b": "meta-llama/Llama-3.3-70B-Instruct-Turbo",
|
|
"claude-3-haiku": "claude-3-haiku-20240307",
|
|
"mixtral-small-24b": "mistralai/Mistral-Small-24B-Instruct-2501",
|
|
}
|
|
|
|
@classmethod
|
|
def create_completion(
|
|
cls,
|
|
model: str,
|
|
messages: Messages,
|
|
proxy: str = None,
|
|
timeout: int = 60,
|
|
**kwargs
|
|
) -> CreateResult:
|
|
if not has_requirements:
|
|
raise ImportError("duckai is not installed. Install it with `pip install -U duckai`.")
|
|
if cls.duck_ai is None:
|
|
cls.duck_ai = DuckAI(proxy=proxy, timeout=timeout)
|
|
model = cls.get_model(model)
|
|
yield cls.duck_ai.chat(get_last_user_message(messages), model, timeout) |