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

fix: change local db component input as required and tool method name (#8560)

Browse source 
* change input as required and tool method name

* [autofix.ci] apply automated fixes

* Delete uv.lock

* add uvlock back

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: Edwin Jose <edwin.jose@datastax.com>
Co-authored-by: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
This commit is contained in:
Yuqi Tang 2025-06-20 11:51:47 -07:00 • committed by GitHub
commit 664cd9d7ed
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/backend/base/langflow/components/vectorstores/local_db.py
View file

@ -1,6 +1,5 @@
from copy import deepcopy from copy import deepcopy
from pathlib import Path from pathlib import Path
from typing import TYPE_CHECKING
from langchain_chroma import Chroma from langchain_chroma import Chroma
from loguru import logger from loguru import logger
@ -11,10 +10,8 @@ from langflow.base.vectorstores.utils import chroma_collection_to_data
from langflow.inputs.inputs import MultilineInput from langflow.inputs.inputs import MultilineInput
from langflow.io import BoolInput, DropdownInput, HandleInput, IntInput, MessageTextInput, TabInput from langflow.io import BoolInput, DropdownInput, HandleInput, IntInput, MessageTextInput, TabInput
from langflow.schema.data import Data from langflow.schema.data import Data
from langflow.template.field.base import Output
if TYPE_CHECKING:
from langflow.schema.dataframe import DataFrame from langflow.schema.dataframe import DataFrame
from langflow.template.field.base import Output
class LocalDBComponent(LCVectorStoreComponent): class LocalDBComponent(LCVectorStoreComponent):
@ -39,6 +36,7 @@ class LocalDBComponent(LCVectorStoreComponent):
name="collection_name", name="collection_name",
display_name="Collection Name", display_name="Collection Name",
value="langflow", value="langflow",
required=True,
), ),
MessageTextInput( MessageTextInput(
name="persist_directory", name="persist_directory",
@ -58,7 +56,7 @@ class LocalDBComponent(LCVectorStoreComponent):
show=False, show=False,
combobox=True, combobox=True,
), ),
HandleInput(name="embedding", display_name="Embedding", input_types=["Embeddings"]), HandleInput(name="embedding", display_name="Embedding", required=True, input_types=["Embeddings"]),
BoolInput( BoolInput(
name="allow_duplicates", name="allow_duplicates",
display_name="Allow Duplicates", display_name="Allow Duplicates",
@ -102,7 +100,7 @@ class LocalDBComponent(LCVectorStoreComponent):
), ),
] ]
outputs = [ outputs = [
Output(display_name="DataFrame", name="dataframe", method="as_dataframe"), Output(display_name="DataFrame", name="dataframe", method="perform_search"),
] ]
def get_vector_store_directory(self, base_dir: str | Path) -> Path: def get_vector_store_directory(self, base_dir: str | Path) -> Path:
@ -257,3 +255,6 @@ class LocalDBComponent(LCVectorStoreComponent):
vector_store.add_documents(documents) vector_store.add_documents(documents)
else: else:
self.log("No documents to add to the Vector Store.") self.log("No documents to add to the Vector Store.")
def perform_search(self) -> DataFrame:
return DataFrame(self.search_documents())