code: def create_context(
      store: metadata_store.MetadataStore,
      context_type_id: int,
      context_name: str,
  ) -> metadata_store_pb2.Context:
      """Put a Context in the MLMD database."""
  
      context = metadata_store_pb2.Context(
          type_id=context_type_id, name=context_name
      )
      [context.id] = store.put_contexts([context])
      return context
  
imports: from typing import Dict, List
  from absl.testing import absltest
  from ml_metadata import metadata_store
  from ml_metadata.proto import metadata_store_pb2
  from ml_metadata.tools.mlmd_resolver import metadata_resolver
  from ml_metadata.tools.mlmd_resolver import metadata_resolver_utils
label_desc: Use image node_id 6 for calling a global function with example usage: create_context(store, context_type_id, context_name) and returns: context