File size: 1,743 Bytes
16d282e 6eb1f69 16d282e 6eb1f69 16d282e 6eb1f69 16d282e 6eb1f69 16d282e 6eb1f69 16d282e 6eb1f69 16d282e f8d91b4 6eb1f69 |
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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
from fastapi import FastAPI
from fastapi.openapi.docs import get_swagger_ui_html
from data_models.models import SearchQuery
from services import SearchService
from data.data_loader import load_data
app = FastAPI(docs_url=None, redoc_url=None)
# Initiating SearchService
search_service = SearchService()
# Loading data from serialized file
embd_id = load_data()
# Ingesting data into VectorDB
search_service.ingest_data(embd_id)
@app.get("/")
async def root():
return {"message": "Welcome to the Search API"}
@app.get("/docs", include_in_schema=False)
async def custom_swagger_ui_html():
return get_swagger_ui_html(
openapi_url="/openapi.json",
title="API Docs",
oauth2_redirect_url=None,
swagger_js_url="https://unpkg.com/[email protected]/swagger-ui-bundle.js",
swagger_css_url="https://unpkg.com/[email protected]/swagger-ui.css",
)
@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=7860)
# # 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=7860)
|