Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-06 03:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update serpapi.py Support baidu list type answer_box (#6386)

Browse Source 
Support baidu list type answer_box

From [this document](https://serpapi.com/baidu-answer-box), we can know
that the answer_box attribute returned by the Baidu interface is a list,
and the list contains only one Object, but an error will occur when the
current code is executed.

So when answer_box is a list, we reset res["answer_box"] so that the
code can execute successfully.
This commit is contained in:
Gavin 2023-06-20 13:48:18 +08:00 committed by GitHub
parent 384fa43fc3
commit b0d80c4b3e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
langchain/utilities/serpapi.py
View File

@ -129,6 +129,8 @@ class SerpAPIWrapper(BaseModel):
"""Process response from SerpAPI.""" """Process response from SerpAPI."""
if "error" in res.keys(): if "error" in res.keys():
raise ValueError(f"Got error from SerpAPI: {res['error']}") raise ValueError(f"Got error from SerpAPI: {res['error']}")
if "answer_box" in res.keys() and type(res["answer_box"]) == list:
res["answer_box"] = res["answer_box"][0]
if "answer_box" in res.keys() and "answer" in res["answer_box"].keys(): if "answer_box" in res.keys() and "answer" in res["answer_box"].keys():
toret = res["answer_box"]["answer"] toret = res["answer_box"]["answer"]
elif "answer_box" in res.keys() and "snippet" in res["answer_box"].keys(): elif "answer_box" in res.keys() and "snippet" in res["answer_box"].keys():