Upload retrieve_docs.py with huggingface_hub

retrieve_docs.py

@@ -3,6 +3,7 @@
 import json
 import faiss
 import numpy as np
+from sentence_transformers import SentenceTransformer
 
 INDEX_PATH = "data/docs/docs.index"
 META_PATH = "data/docs/docs_meta.json"
@@ -17,16 +18,10 @@ QUERY_TEMPLATES = {
     "file_not_found": "Jenkins pipeline workspace file not found",
     "git_authentication_error": "Jenkins git authentication failed checkout"
 }
-
-# Lazy load model
-_model = None
-
-def get_model():
-    global _model
-    if _model is None:
-        from sentence_transformers import SentenceTransformer
-        _model = SentenceTransformer("all-MiniLM-L6-v2", cache_folder="./models")
-    return _model
+model = SentenceTransformer(
+    "paraphrase-MiniLM-L3-v2",
+    cache_folder="./model_cache"
+)
 
 def retrieve_docs(error_category: str):
 
@@ -39,7 +34,6 @@ def retrieve_docs(error_category: str):
         "Jenkins pipeline error"
     )
 
-    model = get_model()
     query_embedding = model.encode([query])
     distances, indices = index.search(query_embedding, TOP_K)