From 6177b9f9ab81e8750292ef5951115e5135909c9e Mon Sep 17 00:00:00 2001 From: Artem Yankov <68096337+strangehelix@users.noreply.github.com> Date: Tue, 25 Feb 2025 15:27:21 -0800 Subject: [PATCH] community: add title, score and raw_content to tavily search results (#29995) **Description:** Tavily search results returned from API include useful information like title, score and (optionally) raw_content that is missed in wrapper although it's documented there properly. Add this data to the result structure. --------- Co-authored-by: Chester Curme --- .../utilities/tavily_search.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/libs/community/langchain_community/utilities/tavily_search.py b/libs/community/langchain_community/utilities/tavily_search.py index 84e9815ee9e..8dfb2b989d3 100644 --- a/libs/community/langchain_community/utilities/tavily_search.py +++ b/libs/community/langchain_community/utilities/tavily_search.py @@ -176,10 +176,13 @@ class TavilySearchAPIWrapper(BaseModel): """Clean results from Tavily Search API.""" clean_results = [] for result in results: - clean_results.append( - { - "url": result["url"], - "content": result["content"], - } - ) + clean_result = { + "title": result["title"], + "url": result["url"], + "content": result["content"], + "score": result["score"], + } + if raw_content := result.get("raw_content"): + clean_result["raw_content"] = raw_content + clean_results.append(clean_result) return clean_results