j-js commited on
Commit
0b3cc65
·
verified ·
1 Parent(s): 63257f4

Update retrieval_engine.py

Browse files
Files changed (1) hide show
  1. retrieval_engine.py +9 -9
retrieval_engine.py CHANGED
@@ -113,15 +113,15 @@ class RetrievalEngine:
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)
 
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)