diff --git a/src/backend/langflow/components/tools/SearchApi.py b/src/backend/langflow/components/tools/SearchApi.py index 53da6c741..8a0bfe0ac 100644 --- a/src/backend/langflow/components/tools/SearchApi.py +++ b/src/backend/langflow/components/tools/SearchApi.py @@ -1,9 +1,14 @@ -from langflow import CustomComponent -from langchain.schema import Document -from langflow.services.database.models.base import orjson_dumps -from langchain_community.utilities.searchapi import SearchApiAPIWrapper from typing import Optional +from langchain.schema import Document +from langchain_community.utilities.searchapi import SearchApiAPIWrapper + +from langflow import CustomComponent +from langflow.schema.schema import Record +from langflow.services.database.models.base import orjson_dumps +from langflow.schema.schema import Record +from langflow.services.database.models.base import orjson_dumps + class SearchApi(CustomComponent): display_name: str = "SearchApi" @@ -35,7 +40,7 @@ class SearchApi(CustomComponent): engine: str, api_key: str, params: Optional[dict] = None, - ) -> Document: + ) -> Record: if params is None: params = {} @@ -48,4 +53,10 @@ class SearchApi(CustomComponent): document = Document(page_content=result) - return document + records = self.to_records(document) + if records: + record = records[0] + self.status = record + return record + else: + return Record()