test_loads_llmchain_env() — langchain Function Reference

Architecture documentation for the test_loads_llmchain_env() function in test_load.py from the langchain codebase.

Function python LangChainCore MessageInterface

Entity Profile

LangChainCore→ MessageInterface→ test_loads_llmchain_env() — langchain Function Reference

Dependency Diagram

graph TD
  efe13b8e_8146_6c6c_b2f0_85e5c02d394b["test_loads_llmchain_env()"]
  0b08876e_5a62_23f2_8ed5_ec59f46366a7["test_load.py"]
  efe13b8e_8146_6c6c_b2f0_85e5c02d394b -->|defined in| 0b08876e_5a62_23f2_8ed5_ec59f46366a7
  style efe13b8e_8146_6c6c_b2f0_85e5c02d394b fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain/tests/unit_tests/load/test_load.py lines 64–86

def test_loads_llmchain_env() -> None:
    import os

    from langchain_openai import OpenAI

    has_env = "OPENAI_API_KEY" in os.environ
    if not has_env:
        os.environ["OPENAI_API_KEY"] = "env_variable"

    llm = OpenAI(model="davinci", temperature=0.5, top_p=0.8)
    prompt = PromptTemplate.from_template("hello {name}!")
    chain = LLMChain(llm=llm, prompt=prompt)
    chain_string = dumps(chain)
    chain2 = loads(chain_string)

    assert chain2 == chain
    assert dumps(chain2) == chain_string
    assert isinstance(chain2, LLMChain)
    assert isinstance(chain2.llm, OpenAI)
    assert isinstance(chain2.prompt, PromptTemplate)

    if not has_env:
        del os.environ["OPENAI_API_KEY"]