Skip to content

Commit

Permalink
fix: 🐛 duplicate
Browse files Browse the repository at this point in the history
  • Loading branch information
StanGirard committed Jan 27, 2024
1 parent d0421f4 commit 7043a28
Showing 1 changed file with 13 additions and 1 deletion.
14 changes: 13 additions & 1 deletion backend/modules/brain/service/brain_service.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,10 @@
from modules.brain.service.utils.validate_brain import validate_api_brain
from modules.knowledge.service.knowledge_service import KnowledgeService

from logger import get_logger

logger = get_logger(__name__)

knowledge_service = KnowledgeService()
# TODO: directly user api_brain_definition repository
api_brain_definition_service = ApiBrainDefinitionService()
Expand Down Expand Up @@ -97,7 +101,15 @@ def find_brain_from_question(
# Calculate the closest brains to the question
list_brains = vector_store.find_brain_closest_query(user.id, question)

metadata["close_brains"] = list_brains[:5]
unique_list_brains = []
seen_brain_ids = set()

for brain in list_brains:
if brain["id"] not in seen_brain_ids:
unique_list_brains.append(brain)
seen_brain_ids.add(brain["id"])

metadata["close_brains"] = unique_list_brains[:5]

if list_brains and not brain_id_to_use:
brain_id_to_use = list_brains[0]["id"]
Expand Down

0 comments on commit 7043a28

Please sign in to comment.