o Zh@sXddlZddlmZddlmZddlmZeeZ eddddGd d d eZ dS) N)urlparse) deprecated) ChatMlflowz0.3.3z1.0z#databricks_langchain.ChatDatabricks)ZsinceZremovalZalternative_importc@sNeZdZUdZdZeed< edefddZedefddZ d d d Z d S) ChatDatabricksa`Databricks` chat models API. To use, you should have the ``mlflow`` python package installed. For more information, see https://mlflow.org/docs/latest/llms/deployments. Example: .. code-block:: python from langchain_community.chat_models import ChatDatabricks chat_model = ChatDatabricks( target_uri="databricks", endpoint="databricks-llama-2-70b-chat", temperature=0.1, ) # single input invocation print(chat_model.invoke("What is MLflow?").content) # single input invocation with streaming response for chunk in chat_model.stream("What is MLflow?"): print(chunk.content, end="|") databricks target_urireturncCdS)zReturn type of chat model.zdatabricks-chatselfr r a/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/chat_models/databricks.py _llm_type,szChatDatabricks._llm_typecCr )Nr r r r r _mlflow_extras1szChatDatabricks._mlflow_extrasNcCs*|jdkrdSt|jjdkrtddS)NrzBInvalid target URI. The target URI must be a valid databricks URI.)rrscheme ValueErrorr r r r _validate_uri5s zChatDatabricks._validate_uri)rN) __name__ __module__ __qualname____doc__rstr__annotations__propertyrrrr r r r r s  r) logging urllib.parserZlangchain_core._apirZ&langchain_community.chat_models.mlflowr getLoggerrloggerrr r r r s