AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-11 16:08:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: typo in upstashVector if id is always true, also fix some type hint (#11183)

Browse Source 
Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
yihong 2024-11-28 14:05:25 +08:00 committed by GitHub
parent 18add94a31
commit d7160ee563
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/core/rag/datasource/vdb/upstash/upstash_vector.py
View File

@ -64,7 +64,7 @@ class UpstashVector(BaseVector):
item_ids = [] item_ids = []
for doc_id in ids: for doc_id in ids:
ids = self.get_ids_by_metadata_field("doc_id", doc_id) ids = self.get_ids_by_metadata_field("doc_id", doc_id)
if id: if ids:
item_ids += ids item_ids += ids
self._delete_by_ids(ids=item_ids) self._delete_by_ids(ids=item_ids)
@ -95,6 +95,7 @@ class UpstashVector(BaseVector):
metadata = record.metadata metadata = record.metadata
text = record.data text = record.data
score = record.score score = record.score
if metadata is not None and text is not None:
metadata["score"] = score metadata["score"] = score
if score > score_threshold: if score > score_threshold:
docs.append(Document(page_content=text, metadata=metadata)) docs.append(Document(page_content=text, metadata=metadata))
@ -123,7 +124,7 @@ class UpstashVectorFactory(AbstractVectorFactory):
return UpstashVector( return UpstashVector(
collection_name=collection_name, collection_name=collection_name,
config=UpstashVectorConfig( config=UpstashVectorConfig(
url=dify_config.UPSTASH_VECTOR_URL, url=dify_config.UPSTASH_VECTOR_URL or "",
token=dify_config.UPSTASH_VECTOR_TOKEN, token=dify_config.UPSTASH_VECTOR_TOKEN or "",
), ),
) )