github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-20 05:43:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Harrison/serp api imp (#444)

Browse Source 
improve serp api

Co-authored-by: Bruno Bornsztein <bruno.bornsztein@gmail.com>
This commit is contained in:
Harrison Chase 2022-12-27 20:27:18 -05:00 committed by GitHub
parent ffe35c396c
commit c994ce6b7f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
langchain/serpapi.py
View File

@ -92,6 +92,16 @@ class SerpAPIWrapper(BaseModel):
toret = res["answer_box"]["snippet_highlighted_words"][0] toret = res["answer_box"]["snippet_highlighted_words"][0]
elif "snippet" in res["organic_results"][0].keys(): elif "snippet" in res["organic_results"][0].keys():
toret = res["organic_results"][0]["snippet"] toret = res["organic_results"][0]["snippet"]
elif (
"sports_results" in res.keys()
and "game_spotlight" in res["sports_results"].keys()
):
toret = res["sports_results"]["game_spotlight"]
elif (
"knowledge_graph" in res.keys()
and "description" in res["knowledge_graph"].keys()
):
toret = res["knowledge_graph"]["description"]
else: else:
toret = "No good search result found" toret = "No good search result found"
return toret return toret