From b888c8bf76965ae2314da4e735bed3cd4a190f91 Mon Sep 17 00:00:00 2001 From: Bryan Qiu Date: Mon, 31 Mar 2025 13:50:21 -0400 Subject: [PATCH 1/3] . Signed-off-by: Bryan Qiu --- .../langchain/src/databricks_langchain/chat_models.py | 11 +---------- 1 file changed, 1 insertion(+), 10 deletions(-) diff --git a/integrations/langchain/src/databricks_langchain/chat_models.py b/integrations/langchain/src/databricks_langchain/chat_models.py index 79f2d94e..9008c1c2 100644 --- a/integrations/langchain/src/databricks_langchain/chat_models.py +++ b/integrations/langchain/src/databricks_langchain/chat_models.py @@ -214,7 +214,7 @@ class GetPopulation(BaseModel): """Name of Databricks Model Serving endpoint to query.""" target_uri: str = "databricks" """The target URI to use. Defaults to ``databricks``.""" - temperature: Optional[float] = 0.0 + temperature: Optional[float] """Sampling temperature. Higher values make the model more creative.""" n: int = 1 """The number of completion choices to generate.""" @@ -252,15 +252,6 @@ def endpoint(self, value: str) -> None: def __init__(self, **kwargs: Any): super().__init__(**kwargs) - if "temperature" not in kwargs: - warnings.warn( - "Currently, temperature defaults to 0.0 if not specified. " - "In the next release, temperature will need to be explicitly set. " - "Please update your code to specify a temperature value. " - "Note: If you are using an o1 or o3 model, you need to set temperature=None.", - DeprecationWarning, - stacklevel=2, - ) self.client = get_deployment_client(self.target_uri) self.extra_params = self.extra_params or {} From 7ca2b4533e1c7811767ce26fab64689027d3b280 Mon Sep 17 00:00:00 2001 From: Bryan Qiu Date: Mon, 31 Mar 2025 14:00:18 -0400 Subject: [PATCH 2/3] . Signed-off-by: Bryan Qiu --- integrations/langchain/src/databricks_langchain/chat_models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/integrations/langchain/src/databricks_langchain/chat_models.py b/integrations/langchain/src/databricks_langchain/chat_models.py index 9008c1c2..4957e481 100644 --- a/integrations/langchain/src/databricks_langchain/chat_models.py +++ b/integrations/langchain/src/databricks_langchain/chat_models.py @@ -214,7 +214,7 @@ class GetPopulation(BaseModel): """Name of Databricks Model Serving endpoint to query.""" target_uri: str = "databricks" """The target URI to use. Defaults to ``databricks``.""" - temperature: Optional[float] + temperature: Optional[float] = None """Sampling temperature. Higher values make the model more creative.""" n: int = 1 """The number of completion choices to generate.""" @@ -302,6 +302,7 @@ def _prepare_inputs( if self.max_tokens is not None: data["max_tokens"] = self.max_tokens + print("request data", data) return data def _convert_response_to_chat_result(self, response: Mapping[str, Any]) -> ChatResult: From c144b6657b3f4416d163d939c22cc88420d435aa Mon Sep 17 00:00:00 2001 From: Bryan Qiu Date: Mon, 31 Mar 2025 15:21:29 -0400 Subject: [PATCH 3/3] . Signed-off-by: Bryan Qiu --- integrations/langchain/src/databricks_langchain/chat_models.py | 1 - 1 file changed, 1 deletion(-) diff --git a/integrations/langchain/src/databricks_langchain/chat_models.py b/integrations/langchain/src/databricks_langchain/chat_models.py index 4957e481..df2760eb 100644 --- a/integrations/langchain/src/databricks_langchain/chat_models.py +++ b/integrations/langchain/src/databricks_langchain/chat_models.py @@ -302,7 +302,6 @@ def _prepare_inputs( if self.max_tokens is not None: data["max_tokens"] = self.max_tokens - print("request data", data) return data def _convert_response_to_chat_result(self, response: Mapping[str, Any]) -> ChatResult: