Prajith04 commited on
Commit
8703987
·
verified ·
1 Parent(s): c20fa4f

Update vectordb_utils.py

Browse files
Files changed (1) hide show
  1. vectordb_utils.py +1 -0
vectordb_utils.py CHANGED
@@ -4,6 +4,7 @@ from qdrant_client import QdrantClient
4
  from qdrant_client.models import VectorParams, Distance, PointStruct
5
  from sentence_transformers import SentenceTransformer
6
  import uuid
 
7
  cache_dir = os.environ.get("MODEL_CACHE_DIR", "/app/cache") # Fallback to /app/cache
8
  os.makedirs(cache_dir, exist_ok=True)
9
  encoder =SentenceTransformer("all-MiniLM-L6-v2", cache_folder=cache_dir)
 
4
  from qdrant_client.models import VectorParams, Distance, PointStruct
5
  from sentence_transformers import SentenceTransformer
6
  import uuid
7
+ import os
8
  cache_dir = os.environ.get("MODEL_CACHE_DIR", "/app/cache") # Fallback to /app/cache
9
  os.makedirs(cache_dir, exist_ok=True)
10
  encoder =SentenceTransformer("all-MiniLM-L6-v2", cache_folder=cache_dir)