feat: chromadb integration

Backend/app/api/deps.py

@@ -8,6 +8,7 @@ from app.models import User
 from app.config import settings
 from fastapi import Request
 from chromadb import AsyncHttpClient
+from chromadb.api.models.Collection import Collection
 
 security  = HTTPBearer()
 
@@ -56,4 +57,10 @@ async def get_chroma_client(request: Request) -> AsyncHttpClient:
     client = getattr(request.app.state, "chroma_client", None)
     if client is None:
         raise RuntimeError("ChromaDB client is not initialized in App State.")
-    return client
+    return client
+
+def get_chroma_collection(request: Request) -> Collection:
+    collection = getattr(request.app.state, "chroma_collection", None)
+    if collection is None:
+        raise RuntimeError("ChromaDB Collection not loaded during application startup.")
+    return collection

Backend/app/api/v1/endpoints/quiz.py

@@ -3,11 +3,30 @@ from chromadb import AsyncHttpClient
 from app.models import User
 from app.api.deps import get_db, get_current_user, get_chroma_client
 from app.schema import Quiz_input
-from prompts import SYSTEM_PROMPT
-
-router = APIRouter()
-
+from .prompts import SYSTEM_PROMPT
 
+from fastapi import APIRouter, Depends, HTTPException
+from chromadb.api.models.Collection import Collection # Import Collection type
+from app.api.deps import get_chroma_collection
+
+router = APIRouter(prefix="/search")
+
+
+@router.get("/")
+async def search_documents(
+    query: str, 
+    # Inject the pre-loaded Collection object directly
+    collection: Collection = Depends(get_chroma_collection) 
+):
+    try:
+        # The Collection object is ready to use immediately.
+        results = await collection.query(
+            query_texts=[query],
+            n_results=5
+        )
+        return results
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"ChromaDB Query Error: {e}")
 
 
 

Backend/app/main.py

@@ -6,7 +6,7 @@ from app.config import settings
 from app.database import engine, Base
 from app.api.v1.api import api_router
 import chromadb
-
+from chromadb.api.models.Collection import Collection
 
 
 @asynccontextmanager
@@ -18,10 +18,20 @@ async def lifespan(app: FastAPI):
     async with engine.begin() as conn:
         await conn.run_sync(Base.metadata.create_all)
     
-    app.state.chroma_client = await chromadb.AsyncHttpClient(
+    client = await chromadb.AsyncHttpClient(
         host=settings.chroma_host,
         port=settings.chroma_port
     )
+    app.state.chroma_client = client
+
+    try:
+        collection: Collection = await client.get_or_create_collection(settings.chroma_collection)
+        app.state.chroma_collection = collection
+
+        count = await collection.count()
+        print(f"Successfully loaded collection '{settings.chroma_collection}' with {count} documents.")
+    except Exception as e:
+        print(f"Failed to load ChromaDB collection: {e}")
 
     print("✅ Tables ready!")
     yield

Backend/requirements.txt

@@ -8,3 +8,4 @@ passlib[argon2]
 python-jose[cryptography]
 email-validator
 
+