Update the chatbot example to use newer SparseVectorStrategy class (e…

…lastic#327)
leemthompo · Sep 11, 2024 · 61c1af5 · 61c1af5
1 parent 6f9a71d
commit 61c1af5
Show file tree

Hide file tree

Showing 2 changed files with 144 additions and 155 deletions.
diff --git a/example-apps/chatbot-rag-app/api/chat.py b/example-apps/chatbot-rag-app/api/chat.py
@@ -1,4 +1,4 @@
-from langchain_elasticsearch import ElasticsearchStore
+from langchain_elasticsearch import ElasticsearchStore, SparseVectorStrategy
 from llm_integrations import get_llm
 from elasticsearch_client import (
     elasticsearch_client,
@@ -20,7 +20,7 @@
 store = ElasticsearchStore(
     es_connection=elasticsearch_client,
     index_name=INDEX,
-    strategy=ElasticsearchStore.SparseVectorRetrievalStrategy(model_id=ELSER_MODEL),
+    strategy=SparseVectorStrategy(model_id=ELSER_MODEL),
 )