Api for listing documents
This commit is contained in:
parent
110e020122
commit
b70fd8b7b4
1 changed files with 14 additions and 0 deletions
|
|
@ -189,6 +189,19 @@ async def health_check(_: Request) -> JSONResponse:
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
async def list_documents(_: Request) -> JSONResponse:
|
||||||
|
"""List all documents in the database."""
|
||||||
|
client = get_client(db_path)
|
||||||
|
docs = await client.document_repository.list_all()
|
||||||
|
return JSONResponse(
|
||||||
|
{
|
||||||
|
"documents": [
|
||||||
|
{"id": doc.id, "title": doc.title, "uri": doc.uri} for doc in docs
|
||||||
|
]
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
async def visualize_chunk(request: Request) -> JSONResponse:
|
async def visualize_chunk(request: Request) -> JSONResponse:
|
||||||
"""Return visual grounding images for a chunk as base64."""
|
"""Return visual grounding images for a chunk as base64."""
|
||||||
import base64
|
import base64
|
||||||
|
|
@ -228,6 +241,7 @@ async def visualize_chunk(request: Request) -> JSONResponse:
|
||||||
app = Starlette(
|
app = Starlette(
|
||||||
routes=[
|
routes=[
|
||||||
Route("/v1/research/stream", stream_research_agent, methods=["POST"]),
|
Route("/v1/research/stream", stream_research_agent, methods=["POST"]),
|
||||||
|
Route("/api/documents", list_documents, methods=["GET"]),
|
||||||
Route("/api/visualize/{chunk_id}", visualize_chunk, methods=["GET"]),
|
Route("/api/visualize/{chunk_id}", visualize_chunk, methods=["GET"]),
|
||||||
Route("/health", health_check, methods=["GET"]),
|
Route("/health", health_check, methods=["GET"]),
|
||||||
],
|
],
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue