mirror of
https://github.com/hwchase17/langchain.git
synced 2025-04-29 12:25:37 +00:00
c2a3021bb0
Signed-off-by: ChengZi <chen.zhang@zilliz.com> Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com> Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com> Co-authored-by: Dan O'Donovan <dan.odonovan@gmail.com> Co-authored-by: Tom Daniel Grande <tomdgrande@gmail.com> Co-authored-by: Grande <Tom.Daniel.Grande@statsbygg.no> Co-authored-by: Bagatur <baskaryan@gmail.com> Co-authored-by: ccurme <chester.curme@gmail.com> Co-authored-by: Harrison Chase <hw.chase.17@gmail.com> Co-authored-by: Tomaz Bratanic <bratanic.tomaz@gmail.com> Co-authored-by: ZhangShenao <15201440436@163.com> Co-authored-by: Friso H. Kingma <fhkingma@gmail.com> Co-authored-by: ChengZi <chen.zhang@zilliz.com> Co-authored-by: Nuno Campos <nuno@langchain.dev> Co-authored-by: Morgante Pell <morgantep@google.com>
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
import pytest
|
|
from pydantic import SecretStr, ValidationError
|
|
|
|
from langchain_community.chat_models.octoai import ChatOctoAI
|
|
|
|
DEFAULT_API_BASE = "https://text.octoai.run/v1/"
|
|
DEFAULT_MODEL = "llama-2-13b-chat"
|
|
|
|
|
|
@pytest.mark.requires("openai")
|
|
def test__default_octoai_api_base() -> None:
|
|
chat = ChatOctoAI(octoai_api_token=SecretStr("test_token")) # type: ignore[call-arg]
|
|
assert chat.octoai_api_base == DEFAULT_API_BASE
|
|
|
|
|
|
@pytest.mark.requires("openai")
|
|
def test__default_octoai_api_token() -> None:
|
|
chat = ChatOctoAI(octoai_api_token=SecretStr("test_token")) # type: ignore[call-arg]
|
|
assert chat.octoai_api_token.get_secret_value() == "test_token"
|
|
|
|
|
|
@pytest.mark.requires("openai")
|
|
def test__default_model_name() -> None:
|
|
chat = ChatOctoAI(octoai_api_token=SecretStr("test_token")) # type: ignore[call-arg]
|
|
assert chat.model_name == DEFAULT_MODEL
|
|
|
|
|
|
@pytest.mark.requires("openai")
|
|
def test__field_aliases() -> None:
|
|
chat = ChatOctoAI(octoai_api_token=SecretStr("test_token"), model="custom-model") # type: ignore[call-arg]
|
|
assert chat.model_name == "custom-model"
|
|
assert chat.octoai_api_token.get_secret_value() == "test_token"
|
|
|
|
|
|
@pytest.mark.requires("openai")
|
|
def test__missing_octoai_api_token() -> None:
|
|
with pytest.raises(ValidationError) as e:
|
|
ChatOctoAI()
|
|
assert "Did not find octoai_api_token" in str(e)
|
|
|
|
|
|
@pytest.mark.requires("openai")
|
|
def test__all_fields_provided() -> None:
|
|
chat = ChatOctoAI( # type: ignore[call-arg]
|
|
octoai_api_token=SecretStr("test_token"),
|
|
model="custom-model",
|
|
octoai_api_base="https://custom.api/base/",
|
|
)
|
|
assert chat.octoai_api_base == "https://custom.api/base/"
|
|
assert chat.octoai_api_token.get_secret_value() == "test_token"
|
|
assert chat.model_name == "custom-model"
|