Skip to content

Commit

Permalink
fix: [2.5]Add is_clustering params for get_compaction_plans in orm (#…
Browse files Browse the repository at this point in the history
…2430)

master pr: #2428

Signed-off-by: Cai Zhang <[email protected]>
  • Loading branch information
xiaocai2333 authored Dec 11, 2024
1 parent 7f702c9 commit eb23c2e
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 @@ -1563,17 +1563,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 eb23c2e

Please sign in to comment.