h78jdZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ej e ZedZe d&d ededededzdedededeeeeffdZe d'd ededededzdededededzdedzdedzdedzdedzdeeeeffdZe d&d ededededzdedededeeeeffdZe d(d ededededzdededededzdedzdedzdeeeeffd Ze d)d ededededeeeeff d!Zed*d#ed$edeeeezffd%ZdS)+zMCP server for DDGS.N)Any)FastMCP)DDGS)_expand_proxy_tb_aliasz ddgs-searchus-enmoderate autoqueryregion safesearch timelimit max_resultspagebackendreturnc xKtjfdd{V}t|S)aPerform a text search using DDGS. Args: query: Search query string region: Region for search (e.g., us-en, uk-en, ru-ru) safesearch: Safe search setting (on, moderate, off) timelimit: Time limit (d, w, m, y) or custom date range max_results: Maximum number of results to return page: Page number of results backend: Search backend (auto, or specific engine) Returns: List of search results with title, href, and body c tttjdSN DDGS_PROXYproxy)r r rrrrr)rrosenvirongettextrrrr r rrs[/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/ddgs/api_server/mcp.pyzsearch_text..-T1"*..2N2NOOPPPUU!#V  Nasyncio to_threadlistr r rrrrrresultss``````` r search_textr)2%                  G ==r"sizecolor type_imagelayout license_imagec  Ktj  f dd{V} t| S)aPerform an image search using DDGS. Args: query: Image search query string region: Region for search (e.g., us-en, uk-en, ru-ru) safesearch: Safe search setting (on, moderate, off) timelimit: Time limit (d, w, m, y) or custom date range max_results: Maximum number of results to return page: Page number of results backend: Search backend (auto, or specific engine) size: Image size (Small, Medium, Large, Wallpaper) color: Image color filter type_image: Image type (photo, clipart, gif, transparent, line) layout: Image layout (Square, Tall, Wide) license_image: Image license filter Returns: List of image search results with title, image URL, and source c tttjd    S)Nrr) r r rrrrrr+r,r-r.r/)rrrrrimages) rr,r.r/rrr r rr+rr-s rr zsearch_images..^sc1"*..2N2NOOPPPWW!#!'X   r"Nr#) r r rrrrrr+r,r-r.r/r(s ```````````` r search_imagesr3:sF% G ==r"c xKtjfdd{V}t|S)aPerform a news search using DDGS. Args: query: News search query string region: Region for search (e.g., us-en, uk-en, ru-ru) safesearch: Safe search setting (on, moderate, off) timelimit: Time limit (d, w, m, y) or custom date range max_results: Maximum number of results to return page: Page number of results backend: Search backend (auto, or specific engine) Returns: List of news results with title, URL, source, and date c tttjdSr)rrrrrnewsrsrr zsearch_news..r!r"Nr#r's``````` r search_newsr7pr*r" resolutiondurationlicense_videosc  Ktj f dd{V} t| S)aPerform a video search using DDGS. Args: query: Video search query string region: Region for search (e.g., us-en, uk-en, ru-ru) safesearch: Safe search setting (on, moderate, off) timelimit: Time limit (d, w, m) or custom date range max_results: Maximum number of results to return page: Page number of results backend: Search backend (auto, or specific engine) resolution: Video resolution (high, standard) duration: Video duration (short, medium, long) license_videos: Video license (creativeCommon, youtube) Returns: List of video search results with title, URL, and metadata c  tttjd  S)Nrr) r r rrrrrr8r9r:)rrrrrvideos) rr9r:rrr r r8rrs rr zsearch_videos..s]1"*..2N2NOOPPPWW!#!)X   r"Nr#) r r rrrrrr8r9r:r(s `````````` r search_videosr>s>%         G ==r"clKtjfdd{V}t|S)aGPerform a book search using DDGS. Args: query: Books search query string max_results: Maximum number of results to return page: Page number of results backend: Search backend (auto, or specific engine) Returns: List of book search results with title, author, and metadata ctttjdS)Nrr)r rrr)rrrrrbooks)rrrr srr zsearch_books..sK1"*..2N2NOOPPPVV# W  r"Nr#)r rrrr(s```` r search_booksrBsm&%       G ==r" text_markdownurlfmtcFKtjfdd{VS)aExtract content from a URL. Args: url: The URL to fetch and extract content from. fmt: Output format: "text_markdown", "text_plain", "text_rich", "text" (raw HTML), "content" (raw bytes). Returns: Dictionary with url and content keys. ctttjdS)NrrrDrE)rrrrrextract)rErDsrr z!extract_content..sE1"*..2N2NOOPPPXXY  r"N)r$r%rHs``rextract_contentrJsV"           r")rrNr r r ) rrNr r r NNNNN) rrNr r r NNN)r r r )rC)__doc__r$loggingrtypingrmcp.server.fastmcprddgsr ddgs.utilsr getLogger__name__loggermcptoolstrintr&dictr)r3r7r>rBbytesrJr"rr[sc &&&&&&------  8 $ $gm  ## # ##Tz #  #  ## $sCx.### #L  ! $22 2 22Tz 2  2  22 *2 :2d 2 $J2:2 $sCx.222 2j  ## # ##Tz #  #  ## $sCx.### #L  !!%,, , ,,Tz ,  ,  ,,d ,Dj,$J, $sCx.,,, ,^      $sCx.  :s4SSX[HXCY r"