Refactor SearchApi class to simplify code
This commit is contained in:
parent
df7b4fc590
commit
c7394b37a3
1 changed files with 3 additions and 12 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue