github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-10 04:37:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:knowledge init

Browse Source
This commit is contained in:
aries-ckt 2023-06-13 19:50:10 +08:00
parent 669e5cc3dc
commit 38ca7c1cb2
2 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
pilot/scene/chat_knowledge/default/chat.py
View File

@ -1,3 +1,5 @@
from chromadb.errors import NoIndexException
from pilot.scene.base_chat import BaseChat, logger, headers from pilot.scene.base_chat import BaseChat, logger, headers
from pilot.scene.base import ChatScene from pilot.scene.base import ChatScene
from pilot.common.sql_database import Database from pilot.common.sql_database import Database
@ -46,12 +48,15 @@ class ChatDefaultKnowledge(BaseChat):
) )
def generate_input_values(self): def generate_input_values(self):
docs = self.knowledge_embedding_client.similar_search( try:
self.current_user_input, CFG.KNOWLEDGE_SEARCH_TOP_SIZE docs = self.knowledge_embedding_client.similar_search(
) self.current_user_input, CFG.KNOWLEDGE_SEARCH_TOP_SIZE
context = [d.page_content for d in docs] )
context = context[:2000] context = [d.page_content for d in docs]
input_values = {"context": context, "question": self.current_user_input} context = context[:2000]
input_values = {"context": context, "question": self.current_user_input}
except NoIndexException:
raise ValueError("you have no default knowledge store, please execute python knowledge_init.py")
return input_values return input_values
def do_with_prompt_response(self, prompt_response): def do_with_prompt_response(self, prompt_response):

1
tools/knowlege_init.py
View File

@ -43,7 +43,6 @@ if __name__ == "__main__":
parser.add_argument("--vector_name", type=str, default="default") parser.add_argument("--vector_name", type=str, default="default")
args = parser.parse_args() args = parser.parse_args()
vector_name = args.vector_name vector_name = args.vector_name
append_mode = args.append
store_type = CFG.VECTOR_STORE_TYPE store_type = CFG.VECTOR_STORE_TYPE
vector_store_config = {"vector_store_name": vector_name} vector_store_config = {"vector_store_name": vector_name}
print(vector_store_config) print(vector_store_config)