search_pt / main.py
bardd's picture
first commit
16d282e verified
raw
history blame
590 Bytes
# 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)