File size: 669 Bytes
e0d9dd8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
from service.data_loader_service import CSVDataLoader
from service.embedded_service import EmbeddingService
from service.vector_store_service import VectorStoreService
class KnowledgeBase:
def __init__(self, csv_path: str):
self.embedder = EmbeddingService()
loader = CSVDataLoader(csv_path)
qa_pairs = loader.load_qa_pairs()
self.documents = [
f"Question: {p['question']}\nAnswer: {p['answer']}"
for p in qa_pairs
]
self.embeddings = self.embedder.embed(self.documents)
self.vector_store = VectorStoreService(
self.embeddings,
self.documents
)
|