feat: tool mode for all vector store components (#5348)

* feat: tool mode for all vector store components

* [autofix.ci] apply automated fixes

* [autofix.ci] apply automated fixes (attempt 2/3)

* Update elasticsearch.py

* Update test_vector_store_rag.py

* Update vector_store_rag.py

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
Eric Hare 2024-12-19 08:49:09 -08:00 • committed by GitHub
commit 22877ae920
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
28 changed files with 412 additions and 572 deletions

View file

@ -49,7 +49,7 @@ def rag_graph():
chat_input.get_output("message").value = "What is the meaning of life?"
rag_vector_store = AstraDBVectorStoreComponent(_id="rag-vector-store-123")
rag_vector_store.set(
search_input=chat_input.message_response,
search_query=chat_input.message_response,
api_endpoint="https://astra.example.com",
token="token", # noqa: S106
embedding_model=openai_embeddings.build_embeddings,