enhance: get and delete to use schema cache (#2606)

enhance: get and delete to use schema cache issue: milvus-io/milvus#39093 Signed-off-by: Xianhui.Lin <[email protected]>
milvus-io · Feb 6, 2025 · 3056999 · 3056999
1 parent c5a3c18
commit 3056999
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pymilvus/milvus_client/milvus_client.py b/pymilvus/milvus_client/milvus_client.py
@@ -710,7 +710,7 @@ def get(
 
         conn = self._get_connection()
         try:
-            schema_dict = conn.describe_collection(collection_name, timeout=timeout, **kwargs)
+            schema_dict, _ = conn._get_schema_from_cache_or_remote(collection_name, timeout=timeout)
         except Exception as ex:
             logger.error("Failed to describe collection: %s", collection_name)
             raise ex from ex
@@ -796,7 +796,9 @@ def delete(
         conn = self._get_connection()
         if len(pks) > 0:
             try:
-                schema_dict = conn.describe_collection(collection_name, timeout=timeout, **kwargs)
+                schema_dict, _ = conn._get_schema_from_cache_or_remote(
+                    collection_name, timeout=timeout
+                )
             except Exception as ex:
                 logger.error("Failed to describe collection: %s", collection_name)
                 raise ex from ex