Change to not modify model_id

langchain-ai · Dec 18, 2024 · 86f6e0d · 86f6e0d
1 parent 1d28fe7
commit 86f6e0d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/aws/langchain_aws/chat_models/bedrock.py b/libs/aws/langchain_aws/chat_models/bedrock.py
@@ -410,9 +410,9 @@ def get_lc_namespace(cls) -> List[str]:
     @model_validator(mode="before")
     @classmethod
     def set_beta_use_converse_api(cls, values: Dict) -> Any:
-        model_id = values.get("model_id", values.get("model")) or ""
+        model_id = values.get("model_id", values.get("model"))
 
-        if "beta_use_converse_api" not in values:
+        if model_id is not None and "beta_use_converse_api" not in values:
             values["beta_use_converse_api"] = "nova" in model_id
         return values