forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
17 lines
515 B
Python
17 lines
515 B
Python
"""Integration test for Dall-E image generator agent."""
|
|
from langchain.agents import initialize_agent, load_tools
|
|
from langchain.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="zero-shot-react-description", verbose=True
|
|
)
|
|
|
|
output = agent.run("Create an image of a volcano island")
|
|
assert output is not None
|