fix: Fix a bug of connection cache for collection schema (#3071)

yhmo · web-flow · commit 097f583177cf · 2025-11-13T19:25:33.000+08:00
Fix: #3058 Signed-off-by: yhmo <yihua.mo@zilliz.com>
diff --git a/pymilvus/client/grpc_handler.py b/pymilvus/client/grpc_handler.py
@@ -328,6 +328,8 @@ def drop_collection(self, collection_name: str, timeout: Optional[float] = None,
             request, timeout=timeout, metadata=_api_level_md(**kwargs)
         )
         check_status(status)
+        if collection_name in self.schema_cache:
+            self.schema_cache.pop(collection_name)
 
     @retry_on_rpc_failure()
     def add_collection_field(

Original file line number	Diff line number	Diff line change
`@@ -328,6 +328,8 @@ def drop_collection(self, collection_name: str, timeout: Optional[float] = None,`
`328`	`328`	`request, timeout=timeout, metadata=_api_level_md(**kwargs)`
`329`	`329`	`)`
`330`	`330`	`check_status(status)`
	`331`	`+ if collection_name in self.schema_cache:`
	`332`	`+ self.schema_cache.pop(collection_name)`
`331`	`333`
`332`	`334`	`@retry_on_rpc_failure()`
`333`	`335`	`def add_collection_field(`