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

Add AstraDB icon and field order to AstraDBVectorStoreComponent and AstraDBSearchComponent

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-27 00:43:02 -03:00
commit 569d384573
6 changed files with 43 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/base/langflow/components/vectorstores/AstraDB.py
View file

@ -10,6 +10,8 @@ from langflow.schema import Record
class AstraDBVectorStoreComponent(CustomComponent):
display_name = "AstraDB Vector Store"
description = "Builds or loads an AstraDB Vector Store"
icon = "AstraDB"
field_order = ["token", "api_endpoint", "collection_name", "inputs", "embedding"]
def build_config(self):
return {

2
src/backend/base/langflow/components/vectorstores/AstraDBSearch.py
View file

@ -9,6 +9,8 @@ from langflow.schema import Record
class AstraDBSearchComponent(AstraDBVectorStoreComponent, LCVectorStoreComponent):
display_name = "AstraDB Search"
description = "Searches an existing AstraDB Vector Store"
icon = "AstraDB"
field_order = ["token", "api_endpoint", "collection_name", "input_value", "embedding"]
def build_config(self):
return {