github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-14 21:51:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: chroma db atexit problem

Browse Source
This commit is contained in:
aries-ckt
2023-06-06 16:56:08 +08:00
parent b4d9127751
commit 5e159c778a
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pilot/server/webserver.py
View File

@@ -1,5 +1,6 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import signal
import threading import threading
import traceback import traceback
import argparse import argparse
@@ -651,6 +652,11 @@ def async_db_summery():
thread = threading.Thread(target=client.init_db_summary) thread = threading.Thread(target=client.init_db_summary)
thread.start() thread.start()
def signal_handler(sig, frame):
print("in order to avoid chroma db atexit problem")
os._exit(0)
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
@@ -668,6 +674,7 @@ if __name__ == "__main__":
cfg = Config() cfg = Config()
dbs = cfg.local_db.get_database_list() dbs = cfg.local_db.get_database_list()
signal.signal(signal.SIGINT, signal_handler)
async_db_summery() async_db_summery()
cfg.set_plugins(scan_plugins(cfg, cfg.debug_mode)) cfg.set_plugins(scan_plugins(cfg, cfg.debug_mode))