@@ -282,7 +282,7 @@ def test_method_wait_for_database_success(self, client: Gradient) -> None:
282282
283283 call_count = [0 ]
284284
285- def mock_retrieve (uuid , ** kwargs ):
285+ def mock_retrieve (_uuid , ** _kwargs ):
286286 call_count [0 ] += 1
287287 response = Mock ()
288288 # Simulate CREATING -> ONLINE transition
@@ -307,7 +307,7 @@ def test_method_wait_for_database_failed_state(self, client: Gradient) -> None:
307307
308308 from gradient .resources .knowledge_bases import KnowledgeBaseDatabaseError
309309
310- def mock_retrieve (uuid , ** kwargs ):
310+ def mock_retrieve (_uuid , ** _kwargs ):
311311 response = Mock ()
312312 response .database_status = "UNHEALTHY"
313313 return response
@@ -328,7 +328,7 @@ def test_method_wait_for_database_timeout(self, client: Gradient) -> None:
328328
329329 from gradient .resources .knowledge_bases import KnowledgeBaseTimeoutError
330330
331- def mock_retrieve (uuid , ** kwargs ):
331+ def mock_retrieve (_uuid , ** _kwargs ):
332332 response = Mock ()
333333 response .database_status = "CREATING"
334334 return response
@@ -349,7 +349,7 @@ def test_method_wait_for_database_decommissioned(self, client: Gradient) -> None
349349
350350 from gradient .resources .knowledge_bases import KnowledgeBaseDatabaseError
351351
352- def mock_retrieve (uuid , ** kwargs ):
352+ def mock_retrieve (_uuid , ** _kwargs ):
353353 response = Mock ()
354354 response .database_status = "DECOMMISSIONED"
355355 return response
@@ -636,7 +636,7 @@ async def test_method_wait_for_database_success(self, async_client: AsyncGradien
636636
637637 call_count = [0 ]
638638
639- async def mock_retrieve (uuid , ** kwargs ):
639+ async def mock_retrieve (_uuid , ** _kwargs ):
640640 call_count [0 ] += 1
641641 response = Mock ()
642642 # Simulate CREATING -> ONLINE transition
@@ -661,7 +661,7 @@ async def test_method_wait_for_database_failed_state(self, async_client: AsyncGr
661661
662662 from gradient .resources .knowledge_bases import KnowledgeBaseDatabaseError
663663
664- async def mock_retrieve (uuid , ** kwargs ):
664+ async def mock_retrieve (_uuid , ** _kwargs ):
665665 response = Mock ()
666666 response .database_status = "UNHEALTHY"
667667 return response
@@ -682,7 +682,7 @@ async def test_method_wait_for_database_timeout(self, async_client: AsyncGradien
682682
683683 from gradient .resources .knowledge_bases import KnowledgeBaseTimeoutError
684684
685- async def mock_retrieve (uuid , ** kwargs ):
685+ async def mock_retrieve (_uuid , ** _kwargs ):
686686 response = Mock ()
687687 response .database_status = "CREATING"
688688 return response
@@ -703,7 +703,7 @@ async def test_method_wait_for_database_decommissioned(self, async_client: Async
703703
704704 from gradient .resources .knowledge_bases import KnowledgeBaseDatabaseError
705705
706- async def mock_retrieve (uuid , ** kwargs ):
706+ async def mock_retrieve (_uuid , ** _kwargs ):
707707 response = Mock ()
708708 response .database_status = "DECOMMISSIONED"
709709 return response
0 commit comments