Spaces:
Sleeping
Sleeping
File size: 1,310 Bytes
ca2fff7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
#This is to init the vector store
from typing import Annotated
from db.vector_store import Store
from document_parsing import generate_documents
from fastapi import APIRouter, Body
from fastapi.responses import JSONResponse
from fastapi import APIRouter, UploadFile, File, Body
router = APIRouter()
@router.put("/v1/docs")
async def recreate_collection(name: Annotated[str, Body(embed=True)]):
""" `name` of the doc to be created.
If one exits, delete and recreate.
"""
print(f"creating collection {name} in db")
return Store.get_instance().create_collection(name)
@router.post("/v1/docs")
async def update(name: Annotated[str, Body()], file_name: Annotated[str, Body()], file: UploadFile = File(...)):
"""Update an existing document with information from the file.
If one doesn't exist with name, it creates a new document to update.
`name` of the collection
`file` to upload.
`fileName` name of the file. This is used for metadata purposes only.
"""
_db = Store.get_instance().get_collection(name)
if not _db:
return JSONResponse(status_code=404, content={})
async for doc in generate_documents(file, file_name):
print(doc)
_db.add_documents([doc])
return JSONResponse(status_code=200, content={"name": name})
|