From c7394b37a3c73525f1f30d3a6a1959c5257a0168 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 24 Mar 2024 20:09:49 -0300 Subject: [PATCH] Refactor SearchApi class to simplify code --- .../langflow/components/tools/SearchApi.py | 15 +++------------ 1 file changed, 3 insertions(+), 12 deletions(-) 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