2 files changed, 4 insertions, 1 deletions
diff --git a/src/services/api/rest/models.py b/src/services/api/rest/models.py
index d3ee3a08e..abe7e7726 100644
--- a/src/services/api/rest/models.py
+++ b/src/services/api/rest/models.py
@@ -332,6 +332,7 @@ class PingModel(ApiModel):
 class TracerouteModel(ApiModel):
     op: StrictStr
     host: StrictStr
+    vrf: StrictStr | None = None
 
     class Config:
         schema_extra = {
@@ -339,6 +340,7 @@ class TracerouteModel(ApiModel):
                 'key': 'id_key',
                 'op': 'traceroute',
                 'host': 'host',
+                'vrf': 'vrf',
             }
         }
 
diff --git a/src/services/api/rest/routers.py b/src/services/api/rest/routers.py
index a241cabf8..7b0762084 100644
--- a/src/services/api/rest/routers.py
+++ b/src/services/api/rest/routers.py
@@ -992,10 +992,11 @@ def traceroute_op(data: TracerouteModel):
 
     op = data.op
     host = data.host
+    vrf = data.vrf
 
     try:
         if op == 'traceroute':
-            res = session.traceroute(host)
+            res = session.traceroute(host, vrf)
         else:
             return error(400, f"'{op}' is not a valid operation")
     except ConfigSessionError as e: