o ưiI@sdZddlmZmZmZmZmZddlZddlm Z ddl m Z ddl mZddlmZddlmZdd lmZmZmZmZmZdd lmZdd lmZGd d d eZdS)a Transformation logic for Perplexity Agent API (Responses API) This module handles the translation between OpenAI's Responses API format and Perplexity's Responses API format, which supports: - Third-party model access (OpenAI, Anthropic, Google, xAI, etc.) - Presets for optimized configurations - Web search and URL fetching tools - Reasoning effort control - Instructions parameter for system-level guidance )AnyDictListOptionalUnionN)verbose_logger)Logging)BaseLLMException)OpenAIResponsesAPIConfigget_secret_str)ResponseAPIUsageResponseInputParam!ResponsesAPIOptionalRequestParamsResponsesAPIResponseResponsesAPIStreamingResponse)GenericLiteLLMParams) LlmProvidersc @seZdZdZedefddZdedefddZ de ded e e de fd d Z d e ed e defd dZdedededefddZdeeeefdeeeeffddZdedeeefded e de def ddZdeeefdeeeeeefffddZdedejdedefddZd eeefdeeeffd!d"Zded#e dede fd$d%Z!d&e de fd'd(Z"defd)d*Z#d+S),PerplexityResponsesConfigz Configuration for Perplexity Agent API (Responses API) Reference: https://docs.perplexity.ai/docs/agent-api/overview returncCstjS)N)rZ PERPLEXITYselfrg/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/perplexity/responses/transformation.pycustom_llm_provider)sz-PerplexityResponsesConfig.custom_llm_providermodelcCsgdS)z Perplexity Responses API supports a different set of parameters Ref: https://docs.perplexity.ai/api-reference/responses-post Params aligned with response-echo fields and Open Responses spec. )max_output_tokensstream temperaturetop_ptools reasoningpreset instructionsmodels tool_choiceparallel_tool_callsmax_tool_callstextprevious_response_idstore background truncationmetadatasafety_identifieruserstream_options top_logprobsprompt_cache_keyfrequency_penaltypresence_penalty service_tierr)rrrrrget_supported_openai_params-sz5PerplexityResponsesConfig.get_supported_openai_paramsheaderslitellm_paramscCs.tdptd}|rd||d<d|d<|S)z'Validate environment and set up headersZPERPLEXITYAI_API_KEYZPERPLEXITY_API_KEYzBearer Authorizationzapplication/jsonz Content-Typer )rr7rr8Zapi_keyrrrvalidate_environmentQs z.PerplexityResponsesConfig.validate_environmentapi_basecCs(|dur tdp d}|d}|dS)z5Get the complete URL for the Perplexity Responses APINZPERPLEXITY_API_BASEzhttps://api.perplexity.ai/z /v1/responses)r rstrip)rr;r8rrrget_complete_urlas    z*PerplexityResponsesConfig.get_complete_urlresponse_api_optional_params drop_paramsc Csbi}|dr |d|d<|dr|d|d<|dr#|d|d<|dr.|d|d<|dr9|d|d<|d}|rD||d<|d}|rO||d<|drZ|d|d<|d }|rod d |D}|||d <|d rz|d |d <|d dur|d |d <|dr|d|d<|d} | r| |d<|dr|d|d<|ddur|d|d<|ddur|d|d<|dr|d|d<|dr|d|d<|dr|d|d<|dr|d|d<|ddur|d|d<|dr|d|d<|ddur|d|d<|ddur#|d|d<|dr/|d|d<|S)a0 Map OpenAI Responses API parameters to Perplexity format Key differences: - Supports 'preset' parameter for predefined configurations - Supports 'instructions' parameter for system-level guidance - Tools are specified differently (web_search, fetch_url) rrrrr0r"r#r!r cSs"g|] }t|dr t|n|qS)__dict__)hasattrdict).0toolrrr s"z?PerplexityResponsesConfig.map_openai_params..r%r&Nr'r(r)r*r+r,r-r.r/r1r2r3r4r5)get_transform_tools) rr?rr@Z mapped_paramsr"r#r Z tools_listZ text_paramrrrmap_openai_paramsrs                               z+PerplexityResponsesConfig.map_openai_paramsr cCsNg}|D] }t|tr$|dd}|dvr||q|dkr$||q|S)z Transform tools to Perplexity format. Perplexity supports (per public OpenAPI spec): - web_search: Performs web searches - fetch_url: Fetches content from URLs - function: Function Calling type)Z web_searchZ fetch_urlfunction) isinstancerCrGappend)rr Zperplexity_toolsrEZ tool_typerrrrHs     z*PerplexityResponsesConfig._transform_toolsinput$response_api_optional_request_paramsc Csx|dr|dd}|||d}n|dr%|d||d}n|||d}|D]\}} | ||<q1|S)zF Transform request to Perplexity Responses API format zpreset/rK)r"rOr")rrO) startswithreplace _format_inputrGpopitems) rrrOrPr8r7Z preset_namedatakeyvaluerrrtransform_responses_api_requests   z9PerplexityResponsesConfig.transform_responses_api_requestcCs`t|tr|St|tr,g}|D]}t|tr)d|d|ddd}||q|St|S)z Format input for Perplexity Responses API The API accepts either: - A simple string for single-turn queries - An array of message objects for multi-turn conversations messagerolecontentrK)rJr[r\)rMstrlistrCrGrN)rrOZformatted_messagesitemZformatted_messagerrrrS#s     z'PerplexityResponsesConfig._format_input raw_response logging_objc Csz|}Wnty}z t|jdt|dd}~ww|d}|dkr;|di}|dd}t|j|d|d i} || } | rOtdi| nd} t|d d d |d d|dd|d||dg| d} | S)z\ Transform Perplexity Responses API response to OpenAI Responses API format zFailed to parse response: ) status_coderZNstatusfailederrorrZz Unknown errorusageidrKresponse created_atr completedroutput)rgobjectrircrrkrfr) json Exceptionr rbr]rG_transform_usager r) rrr`raZraw_response_jsonercre error_message usage_dataZtransformed_usage_dict usage_objrhrrrtransform_response_api_response?sB             z9PerplexityResponsesConfig.transform_response_api_responserrcCs|dd|dd|ddd}|d}t|tr0d|vr0|d|d<td||dn|d ur8||d<d |vrB|d |d <d |vrL|d |d <|S) a< Transform Perplexity usage data to OpenAI format Perplexity returns: { "input_tokens": 100, "output_tokens": 200, "total_tokens": 300, "cost": { "currency": "USD", "input_cost": 0.0001, "output_cost": 0.0002, "total_cost": 0.0003 } } OpenAI expects: { "input_tokens": 100, "output_tokens": 200, "total_tokens": 300, "cost": 0.0003 } input_tokensr output_tokens total_tokens)rurvrwcost total_cost5Transformed Perplexity cost object to float: %s -> %sNZinput_tokens_detailsZoutput_tokens_details)rGrMrCrdebug)rrrZ transformedcost_objrrrrors&       z*PerplexityResponsesConfig._transform_usage parsed_chunkcCstd|t|d}tj|d}||}z#|d}t|tr:|ddur:t|}t||d<d|dd<Wnt yItdYnw|d i|S) zb Transform a parsed streaming response chunk into a ResponsesAPIStreamingResponse zRaw Perplexity Chunk=%srJ) event_typerecodeNZ unknown_errorz-Failed to coalesce error.code in parsed_chunkr) rr{r]rGrZget_event_model_class_transform_perplexity_chunkrMrCrn)rrr}rar~Zevent_pydantic_modelZ error_objrrrtransform_streaming_responses"      z6PerplexityResponsesConfig.transform_streaming_responsechunkc Cst|}zV|d}t|trO|d}t|trR|d}t|trUd|vrXt|}t||d<t||dd<|d|ddd<td||dW|SW|SW|SW|SW|Stys}z td|WYd}~|Sd}~ww)z Transform Perplexity-specific fields in a streaming chunk to OpenAI format. This handles: - Converting Perplexity's cost object to a simple float rhrfrxryrzz.Failed to transform Perplexity cost object: %sN)rCrGrMrr{rn)rrZ response_objrsr|rprrrrs>       z5PerplexityResponsesConfig._transform_perplexity_chunkcCsdS)z>Perplexity does not support native WebSocket for Responses APIFrrrrrsupports_native_websocketsz3PerplexityResponsesConfig.supports_native_websocketN)$__name__ __module__ __qualname____doc__propertyrrr]r^r6rCrrr:r>rboolrrIrrrHrrrYrShttpxResponseLiteLLMLoggingObjrrtrorrrrrrrrr!s$   *s  %   "36 "$r)rtypingrrrrrrZlitellm._loggingrZ*litellm.litellm_core_utils.litellm_loggingrrZ)litellm.llms.base_llm.chat.transformationr Z,litellm.llms.openai.responses.transformationr Zlitellm.secret_managers.mainr Zlitellm.types.llms.openair rrrrZlitellm.types.routerrZlitellm.types.utilsrrrrrrs