github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-07 03:14:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

WEB API independent

Browse Source
This commit is contained in:
tuyang.yhj 2023-07-04 11:34:12 +08:00
parent 92dbe22504
commit c23a32b906
2 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pilot/scene/chat_db/auto_execute/out_parser.py
View File

@ -35,15 +35,15 @@ class DbChatOutputParser(BaseOutputParser):
if len(data) <= 1:
data.insert(0, ["result"])
df = pd.DataFrame(data[1:], columns=data[0])
if CFG.NEW_SERVER_MODE:
html = df.to_html(index=False, escape=False, sparsify=False)
html = "".join(html.split())
else:
if not CFG.NEW_SERVER_MODE:
table_style = """<style>
table{border-collapse:collapse;width:100%;height:80%;margin:0 auto;float:center;border: 1px solid #007bff; background-color:#333; color:#fff}th,td{border:1px solid #ddd;padding:3px;text-align:center}th{background-color:#C9C3C7;color: #fff;font-weight: bold;}tr:nth-child(even){background-color:#444}tr:hover{background-color:#444}
</style>"""
html_table = df.to_html(index=False, escape=False)
html = f"<html><head>{table_style}</head><body>{html_table}</body></html>"
else:
html = df.to_html(index=False, escape=False, sparsify=False)
html = "".join(html.split())
view_text = f"##### {str(speak)}" + "\n" + html.replace("\n", " ")
return view_text

14
pilot/server/dbgpt_server.py
View File

@ -91,15 +91,19 @@ if __name__ == "__main__":
parser.add_argument("--port", type=int, default=5000)
parser.add_argument("--concurrency-count", type=int, default=10)
parser.add_argument("--share", default=False, action="store_true")
parser.add_argument("-light", "--light", default=False,action="store_true", help="enable light mode")
signal.signal(signal.SIGINT, signal_handler)
# init server config
args = parser.parse_args()
from pilot.server.llmserver import worker
worker.start_check()
server_init(args)
CFG.NEW_SERVER_MODE = True
import uvicorn
if not args.light:
print("Model Unified Deployment Mode!")
from pilot.server.llmserver import worker
worker.start_check()
CFG.NEW_SERVER_MODE = True
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=args.port)