langchain-anthropic[patch]: Add ruff bandit rules (#31789)

Author: mdrxy

libs/partners/anthropic/langchain_anthropic/_client_utils.py

@@ -25,7 +25,7 @@ def __del__(self) -> None:
 
         try:
             self.close()
-        except Exception:
+        except Exception:  # noqa: S110
             pass
 
 
@@ -39,7 +39,7 @@ def __del__(self) -> None:
         try:
             # TODO(someday): support non asyncio runtimes here
             asyncio.get_running_loop().create_task(self.aclose())
-        except Exception:
+        except Exception:  # noqa: S110
             pass
 
 

libs/partners/anthropic/langchain_anthropic/chat_models.py

@@ -327,9 +327,10 @@ def _format_messages(
 
         if not isinstance(message.content, str):
             # parse as dict
-            assert isinstance(
-                message.content, list
-            ), "Anthropic message content must be str or list of dicts"
+            if not isinstance(message.content, list):
+                raise ValueError(
+                    "Anthropic message content must be str or list of dicts"
+                )
 
             # populate content
             content = []

libs/partners/anthropic/pyproject.toml

@@ -60,7 +60,7 @@ plugins = ['pydantic.mypy']
 target-version = "py39"
 
 [tool.ruff.lint]
-select = ["E", "F", "I", "T201", "UP"]
+select = ["E", "F", "I", "T201", "UP", "S"]
 ignore = [ "UP007", ]
 
 [tool.coverage.run]
@@ -73,3 +73,9 @@ markers = [
     "compile: mark placeholder test used to compile integration tests without running them",
 ]
 asyncio_mode = "auto"
+
+[tool.ruff.lint.extend-per-file-ignores]
+"tests/**/*.py" = [
+    "S101", # Tests need assertions
+    "S311", # Standard pseudo-random generators are not suitable for cryptographic purposes
+]

libs/partners/anthropic/tests/integration_tests/test_chat_models.py

@@ -651,7 +651,7 @@ def test_anthropic_bind_tools_tool_choice(tool_choice: str) -> None:
 
 def test_pdf_document_input() -> None:
     url = "https://www.w3.org/WAI/ER/tests/xhtml/testfiles/resources/pdf/dummy.pdf"
-    data = b64encode(requests.get(url).content).decode()
+    data = b64encode(requests.get(url, timeout=10).content).decode()
 
     result = ChatAnthropic(model=IMAGE_MODEL_NAME).invoke(
         [

libs/partners/anthropic/tests/unit_tests/test_chat_models.py

@@ -1071,4 +1071,4 @@ def mock_create(*args: Any, **kwargs: Any) -> Message:
 
     # Test headers are correctly propagated to request
     payload = llm._get_request_payload([input_message])
-    assert payload["mcp_servers"][0]["authorization_token"] == "PLACEHOLDER"
+    assert payload["mcp_servers"][0]["authorization_token"] == "PLACEHOLDER"  # noqa: S105