NEXAS commited on
Commit
5eb4729
·
verified ·
1 Parent(s): 29df71b

Update utils/qa.py

Browse files
Files changed (1) hide show
  1. utils/qa.py +3 -1
utils/qa.py CHANGED
@@ -11,7 +11,9 @@ class QAEngine:
11
 
12
  def query(self, question: str, k: int = 5) -> str:
13
  """Query the document using semantic search and generate an answer"""
14
- query_embedding = self.llm_processor.embed_model.encode(question)
 
 
15
 
16
  results = self.collection.query(
17
  query_embeddings=[query_embedding],
 
11
 
12
  def query(self, question: str, k: int = 5) -> str:
13
  """Query the document using semantic search and generate an answer"""
14
+
15
+ # ✅ Correct method for getting embeddings
16
+ query_embedding = self.llm_processor.embed_model.embed_query(question)
17
 
18
  results = self.collection.query(
19
  query_embeddings=[query_embedding],