chore: Remove unused vector search component and update vector store parameters format
This commit is contained in:
parent
4b495fc0a1
commit
6e88eab7f5
3 changed files with 7 additions and 6 deletions
|
|
@ -12,6 +12,5 @@ __all__ = [
|
||||||
"textsplitters",
|
"textsplitters",
|
||||||
"toolkits",
|
"toolkits",
|
||||||
"tools",
|
"tools",
|
||||||
"vectorsearch",
|
|
||||||
"vectorstores",
|
"vectorstores",
|
||||||
]
|
]
|
||||||
|
|
|
||||||
|
|
@ -221,5 +221,6 @@ class Component(CustomComponent):
|
||||||
return [field.name for field in inputs]
|
return [field.name for field in inputs]
|
||||||
except KeyError:
|
except KeyError:
|
||||||
return []
|
return []
|
||||||
return []
|
|
||||||
return []
|
def build(self, **kwargs):
|
||||||
|
self.set_attributes(kwargs)
|
||||||
|
|
|
||||||
|
|
@ -4,7 +4,6 @@ import pytest
|
||||||
from langchain_core.documents import Document
|
from langchain_core.documents import Document
|
||||||
from langflow.components.memories.AstraDBMessageReader import AstraDBMessageReaderComponent
|
from langflow.components.memories.AstraDBMessageReader import AstraDBMessageReaderComponent
|
||||||
from langflow.components.memories.AstraDBMessageWriter import AstraDBMessageWriterComponent
|
from langflow.components.memories.AstraDBMessageWriter import AstraDBMessageWriterComponent
|
||||||
from langflow.components.vectorsearch.AstraDBSearch import AstraDBSearchComponent
|
|
||||||
from langflow.components.vectorstores.AstraDB import AstraVectorStoreComponent
|
from langflow.components.vectorstores.AstraDB import AstraVectorStoreComponent
|
||||||
from langflow.schema.data import Data
|
from langflow.schema.data import Data
|
||||||
|
|
||||||
|
|
@ -78,10 +77,11 @@ def test_astra_embeds_and_search(astra_fixture):
|
||||||
collection_name=SEARCH_COLLECTION,
|
collection_name=SEARCH_COLLECTION,
|
||||||
embedding=embedding,
|
embedding=embedding,
|
||||||
inputs=records,
|
inputs=records,
|
||||||
|
add_to_vector_store=True,
|
||||||
)
|
)
|
||||||
|
component.build_vector_store()
|
||||||
|
|
||||||
component = AstraDBSearchComponent()
|
component.build(
|
||||||
records = component.build(
|
|
||||||
token=application_token,
|
token=application_token,
|
||||||
api_endpoint=api_endpoint,
|
api_endpoint=api_endpoint,
|
||||||
collection_name=SEARCH_COLLECTION,
|
collection_name=SEARCH_COLLECTION,
|
||||||
|
|
@ -89,6 +89,7 @@ def test_astra_embeds_and_search(astra_fixture):
|
||||||
input_value="test1",
|
input_value="test1",
|
||||||
number_of_results=1,
|
number_of_results=1,
|
||||||
)
|
)
|
||||||
|
records = component.search_documents()
|
||||||
|
|
||||||
assert len(records) == 1
|
assert len(records) == 1
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue