uPj(ddlmZddlmZmZmZmZddlmZerddl m Z m Z m Z ddl mZddlmZddlmZGd d eZy ) )abstractmethod) TYPE_CHECKINGFinalOptionalUnion)BaseLLMModelInfo)URLHeadersResponse)Logging)CostResponseTypesBaseLLMExceptioncfeZdZedededefdZdedededzddfd Zed edzd edzd edededzd ede deffdZ d"ded ededzd ed edzde ee dzff dZ dede deedfddfdZd ededddedddededfdZdeeddd ededededf dZd ee deefd!Zy)#BasePassthroughConfigendpoint request_datareturncy)z= Check if the request is a streaming request N)selfrrs u/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/base_llm/passthrough/transformation.pyis_streaming_requestz*BasePassthroughConfig.is_streaming_requestbase_target_urlrequest_query_paramsNr cddlm}ddl}|jd}|j d}|d|}|j |}|r'|j ||jd}|S)aa Helper function to add query params to the url Args: endpoint: str - the endpoint to add to the url base_target_url: str - the base url to add the endpoint to request_query_params: Optional[dict] - the query params to add to the url Returns: httpx.URL - the formatted url r) urlencodeN/ascii)query) urllib.parser httpxrstriplstripr copy_withencode) rrrrr r%basefull_urlurls r format_urlz BasePassthroughConfig.format_urlso +%,,S1??3'!F!H:.ii! --i0D&E&L&LW&U-VC rapi_baseapi_keymodellitellm_paramscy)z Get the complete url for the request Returns: - complete_url: URL - the complete url for the request - base_target_url: str - the base url to add the endpoint to. Useful for auth headers. Nr)rr.r/r0rrr1s rget_complete_urlz&BasePassthroughConfig.get_complete_url4rrheadersc |dfS)a0 Some providers like Bedrock require signing the request. The sign request funtion needs access to `request_data` and `complete_url` Args: headers: dict optional_params: dict request_data: dict - the request body being sent in http request api_base: str - the complete url being sent in http request Returns: dict - the signed headers Update the headers with the signed headers in this function. The return values will be sent as headers in the http request. Nr)rr4r1rr.r0s r sign_requestz"BasePassthroughConfig.sign_requestEs(}r error_message status_coder rc$ddlm}||||S)Nrr)r8messager4))litellm.llms.base_llm.chat.transformationr)rr7r8r4rs rget_error_classz%BasePassthroughConfig.get_error_class[s OKX_``rcustom_llm_providerhttpx_responser logging_objLiteLLMLoggingObjrcyNr)rr0r=r>rr?rs rlogging_non_streaming_responsez4BasePassthroughConfig.logging_non_streaming_responsebs r all_chunkslitellm_logging_objcyrBr)rrDrEr0r=rs rhandle_logging_collected_chunksz5BasePassthroughConfig.handle_logging_collected_chunksmsr raw_bytescdj|jd}|jdDcgc]#}|js|j%}}|Scc}w)a Converts a list of raw bytes into a list of string lines, similar to aiter_lines() Args: raw_bytes: List of bytes chunks from aiter.bytes() Returns: List of string lines, with each line being a complete data: {} chunk rzutf-8 )joindecodesplitstrip)rrH combined_strlineliness r_convert_raw_bytes_to_str_linesz5BasePassthroughConfig._convert_raw_bytes_to_str_lineswsV"hhy188A 2>1C1CD1IZTZZ\ ZZ [s A  A rB)__name__ __module__ __qualname__rstrdictboolrr-tupler3bytesr6intrr<rrClistrGrRrrrrrs S    #Tk   < * t     #Tk    ucz   ,! Tk    Tz  tUT\! ",a a/2a=B4?=Sa a    !  #    )     % &  I1  !    % &ecrrN)abcrtypingrrrr base_utilsr r%r r r *litellm.litellm_core_utils.litellm_loggingr r@litellm.types.utilsrchat.transformationrrrrrrcs288),,W56x,xr