uPjWdZddlmZmZddlZddlmZddlm Z m Z m Z ddl m Z GddeZGd d ed ZGd de Zy)z Calls Firecrawl's /search endpoint to search the web. Firecrawl API Reference: https://docs.firecrawl.dev/api-reference/endpoint/search )Final TypedDictN)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strceZdZUdZeed<y)_FirecrawlSearchRequestRequiredz1Required fields for Firecrawl Search API request.queryN)__name__ __module__ __qualname____doc__str__annotations__q/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/firecrawl/search/transformation.pyr r s ; Jrr ceZdZUdZeed<eeed<eeeefed<eed<eed<eed<eed<e ed <eed <y ) FirecrawlSearchRequestzu Firecrawl Search API request format. Based on: https://docs.firecrawl.dev/api-reference/endpoint/search limitsources categoriestbslocationcountrytimeoutignoreInvalidURLs scrapeOptionsN) r rrrintrlistrdictboolrrrrrsK J #YT#s(^$$ HM L LrrF)totalc eZdZdZedefdZ ddededzdedzdefdZ ddedzd ed ee ezdzdefd Z d ee ezd edefd Z de jdedefdZy)FirecrawlSearchConfigzhttps://api.firecrawl.dev/v2returncy)N Firecrawlrrrrui_friendly_namez&FirecrawlSearchConfig.ui_friendly_name.srNheadersapi_keyapi_basec z|j||dd|j}|s tdd||d<d|d<|S) z: Validate environment and return headers. )FIRECRAWL_API_KEYFIRECRAWL_API_BASE)caller_api_keycaller_api_base key_env_vars base_env_vardefault_api_basezKFIRECRAWL_API_KEY is not set. Set `FIRECRAWL_API_KEY` environment variable.zBearer Authorizationzapplication/jsonz Content-Type)resolve_server_api_keyr1 ValueError)selfr,r-r.kwargss rvalidate_environmentz*FirecrawlSearchConfig.validate_environment2s_--"$/-!44 . jk k%,WI#6 "4roptional_paramsdatac l|xstdxs |j}|jds|d}|S)z7 Get complete URL for Search endpoint. r1z/search)r r1endswith)r:r.r=r>r;s rget_complete_urlz&FirecrawlSearchConfig.get_complete_urlIs@^~.BC^tG^G^  +"7+Hrr c t|trdj|}d|i}d|vr|d|d<d|vr|d|d<t|}|j D]"\}}||j vs||vs|||<$d|vr dgdd |d<|S) u Transform Search request to Firecrawl API format. Transforms Perplexity unified spec parameters: - query → query (same) - max_results → limit - search_domain_filter → (not directly supported, can use scrapeOptions) - country → country - max_tokens_per_page → (not applicable, ignored) All other Firecrawl-specific parameters are passed through as-is. Args: query: Search query (string or list of strings). Firecrawl only supports single string queries. optional_params: Optional parameters for the request Returns: Dict with typed request data following FirecrawlSearchRequest spec  r max_resultsrrr markdownT)formatsonlyMainContent) isinstancer"joinr#items(get_supported_perplexity_optional_params)r:r r=r; request_data result_dataparamvalues rtransform_search_requestz.FirecrawlSearchConfig.transform_search_request[s2 eT "HHUOE U7 O +$3M$BL !  '&5i&@L #",/ ,113 +LE5DIIKKPU]hPh%* E" + + -&<#',K ( r raw_response logging_objc |j}g}|jdi}t|tro|D]h}|jdxs|jdd}t |jdd|jdd|dd} |j | jnt|t r |jd g} | D]h}|jdxs|jdd}t |jdd|jdd|dd} |j | j|jd g} | D]w}|jdxs|jd d}t |jdd|jdd||jd d} |j | yt|d S)a Transform Firecrawl API response to LiteLLM unified SearchResponse format. Supports both response formats: Firecrawl Cloud (v2): {"data": {"web": [...], "news": [...]}} Firecrawl Self-Hosted (v1): {"success": true, "data": [{"url": "...", "title": "...", ...}, ...]} Args: raw_response: Raw httpx response from Firecrawl API logging_obj: Logging object for tracking Returns: SearchResponse with standardized format r>rE descriptiontitleurlN)rVrWsnippetdate last_updatedwebnewsrXrYsearch)resultsobject)jsongetrHr"rappendr#r) r:rQrRr; response_jsonr^r>resultrX search_result web_results news_resultss rtransform_search_responsez/FirecrawlSearchConfig.transform_search_responses0 ,002 #''3 dD ! . **Z0QFJJ}b4Q , **Wb1 5"-#!% ! }- .d #!%%!4K% . **Z0QFJJ}b4Q , **Wb1 5"-#!% ! }- .#'((62"6L& . **Z0MFJJy"4M , **Wb1 5"-#F+!% ! }- .  r)NN)N)r rrr1 staticmethodrr+r#r<r"rArPhttpxResponseLiteLLMLoggingObjrrhrrrr'r'+s7c ## t*   6*. *T$Z$&  $8T#Y88  8tM nnM 'M  M rr')rtypingrrrj*litellm.litellm_core_utils.litellm_loggingrrl+litellm.llms.base_llm.search.transformationrrrlitellm.secret_managers.mainr r rr'rrrrqsN $ S 8i