hHddlZddlmZmZmZmZmZddlmZGddZ y)N)ListDictAnyOptionalUnion)UnauthorizedErrorceZdZddedeefdZdeeeffdZ ddedeeeefd ee d ee d ee d ee d ee dee deede de eee fejffdZy) ChatClientNbase_urlapi_keyc>|jd|_||_y)a Initialize the ChatClient. Args: base_url (str): The base URL of the LiteLLM proxy server (e.g., "http://localhost:8000") api_key (Optional[str]): API key for authentication. If provided, it will be sent as a Bearer token. /N)rstrip _base_url_api_key)selfr r s U/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/client/chat.py__init__zChatClient.__init__s"- returncJddi}|jrd|j|d<|S)z Get the headers for API requests, including authorization if api_key is set. Returns: Dict[str, str]: Headers to use for API requests z Content-Typezapplication/jsonzBearer Authorization)r)rheaderss r _get_headerszChatClient._get_headerss0"#56 ==)0'@GO $rmodelmessages temperaturetop_pn max_tokenspresence_penaltyfrequency_penaltyuserreturn_requestc |jd} ||d} ||| d<||| d<||| d<||| d<||| d<||| d <| | | d <tjd | |j| } | r| Stj} |j | j }|j|jS#tjj$r*}|jjd k(r t|d}~wwxYw)a Create a chat completion. Args: model (str): The model to use for completion messages (List[Dict[str, str]]): The messages to generate a completion for temperature (Optional[float]): Sampling temperature between 0 and 2 top_p (Optional[float]): Nucleus sampling parameter between 0 and 1 n (Optional[int]): Number of completions to generate max_tokens (Optional[int]): Maximum number of tokens to generate presence_penalty (Optional[float]): Presence penalty between -2.0 and 2.0 frequency_penalty (Optional[float]): Frequency penalty between -2.0 and 2.0 user (Optional[str]): Unique identifier for the end user return_request (bool): If True, returns the prepared request object instead of executing it Returns: Union[Dict[str, Any], requests.Request]: Either the completion response from the server or a prepared request object if return_request is True Raises: UnauthorizedError: If the request fails with a 401 status code requests.exceptions.RequestException: If the request fails with any other error z/chat/completions)rrNrrr r!r"r#r$POST)rjsoni)rrequestsRequestrSessionsendprepareraise_for_statusr( exceptions HTTPErrorresponse status_coder )rrrrrr r!r"r#r$r%urldatarequestsessionr1es r completionszChatClient.completionss9H 12*/HE  ""-D   !DM =DI  !!+D   ''7D# $  ((9D$ %  DL""638I8I8KRVW N""$ ||GOO$56H  % % '==? """,, zz%%,'**  s>CD "%DD )N)NNNNNNNF)__name__ __module__ __qualname__strrrrrrfloatintboolrrr)r*r8rrr r s    x}   d38n  (,!%$(,0-1"$GGtCH~&Ge_ G  G C= GSMG#5/G$E?GsmGG tCH~x/// 0Grr ) r)typingrrrrrr/r r r@rrrBs33)__r