From c994ce6b7ff7a4e30565c742f313e163f0b85f2c Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Tue, 27 Dec 2022 20:27:18 -0500 Subject: [PATCH] Harrison/serp api imp (#444) improve serp api Co-authored-by: Bruno Bornsztein --- langchain/serpapi.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/langchain/serpapi.py b/langchain/serpapi.py index 6224939d58..cb9e68e27c 100644 --- a/langchain/serpapi.py +++ b/langchain/serpapi.py @@ -92,6 +92,16 @@ class SerpAPIWrapper(BaseModel): toret = res["answer_box"]["snippet_highlighted_words"][0] elif "snippet" in res["organic_results"][0].keys(): 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: toret = "No good search result found" return toret