Update retrieval_engine.py
Browse files- retrieval_engine.py +9 -9
retrieval_engine.py
CHANGED
|
@@ -113,15 +113,15 @@ class RetrievalEngine:
|
|
| 113 |
|
| 114 |
return bonus
|
| 115 |
|
| 116 |
-
def search(
|
| 117 |
-
|
| 118 |
-
|
| 119 |
-
|
| 120 |
-
|
| 121 |
-
|
| 122 |
-
) -> List[RetrievedChunk]:
|
| 123 |
-
|
| 124 |
-
|
| 125 |
return []
|
| 126 |
|
| 127 |
combined_query = clean_math_text(query)
|
|
|
|
| 113 |
|
| 114 |
return bonus
|
| 115 |
|
| 116 |
+
def search(
|
| 117 |
+
self,
|
| 118 |
+
query: str,
|
| 119 |
+
topic: str = "",
|
| 120 |
+
intent: str = "answer",
|
| 121 |
+
k: int = 3,
|
| 122 |
+
) -> List[RetrievedChunk]:
|
| 123 |
+
|
| 124 |
+
if not self.rows:
|
| 125 |
return []
|
| 126 |
|
| 127 |
combined_query = clean_math_text(query)
|