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)