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

fix: lint

Browse Source
This commit is contained in:
csunny 2023-05-25 23:56:03 +08:00
parent 310e1c42c5
commit b00a47a0ee
3 changed files with 34 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/server/vectordb_qa.py
View File

@ -52,4 +52,4 @@ class KnownLedgeBaseQA:
prompt = state.get_prompt() prompt = state.get_prompt()
print("new prompt length:" + str(len(prompt))) print("new prompt length:" + str(len(prompt)))
return prompt return prompt

29
pilot/server/webserver.py
View File

@ -385,13 +385,16 @@ def http_bot(
cfg.set_last_plugin_return(plugin_resp) cfg.set_last_plugin_return(plugin_resp)
print(plugin_resp) print(plugin_resp)
state.messages[-1][-1] = ( state.messages[-1][-1] = (
"Model推理信息:\n" + ai_response + "\n\nDB-GPT执行结果:\n" + plugin_resp "Model推理信息:\n"
+ ai_response
+ "\n\nDB-GPT执行结果:\n"
+ plugin_resp
) )
yield (state, state.to_gradio_chatbot()) + (no_change_btn,) * 5 yield (state, state.to_gradio_chatbot()) + (no_change_btn,) * 5
except NotCommands as e: except NotCommands as e:
print("命令执行:" + e.message) print("命令执行:" + e.message)
state.messages[-1][-1] = ( state.messages[-1][-1] = (
"命令执行:" + e.message + "\n模型输出:\n" + str(ai_response) "命令执行:" + e.message + "\n模型输出:\n" + str(ai_response)
) )
yield (state, state.to_gradio_chatbot()) + (no_change_btn,) * 5 yield (state, state.to_gradio_chatbot()) + (no_change_btn,) * 5
else: else:
@ -422,18 +425,32 @@ def http_bot(
output = post_process_code(output) output = post_process_code(output)
state.messages[-1][-1] = output + "▌" state.messages[-1][-1] = output + "▌"
yield (state, state.to_gradio_chatbot()) + (disable_btn,) * 5 yield (state, state.to_gradio_chatbot()) + (
disable_btn,
) * 5
else: else:
output = data["text"] + f" (error_code: {data['error_code']})" output = (
data["text"] + f" (error_code: {data['error_code']})"
)
state.messages[-1][-1] = output state.messages[-1][-1] = output
yield (state, state.to_gradio_chatbot()) + ( yield (state, state.to_gradio_chatbot()) + (
disable_btn, disable_btn, disable_btn, enable_btn, enable_btn) disable_btn,
disable_btn,
disable_btn,
enable_btn,
enable_btn,
)
return return
except requests.exceptions.RequestException as e: except requests.exceptions.RequestException as e:
state.messages[-1][-1] = server_error_msg + f" (error_code: 4)" state.messages[-1][-1] = server_error_msg + f" (error_code: 4)"
yield (state, state.to_gradio_chatbot()) + ( yield (state, state.to_gradio_chatbot()) + (
disable_btn, disable_btn, disable_btn, enable_btn, enable_btn) disable_btn,
disable_btn,
disable_btn,
enable_btn,
enable_btn,
)
return return
state.messages[-1][-1] = state.messages[-1][-1][:-1] state.messages[-1][-1] = state.messages[-1][-1][:-1]

20
pilot/vector_store/milvus_store.py
View File

@ -32,7 +32,6 @@ class MilvusStore(VectorStoreBase):
self.fields = [] self.fields = []
self.alias = "default" self.alias = "default"
# use HNSW by default. # use HNSW by default.
self.index_params = { self.index_params = {
"metric_type": "L2", "metric_type": "L2",
@ -105,9 +104,7 @@ class MilvusStore(VectorStoreBase):
embeddings = self.embedding.embed_query(texts[0]) embeddings = self.embedding.embed_query(texts[0])
if utility.has_collection(self.collection_name): if utility.has_collection(self.collection_name):
self.col = Collection( self.col = Collection(self.collection_name, using=self.alias)
self.collection_name, using=self.alias
)
self.fields = [] self.fields = []
for x in self.col.schema.fields: for x in self.col.schema.fields:
self.fields.append(x.name) self.fields.append(x.name)
@ -115,7 +112,10 @@ class MilvusStore(VectorStoreBase):
self.fields.remove(x.name) self.fields.remove(x.name)
if x.is_primary: if x.is_primary:
self.primary_field = x.name self.primary_field = x.name
if x.dtype == DataType.FLOAT_VECTOR or x.dtype == DataType.BINARY_VECTOR: if (
x.dtype == DataType.FLOAT_VECTOR
or x.dtype == DataType.BINARY_VECTOR
):
self.vector_field = x.name self.vector_field = x.name
self._add_documents(texts, metadatas) self._add_documents(texts, metadatas)
return self.collection_name return self.collection_name
@ -132,9 +132,7 @@ class MilvusStore(VectorStoreBase):
for y in texts: for y in texts:
max_length = max(max_length, len(y)) max_length = max(max_length, len(y))
# Create the text field # Create the text field
fields.append( fields.append(FieldSchema(text_field, DataType.VARCHAR, max_length=65535))
FieldSchema(text_field, DataType.VARCHAR, max_length= 65535)
)
# primary key field # primary key field
fields.append( fields.append(
FieldSchema(primary_field, DataType.INT64, is_primary=True, auto_id=True) FieldSchema(primary_field, DataType.INT64, is_primary=True, auto_id=True)
@ -252,7 +250,9 @@ class MilvusStore(VectorStoreBase):
"""load document in vector database.""" """load document in vector database."""
# self.init_schema_and_load(self.collection_name, documents) # self.init_schema_and_load(self.collection_name, documents)
batch_size = 500 batch_size = 500
batched_list = [documents[i:i + batch_size] for i in range(0, len(documents), batch_size)] batched_list = [
documents[i : i + batch_size] for i in range(0, len(documents), batch_size)
]
# docs = [] # docs = []
for doc_batch in batched_list: for doc_batch in batched_list:
self.init_schema_and_load(self.collection_name, doc_batch) self.init_schema_and_load(self.collection_name, doc_batch)
@ -320,4 +320,4 @@ class MilvusStore(VectorStoreBase):
return data[0], ret return data[0], ret
def close(self): def close(self):
connections.disconnect() connections.disconnect()