SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 20:12:55 +01:00
Code Issues Projects Releases Packages Activity

Added delete schema

Browse source
This commit is contained in:
Saurab-Shrestha 2024-02-06 13:28:09 +05:45
parent 0a326c1e43
commit 2fe78b4b5f
8 changed files with 35 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

11
private_gpt/server/ingest/ingest_router.py
View file

@ -2,7 +2,7 @@ import logging
from pathlib import Path
from typing import Literal
from fastapi import APIRouter, Depends, HTTPException, Request, UploadFile, File, status, Security
from fastapi import APIRouter, Depends, HTTPException, Request, UploadFile, File, status, Security, Body
from fastapi.responses import JSONResponse
from pydantic import BaseModel, Field
@ -35,6 +35,8 @@ class IngestResponse(BaseModel):
model: Literal["private-gpt"]
data: list[IngestedDoc]
class DeleteFilename(BaseModel):
filename: str
@ingest_router.post("/ingest", tags=["Ingestion"], deprecated=True)
def ingest(request: Request, file: UploadFile) -> IngestResponse:
@ -121,10 +123,10 @@ def delete_ingested(request: Request, doc_id: str) -> None:
service.delete(doc_id)
@ingest_router.delete("/ingest/file/{filename}", tags=["Ingestion"])
@ingest_router.post("/ingest/file/delete", tags=["Ingestion"])
def delete_file(
request: Request,
filename: str,
delete_input: DeleteFilename,
current_user: models.User = Security(
deps.get_current_user,
)) -> dict:
@ -133,6 +135,9 @@ def delete_file(
The `filename` can be obtained from the `GET /ingest/list` endpoint.
The document will be effectively deleted from your storage context.
"""
filename = delete_input.filename
print(filename)
service = request.state.injector.get(IngestService)
try:
doc_ids = service.get_doc_ids_by_filename(filename)