github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-22 14:49:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

langchain_community: add image support to DuckDuckGoSearchAPIWrapper (#29816)

Browse Source 
- [ ] **PR title**: langchain_community: add image support to
DuckDuckGoSearchAPIWrapper

- **Description:** This PR enhances the DuckDuckGoSearchAPIWrapper
within the langchain_community package by introducing support for image
searches. The enhancement includes:
  - Adding a new method _ddgs_images to handle image search queries.
- Updating the run and results methods to process and return image
search results appropriately.
- Modifying the source parameter to accept "images" as a valid option,
alongside "text" and "news".
- **Dependencies:** No additional dependencies are required for this
change.
This commit is contained in:
Krishna Kulkarni 2025-02-16 08:02:14 +05:30 committed by GitHub
parent 0d9f0b4215
commit a98c5f1c4b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 33 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
libs/community/langchain_community/utilities/duckduckgo_search.py
View File

@ -34,7 +34,7 @@ class DuckDuckGoSearchAPIWrapper(BaseModel):
"""
source: str = "text"
"""
Options: text, news
Options: text, news, images
"""
model_config = ConfigDict(
@ -91,12 +91,31 @@ class DuckDuckGoSearchAPIWrapper(BaseModel):
return [r for r in ddgs_gen]
return []
def _ddgs_images(
self, query: str, max_results: Optional[int] = None
) -> List[Dict[str, str]]:
"""Run query through DuckDuckGo image search and return results."""
from duckduckgo_search import DDGS
with DDGS() as ddgs:
ddgs_gen = ddgs.images(
query,
region=self.region, # type: ignore[arg-type]
safesearch=self.safesearch,
max_results=max_results or self.max_results,
)
if ddgs_gen:
return [r for r in ddgs_gen]
return []
def run(self, query: str) -> str:
"""Run query through DuckDuckGo and return concatenated results."""
if self.source == "text":
results = self._ddgs_text(query)
elif self.source == "news":
results = self._ddgs_news(query)
elif self.source == "images":
results = self._ddgs_images(query)
else:
results = []
@ -137,6 +156,19 @@ class DuckDuckGoSearchAPIWrapper(BaseModel):
}
for r in self._ddgs_news(query, max_results=max_results)
]
elif source == "images":
results = [
{
"title": r["title"],
"thumbnail": r["thumbnail"],
"image": r["image"],
"url": r["url"],
"height": r["height"],
"width": r["width"],
"source": r["source"],
}
for r in self._ddgs_images(query, max_results=max_results)
]
else:
results = []