helloparthshah commited on
Commit
344c9c4
·
1 Parent(s): e7a610c

Updating threshold

Browse files
Files changed (1) hide show
  1. src/manager/manager.py +1 -1
src/manager/manager.py CHANGED
@@ -176,7 +176,7 @@ class GeminiManager:
176
  return results
177
 
178
  def run(self, messages):
179
- memories = self.get_k_memories(messages[-1]['content'], k=5, threshold=0.0)
180
  if len(memories) > 0:
181
  messages.append({
182
  "role": "memories",
 
176
  return results
177
 
178
  def run(self, messages):
179
+ memories = self.get_k_memories(messages[-1]['content'], k=5, threshold=0.1)
180
  if len(memories) > 0:
181
  messages.append({
182
  "role": "memories",