weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

split api_start to create_app & run_api. user can run api with uvicorn in console: uvicorn server.api:app --port 7861

This commit is contained in:
liunux4odoo 2023-07-31 10:05:19 +08:00
parent 463659f0ba
commit 946f10e1f2
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
server/api.py
View File

@ -20,9 +20,7 @@ async def document():
return RedirectResponse(url="/docs") return RedirectResponse(url="/docs")
def api_start(host, port, **kwargs): def create_app():
global app
app = FastAPI() app = FastAPI()
# Add CORS middleware to allow all origins # Add CORS middleware to allow all origins
# 在config.py中设置OPEN_DOMAIN=True，允许跨域 # 在config.py中设置OPEN_DOMAIN=True，允许跨域
@ -94,7 +92,11 @@ def api_start(host, port, **kwargs):
response_model=BaseResponse, response_model=BaseResponse,
summary="上传文件到知识库，并删除另一个文件" summary="上传文件到知识库，并删除另一个文件"
)(update_doc) )(update_doc)
return app
app = create_app()
def run_api(host, port, **kwargs):
if kwargs.get("ssl_keyfile") and kwargs.get("ssl_certfile"): if kwargs.get("ssl_keyfile") and kwargs.get("ssl_certfile"):
uvicorn.run(app, host=host, port=port, ssl_keyfile=kwargs.get("ssl_keyfile"), uvicorn.run(app, host=host, port=port, ssl_keyfile=kwargs.get("ssl_keyfile"),
ssl_certfile=kwargs.get("ssl_certfile")) ssl_certfile=kwargs.get("ssl_certfile"))
@ -113,4 +115,4 @@ if __name__ == "__main__":
# 初始化消息 # 初始化消息
args = parser.parse_args() args = parser.parse_args()
args_dict = vars(args) args_dict = vars(args)
api_start(args.host, args.port, ssl_keyfile=args.ssl_keyfile, ssl_certfile=args.ssl_certfile) run_api(args.host, args.port, ssl_keyfile=args.ssl_keyfile, ssl_certfile=args.ssl_certfile)