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

remove retriever component and outputs (#4979)

Browse source 
* remove retriever component and outputs

* [autofix.ci] apply automated fixes

* ruff check fix

* [autofix.ci] apply automated fixes

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: cristhianzl <cristhian.lousa@gmail.com>
Co-authored-by: Eric Hare <ericrhare@gmail.com>
This commit is contained in:
Rodrigo Nader 2024-12-03 12:38:02 -03:00 • committed by GitHub
commit 53e00c8092
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 1 additions and 59 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/tests/integration/components/astra/test_astra_component.py
View file

@ -55,9 +55,7 @@ async def test_base(astradb_client: AstraDB):
),
},
)
from langchain_core.vectorstores import VectorStoreRetriever
assert isinstance(results["base_retriever"], VectorStoreRetriever)
assert results["vector_store"] is not None
assert results["search_results"] == []
assert astradb_client.collection(BASIC_COLLECTION)

2
src/backend/tests/unit/initial_setup/starter_projects/test_vector_store_rag.py
View file

@ -36,7 +36,6 @@ def ingestion_graph():
api_endpoint="https://astra.example.com",
token="token", # noqa: S106
)
vector_store.set_on_output(name="base_retriever", value="mock_retriever", cache=True)
vector_store.set_on_output(name="search_results", value=[Data(text="This is a test file.")], cache=True)
return Graph(file_component, vector_store)
@ -64,7 +63,6 @@ def rag_graph():
],
cache=True,
)
rag_vector_store.set_on_output(name="base_retriever", value="mock_retriever", cache=True)
parse_data = ParseDataComponent(_id="parse-data-123")
parse_data.set(data=rag_vector_store.search_documents)
prompt_component = PromptComponent(_id="prompt-123")