o
�iK
� @ st d Z ddlZddlZddlZddlmZmZmZ ddlm Z ddl
m Z dd e d
e
d e
d eee ef fd
d�ZdS )z&DuckDuckGo search tool implementation.� N)�Any�Dict�List)�DDGS� )� PROXY_URL� � �query� max_results�retries�returnc s< d}g d�}g d�}g d�}t |�D �]}z�|dkr6|d| t�dd� }t�d |d
�d �� t�|� nt�dd�} t�d | d
�d �� t�| � t�d
|d � d|� d| � �� ttd�}
t |
j
| |d��} | sut�d� g W S g } | D ]}
| � |
� dd�|
� dd�|
� dd�d�� qyt�dt
| �� �� | W S t�y } zot|�}|�� � t�d|� �� t� fdd�|D ��r�t�d � � t� fd d�|D ��r�||d k r�d t�dd � }t�d |d
�d!�� t�|� W Y d"}~qtd#��t� fd$d�|D ��}|�r||d k�r� W Y d"}~qd"}~ww g S )%aV Execute DuckDuckGo search.
This function performs web searches using DuckDuckGo with retry logic,
rate limiting handling, and proxy support.
Args:
query: The search query string.
max_results: Maximum number of results to return (default: 5).
retries: Number of retry attempts on failure (default: 3).
Returns:
A list of search result dictionaries containing:
- title: The result title
- url: The result URL
- snippet: The result body/snippet
Raises:
Exception: On TLS errors or when retries are exhausted.
r )ztls handshake failedzunexpected eofzclient error (connect)�ssl)Z ratelimitZ429Z202)�timeout�
connectionz
connect errorr r � r z Retry wait z.1fz secondszInitial delay zSearch query (�/z): )�proxy)r z Search successful but no results�title� �href�body)r �url�snippetz Search successful, results: zSearch error: c 3 � � | ]}|� v V qd S �N� ��.0�k�Z error_lowerr �_C:\Users\Rainy_Universe\Desktop\claude-code\mcp-server-ddgs\src\mcp_server_ddgs\tools\search.py� <genexpr>Z � � z search_ddgs.<locals>.<genexpr>z#TLS layer failure, possibly blockedc 3 r r r r r r r! r" ^ r# � �
zRate limit hit, waiting z seconds before retryNzFRate limit still active after multiple retries, please try again laterc 3 r r r r r r r! r" g r# )�range�random�uniform�logging�info�time�sleepr r �list�text�append�get�len� Exception�str�lower�error�any�warning)r
r r Z
base_delayZfatal_tls_errorsZrate_limit_errorsZnetwork_errorsZattempt�delayZ
initial_delay�ddgsZsearch_results�results�r�e� error_msgZ wait_timeZ retryabler r r! � search_ddgs
sb
�
���r> )r r )�__doc__r) r' r+ �typingr r r r9 r � constantsr r3 �intr- �dictr> r r r r! �<module> s ,
