mirror of
https://github.com/hwchase17/langchain
synced 2024-11-06 03:20:49 +00:00
b3e74f2b98
**Description:** This pull request introduces a new feature to community tools, enhancing its search capabilities by integrating the Mojeek search engine **Dependencies:** None --------- Co-authored-by: Igor Brai <igor@mojeek.com> Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com> Co-authored-by: ccurme <chester.curme@gmail.com>
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
import json
|
|
from typing import List
|
|
|
|
import requests
|
|
from langchain_core.pydantic_v1 import BaseModel, Field
|
|
|
|
|
|
class MojeekSearchAPIWrapper(BaseModel):
|
|
api_key: str
|
|
search_kwargs: dict = Field(default_factory=dict)
|
|
api_url = "https://api.mojeek.com/search"
|
|
|
|
def run(self, query: str) -> str:
|
|
search_results = self._search(query)
|
|
|
|
results = []
|
|
|
|
for result in search_results:
|
|
title = result.get("title", "")
|
|
url = result.get("url", "")
|
|
desc = result.get("desc", "")
|
|
results.append({"title": title, "url": url, "desc": desc})
|
|
|
|
return json.dumps(results)
|
|
|
|
def _search(self, query: str) -> List[dict]:
|
|
headers = {
|
|
"Accept": "application/json",
|
|
}
|
|
|
|
req = requests.PreparedRequest()
|
|
request = {
|
|
**self.search_kwargs,
|
|
**{"q": query, "fmt": "json", "api_key": self.api_key},
|
|
}
|
|
req.prepare_url(self.api_url, request)
|
|
if req.url is None:
|
|
raise ValueError("prepared url is None, this should not happen")
|
|
|
|
response = requests.get(req.url, headers=headers)
|
|
if not response.ok:
|
|
raise Exception(f"HTTP error {response.status_code}")
|
|
|
|
return response.json().get("response", {}).get("results", [])
|