o ưi' @stdZddlmZmZmZmZmZddlZddlm Z ddl m Z m Z ddl mZddlmZmZGdd d e ZdS) zY Re rank api LiteLLM supports the re rank API format, no paramter transformation occurs )AnyDictListOptionalUnionN)BaseLLM)_get_httpx_clientget_async_httpx_client)TogetherAIRerankConfig) RerankRequestRerankResponsec@seZdZ     ddedededeeeeeeffdee d eeed ee d ee d ee d e fddZ deeefded e fddZ dS)TogetherAIRerankNTFmodelapi_keyquery documentstop_n rank_fieldsreturn_documentsmax_chunks_per_doc _is_asyncreturnc Cst} t||||||d} | jdd} |durtd| r#|| |S| jdddd|d| d } | jd krs