github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-29 21:26:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:prompt

Browse Source
This commit is contained in:
aries-ckt 2023-05-18 15:39:28 +08:00
parent e8a7fd4546
commit b92ecf2199
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/server/webserver.py
View File

@ -254,7 +254,7 @@ def http_bot(state, mode, sql_mode, db_selector, temperature, max_new_tokens, re
result = prompt_template.format(context="\n".join(context), question=query) result = prompt_template.format(context="\n".join(context), question=query)
state.messages[-2][1] = result state.messages[-2][1] = result
prompt = state.get_prompt() prompt = state.get_prompt()
print("prompt length:" + len(prompt)) print("prompt length:" + str(len(prompt)))
if len(prompt) > 4000: if len(prompt) > 4000:
logger.info("prompt length greater than 4000, rebuild") logger.info("prompt length greater than 4000, rebuild")