From d7eedc75d1d74e7d0e3f3007ffd256e9d2b4eaf3 Mon Sep 17 00:00:00 2001
From: blob42 <spike@w530>
Date: Sun, 12 Feb 2023 21:21:28 +0100
Subject: [PATCH] [searx-search] helper parameter for selecting engines

---
 langchain/utilities/searx_search.py | 20 +++++++++++++++++---
 1 file changed, 17 insertions(+), 3 deletions(-)

diff --git a/langchain/utilities/searx_search.py b/langchain/utilities/searx_search.py
index 1263f840..e7cf88e7 100644
--- a/langchain/utilities/searx_search.py
+++ b/langchain/utilities/searx_search.py
@@ -90,6 +90,7 @@ class SearxSearchWrapper(BaseModel):
     unsecure: bool = False
     params: dict = Field(default_factory=_get_default_params)
     headers: Optional[dict] = None
+    engines: Optional[List[str]] = []
     k: int = 10
 
     @validator("unsecure")
@@ -113,6 +114,10 @@ class SearxSearchWrapper(BaseModel):
         default = _get_default_params()
         values["params"] = {**default, **user_params}
 
+        engines = values.get("engines")
+        if engines:
+            values["params"]["engines"] = ','.join(engines)
+
         searx_host = get_from_dict_or_env(values, "searx_host", "SEARX_HOST")
         if not searx_host.startswith("http"):
             print(
@@ -144,7 +149,7 @@ class SearxSearchWrapper(BaseModel):
         self._result = res
         return res
 
-    def run(self, query: str, **kwargs: Any) -> str:
+    def run(self, query: str, engines: List[str] = [], **kwargs: Any) -> str:
         """Run query through Searx API and parse results.
 
         You can pass any other params to the searx query API.
@@ -162,12 +167,15 @@ class SearxSearchWrapper(BaseModel):
                 searx = SearxSearchWrapper(searx_host="http://my.searx.host")
                 searx.run("what is the weather in France ?", engine="qwant")
 
-
         """
         _params = {
             "q": query,
         }
         params = {**self.params, **_params, **kwargs}
+
+        if isinstance(engines, list) and len(engines) > 0:
+            params['engines'] = ','.join(engines)
+
         res = self._searx_api_query(params)
 
         if len(res.answers) > 0:
@@ -183,7 +191,7 @@ class SearxSearchWrapper(BaseModel):
 
         return toret
 
-    def results(self, query: str, num_results: int, **kwargs: Any) -> List[Dict]:
+    def results(self, query: str, num_results: int, engines: List[str] = [],  **kwargs: Any) -> List[Dict]:
         """Run query through Searx API and returns the results with metadata.
 
         Args:
@@ -195,12 +203,18 @@ class SearxSearchWrapper(BaseModel):
                 snippet - The description of the result.
                 title - The title of the result.
                 link - The link to the result.
+                engines - The engines used for the result.
+                category - Searx category of the result.
+
+
         """
         metadata_results = []
         _params = {
             "q": query,
         }
         params = {**self.params, **_params, **kwargs}
+        if isinstance(engines, list) and len(engines) > 0:
+            params['engines'] = ','.join(engines)
         results = self._searx_api_query(params).results[:num_results]
         if len(results) == 0:
             return [{"Result": "No good Search Result was found"}]