Spaces:
Sleeping
Sleeping
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],
|