From 18b5143cd638abb917ded4262b2b14659d7712b3 Mon Sep 17 00:00:00 2001
From: gifrerenom <lluis.gifre@cttc.es>
Date: Tue, 25 Mar 2025 10:28:42 +0000
Subject: [PATCH] Device component - OpenConfig driver:

- Added missing AFI-SAFI in BGP config
---
 .../templates/VPN/Network_instance_multivendor.py        | 6 ++++++
 .../templates/network_instance/protocols/edit_config.xml | 9 +++++++++
 2 files changed, 15 insertions(+)

diff --git a/src/device/service/drivers/openconfig/templates/VPN/Network_instance_multivendor.py b/src/device/service/drivers/openconfig/templates/VPN/Network_instance_multivendor.py
index 71581640e..761312c47 100644
--- a/src/device/service/drivers/openconfig/templates/VPN/Network_instance_multivendor.py
+++ b/src/device/service/drivers/openconfig/templates/VPN/Network_instance_multivendor.py
@@ -134,6 +134,12 @@ def add_protocol_NI(parameters,vendor, DEL):
                                         with tag('as')       :text(parameters['as'])
                                         if "router_id" in parameters: 
                                             with tag('router-id'):text(parameters['router_id'])
+                                    with tag('afi-safis'):
+                                        with tag('afi-safi', 'xmlns:oc-bgp-types="http://openconfig.net/yang/bgp-types"'):
+                                            with tag('afi-safi-name'): text('oc-bgp-types:IPV4_UNICAST')
+                                            with tag('config'):
+                                                with tag('afi-safi-name'): text('oc-bgp-types:IPV4_UNICAST')
+                                                with tag('enabled'): text('true')
                                 if 'neighbors' in parameters:
                                     with tag('neighbors'):
                                         for neighbor in parameters['neighbors']:
diff --git a/src/device/service/drivers/openconfig/templates/network_instance/protocols/edit_config.xml b/src/device/service/drivers/openconfig/templates/network_instance/protocols/edit_config.xml
index da66d97f0..7c36dfb45 100644
--- a/src/device/service/drivers/openconfig/templates/network_instance/protocols/edit_config.xml
+++ b/src/device/service/drivers/openconfig/templates/network_instance/protocols/edit_config.xml
@@ -19,6 +19,15 @@
                             <as>{{as}}</as>
                             <router-id>{{router_id}}</router-id>
                         </config>
+                        <afi-safis>
+                            <afi-safi xmlns:oc-bgp-types="http://openconfig.net/yang/bgp-types">
+                                <afi-safi-name>oc-bgp-types:IPV4_UNICAST</afi-safi-name>
+                                <config>
+                                    <afi-safi-name>oc-bgp-types:IPV4_UNICAST</afi-safi-name>
+                                    <enabled>true</enabled>
+                                </config>
+                            </afi-safi>
+                        </afi-safis>
                     </global>
                     {% if neighbors is defined %}
                     <neighbors>
-- 
GitLab