From 67808bad0e2a4be7c8d1f24d4782142500f37a63 Mon Sep 17 00:00:00 2001
From: Harrison Chase <hw.chase.17@gmail.com>
Date: Fri, 13 Jan 2023 17:36:10 -0800
Subject: [PATCH] expose more serpapi parameters (#609)

---
 langchain/serpapi.py | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/langchain/serpapi.py b/langchain/serpapi.py
index 7b2d0cca..28221852 100644
--- a/langchain/serpapi.py
+++ b/langchain/serpapi.py
@@ -6,7 +6,7 @@ import os
 import sys
 from typing import Any, Dict, Optional
 
-from pydantic import BaseModel, Extra, root_validator
+from pydantic import BaseModel, Extra, Field, root_validator
 
 from langchain.utils import get_from_dict_or_env
 
@@ -25,6 +25,15 @@ class HiddenPrints:
         sys.stdout = self._original_stdout
 
 
+def _get_default_params() -> dict:
+    return {
+        "engine": "google",
+        "google_domain": "google.com",
+        "gl": "us",
+        "hl": "en",
+    }
+
+
 class SerpAPIWrapper(BaseModel):
     """Wrapper around SerpAPI.
 
@@ -40,7 +49,7 @@ class SerpAPIWrapper(BaseModel):
     """
 
     search_engine: Any  #: :meta private:
-
+    params: dict = Field(default_factory=_get_default_params)
     serpapi_api_key: Optional[str] = None
 
     class Config:
@@ -68,14 +77,11 @@ class SerpAPIWrapper(BaseModel):
 
     def run(self, query: str) -> str:
         """Run query through SerpAPI and parse result."""
-        params = {
+        _params = {
             "api_key": self.serpapi_api_key,
-            "engine": "google",
             "q": query,
-            "google_domain": "google.com",
-            "gl": "us",
-            "hl": "en",
         }
+        params = {**self.params, **_params}
         with HiddenPrints():
             search = self.search_engine(params)
             res = search.get_dict()