File size: 590 Bytes
16d282e |
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 |
# main.py
from fastapi import FastAPI
from data_models.models import SearchQuery
from services import SearchService
from data.data_loader import load_data
#Initiating FastAPI
app = FastAPI()
#Initaiting SearchService
search_service = SearchService()
#Loading data from serialized file
embd_id = load_data()
#Ingesting data into VectorDB
search_service.ingest_data(embd_id)
@app.post("/search")
async def search(query: SearchQuery):
return search_service.search(query.user_search_query)
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=port)
|