Co-authored-by: jyong <jyong@dify.ai>tags/0.3.34
| "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'] |
| 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] |
| 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, |