mirror of
https://github.com/hwchase17/langchain
synced 2024-11-06 03:20:49 +00:00
c9999557bf
<!-- Thank you for contributing to LangChain! Replace this entire comment with: - **Description:** a description of the change, - **Issue:** the issue # it fixes (if applicable), - **Dependencies:** any dependencies required for this change, - **Tag maintainer:** for a quicker response, tag the relevant maintainer (see below), - **Twitter handle:** we announce bigger features on Twitter. If your PR gets announced, and you'd like a mention, we'll gladly shout you out! Please make sure your PR is passing linting and testing before submitting. Run `make format`, `make lint` and `make test` to check this locally. See contribution guidelines for more information on how to write/run tests, lint, etc: https://github.com/langchain-ai/langchain/blob/master/.github/CONTRIBUTING.md If you're adding a new integration, please include: 1. a test for the integration, preferably unit tests that do not rely on network access, 2. an example notebook showing its use. It lives in `docs/extras` directory. If no one reviews your PR within a few days, please @-mention one of @baskaryan, @eyurtsev, @hwchase17. --> - **Description:** 1. Modify LLMs/Anyscale to work with OAI v1 2. Get rid of openai_ prefixed variables in Chat_model/ChatAnyscale 3. Modify `anyscale_api_base` to `anyscale_base_url` to follow OAI name convention (reverted) --------- Co-authored-by: Erick Friis <erick@langchain.dev>
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
"""Test Anyscale llm"""
|
|
import pytest
|
|
from langchain_core.pydantic_v1 import SecretStr
|
|
from pytest import CaptureFixture, MonkeyPatch
|
|
|
|
from langchain_community.llms.anyscale import Anyscale
|
|
|
|
|
|
@pytest.mark.requires("openai")
|
|
def test_api_key_is_secret_string() -> None:
|
|
llm = Anyscale(anyscale_api_key="secret-api-key", anyscale_api_base="test")
|
|
assert isinstance(llm.anyscale_api_key, SecretStr)
|
|
|
|
|
|
@pytest.mark.requires("openai")
|
|
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("ANYSCALE_API_KEY", "secret-api-key")
|
|
llm = Anyscale(anyscale_api_base="test")
|
|
print(llm.anyscale_api_key, end="")
|
|
captured = capsys.readouterr()
|
|
|
|
assert captured.out == "**********"
|
|
|
|
|
|
@pytest.mark.requires("openai")
|
|
def test_api_key_masked_when_passed_via_constructor(
|
|
capsys: CaptureFixture,
|
|
) -> None:
|
|
"""Test initialization with an API key provided via the initializer"""
|
|
llm = Anyscale(anyscale_api_key="secret-api-key", anyscale_api_base="test")
|
|
print(llm.anyscale_api_key, end="")
|
|
captured = capsys.readouterr()
|
|
|
|
assert captured.out == "**********"
|