h FdZddlmZmZddlmZddlmZGddeZy) z{ Support for OpenAI's `/v1/chat/completions` endpoint. Calls done in OpenAI/openai.py as DataRobot is openai-compatible. )OptionalTupleget_secret_str)OpenAILikeChatConfigceZdZed deedefdZed deedeefdZdeedeedeeeeeffdZ d deedeeded e d e d ee defd Z y)DataRobotConfigNapi_keyreturnc(|xstdxsdS)zAttempt to ensure that the API key is set, preferring the user-provided key over the secret manager key (``DATAROBOT_API_TOKEN``). If both are None, a fake API key is returned for testing. DATAROBOT_API_TOKENz fake-api-keyr)r s f/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/llms/datarobot/chat/transformation.py_resolve_api_keyz DataRobotConfig._resolve_api_key sQ.)>?Q>Qapi_basecr|xs td}|d}d|vr d|vr|dz }|jds|dz }|S)aMAttempt to ensure that the API base is set, preferring the user-provided key over the secret manager key (``DATAROBOT_ENDPOINT``). If both are None, a default Llamafile server URL is returned. See: https://github.com/Mozilla-Ocho/llamafile/blob/bd1bbe9aabb1ee12dbdcafa8936db443c571eb9d/README.md#L61 DATAROBOT_ENDPOINTzhttps://app.datarobot.comzapi/v2/deploymentsz#api/v2/genai/llmgw/chat/completionsz$/api/v2/genai/llmgw/chat/completions/)rendswith)rs r_resolve_api_basez!DataRobotConfig._resolve_api_basesZC~.BC  2H x /4HDBB  % OHrc^tj|}tj|}||fS)a4Attempts to ensure that the API base and key are set, preferring user-provided values, before falling back to secret manager values (``DATAROBOT_ENDPOINT`` and ``DATAROBOT_API_TOKEN`` respectively). If an API key cannot be resolved via either method, a fake key is returned. )r rr)selfrr dynamic_api_keys r$_get_openai_compatible_provider_infoz4DataRobotConfig._get_openai_compatible_provider_info/s/#44X>)::7C((rmodeloptional_paramslitellm_paramsstreamct|S)a  Get the complete URL for the API call. Datarobot's API base is set to the complete value, so it does not need to be updated to additionally add chat completions. Returns: str: The complete URL for the API call. )str)rrr rrrrs rget_complete_urlz DataRobotConfig.get_complete_url?s"8}r)N) __name__ __module__ __qualname__ staticmethodrr!rrrrdictboolr"rrr r sR(3-R3RRHSMXc]0)3-)#) x}hsm+ , )."&3-#      rr N) __doc__typingrrlitellm.secret_managers.mainropenai_like.chat.transformationrr r)rrr.s$ #7CD*Dr