fix clab for ceos

Author: petercrocker

templates/clab_topology.j2

@@ -39,6 +39,10 @@ topology:
 
   links:
     {% set processed_endpoints = [] %}
+    {% set device_kinds = {} %}
+    {% for device in topology.node.devices.edges %}
+      {% set _ = device_kinds.update({device.node.name.value: device.node.platform.node.containerlab_os.value}) %}
+    {% endfor %}
     {% for device in topology.node.devices.edges %}
       {% for interface in device.node.interfaces.edges %}
         {% if interface.node.connector is defined and interface.node.connector.node is not none and interface.node.role.value not in ["console", "management"] %}
@@ -51,6 +55,14 @@ topology:
             {% set _ = processed_endpoints.append(endpoint_key) %}
             {% set endpoint1 = endpoint1.replace("Ethernet", "eth") %}
             {% set endpoint2 = endpoint2.replace("Ethernet", "eth") %}
+            {% set device1_name = endpoint1.split(":")[0] %}
+            {% set device2_name = endpoint2.split(":")[0] %}
+            {% if device_kinds.get(device1_name) == "arista_ceos" %}
+              {% set endpoint1 = endpoint1.split("/")[0] %}
+            {% endif %}
+            {% if device_kinds.get(device2_name) == "arista_ceos" %}
+              {% set endpoint2 = endpoint2.split("/")[0] %}
+            {% endif %}
     - endpoints: ["{{ endpoint1 }}", "{{ endpoint2 }}"]
           {% endif %}
         {% endif %}