uPjdZddlmZmZddlZddlmZddlm Z m Z m Z ddl m Z GddeZGd d ed ZGd de Zy)a[ Calls fastCRW's /v1/search endpoint to search the web. fastCRW is a Firecrawl-compatible web data engine (single Rust binary; self-host or cloud). The search response uses the Firecrawl-compatible envelope { "success": true, "data": [ { "title", "url", "description", "markdown"? } ] }. fastCRW API Reference: https://fastcrw.com/docs/rest-api )Final TypedDictN)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strceZdZUdZeed<y)_FastCRWSearchRequestRequiredz/Required fields for fastCRW Search API request.queryN)__name__ __module__ __qualname____doc__str__annotations__o/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/fastcrw/search/transformation.pyr r s 9 Jrr c6eZdZUdZeed<eeed<eed<y)FastCRWSearchRequestz\ fastCRW Search API request format. Based on: https://fastcrw.com/docs/rest-api limitsources scrapeOptionsN) r rrrintrlistrdictrrrrrs J #YrrF)totalc eZdZdZedefdZ ddededzdedzdefdZ ddedzd ed ee ezdzdefd Z d ee ezd edefd Z de jdedefdZy)FastCRWSearchConfigzhttps://fastcrw.com/api/v1returncy)NfastCRWrrrrui_friendly_namez$FastCRWSearchConfig.ui_friendly_name,srNheadersapi_keyapi_basec z|j||dd|j}|s tdd||d<d|d<|S) z: Validate environment and return headers. ) CRW_API_KEY CRW_API_BASE)caller_api_keycaller_api_base key_env_vars base_env_vardefault_api_basez?CRW_API_KEY is not set. Set `CRW_API_KEY` environment variable.zBearer Authorizationzapplication/jsonz Content-Type)resolve_server_api_keyFASTCRW_API_BASE ValueError)selfr%r&r'kwargss rvalidate_environmentz(FastCRWSearchConfig.validate_environment0s_--"$)'!22 . ^_ _%,WI#6 "4roptional_paramsdatac l|xstdxs |j}|jds|d}|S)z7 Get complete URL for Search endpoint. r*z/search)r r2endswith)r4r'r7r8r5s rget_complete_urlz$FastCRWSearchConfig.get_complete_urlGs?V~n=VAVAV  +"7+Hrr c t|trdj|}d|i}d|vr|d|d<t|}|j D]"\}}||j vs||vs|||<$d|vr dgdd|d<|S) a Transform Search request to fastCRW API format. Transforms Perplexity unified spec parameters: - query -> query (same) - max_results -> limit All other fastCRW-specific parameters are passed through as-is. Args: query: Search query (string or list of strings). fastCRW only supports single string queries. optional_params: Optional parameters for the request Returns: Dict with typed request data following FastCRWSearchRequest spec  r max_resultsrrmarkdownT)formatsonlyMainContent) isinstancerjoinritems(get_supported_perplexity_optional_params)r4r r7r5 request_data result_dataparamvalues rtransform_search_requestz,FastCRWSearchConfig.transform_search_requestYs, eT "HHUOE U5 O +$3M$BL !",/ ,113 +LE5DIIKKPU]hPh%* E" + + -&<#',K ( r raw_response logging_objc ^|j}g}|jdg}t|trm|D]h}|jdxs|jdd}t |jdd|jdd|dd} |j | jt |d S) a Transform fastCRW API response to LiteLLM unified SearchResponse format. fastCRW (Firecrawl-compatible) returns: {"success": true, "data": [{"url": "...", "title": "...", "description": "...", "markdown"?: "..."}, ...]} Args: raw_response: Raw httpx response from fastCRW API logging_obj: Logging object for tracking Returns: SearchResponse with standardized format r8r? descriptiontitleurlN)rPrQsnippetdate last_updatedsearch)resultsobject)jsongetrBrrappendr) r4rKrLr5 response_jsonrVr8resultrR search_results rtransform_search_responsez-FastCRWSearchConfig.transform_search_responses& ,002 #''3 dD ! . **Z0QFJJ}b4Q , **Wb1 5"-#!% ! }- .  r)NN)N)r rrr2 staticmethodrr$rr6rr;rJhttpxResponseLiteLLMLoggingObjrr^rrrr r )s3c ## t*   6*. *T$Z$&  $2T#Y22  2h( nn( '(  ( rr )rtypingrrr`*litellm.litellm_core_utils.litellm_loggingrrb+litellm.llms.base_llm.search.transformationrrrlitellm.secret_managers.mainr r rr rrrrgsN$ S 8I 8L *L r