f799b41ed9
- #11 VectorIndexProtocol for typed duck-typing of vector_index param - #10 Wrap all sync VectorIndex.query() calls in asyncio.to_thread - #9 Make vector_search async (consistent with to_thread wrapping) - #12 Replace module-level singleton with app.state.query_service - #13 InMemoryGraphDB.close(destroy=False) guard for test safety - #14 Parse dates with datetime.fromisoformat in list_cases sort Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
"""Route definitions for the read-only query API."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from fastapi import APIRouter, HTTPException, Query, Depends
|
|
|
|
from aucourt_ingest.api.dependencies import get_query_service
|
|
from aucourt_ingest.api.schemas import (
|
|
CaseGraphSummary,
|
|
CaseSummary,
|
|
ChunkResult,
|
|
HealthResponse,
|
|
HybridResult,
|
|
JurorContextResponse,
|
|
PersonaInfo,
|
|
)
|
|
|
|
router = APIRouter(prefix="/api/v1")
|
|
|
|
|
|
@router.get("/health", response_model=HealthResponse)
|
|
async def health(svc=Depends(get_query_service)):
|
|
return await svc.health()
|
|
|
|
|
|
@router.get("/personas", response_model=list[PersonaInfo])
|
|
async def list_personas(svc=Depends(get_query_service)):
|
|
return svc.list_personas()
|
|
|
|
|
|
@router.get("/cases", response_model=list[CaseSummary])
|
|
async def list_cases(
|
|
court: str | None = Query(None, description="Filter by court code"),
|
|
limit: int = Query(50, ge=1, le=200),
|
|
offset: int = Query(0, ge=0),
|
|
svc=Depends(get_query_service),
|
|
):
|
|
return await svc.list_cases(court=court, limit=limit, offset=offset)
|
|
|
|
|
|
@router.get("/cases/search", response_model=list[CaseSummary])
|
|
async def search_cases(
|
|
q: str = Query(..., description="Search query text"),
|
|
limit: int = Query(10, ge=1, le=50),
|
|
svc=Depends(get_query_service),
|
|
):
|
|
return await svc.search_cases(q=q, limit=limit)
|
|
|
|
|
|
@router.get("/cases/{case_mnc}/juror/{persona}", response_model=JurorContextResponse)
|
|
async def get_juror_context(
|
|
case_mnc: str,
|
|
persona: str,
|
|
max_tokens: int | None = Query(None, ge=100, le=16000),
|
|
svc=Depends(get_query_service),
|
|
):
|
|
try:
|
|
return await svc.get_juror_context(
|
|
case_mnc, persona, max_tokens,
|
|
)
|
|
except KeyError as e:
|
|
raise HTTPException(status_code=404, detail=str(e))
|
|
|
|
|
|
@router.get("/cases/{case_mnc}", response_model=CaseGraphSummary)
|
|
async def get_case_graph(case_mnc: str, svc=Depends(get_query_service)):
|
|
result = await svc.get_case_graph(case_mnc)
|
|
if result is None:
|
|
raise HTTPException(status_code=404, detail=f"Case not found: {case_mnc}")
|
|
return result
|
|
|
|
|
|
@router.get("/search", response_model=list[ChunkResult])
|
|
async def vector_search(
|
|
q: str = Query(..., description="Query text"),
|
|
top_k: int = Query(10, ge=1, le=50),
|
|
chunk_types: str | None = Query(None, description="Comma-separated chunk types"),
|
|
doc_ids: str | None = Query(None, description="Comma-separated doc IDs"),
|
|
svc=Depends(get_query_service),
|
|
):
|
|
types = [t.strip() for t in chunk_types.split(",") if t.strip()] if chunk_types else None
|
|
docs = [d.strip() for d in doc_ids.split(",") if d.strip()] if doc_ids else None
|
|
return await svc.vector_search(q, top_k=top_k, chunk_types=types, doc_ids=docs)
|
|
|
|
|
|
@router.get("/hybrid", response_model=list[HybridResult])
|
|
async def hybrid_search(
|
|
q: str = Query(..., description="Query text"),
|
|
persona: str | None = Query(None, description="Juror persona name"),
|
|
top_k: int = Query(10, ge=1, le=50),
|
|
max_tokens: int | None = Query(None, ge=100, le=16000),
|
|
svc=Depends(get_query_service),
|
|
):
|
|
return await svc.hybrid_search(q, persona_name=persona, top_k=top_k, max_tokens=max_tokens)
|