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

Refactor SerpAPIWrapperComponent class in SerpAPIWrapper.py

Browse source
This commit is contained in:
anovazzi1 2024-01-09 18:42:45 -03:00
commit 5cc8afc557
1 changed files with 3 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,9 +1,6 @@
from langflow import CustomComponent from langflow import CustomComponent
from typing import Callable, Union from typing import Callable, Union
from langchain_community.utilities import SerpAPIWrapper
# Assuming SerpAPIWrapper is a predefined class within the langflow context.
# If it's not, it must be defined or imported from the appropriate module.
class SerpAPIWrapperComponent(CustomComponent): class SerpAPIWrapperComponent(CustomComponent):
display_name = "SerpAPIWrapper" display_name = "SerpAPIWrapper"
@ -17,7 +14,7 @@ class SerpAPIWrapperComponent(CustomComponent):
def build( def build(
self, self,
serpapi_api_key: str, serpapi_api_key: str,
) -> Union['SerpAPIWrapper', Callable]: ) -> Union[SerpAPIWrapper, Callable]: # Removed quotes around SerpAPIWrapper
# Default parameters as defined in the JSON template. # Default parameters as defined in the JSON template.
default_params = { default_params = {
"engine": "google", "engine": "google",
@ -29,4 +26,4 @@ class SerpAPIWrapperComponent(CustomComponent):
return SerpAPIWrapper( return SerpAPIWrapper(
serpapi_api_key=serpapi_api_key, serpapi_api_key=serpapi_api_key,
params=default_params params=default_params
) )