g6 8ddlmZmZmZddlmZGddZy))LiteralOptionalTuple)DatabricksErrorceZdZdeedeedeedeeeffdZdeedeededdee deedeeeff d Z y ) DatabricksBaseapi_keyapi_baseheadersreturnc|xsddi} ddlm}|}|xs|jjd}| |jj }i||}||fS#t $rt ddwxYw) N Content-Typeapplication/jsonr)WorkspaceClientz/serving-endpointsa If the Databricks base URL and API key are not set, the databricks-sdk Python library must be installed. Please install the databricks-sdk, set {LLM_PROVIDER}_API_BASE and {LLM_PROVIDER}_API_KEY environment variables, or provide the base URL and API key as arguments. status_codemessage)databricks.sdkrconfighost authenticate ImportErrorr)selfr r r rdatabricks_clientdatabricks_auth_headerss Y/var/www/openai/venv/lib/python3.12/site-packages/litellm/llms/databricks/common_utils.py_get_databricks_credentialsz*DatabricksBase._get_databricks_credentialssAn.@A  6 / 1 W&7&>&>&C&C%DDV#WH%,,99;(A4@@W$ $ !H  s AAA1 endpoint_type)chat_completions embeddingscustom_endpointc|(|&| tdd|j|||\}}|&|r tdd|j|||\}}|dj|dd}n$|"|jd dj|i|d ||d <|d k(r|d urd j|}||fS|dk(r|d urdj|}||fS)NrzMissing API Key - A call is being made to LLM Provider but no key is set either in the environment variables ({LLM_PROVIDER}_API_KEY) or via paramsr)r r r zMissing API Base - A call is being made to LLM Provider but no api base is set either in the environment variables ({LLM_PROVIDER}_API_KEY) or via paramsz Bearer {}r) Authorizationrr%zBearer r!Tz{}/chat/completionsr"z {}/embeddings)rrformatupdate)rr r r r#r s rdatabricks_validate_environmentz.DatabricksBase.databricks_validate_environment$sE ?w*% #r %)$D$D%w%E%!'  % #x %)$D$D%w%E%!' ?!,!3!3G!< 2G "1C1CG1LMN  )0 ':GO $ . .?$3N,33H=H  l *d/J&--h7H  N) __name__ __module__ __qualname__rstrdictrrrboolr(r)rr r s}08 HPQU sDy :.!#.!3-.!?@ .! "$ .! $ .! sDy .!r)r N)typingrrr exceptionsrr r0r)rr3s++'L!L!r)