test_get_ls_params() — langchain Function Reference

Architecture documentation for the test_get_ls_params() function in test_base.py from the langchain codebase.

Function python LangChainCore LanguageModelBase calls 1

Entity Profile

LangChainCore→ LanguageModelBase→ test_get_ls_params() — langchain Function Reference

Dependency Diagram

graph TD
  c2aa9f76_ba04_7686_5375_7926aadc7064["test_get_ls_params()"]
  8cb88ac4_61d9_baf3_9df4_9b3f5095927e["test_base.py"]
  c2aa9f76_ba04_7686_5375_7926aadc7064 -->|defined in| 8cb88ac4_61d9_baf3_9df4_9b3f5095927e
  0cde1f99_9131_e33d_43b3_4c705000e434["_stream()"]
  c2aa9f76_ba04_7686_5375_7926aadc7064 -->|calls| 0cde1f99_9131_e33d_43b3_4c705000e434
  style c2aa9f76_ba04_7686_5375_7926aadc7064 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/language_models/chat_models/test_base.py lines 1162–1213

def test_get_ls_params() -> None:
    class LSParamsModel(BaseChatModel):
        model: str = "foo"
        temperature: float = 0.1
        max_tokens: int = 1024

        def _generate(
            self,
            messages: list[BaseMessage],
            stop: list[str] | None = None,
            run_manager: CallbackManagerForLLMRun | None = None,
            **kwargs: Any,
        ) -> ChatResult:
            raise NotImplementedError

        @override
        def _stream(
            self,
            messages: list[BaseMessage],
            stop: list[str] | None = None,
            run_manager: CallbackManagerForLLMRun | None = None,
            **kwargs: Any,
        ) -> Iterator[ChatGenerationChunk]:
            raise NotImplementedError

        @property
        def _llm_type(self) -> str:
            return "fake-chat-model"

    llm = LSParamsModel()

    # Test standard tracing params
    ls_params = llm._get_ls_params()
    assert ls_params == {
        "ls_provider": "lsparamsmodel",
        "ls_model_type": "chat",
        "ls_model_name": "foo",
        "ls_temperature": 0.1,
        "ls_max_tokens": 1024,
    }

    ls_params = llm._get_ls_params(model="bar")
    assert ls_params["ls_model_name"] == "bar"

    ls_params = llm._get_ls_params(temperature=0.2)
    assert ls_params["ls_temperature"] == 0.2

    ls_params = llm._get_ls_params(max_tokens=2048)
    assert ls_params["ls_max_tokens"] == 2048

    ls_params = llm._get_ls_params(stop=["stop"])
    assert ls_params["ls_stop"] == ["stop"]