Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
D
dify
Project
Project
Details
Activity
Releases
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
ai-tech
dify
Commits
e14ffcca
Commit
e14ffcca
authored
Jun 20, 2023
by
John Wang
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix: delete document bug
parent
9e40dbd6
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
4 additions
and
1 deletion
+4
-1
base.py
api/core/index/vector_index/base.py
+2
-1
qdrant_vector_index.py
api/core/index/vector_index/qdrant_vector_index.py
+1
-0
weaviate_vector_index.py
api/core/index/vector_index/weaviate_vector_index.py
+1
-0
No files found.
api/core/index/vector_index/base.py
View file @
e14ffcca
import
json
import
json
import
logging
import
logging
from
abc
import
abstractmethod
from
abc
import
abstractmethod
from
typing
import
List
,
Any
,
Tuple
,
cast
from
typing
import
List
,
Any
,
cast
from
langchain.embeddings.base
import
Embeddings
from
langchain.embeddings.base
import
Embeddings
from
langchain.schema
import
Document
,
BaseRetriever
from
langchain.schema
import
Document
,
BaseRetriever
...
@@ -101,6 +101,7 @@ class BaseVectorIndex(BaseIndex):
...
@@ -101,6 +101,7 @@ class BaseVectorIndex(BaseIndex):
def
delete_by_ids
(
self
,
ids
:
list
[
str
])
->
None
:
def
delete_by_ids
(
self
,
ids
:
list
[
str
])
->
None
:
if
self
.
_is_origin
():
if
self
.
_is_origin
():
self
.
recreate_dataset
(
self
.
dataset
)
self
.
recreate_dataset
(
self
.
dataset
)
return
vector_store
=
self
.
_get_vector_store
()
vector_store
=
self
.
_get_vector_store
()
vector_store
=
cast
(
self
.
_get_vector_store_class
(),
vector_store
)
vector_store
=
cast
(
self
.
_get_vector_store_class
(),
vector_store
)
...
...
api/core/index/vector_index/qdrant_vector_index.py
View file @
e14ffcca
...
@@ -90,6 +90,7 @@ class QdrantVectorIndex(BaseVectorIndex):
...
@@ -90,6 +90,7 @@ class QdrantVectorIndex(BaseVectorIndex):
def
delete_by_document_id
(
self
,
document_id
:
str
):
def
delete_by_document_id
(
self
,
document_id
:
str
):
if
self
.
_is_origin
():
if
self
.
_is_origin
():
self
.
recreate_dataset
(
self
.
dataset
)
self
.
recreate_dataset
(
self
.
dataset
)
return
vector_store
=
self
.
_get_vector_store
()
vector_store
=
self
.
_get_vector_store
()
vector_store
=
cast
(
self
.
_get_vector_store_class
(),
vector_store
)
vector_store
=
cast
(
self
.
_get_vector_store_class
(),
vector_store
)
...
...
api/core/index/vector_index/weaviate_vector_index.py
View file @
e14ffcca
...
@@ -111,6 +111,7 @@ class WeaviateVectorIndex(BaseVectorIndex):
...
@@ -111,6 +111,7 @@ class WeaviateVectorIndex(BaseVectorIndex):
def
delete_by_document_id
(
self
,
document_id
:
str
):
def
delete_by_document_id
(
self
,
document_id
:
str
):
if
self
.
_is_origin
():
if
self
.
_is_origin
():
self
.
recreate_dataset
(
self
.
dataset
)
self
.
recreate_dataset
(
self
.
dataset
)
return
vector_store
=
self
.
_get_vector_store
()
vector_store
=
self
.
_get_vector_store
()
vector_store
=
cast
(
self
.
_get_vector_store_class
(),
vector_store
)
vector_store
=
cast
(
self
.
_get_vector_store_class
(),
vector_store
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment