From b6a7f40ad38c18e2a9aa9be9cccbe37bef950533 Mon Sep 17 00:00:00 2001 From: Chang Sau Sheong Date: Sat, 15 Jul 2023 22:04:18 +0800 Subject: [PATCH] added support for Google Images search (#7751) - Description: Added Google Image Search support for SerpAPIWrapper - Issue: NA - Dependencies: None - Tag maintainer: @hinthornw - Twitter handle: @sausheong --------- Co-authored-by: Bagatur --- langchain/utilities/serpapi.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/langchain/utilities/serpapi.py b/langchain/utilities/serpapi.py index db6ddf79cc..98f4214ce0 100644 --- a/langchain/utilities/serpapi.py +++ b/langchain/utilities/serpapi.py @@ -159,7 +159,12 @@ class SerpAPIWrapper(BaseModel): toret = res["organic_results"][0]["snippet"] elif "link" in res["organic_results"][0].keys(): toret = res["organic_results"][0]["link"] - + elif ( + "images_results" in res.keys() + and "thumbnail" in res["images_results"][0].keys() + ): + thumbnails = [item["thumbnail"] for item in res["images_results"][:10]] + toret = thumbnails else: toret = "No good search result found" return toret