Feat/improve vector database logic (#1193)

Co-authored-by: jyong <jyong@dify.ai>
2023-09-18 18:15:41 +08:00
parent 60e0bbd713
commit 269a465fc4
14 changed files with 463 additions and 46 deletions
--- a/api/core/tool/dataset_retriever_tool.py
+++ b/api/core/tool/dataset_retriever_tool.py
@@ -33,7 +33,6 @@ class DatasetRetrieverTool(BaseTool):
    return_resource: str
    retriever_from: str

-
    @classmethod
    def from_dataset(cls, dataset: Dataset, **kwargs):
        description = dataset.description
@@ -94,7 +93,10 @@ class DatasetRetrieverTool(BaseTool):
                    query,
                    search_type='similarity_score_threshold',
                    search_kwargs={
-                        'k': self.k
+                        'k': self.k,
+                        'filter': {
+                            'group_id': [dataset.id]
+                        }
                    }
                )
            else: