jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Update SerpAPIWrapper import and parameters

Browse source
This commit is contained in:
anovazzi1 2024-01-16 00:16:16 -03:00
commit c215491549
1 changed files with 5 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/backend/langflow/components/utilities/SerpAPIWrapper.py
View file

@ -1,6 +1,6 @@
from langflow import CustomComponent
from typing import Callable, Union
from langchain_community.utilities import SerpAPIWrapper
from langchain_community.utilities.serpapi import SerpAPIWrapper
class SerpAPIWrapperComponent(CustomComponent):
display_name = "SerpAPIWrapper"
@ -8,22 +8,16 @@ class SerpAPIWrapperComponent(CustomComponent):
def build_config(self):
return {
"serpapi_api_key": {"display_name": "SerpAPI API Key", "type": "password"},
"serpapi_api_key": {"display_name": "SerpAPI API Key", "type": "str", "password": True},
"params": {"display_name": "Parameters", "type": "dict","advanced":True, "multiline": True,"value": '{"engine": "google","google_domain": "google.com","gl": "us","hl": "en"}'},
}
def build(
self,
serpapi_api_key: str,
params: dict,
) -> Union[SerpAPIWrapper, Callable]: # Removed quotes around SerpAPIWrapper
# Default parameters as defined in the JSON template.
default_params = {
"engine": "google",
"google_domain": "google.com",
"gl": "us",
"hl": "en"
}
return SerpAPIWrapper(
serpapi_api_key=serpapi_api_key,
params=default_params
params=params
)