Red-tech-hub commited on
Commit
6b8ece8
1 Parent(s): 945cca4

[update] cors middleware

Browse files
Files changed (1) hide show
  1. server.py +9 -0
server.py CHANGED
@@ -1,4 +1,5 @@
1
  from fastapi import FastAPI, HTTPException, Request, Body
 
2
  from langserve import add_routes
3
  from run import chain
4
  import os
@@ -8,6 +9,14 @@ from titleGenerator import generate_title
8
 
9
  app = FastAPI(title="Code Vulnerability AI")
10
 
 
 
 
 
 
 
 
 
11
  add_routes(app, chain)
12
 
13
  class MessageInput(BaseModel):
 
1
  from fastapi import FastAPI, HTTPException, Request, Body
2
+ from fastapi.middleware.cors import CORSMiddleware
3
  from langserve import add_routes
4
  from run import chain
5
  import os
 
9
 
10
  app = FastAPI(title="Code Vulnerability AI")
11
 
12
+ app.add_middleware(
13
+ CORSMiddleware,
14
+ allow_origins=["*"], # Allows all origins for testing purposes
15
+ allow_credentials=True,
16
+ allow_methods=["*"],
17
+ allow_headers=["*"],
18
+ )
19
+
20
  add_routes(app, chain)
21
 
22
  class MessageInput(BaseModel):