absence404 commited on
Commit
0360d9a
·
verified ·
1 Parent(s): 2fd7279

Update backend/knowledge_base.py

Browse files
Files changed (1) hide show
  1. backend/knowledge_base.py +1 -1
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 = 3) -> 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 []
 
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 []