Unverified Commit 8c451eb0 authored by Jyong's avatar Jyong Committed by GitHub

fix only full text search in app issue (#1736)

Co-authored-by: 's avatarjyong <jyong@dify.ai>
parent 79b43662
...@@ -141,6 +141,16 @@ class WeaviateVectorIndex(BaseVectorIndex): ...@@ -141,6 +141,16 @@ class WeaviateVectorIndex(BaseVectorIndex):
"valueText": document_id "valueText": document_id
}) })
def delete_by_group_id(self, group_id: str):
if self._is_origin():
self.recreate_dataset(self.dataset)
return
vector_store = self._get_vector_store()
vector_store = cast(self._get_vector_store_class(), vector_store)
vector_store.delete()
def _is_origin(self): def _is_origin(self):
if self.dataset.index_struct_dict: if self.dataset.index_struct_dict:
class_prefix: str = self.dataset.index_struct_dict['vector_store']['class_prefix'] class_prefix: str = self.dataset.index_struct_dict['vector_store']['class_prefix']
......
...@@ -82,7 +82,8 @@ class DatasetMultiRetrieverTool(BaseTool): ...@@ -82,7 +82,8 @@ class DatasetMultiRetrieverTool(BaseTool):
hit_callback.on_tool_end(all_documents) hit_callback.on_tool_end(all_documents)
document_score_list = {} document_score_list = {}
for item in all_documents: for item in all_documents:
document_score_list[item.metadata['doc_id']] = item.metadata['score'] if 'score' in item.metadata and item.metadata['score']:
document_score_list[item.metadata['doc_id']] = item.metadata['score']
document_context_list = [] document_context_list = []
index_node_ids = [document.metadata['doc_id'] for document in all_documents] index_node_ids = [document.metadata['doc_id'] for document in all_documents]
......
...@@ -158,7 +158,8 @@ class DatasetRetrieverTool(BaseTool): ...@@ -158,7 +158,8 @@ class DatasetRetrieverTool(BaseTool):
document_score_list = {} document_score_list = {}
if dataset.indexing_technique != "economy": if dataset.indexing_technique != "economy":
for item in documents: for item in documents:
document_score_list[item.metadata['doc_id']] = item.metadata['score'] if 'score' in item.metadata and item.metadata['score']:
document_score_list[item.metadata['doc_id']] = item.metadata['score']
document_context_list = [] document_context_list = []
index_node_ids = [document.metadata['doc_id'] for document in documents] index_node_ids = [document.metadata['doc_id'] for document in documents]
segments = DocumentSegment.query.filter(DocumentSegment.dataset_id == self.dataset_id, segments = DocumentSegment.query.filter(DocumentSegment.dataset_id == self.dataset_id,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment