forked from Archives/langchain
19 lines
641 B
Python
19 lines
641 B
Python
"""Integration test for self ask with search."""
|
|
|
|
from langchain.agents.react.base import ReActChain
|
|
from langchain.docstore.wikipedia import Wikipedia
|
|
from langchain.llms.openai import OpenAI
|
|
|
|
|
|
def test_react() -> None:
|
|
"""Test functionality on a prompt."""
|
|
llm = OpenAI(temperature=0, model_name="text-davinci-002")
|
|
react = ReActChain(llm=llm, docstore=Wikipedia())
|
|
question = (
|
|
"Author David Chanoff has collaborated with a U.S. Navy admiral "
|
|
"who served as the ambassador to the United Kingdom under "
|
|
"which President?"
|
|
)
|
|
output = react.run(question)
|
|
assert output == "Bill Clinton"
|