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

Refactor SearchApi class to simplify code

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-24 20:09:49 -03:00
commit c7394b37a3
1 changed files with 3 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/backend/langflow/components/tools/SearchApi.py
View file

@ -1,13 +1,10 @@
from typing import Optional from typing import Optional
from langchain.schema import Document
from langchain_community.utilities.searchapi import SearchApiAPIWrapper from langchain_community.utilities.searchapi import SearchApiAPIWrapper
from langflow import CustomComponent from langflow import CustomComponent
from langflow.schema.schema import Record from langflow.schema.schema import Record
from langflow.services.database.models.base import orjson_dumps 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): class SearchApi(CustomComponent):
@ -51,12 +48,6 @@ class SearchApi(CustomComponent):
result = orjson_dumps(results, indent_2=False) result = orjson_dumps(results, indent_2=False)
document = Document(page_content=result) record = Record(data=result)
self.status = record
records = self.to_records(document) return record
if records:
record = records[0]
self.status = record
return record
else:
return Record()