Skip to content

Commit

Permalink
fix: Add is_clustering params for get_compaction_plans in orm (#2428)
Browse files Browse the repository at this point in the history
Signed-off-by: Cai Zhang <[email protected]>
  • Loading branch information
xiaocai2333 authored Dec 11, 2024
1 parent 1519baf commit c2d328d
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion pymilvus/orm/collection.py
Original file line number Diff line number Diff line change
Expand Up @@ -1557,17 +1557,26 @@ def wait_for_compaction_completed(
)
return conn.wait_for_compaction_completed(self.compaction_id, timeout=timeout, **kwargs)

def get_compaction_plans(self, timeout: Optional[float] = None, **kwargs) -> CompactionPlans:
def get_compaction_plans(
self, timeout: Optional[float] = None, is_clustering: Optional[bool] = False, **kwargs
) -> CompactionPlans:
"""Get the current compaction plans
Args:
timeout (``float``, optional): An optional duration of time in seconds to allow
for the RPC. When timeout is set to None, client waits until server response
or error occur.
is_clustering (``bool``, optional): Option to get clustering compaction plan.
Returns:
CompactionPlans: All the plans' states of this compaction.
"""
conn = self._get_connection()
if is_clustering:
return conn.get_compaction_plans(
self.clustering_compaction_id, timeout=timeout, **kwargs
)
return conn.get_compaction_plans(self.compaction_id, timeout=timeout, **kwargs)

def get_replicas(self, timeout: Optional[float] = None, **kwargs) -> Replica:
Expand Down

0 comments on commit c2d328d

Please sign in to comment.