github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-09 06:53:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: upgrade to recent version of mypy (#21616)

Browse Source 
This PR upgrades community to a recent version of mypy. It inserts type:
ignore on all existing failures.
This commit is contained in:
Eugene Yurtsev
2024-05-13 14:55:07 -04:00
committed by GitHub
parent b923951062
commit 25fbe356b4
243 changed files with 718 additions and 710 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libs/community/tests/unit_tests/llms/test_ai21.py
View File

@@ -9,7 +9,7 @@ from langchain_community.llms.ai21 import AI21
def test_api_key_is_secret_string() -> None:
llm = AI21(ai21_api_key="secret-api-key")
llm = AI21(ai21_api_key="secret-api-key") # type: ignore[arg-type]
assert isinstance(llm.ai21_api_key, SecretStr)
@@ -29,7 +29,7 @@ def test_api_key_masked_when_passed_via_constructor(
capsys: CaptureFixture,
) -> None:
"""Test initialization with an API key provided via the initializer"""
llm = AI21(ai21_api_key="secret-api-key")
llm = AI21(ai21_api_key="secret-api-key") # type: ignore[arg-type]
print(llm.ai21_api_key, end="") # noqa: T201
captured = capsys.readouterr()
@@ -38,5 +38,5 @@ def test_api_key_masked_when_passed_via_constructor(
def test_uses_actual_secret_value_from_secretstr() -> None:
"""Test that actual secret is retrieved using `.get_secret_value()`."""
llm = AI21(ai21_api_key="secret-api-key")
llm = AI21(ai21_api_key="secret-api-key") # type: ignore[arg-type]
assert cast(SecretStr, llm.ai21_api_key).get_secret_value() == "secret-api-key"