fixing how tag mutation is handeled for terms and categories

Author: vaibhavatlan

pyatlan/client/asset.py

@@ -905,14 +905,21 @@ def add_atlan_tags(
         from pyatlan.client.atlan import AtlanClient
 
         client = AtlanClient.get_current_client()
-
         name_asset = client.asset.get_by_qualified_name(
-            qualified_name=qualified_name, asset_type=asset_type
-        )
-
-        updated_asset = asset_type.updater(
-            qualified_name=qualified_name, name=name_asset.name
+            qualified_name=qualified_name,
+            asset_type=asset_type,
+            ignore_relationships=False,
         )
+        if asset_type in (AtlasGlossaryTerm, AtlasGlossaryCategory):
+            updated_asset = asset_type.updater(
+                qualified_name=qualified_name,
+                name=name_asset.name,
+                glossary_guid=name_asset.anchor.guid,  # type: ignore
+            )
+        else:
+            updated_asset = asset_type.updater(
+                qualified_name=qualified_name, name=name_asset.name
+            )
 
         atlan_tag = [
             AtlanTag(
@@ -964,11 +971,20 @@ def update_atlan_tags(
 
         client = AtlanClient.get_current_client()
         name_asset = client.asset.get_by_qualified_name(
-            qualified_name=qualified_name, asset_type=asset_type
-        )
-        updated_asset = asset_type.updater(
-            qualified_name=qualified_name, name=name_asset.name
+            qualified_name=qualified_name,
+            asset_type=asset_type,
+            ignore_relationships=False,
         )
+        if asset_type in (AtlasGlossaryTerm, AtlasGlossaryCategory):
+            updated_asset = asset_type.updater(
+                qualified_name=qualified_name,
+                name=name_asset.name,
+                glossary_guid=name_asset.anchor.guid,  # type: ignore
+            )
+        else:
+            updated_asset = asset_type.updater(
+                qualified_name=qualified_name, name=name_asset.name
+            )
 
         atlan_tag = [
             AtlanTag(
@@ -1005,12 +1021,22 @@ def remove_atlan_tags(
 
         client = AtlanClient.get_current_client()
         name_asset = client.asset.get_by_qualified_name(
-            qualified_name=qualified_name, asset_type=asset_type
-        )
-        updated_asset = asset_type.updater(
-            qualified_name=qualified_name, name=name_asset.name
+            qualified_name=qualified_name,
+            asset_type=asset_type,
+            ignore_relationships=False,
         )
 
+        if asset_type in (AtlasGlossaryTerm, AtlasGlossaryCategory):
+            updated_asset = asset_type.updater(
+                qualified_name=qualified_name,
+                name=name_asset.name,
+                glossary_guid=name_asset.anchor.guid,  # type: ignore
+            )
+        else:
+            updated_asset = asset_type.updater(
+                qualified_name=qualified_name, name=name_asset.name
+            )
+
         atlan_tag = [
             AtlanTag(type_name=AtlanTagName(display_text=name))
             for name in atlan_tag_names