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 2

Entity Profile

LangChainCore→ LanguageModelBase→ test_get_ls_params() — langchain Function Reference

Dependency Diagram

graph TD
  eda3cbfe_24b5_3e86_f3b7_da926c668103["test_get_ls_params()"]
  0cad5588_a6f2_d365_b61b_841ca3437132["test_base.py"]
  eda3cbfe_24b5_3e86_f3b7_da926c668103 -->|defined in| 0cad5588_a6f2_d365_b61b_841ca3437132
  3c96260d_040e_902b_2b08_6bf3e26432d4["_generate()"]
  eda3cbfe_24b5_3e86_f3b7_da926c668103 -->|calls| 3c96260d_040e_902b_2b08_6bf3e26432d4
  b19deba4_e834_4fe0_3b1e_d1687a455bd1["_llm_type()"]
  eda3cbfe_24b5_3e86_f3b7_da926c668103 -->|calls| b19deba4_e834_4fe0_3b1e_d1687a455bd1
  style eda3cbfe_24b5_3e86_f3b7_da926c668103 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/language_models/llms/test_base.py lines 237–279

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

        @override
        def _generate(
            self,
            prompts: list[str],
            stop: list[str] | None = None,
            run_manager: CallbackManagerForLLMRun | None = None,
            **kwargs: Any,
        ) -> LLMResult:
            raise NotImplementedError

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

    llm = LSParamsModel()

    # Test standard tracing params
    ls_params = llm._get_ls_params()
    assert ls_params == {
        "ls_provider": "lsparamsmodel",
        "ls_model_type": "llm",
        "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"]