codelion commited on
Commit
9c6f575
Β·
verified Β·
1 Parent(s): 5516e24

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -3
main.py CHANGED
@@ -29,12 +29,13 @@ logger = get_logger(__name__)
29
 
30
  # ─────── embeddings ─────────────────────────────────────────────────────────────
31
  # Switch to local BGE embeddings (no JSONDecode errors, no HTTP‑batch issues) :contentReference[oaicite:0]{index=0}
32
- embeddings = HuggingFaceBgeEmbeddings(
 
 
33
  model_name="BAAI/bge-large-en-v1.5",
34
  model_kwargs={"device": "cpu"},
35
- encode_kwargs={"normalize_embeddings": True},
36
  )
37
-
38
  # ─────── vector store + memory ─────────────────────────────────────────────────
39
  vector_store = SupabaseVectorStore(
40
  client=supabase,
 
29
 
30
  # ─────── embeddings ─────────────────────────────────────────────────────────────
31
  # Switch to local BGE embeddings (no JSONDecode errors, no HTTP‑batch issues) :contentReference[oaicite:0]{index=0}
32
+ from langchain_community.embeddings import SentenceTransformerEmbeddings
33
+
34
+ embeddings = SentenceTransformerEmbeddings(
35
  model_name="BAAI/bge-large-en-v1.5",
36
  model_kwargs={"device": "cpu"},
37
+ encode_kwargs={"normalize_embeddings": True}
38
  )
 
39
  # ─────── vector store + memory ─────────────────────────────────────────────────
40
  vector_store = SupabaseVectorStore(
41
  client=supabase,