From 63c69128413badbca10d2ff456a0bf06119f75c6 Mon Sep 17 00:00:00 2001
From: Alex <a@tushynski.me>
Date: Wed, 14 May 2025 21:45:30 +0100
Subject: [PATCH] lazy load elasticsearch

---
 application/requirements.txt             | 3 ---
 application/vectorstore/elasticsearch.py | 9 ++-------
 2 files changed, 2 insertions(+), 10 deletions(-)

diff --git a/application/requirements.txt b/application/requirements.txt
index b503b896..c51c2324 100644
--- a/application/requirements.txt
+++ b/application/requirements.txt
@@ -6,8 +6,6 @@ dataclasses-json==0.6.7
 docx2txt==0.8
 duckduckgo-search==7.5.2
 ebooklib==0.18
-elastic-transport==8.17.0
-elasticsearch==8.17.1
 escodegen==1.0.11
 esprima==4.0.1
 esutils==1.0.1
@@ -18,7 +16,6 @@ google-genai==1.3.0
 google-generativeai==0.8.5
 gTTS==2.5.4
 gunicorn==23.0.0
-html2text==2024.2.26
 javalang==0.13.0
 jinja2==3.1.6
 jiter==0.8.2
diff --git a/application/vectorstore/elasticsearch.py b/application/vectorstore/elasticsearch.py
index e393e4a5..dfa0314f 100644
--- a/application/vectorstore/elasticsearch.py
+++ b/application/vectorstore/elasticsearch.py
@@ -1,9 +1,6 @@
 from application.vectorstore.base import BaseVectorStore
 from application.core.settings import settings
 from application.vectorstore.document_class import Document
-import elasticsearch
-
-
 
 
 class ElasticsearchStore(BaseVectorStore):
@@ -26,8 +23,7 @@ class ElasticsearchStore(BaseVectorStore):
             else:
                 raise ValueError("Please provide either elasticsearch_url or cloud_id.")
 
-            
-
+            import elasticsearch
             ElasticsearchStore._es_connection = elasticsearch.Elasticsearch(**connection_params)
             
         self.docsearch = ElasticsearchStore._es_connection
@@ -155,8 +151,6 @@ class ElasticsearchStore(BaseVectorStore):
         **kwargs,
         ):
         
-        from elasticsearch.helpers import BulkIndexError, bulk
-
         bulk_kwargs = bulk_kwargs or {}
         import uuid
         embeddings = []
@@ -189,6 +183,7 @@ class ElasticsearchStore(BaseVectorStore):
 
 
         if len(requests) > 0:
+            from elasticsearch.helpers import BulkIndexError, bulk
             try:
                 success, failed = bulk(
                     self._es_connection,