o ưi2@sddlZddlmZmZmZmZmZmZddlZddl Z ddl m Z ddl m Z ddlmZddlmZddlmZddlmZdd lmZmZmZdd lmZdd lmZeradd lmZ e Z!neZ!d Z"GdddeZ#dS)N) TYPE_CHECKINGAnyDictOptionalTupleUnion)verbose_logger)process_response_headers)_safe_convert_created_field) OpenAIError)OpenAIResponsesAPIConfig)get_secret_str)ResponseAPIUsageResponseInputParamResponsesAPIResponse)GenericLiteLLMParams) LlmProviders)Loggingzhttps://api.manus.imc s0eZdZdZedefddZ d#deedee deede fd d Z dedefd d Z d e dedee de fddZdeede defddZdedeeefdede d e def fdd ZdedejdedefddZde fddZdedede d e deeeff dd Zdejdedefd!d"ZZS)$ManusResponsesAPIConfiga Configuration for Manus API's Responses API. Manus API is OpenAI-compatible but has some differences: - API key passed via `API_KEY` header (not `Authorization: Bearer`) - Model format: `manus/{agent_profile}` (e.g., `manus/manus-1.6`) - Requires `extra_body` with `task_mode: "agent"` and `agent_profile` Reference: https://open.manus.im/docs/openai-compatibility returncCstjSN)rZMANUSselfrb/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/manus/responses/transformation.pycustom_llm_provider-sz+ManusResponsesAPIConfig.custom_llm_providerNmodelstreamrcCs|duS)z Manus API doesn't support real-time streaming. It returns a task that runs asynchronously. We fake streaming by converting the response into streaming events. Tr)rrrrrrrshould_fake_stream1s z*ManusResponsesAPIConfig.should_fake_streamcCsd|vr |dddS|S)a Extract agent profile from model name. Model format: `manus/{agent_profile}` Examples: `manus/manus-1.6`, `manus/manus-1.6-lite`, `manus/manus-1.6-max` Returns: str: The agent profile (e.g., "manus-1.6") /)split)rrrrr_extract_agent_profile>s z.ManusResponsesAPIConfig._extract_agent_profileheaderslitellm_paramscCs>|pt}|jptjptd}|std||dd|S)z Validate environment and set up headers for Manus API. Manus uses `API_KEY` header instead of `Authorization: Bearer`. Z MANUS_API_KEYz\Manus API key is required. Set MANUS_API_KEY environment variable or pass api_key parameter.zapplication/json)ZAPI_KEYz Content-Type)rapi_keylitellmr ValueErrorupdate)rr#rr$r%rrrvalidate_environmentMs z,ManusResponsesAPIConfig.validate_environmentapi_basecCs>|p tjp tdp t}|d}|dr|dS|dS)z Get the complete URL for Manus Responses API endpoint. Returns: str: The full URL for the Manus /v1/responses endpoint MANUS_API_BASErz/v1z /responsesz /v1/responses)r&r*r r+rstripendswith)rr*r$rrrget_complete_urlks     z(ManusResponsesAPIConfig.get_complete_urlinput$response_api_optional_request_paramsc sftj|||||d}|j|d}d|d<||d<|dip i}|r(||td|d|S) z Transform the request for Manus API. Manus requires: - `task_mode: "agent"` in the request body - `agent_profile` extracted from model name in the request body )rr/r0r$r#)rZagentZ task_mode agent_profile extra_bodyzManus: Using agent_profile=z, task_mode=agent)supertransform_responses_api_requestr"getr(rdebug) rrr/r0r$r#Z base_requestr1r2 __class__rrr4s"   z7ManusResponsesAPIConfig.transform_responses_api_request raw_response logging_objcCsz,|j|jdiid|}d|vrd|vrt|d|d<d|vr+t|d|d<Wnty;t|j|jdwt|j}t |}d|vsP| ddurTi|d<d|vs_| ddurci|d<d |vsn| d durrg|d <d |vs}| d durt d d d d |d <d |vs| d durdt jdd|d <z tdi|}Wntytd|dtjdi|}Ynw||jd<||jd<|S)z Transform Manus API response to OpenAI-compatible format. Manus uses camelCase (createdAt) instead of snake_case (created_at). complete_input_dictoriginal_responseZadditional_args createdAt created_atmessage status_code reasoningNtextoutputusagerZ input_tokensZ output_tokensZ total_tokensidunknown-)Error constructing ResponsesAPIResponse: , using model_constructadditional_headersr#rZ post_callrDjsonr Exceptionr rBdictr#r r5ruuiduuid4hexrrr6Zmodel_constructZ_hidden_params)rrr9r:raw_response_jsonraw_response_headersprocessed_headersresponserrrtransform_response_api_responses^        z7ManusResponsesAPIConfig.transform_response_api_responsecCsdS)z9Manus does not support native WebSocket for Responses APIFrrrrrsupports_native_websocketsz1ManusResponsesAPIConfig.supports_native_websocket response_idcCs|d|}i}||fS)a Transform the get response API request into a URL and data. Manus API follows OpenAI-compatible format: - GET /v1/responses/{response_id} Reference: https://open.manus.im/docs/openai-compatibility rr)rr[r*r$r#urldatarrr"transform_get_response_api_requestsz:ManusResponsesAPIConfig.transform_get_response_api_requestcCsz,|j|jdiid|}d|vrd|vrt|d|d<d|vr+t|d|d<Wnty;t|j|jdwt|j}t |}d|vsP| ddurTi|d<d|vs_| ddurci|d<d |vsn| d durrg|d <d |vs}| d durt d d d d |d <d |vs| d durdt jdd|d <z tdi|}Wntytd|dtjdi|}Ynw||jd<||jd<|S)z Transform Manus API GET response to OpenAI-compatible format. Manus uses camelCase (createdAt) instead of snake_case (created_at). Same transformation as transform_response_api_response. r;r<r>r?r@rCNrDrErFrrGrHrIrJrKrLrMr#rrN)rr9r:rUrVrWrXrrr#transform_get_response_api_responses^        z;ManusResponsesAPIConfig.transform_get_response_api_responser)__name__ __module__ __qualname____doc__propertyrrrstrboolrr"rQrr)r.rrrr4httpxResponseLiteLLMLoggingObjrrYrZrr^r_ __classcell__rrr7rr!s      + J  r)$rRtypingrrrrrrrgr&Zlitellm._loggingrZ'litellm.litellm_core_utils.core_helpersr ZFlitellm.litellm_core_utils.llm_response_utils.convert_dict_to_responser Z litellm.llms.openai.common_utilsr Z,litellm.llms.openai.responses.transformationr Zlitellm.secret_managers.mainr Zlitellm.types.llms.openairrrZlitellm.types.routerrZlitellm.types.utilsrZ*litellm.litellm_core_utils.litellm_loggingrZ_LiteLLMLoggingObjrir+rrrrrs&