فهرست منبع

Add CORS for local testing

Daoud Clarke 3 سال پیش
والد
کامیت
a93fbe9d66
1فایلهای تغییر یافته به همراه12 افزوده شده و 5 حذف شده
  1. 12 5
      mwmbl/main.py

+ 12 - 5
mwmbl/main.py

@@ -5,6 +5,7 @@ from multiprocessing import Process
 
 
 import uvicorn
 import uvicorn
 from fastapi import FastAPI
 from fastapi import FastAPI
+from starlette.middleware.cors import CORSMiddleware
 
 
 from mwmbl import background
 from mwmbl import background
 from mwmbl.indexer import historical, retrieve, preprocess, update_pages
 from mwmbl.indexer import historical, retrieve, preprocess, update_pages
@@ -41,11 +42,7 @@ def run():
         print("Creating a new index")
         print("Creating a new index")
         TinyIndex.create(item_factory=Document, index_path=args.index, num_pages=NUM_PAGES, page_size=PAGE_SIZE)
         TinyIndex.create(item_factory=Document, index_path=args.index, num_pages=NUM_PAGES, page_size=PAGE_SIZE)
 
 
-    Process(target=background.run, args=(args.index,)).start()
-    # Process(target=historical.run).start()
-    # Process(target=retrieve.run).start()
-    # Process(target=preprocess.run, args=(args.index,)).start()
-    # Process(target=update_pages.run, args=(args.index,)).start()
+    # Process(target=background.run, args=(args.index,)).start()
 
 
     completer = Completer()
     completer = Completer()
 
 
@@ -55,6 +52,16 @@ def run():
         # Initialize FastApi instance
         # Initialize FastApi instance
         app = FastAPI()
         app = FastAPI()
 
 
+        origins = ["*"]
+
+        app.add_middleware(
+            CORSMiddleware,
+            allow_origins=origins,
+            allow_credentials=True,
+            allow_methods=["*"],
+            allow_headers=["*"],
+        )
+
         search_router = search.create_router(ranker)
         search_router = search.create_router(ranker)
         app.include_router(search_router)
         app.include_router(search_router)
         app.include_router(crawler_router)
         app.include_router(crawler_router)