added octoai test (#21793)

- [ ] **PR title**: community: add tests for ChatOctoAI

- [ ] **PR message**: 
Description: Added unit tests for the ChatOctoAI class in the community
package to ensure proper validation and default values. These tests
verify the correct initialization of fields, the handling of missing
required parameters, and the proper setting of aliases.
Issue: N/A
Dependencies: None

---------

Co-authored-by: ccurme <chester.curme@gmail.com>
Co-authored-by: Eugene Yurtsev <eugene@langchain.dev>

libs/community/langchain_community/chat_models/octoai.py

@@ -32,8 +32,8 @@ class ChatOctoAI(ChatOpenAI):
     """
 
     octoai_api_base: str = Field(default=DEFAULT_API_BASE)
-    octoai_api_token: SecretStr = Field(default=None)
-    model_name: str = Field(default=DEFAULT_MODEL)
+    octoai_api_token: SecretStr = Field(default=None, alias="api_key")
+    model_name: str = Field(default=DEFAULT_MODEL, alias="model")
 
     @property
     def _llm_type(self) -> str:

libs/community/tests/unit_tests/chat_models/test_octoai.py

@@ -0,0 +1,51 @@
+import pytest
+from langchain_core.pydantic_v1 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"