Initial commit: breakpilot-core - Shared Infrastructure
Docker Compose with 24+ services: - PostgreSQL (PostGIS), Valkey, MinIO, Qdrant - Vault (PKI/TLS), Nginx (Reverse Proxy) - Backend Core API, Consent Service, Billing Service - RAG Service, Embedding Service - Gitea, Woodpecker CI/CD - Night Scheduler, Health Aggregator - Jitsi (Web/XMPP/JVB/Jicofo), Mailpit Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
77
rag-service/api/collections.py
Normal file
77
rag-service/api/collections.py
Normal file
@@ -0,0 +1,77 @@
|
||||
import logging
|
||||
from typing import Optional
|
||||
|
||||
from fastapi import APIRouter, HTTPException, Request
|
||||
from pydantic import BaseModel
|
||||
|
||||
from api.auth import optional_jwt_auth
|
||||
from qdrant_client_wrapper import qdrant_wrapper, ALL_DEFAULT_COLLECTIONS
|
||||
|
||||
logger = logging.getLogger("rag-service.api.collections")
|
||||
|
||||
router = APIRouter(prefix="/api/v1/collections")
|
||||
|
||||
|
||||
# ---- Request / Response models --------------------------------------------
|
||||
|
||||
class CreateCollectionRequest(BaseModel):
|
||||
name: str
|
||||
vector_size: int = 1536
|
||||
|
||||
|
||||
class CollectionInfoResponse(BaseModel):
|
||||
name: str
|
||||
vectors_count: Optional[int] = None
|
||||
points_count: Optional[int] = None
|
||||
status: Optional[str] = None
|
||||
vector_size: Optional[int] = None
|
||||
|
||||
|
||||
# ---- Endpoints ------------------------------------------------------------
|
||||
|
||||
@router.post("", status_code=201)
|
||||
async def create_collection(body: CreateCollectionRequest, request: Request):
|
||||
"""Create a new Qdrant collection."""
|
||||
optional_jwt_auth(request)
|
||||
try:
|
||||
created = await qdrant_wrapper.create_collection(body.name, body.vector_size)
|
||||
return {
|
||||
"collection": body.name,
|
||||
"vector_size": body.vector_size,
|
||||
"created": created,
|
||||
}
|
||||
except Exception as exc:
|
||||
logger.error("Failed to create collection '%s': %s", body.name, exc)
|
||||
raise HTTPException(status_code=500, detail=str(exc))
|
||||
|
||||
|
||||
@router.get("")
|
||||
async def list_collections(request: Request):
|
||||
"""List all Qdrant collections."""
|
||||
optional_jwt_auth(request)
|
||||
try:
|
||||
result = qdrant_wrapper.client.get_collections()
|
||||
names = [c.name for c in result.collections]
|
||||
return {"collections": names, "count": len(names)}
|
||||
except Exception as exc:
|
||||
logger.error("Failed to list collections: %s", exc)
|
||||
raise HTTPException(status_code=500, detail=str(exc))
|
||||
|
||||
|
||||
@router.get("/defaults")
|
||||
async def list_default_collections(request: Request):
|
||||
"""Return the pre-configured default collections and their dimensions."""
|
||||
optional_jwt_auth(request)
|
||||
return {"defaults": ALL_DEFAULT_COLLECTIONS}
|
||||
|
||||
|
||||
@router.get("/{collection_name}")
|
||||
async def get_collection_info(collection_name: str, request: Request):
|
||||
"""Get stats for a single collection."""
|
||||
optional_jwt_auth(request)
|
||||
try:
|
||||
info = await qdrant_wrapper.get_collection_info(collection_name)
|
||||
return info
|
||||
except Exception as exc:
|
||||
logger.error("Failed to get collection info for '%s': %s", collection_name, exc)
|
||||
raise HTTPException(status_code=404, detail=f"Collection '{collection_name}' not found or error: {exc}")
|
||||
Reference in New Issue
Block a user