Update retrieval_engine.py
Browse files- retrieval_engine.py +2 -1
retrieval_engine.py
CHANGED
|
@@ -113,13 +113,14 @@ class RetrievalEngine:
|
|
| 113 |
|
| 114 |
return bonus
|
| 115 |
|
| 116 |
-
|
| 117 |
self,
|
| 118 |
query: str,
|
| 119 |
topic: str = "",
|
| 120 |
intent: str = "answer",
|
| 121 |
k: int = 3,
|
| 122 |
) -> List[RetrievedChunk]:
|
|
|
|
| 123 |
if not self.rows:
|
| 124 |
return []
|
| 125 |
|
|
|
|
| 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 |
|