Spaces:
Running
Running
Update backend/knowledge_base.py
Browse files
backend/knowledge_base.py
CHANGED
|
@@ -46,7 +46,7 @@ def initialize_knowledge_base():
|
|
| 46 |
else:
|
| 47 |
print(f"Knowledge base already loaded with {existing} documents.")
|
| 48 |
|
| 49 |
-
def search_knowledge_base(query: str, n_results: int =
|
| 50 |
query_embedding = model.encode([query]).tolist()
|
| 51 |
results = collection.query(query_embeddings=query_embedding, n_results=n_results)
|
| 52 |
docs = results['documents'][0] if results['documents'] else []
|
|
|
|
| 46 |
else:
|
| 47 |
print(f"Knowledge base already loaded with {existing} documents.")
|
| 48 |
|
| 49 |
+
def search_knowledge_base(query: str, n_results: int = 5) -> str:
|
| 50 |
query_embedding = model.encode([query]).tolist()
|
| 51 |
results = collection.query(query_embeddings=query_embedding, n_results=n_results)
|
| 52 |
docs = results['documents'][0] if results['documents'] else []
|