langchain/libs/community/tests/integration_tests/chains/test_dalle_agent.py

"""Integration test for Dall-E image generator agent."""
from langchain.agents import AgentType, initialize_agent

from langchain_community.agent_toolkits.load_tools import load_tools
from langchain_community.llms import OpenAI


def test_call() -> None:
    """Test that the agent runs and returns output."""
    llm = OpenAI(temperature=0.9)
    tools = load_tools(["dalle-image-generator"])

    agent = initialize_agent(
        tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True
    )

    output = agent.run("Create an image of a volcano island")
    assert output is not None