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

Feat: add a support for OpenSearch and AstraDB components to yield the langchain vector_store connection object (#6998)

Browse source 
* Added decorator, decorator test, and modified supported vector stores

* Renamed module file name to reflect that this is for generic use, not use for graph rag

* Updated docsstring

* Improved documentation and modification to UT to support graph rag

* Remove extra file from PR

* rollback vector store template

* [autofix.ci] apply automated fixes

---------

Co-authored-by: Nadir J <31660040+NadirJ@users.noreply.github.com>
Co-authored-by: cristhianzl <cristhian.lousa@gmail.com>
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
Pedro Pacheco 2025-03-11 18:41:34 -06:00 • committed by GitHub
commit eedef1efae
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 127 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -30,7 +30,13 @@ def ingestion_graph():
openai_embeddings.set(
openai_api_key="sk-123", openai_api_base="https://api.openai.com/v1", openai_api_type="openai"
)
vector_store = AstraDBVectorStoreComponent(_id="ingestion-vector-store-123")
# Mock search_documents by changing the value otherwise set by the vector_store_connection_decorator
vector_store.set_on_output(name="vectorstoreconnection", value=[Data(text="This is a test file.")], cache=True)
vector_store.set_on_output(name="vectorstoreconnection", value=[Data(text="This is a test file.")], cache=True)
vector_store.set_on_output(name="search_results", value=[Data(text="This is a test file.")], cache=True)
vector_store.set_on_output(name="dataframe", value=DataFrame(data=[Data(text="This is a test file.")]), cache=True)
vector_store.set(