github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2026-01-29 21:49:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat:embedding api

Browse Source 
1.embedding_engine add source_reader param
2.docs update
3.fix chroma exit bug
This commit is contained in:
aries_ckt
2023-07-13 15:45:25 +08:00
parent ff89e2e085
commit 03b507a4e9
15 changed files with 100 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pilot/server/dbgpt_server.py
View File

@@ -1,3 +1,4 @@
import atexit
import traceback
import os
import shutil
@@ -36,7 +37,7 @@ CFG = Config()
logger = build_logger("webserver", LOGDIR + "webserver.log")
def signal_handler(sig, frame):
def signal_handler():
print("in order to avoid chroma db atexit problem")
os._exit(0)
@@ -96,7 +97,6 @@ if __name__ == "__main__":
action="store_true",
help="enable light mode",
)
signal.signal(signal.SIGINT, signal_handler)
# init server config
args = parser.parse_args()
@@ -114,3 +114,4 @@ if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=args.port)
signal.signal(signal.SIGINT, signal_handler())