Initial project setup

This commit is contained in:
Ayomide
2025-07-07 22:08:02 +01:00
parent c158262a49
commit b76a3e75f3
11 changed files with 208 additions and 0 deletions
+14
View File
@@ -0,0 +1,14 @@
import numpy as np
import faiss
from backend.config import Config
class VectorDB:
def __init__(self):
self.index = faiss.IndexFlatL2(768) # Cohere embedding dim
def add_vectors(self, ids, embeddings):
self.index.add(np.array(embeddings).astype('float32'))
def search(self, query_embedding, k=5):
distances, indices = self.index.search(np.array([query_embedding]), k)
return indices[0]