Spaces:
Running
Running
Update utils/qa.py
Browse files- 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 |
-
|
|
|
|
|
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],
|