Skip to content

Commit 4b59f18

Browse files
committed
fixed bug with adding a single embedding
1 parent 31cf5c9 commit 4b59f18

File tree

1 file changed

+14
-6
lines changed

1 file changed

+14
-6
lines changed

backend/src/app/core/vector/crud/crud_base.py

Lines changed: 14 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -97,12 +97,20 @@ def add_embedding(
9797
embedding: Vector embedding
9898
"""
9999
collection = self._get_collection(client=client, project_id=project_id)
100-
collection.data.replace(
101-
uuid=id.uuidv5(),
102-
properties=id.model_dump(),
103-
references=None,
104-
vector=embedding,
105-
)
100+
if collection.data.exists(uuid=id.uuidv5()):
101+
collection.data.replace(
102+
uuid=id.uuidv5(),
103+
properties=id.model_dump(),
104+
references=None,
105+
vector=embedding,
106+
)
107+
else:
108+
collection.data.insert(
109+
uuid=id.uuidv5(),
110+
properties=id.model_dump(),
111+
references=None,
112+
vector=embedding,
113+
)
106114

107115
def add_embedding_batch(
108116
self,

0 commit comments

Comments
 (0)