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

Merge pull request #14 from csunny/dev

Browse Source 
fix
This commit is contained in:
magic.chen 2023-05-06 01:14:55 +08:00 committed by GitHub
commit e672df785c
2 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pilot/app.py
View File

@ -40,13 +40,13 @@ def get_answer(q):
return response.response return response.response
def get_similar(q): def get_similar(q):
from pilot.vector_store.extract_tovec import knownledge_tovec, load_knownledge_from_doc from pilot.vector_store.extract_tovec import knownledge_tovec, knownledge_tovec_st
docsearch = load_knownledge_from_doc() docsearch = knownledge_tovec_st("./datasets/plan.md")
docs = docsearch.similarity_search_with_score(q, k=1) docs = docsearch.similarity_search_with_score(q, k=1)
for doc in docs: for doc in docs:
dc, s = doc dc, s = doc
print(dc.page_content) print(s)
yield dc.page_content yield dc.page_content
if __name__ == "__main__": if __name__ == "__main__":

18
pilot/server/webserver.py
View File

@ -160,7 +160,7 @@ def http_bot(state, db_selector, temperature, max_new_tokens, request: gr.Reques
yield (state, state.to_gradio_chatbot()) + (no_change_btn,) * 5 yield (state, state.to_gradio_chatbot()) + (no_change_btn,) * 5
return return
query = state.messages[-2][1]
if len(state.messages) == state.offset + 2: if len(state.messages) == state.offset + 2:
# 第一轮对话需要加入提示Prompt # 第一轮对话需要加入提示Prompt
@ -168,6 +168,8 @@ def http_bot(state, db_selector, temperature, max_new_tokens, request: gr.Reques
new_state = conv_templates[template_name].copy() new_state = conv_templates[template_name].copy()
new_state.conv_id = uuid.uuid4().hex new_state.conv_id = uuid.uuid4().hex
query = state.messages[-2][1]
# prompt 中添加上下文提示 # prompt 中添加上下文提示
if db_selector: if db_selector:
new_state.append_message(new_state.roles[0], gen_sqlgen_conversation(dbname) + query) new_state.append_message(new_state.roles[0], gen_sqlgen_conversation(dbname) + query)
@ -178,13 +180,13 @@ def http_bot(state, db_selector, temperature, max_new_tokens, request: gr.Reques
new_state.append_message(new_state.roles[1], None) new_state.append_message(new_state.roles[1], None)
state = new_state state = new_state
try: # try:
if not db_selector: # if not db_selector:
sim_q = get_simlar(query) # sim_q = get_simlar(query)
print("********vector similar info*************: ", sim_q) # print("********vector similar info*************: ", sim_q)
state.append_message(new_state.roles[0], sim_q + query) # state.append_message(new_state.roles[0], sim_q + query)
except Exception as e: # except Exception as e:
print(e) # print(e)
prompt = state.get_prompt() prompt = state.get_prompt()