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

Update AstraDB and AstraDBSearch components

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-26 00:33:26 -03:00
commit 9c0bad3883
2 changed files with 51 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -16,7 +16,7 @@ class AstraDBVectorStoreComponent(CustomComponent):
"inputs": {"display_name": "Inputs", "info": "Inputs to AstraDB"}, "inputs": {"display_name": "Inputs", "info": "Inputs to AstraDB"},
"embedding": {"display_name": "Embedding", "info": "Embedding to use"}, "embedding": {"display_name": "Embedding", "info": "Embedding to use"},
"collection_name": {"display_name": "Collection Name", "info": "Collection name"}, "collection_name": {"display_name": "Collection Name", "info": "Collection name"},
"token": {"display_name": "Token", "info": "Token to use"}, "token": {"display_name": "Token", "info": "Token to use", "password": True},
"api_endpoint": {"display_name": "API Endpoint", "info": "API Endpoint to use"}, "api_endpoint": {"display_name": "API Endpoint", "info": "API Endpoint to use"},
"namespace": {"display_name": "Namespace", "info": "Namespace to use", "advanced": True}, "namespace": {"display_name": "Namespace", "info": "Namespace to use", "advanced": True},
"metric": {"display_name": "Metric", "info": "Metric to use", "advanced": True}, "metric": {"display_name": "Metric", "info": "Metric to use", "advanced": True},

52
src/backend/langflow/components/vectorstores/AstraDBSearch.py
View file

@ -2,7 +2,7 @@ from typing import List, Optional
from langflow.components.vectorstores.AstraDB import AstraDBVectorStoreComponent from langflow.components.vectorstores.AstraDB import AstraDBVectorStoreComponent
from langflow.components.vectorstores.base.model import LCVectorStoreComponent from langflow.components.vectorstores.base.model import LCVectorStoreComponent
from langflow.field_typing import Embeddings from langflow.field_typing import Embeddings, Text
from langflow.schema import Record from langflow.schema import Record
@ -20,13 +20,61 @@ class AstraDBSearchComponent(AstraDBVectorStoreComponent, LCVectorStoreComponent
"display_name": "Input Value", "display_name": "Input Value",
"info": "Input value to search", "info": "Input value to search",
}, },
"embedding": {"display_name": "Embedding", "info": "Embedding to use"},
"collection_name": {"display_name": "Collection Name", "info": "Collection name"},
"token": {"display_name": "Token", "info": "Token to use", "password": True},
"api_endpoint": {"display_name": "API Endpoint", "info": "API Endpoint to use"},
"namespace": {"display_name": "Namespace", "info": "Namespace to use", "advanced": True},
"metric": {"display_name": "Metric", "info": "Metric to use", "advanced": True},
"batch_size": {"display_name": "Batch Size", "info": "Batch size to use", "advanced": True},
"bulk_insert_batch_concurrency": {
"display_name": "Bulk Insert Batch Concurrency",
"info": "Bulk Insert Batch Concurrency to use",
"advanced": True,
},
"bulk_insert_overwrite_concurrency": {
"display_name": "Bulk Insert Overwrite Concurrency",
"info": "Bulk Insert Overwrite Concurrency to use",
"advanced": True,
},
"bulk_delete_concurrency": {
"display_name": "Bulk Delete Concurrency",
"info": "Bulk Delete Concurrency to use",
"advanced": True,
},
"setup_mode": {
"display_name": "Setup Mode",
"info": "Setup mode for the vector store",
"options": ["Sync", "Async", "Off"],
"advanced": True,
},
"pre_delete_collection": {
"display_name": "Pre Delete Collection",
"info": "Pre delete collection",
"advanced": True,
},
"metadata_indexing_include": {
"display_name": "Metadata Indexing Include",
"info": "Metadata Indexing Include",
"advanced": True,
},
"metadata_indexing_exclude": {
"display_name": "Metadata Indexing Exclude",
"info": "Metadata Indexing Exclude",
"advanced": True,
},
"collection_indexing_policy": {
"display_name": "Collection Indexing Policy",
"info": "Collection Indexing Policy",
"advanced": True,
},
} }
def build( def build(
self, self,
embedding: Embeddings, embedding: Embeddings,
collection_name: str, collection_name: str,
input_value: Optional[List[Record]] = None, input_value: Optional[Text] = None,
search_type: str = "Similarity", search_type: str = "Similarity",
token: Optional[str] = None, token: Optional[str] = None,
api_endpoint: Optional[str] = None, api_endpoint: Optional[str] = None,