mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-12 12:59:07 +00:00
@@ -1,13 +1,18 @@
|
||||
"""Integration test for self ask with search."""
|
||||
|
||||
from langchain.chains.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)
|
||||
react = ReActChain(llm=llm)
|
||||
question = "Were Scott Derrickson and Ed Wood of the same nationality?"
|
||||
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 == "yes"
|
||||
assert output == "Bill Clinton"
|
||||
|
Reference in New Issue
Block a user