forked from langchain-ai/langchain
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Mask API key for Minimax LLM (langchain-ai#14309)
- **Description:** Added masking for the API key for Minimax LLM + tests inspired by langchain-ai#12418. - **Issue:** the issue # fixes langchain-ai#12165 - **Dependencies:** this fix is dependent on Minimax instantiation fix which is introduced in langchain-ai#13439, so merge this one after. - **Tag maintainer:** @eyurtsev --------- Co-authored-by: Harrison Chase <hw.chase.17@gmail.com>
- Loading branch information
1 parent
22c44c2
commit 2237217
Showing
2 changed files
with
55 additions
and
16 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,42 @@ | ||
"""Test Minimax llm""" | ||
from typing import cast | ||
|
||
from langchain_core.pydantic_v1 import SecretStr | ||
from pytest import CaptureFixture, MonkeyPatch | ||
|
||
from langchain.llms.minimax import Minimax | ||
|
||
|
||
def test_api_key_is_secret_string() -> None: | ||
llm = Minimax(minimax_api_key="secret-api-key", minimax_group_id="group_id") | ||
assert isinstance(llm.minimax_api_key, SecretStr) | ||
|
||
|
||
def test_api_key_masked_when_passed_from_env( | ||
monkeypatch: MonkeyPatch, capsys: CaptureFixture | ||
) -> None: | ||
"""Test initialization with an API key provided via an env variable""" | ||
monkeypatch.setenv("MINIMAX_API_KEY", "secret-api-key") | ||
monkeypatch.setenv("MINIMAX_GROUP_ID", "group_id") | ||
llm = Minimax() | ||
print(llm.minimax_api_key, end="") | ||
captured = capsys.readouterr() | ||
|
||
assert captured.out == "**********" | ||
|
||
|
||
def test_api_key_masked_when_passed_via_constructor( | ||
capsys: CaptureFixture, | ||
) -> None: | ||
"""Test initialization with an API key provided via the initializer""" | ||
llm = Minimax(minimax_api_key="secret-api-key", minimax_group_id="group_id") | ||
print(llm.minimax_api_key, end="") | ||
captured = capsys.readouterr() | ||
|
||
assert captured.out == "**********" | ||
|
||
|
||
def test_uses_actual_secret_value_from_secretstr() -> None: | ||
"""Test that actual secret is retrieved using `.get_secret_value()`.""" | ||
llm = Minimax(minimax_api_key="secret-api-key", minimax_group_id="group_id") | ||
assert cast(SecretStr, llm.minimax_api_key).get_secret_value() == "secret-api-key" |