Refactor SearchApi class to simplify code

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-24 20:09:49 -03:00
commit c7394b37a3

View file

@ -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