code: def task_1(data_size, host):
      """
      task_1:
          before upgrade: create collection and insert data with flush, create index, load and search
          after upgrade: get collection, load, search, insert data with flush, release, create index, load, and search
      """
      prefix = "task_1_"
      connections.connect(host=host, port=19530, timeout=60)
      col_list = get_collections(prefix, check=True)
      assert len(col_list) > 0
      create_index(prefix)
      load_and_search(prefix)
      create_collections_and_insert_data(prefix, count=data_size)
      release_collection(prefix)
      create_index(prefix)
      load_and_search(prefix)
  
file_path: milvus/tests/python_client/deploy/scripts/action_after_upgrade.py
imports: from pymilvus import connections
  import sys
  from common.milvus_sys import MilvusSys
  from utils import *
label_desc: Use image node_id 1 for calling a global function with example usage: task_1(data_size, host) without return types