knowledge/retrieval.py

# Retrieval simulation
from knowledge.knowledge_base import KnowledgeBase

class Retriever:
    def __init__(self):
        self.kb = KnowledgeBase()

    def retrieve(self, topic):
        return self.kb.fetch(topic)