Spaces:
Sleeping
Sleeping
Update back.py
Browse files
back.py
CHANGED
@@ -306,7 +306,7 @@ class ProductDatabase:
|
|
306 |
except Exception as e:
|
307 |
raise Exception(f"Error processing markdown content: {str(e)}")
|
308 |
|
309 |
-
def search(self, query: str, k: int =
|
310 |
"""Search for relevant documents"""
|
311 |
if not self.vectorstore:
|
312 |
raise ValueError("Database not initialized. Please process documents first.")
|
|
|
306 |
except Exception as e:
|
307 |
raise Exception(f"Error processing markdown content: {str(e)}")
|
308 |
|
309 |
+
def search(self, query: str, k: int = 5) -> List[Dict[str, Any]]:
|
310 |
"""Search for relevant documents"""
|
311 |
if not self.vectorstore:
|
312 |
raise ValueError("Database not initialized. Please process documents first.")
|