test_load_llmchain_env() — langchain Function Reference

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

Function python LangChainCore ApiManagement

Entity Profile

LangChainCore→ ApiManagement→ test_load_llmchain_env() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/langchain/tests/unit_tests/load/test_load.py lines 135–157

def test_load_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 = CommunityOpenAI(model="davinci", temperature=0.5)
    prompt = PromptTemplate.from_template("hello {name}!")
    chain = LLMChain(llm=llm, prompt=prompt)
    chain_obj = dumpd(chain)
    chain2 = load(chain_obj)

    assert chain2 == chain
    assert dumpd(chain2) == chain_obj
    assert isinstance(chain2, LLMChain)
    assert isinstance(chain2.llm, OpenAI)
    assert isinstance(chain2.prompt, PromptTemplate)

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