diff --git a/src/backend/langflow/components/tools/SearchApi.py b/src/backend/langflow/components/tools/SearchApi.py index 8a0bfe0ac..a80cd22d5 100644 --- a/src/backend/langflow/components/tools/SearchApi.py +++ b/src/backend/langflow/components/tools/SearchApi.py @@ -1,13 +1,10 @@ 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): @@ -51,12 +48,6 @@ class SearchApi(CustomComponent): result = orjson_dumps(results, indent_2=False) - document = Document(page_content=result) - - records = self.to_records(document) - if records: - record = records[0] - self.status = record - return record - else: - return Record() + record = Record(data=result) + self.status = record + return record