uPj!1UddlZddlmZmZmZddlZddlZddlmZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZmZmZdd lmZdd lmZer dd lmZeZ neZ dZ!ee"d<GddeZ#y)N) TYPE_CHECKINGAnyFinal)verbose_logger)process_response_headers)_safe_convert_created_fieldencode_url_path_segment) OpenAIError)OpenAIResponsesAPIConfig)get_secret_str)ResponseAPIUsageResponseInputParamResponsesAPIResponse)GenericLiteLLMParams) LlmProviders)Loggingzhttps://api.manus.imMANUS_API_BASEc TeZdZdZedefdZ ddedzdedzdedzdefdZ dedefd Z d e ded e dzde fd Z d edzd e defdZdedeezde d e d e de f fd Zdedej$dedefdZdefdZded ed e d e deee ff dZdej$dedefdZxZS)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 returnc"tjSN)rMANUSselfs p/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/manus/responses/transformation.pycustom_llm_providerz+ManusResponsesAPIConfig.custom_llm_provider.s!!!Nmodelstreamrc |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. T)rr r!rs rshould_fake_streamz*ManusResponsesAPIConfig.should_fake_stream2s~rc8d|vr|jdddS|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)rr s r_extract_agent_profilez.ManusResponsesAPIConfig._extract_agent_profile?s& %<;;sA&q) ) rheaderslitellm_paramsc|xs t}|jxstjxs td}|s t d|j |dd|S)z Validate environment and set up headers for Manus API. Manus uses `API_KEY` header instead of `Authorization: Bearer`. MANUS_API_KEYz\Manus API key is required. Set MANUS_API_KEY environment variable or pass api_key parameter.zapplication/json)API_KEYz Content-Type)rapi_keylitellmr ValueErrorupdate)rr*r r+r/s rvalidate_environmentz,ManusResponsesAPIConfig.validate_environmentNsd (A+?+A'//e7??enUdFen  " 2  rapi_basec|xs%tjxstdxst}|j d}|j dr|dS|dS)z Get the complete URL for Manus Responses API endpoint. Returns: str: The full URL for the Manus /v1/responses endpoint rr&z/v1z /responsesz /v1/responses)r0r4r rrstripendswith)rr4r+s rget_complete_urlz(ManusResponsesAPIConfig.get_complete_urlfs_ew//e>BR3SeWe??3'   U #Zz* *=))rinput$response_api_optional_request_paramsct ||||||}|j|}d|d<||d<|jdixsi}|r|j |t j 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 )r r9r:r+r*)r agent task_mode agent_profile extra_bodyz.Manus: Using agent_profile=%s, task_mode=agent)supertransform_responses_api_requestr)getr2rdebug) rr r9r:r+r* base_requestr>r? __class__s rrAz7ManusResponsesAPIConfig.transform_responses_api_request{s $gE1U) F  $:::G %, [!(5 _%ADD\SUV\Z\     +M}]r raw_response logging_objc |j|jdii|j}d|vrd|vrt|d|d<d|vrt|d|d<t|j}t|}d|vs|jdi|d<d|vs|jdi|d<d |vs|jd g|d <d |vs|jd td d d |d <d |vs|jd 'dtjjdd|d < tj |}||j(d<||j(d<|S#t$r"t |j|j wxYw#t$r.t#j$d|tj&di|}YwxYw)z Transform Manus API response to OpenAI-compatible format. Manus uses camelCase (createdAt) instead of snake_case (created_at). complete_input_dictoriginal_responseadditional_args createdAt created_atmessage status_code reasoningNtextoutputusager input_tokens output_tokens total_tokensidunknown-BError constructing ResponsesAPIResponse: %s, using model_constructadditional_headersr*r# post_callrSjsonr Exceptionr rQdictr*rrBruuiduuid4hexrmodel_validaterrCmodel_construct_hidden_params)rr rFrGraw_response_jsonraw_response_headersprocessed_headersresponses rtransform_response_api_responsez7ManusResponsesAPIConfig.transform_response_api_responses< _  ! !"."3"3!6 ; " (4'8'8': //LHY4Y2MN_`kNl2m!,/002MN_`lNm2n!,/'+<+?+?&@#;OPH  QAE%0F%+F4G  G cy)z9Manus does not support native WebSocket for Responses APIFr#rs rsupports_native_websocketz1ManusResponsesAPIConfig.supports_native_websocketsr response_idc6t|d}|d|}i}||fS)z 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) field_namer&r )rrrr4r+r*encoded_response_idurldatas r"transform_get_response_api_requestz:ManusResponsesAPIConfig.transform_get_response_api_requests3&=[Ub%c z#6"78Dyrc |j|jdii|j}d|vrd|vrt|d|d<d|vrt|d|d<t|j}t|}d|vs|jdi|d<d|vs|jdi|d<d |vs|jd g|d <d |vs|jd td d d |d <d |vs|jd 'dtjjdd|d < tj |}||j(d<||j(d<|S#t$r"t |j|j wxYw#t$r.t#j$d|tj&di|}YwxYw)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. rIrJrMrNrOrRNrSrTrUrrVrZr[r\r]r^r*r#r_)rrFrGrjrkrlrms r#transform_get_response_api_responsez;ManusResponsesAPIConfig.transform_get_response_api_responses7 _  ! !"."3"3!6 ; " (4'8'8': //LHY4Y2MN_`kNl2m!,/002MN_`lNm2n!,/'+<+?+?&@#;OPH  Qror)__name__ __module__ __qualname____doc__propertyrrstrboolr$r)rcrr3r8rrAhttpxResponseLiteLLMLoggingObjrrnrqtuplerxrz __classcell__)rEs@rrr"s "\""+/ Tz t  !4Z    C C DNbeiNinr0**** **'''''/3 ' - '  ' 'RBBnnB' B  BH4-    sDy  (@nn@'@  @rr)$rdtypingrrrrr0litellm._loggingr'litellm.litellm_core_utils.core_helpersrFlitellm.litellm_core_utils.llm_response_utils.convert_dict_to_responser$litellm.litellm_core_utils.url_utilsr litellm.llms.openai.common_utilsr ,litellm.llms.openai.responses.transformationr litellm.secret_managers.mainr litellm.types.llms.openairrrlitellm.types.routerrlitellm.types.utilsr*litellm.litellm_core_utils.litellm_loggingr_LiteLLMLoggingObjrr__annotations__rr#rrrsk ,, +LI8Q7 6,X*..^6^r