github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-19 11:08:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: add title, score and raw_content to tavily search results (#29995)

Browse Source 
**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 <chester.curme@gmail.com>
This commit is contained in:
Artem Yankov 2025-02-25 15:27:21 -08:00 committed by GitHub
parent b525226531
commit 6177b9f9ab
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
libs/community/langchain_community/utilities/tavily_search.py
View File

@ -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