github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-14 13:40:54 +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
# -*- coding: utf-8 -*-
import signal
import threading
import traceback
import argparse
@@ -651,6 +652,11 @@ def async_db_summery():
thread = threading.Thread(target=client.init_db_summary)
thread.start()
def signal_handler(sig, frame):
print("in order to avoid chroma db atexit problem")
os._exit(0)
if __name__ == "__main__":
parser = argparse.ArgumentParser()
@@ -668,6 +674,7 @@ if __name__ == "__main__":
cfg = Config()
dbs = cfg.local_db.get_database_list()
signal.signal(signal.SIGINT, signal_handler)
async_db_summery()
cfg.set_plugins(scan_plugins(cfg, cfg.debug_mode))