mengzhx/semantica
1
0
Fork 0
mirror of https://github.com/semantica-agi/semantica.git synced 2026-08-29 04:26:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(milvus): add backward compatibility alias and sanitize query

Browse Source 
- Added insert_vectors alias to add_vectors for backward compatibility.
- Sanitized vector_id in get_vector and get_metadata to prevent query injection.
This commit is contained in:
Sameer6305
2026-08-07 12:11:48 +05:30
parent 248d028b09
commit dd42b7fa95
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
semantica/vector_store/milvus_store.py
+8 -2
View File
@@ -399,6 +399,10 @@ class MilvusStore:
except Exception as e:
raise ProcessingError(f"Failed to get collection: {str(e)}")
def insert_vectors(self, vectors: List[Union[np.ndarray, List[float]]], **options) -> Any:
"""Backward compatibility alias for add_vectors."""
return self.add_vectors(vectors, **options)
def add_vectors(
self,
vectors: List[Union[np.ndarray, List[float]]],
@@ -544,8 +548,9 @@ class MilvusStore:
return None
try:
safe_id = vector_id.replace('"', '\\"')
res = self.collection.collection.query(
expr=f'id == "{vector_id}"',
expr=f'id == "{safe_id}"',
output_fields=["vector"]
)
if res and len(res) > 0:
@@ -560,8 +565,9 @@ class MilvusStore:
return None
try:
safe_id = vector_id.replace('"', '\\"')
res = self.collection.collection.query(
expr=f'id == "{vector_id}"',
expr=f'id == "{safe_id}"',
output_fields=["metadata"]
)
if res and len(res) > 0: