Modified generator to generate plugins

Signed-off-by: saimedhi <saimedhi@amazon.com>

Author: saimedhi

test_opensearchpy/test_server/test_plugins/test_knn.py

@@ -85,11 +85,12 @@ def test_knn_model(self) -> None:
 
             # Fetch the model state
             try:
-                response = self.client.plugins.knn.get_model(model_id=model_id, filter_path='state')['state']
+                response = self.client.plugins.knn.get_model(model_id='my-model')
                 if response['state'] != 'training':
                     pass 
-            except Exception as e:
-                print(f"Error fetching knn model: {e}")
+            except Exception as error:
+                print(f"Error fetching knn model: {error}")
+                break
 
             # Sleep for 1 minute before the next attempt
             time.sleep(10)