From dd12f91c802c1466ca3035be295f681c2f79a32e Mon Sep 17 00:00:00 2001
From: Szymon Basan <116343782+sbasan@users.noreply.github.com>
Date: Thu, 5 Sep 2024 13:05:29 +0200
Subject: [PATCH] draft (#813)

---
 ENDPOINTS.md                                             | 2 +-
 .../sdwan/application_priority/__init__.py               | 2 --
 .../sdwan/application_priority/traffic_policy.py         | 9 +++++++--
 3 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/ENDPOINTS.md b/ENDPOINTS.md
index 5c6360ee..0036c715 100644
--- a/ENDPOINTS.md
+++ b/ENDPOINTS.md
@@ -1,6 +1,6 @@
 **THIS FILE WAS AUTO-GENERATED DO NOT EDIT**
 
-Generated for: catalystwan-0.35.0.dev4
+Generated for: catalystwan-0.35.0.dev6
 
 All URIs are relative to */dataservice*
 HTTP request | Supported Versions | Method | Payload Type | Return Type | Tenancy Mode
diff --git a/catalystwan/models/configuration/feature_profile/sdwan/application_priority/__init__.py b/catalystwan/models/configuration/feature_profile/sdwan/application_priority/__init__.py
index 44f00c5c..3f5270c0 100644
--- a/catalystwan/models/configuration/feature_profile/sdwan/application_priority/__init__.py
+++ b/catalystwan/models/configuration/feature_profile/sdwan/application_priority/__init__.py
@@ -93,7 +93,6 @@
     TrafficCategoryMatch,
     TrafficClass,
     TrafficClassMatch,
-    TrafficDataDirection,
     TrafficPolicyParcel,
     TrafficPolicyTarget,
     TrafficTargetType,
@@ -195,7 +194,6 @@
     "ServiceAreaValue",
     "SequenceIpType",
     "TrafficPolicyTarget",
-    "TrafficDataDirection",
     "SetVpn",
     "SetTlocList",
     "SetDscp",
diff --git a/catalystwan/models/configuration/feature_profile/sdwan/application_priority/traffic_policy.py b/catalystwan/models/configuration/feature_profile/sdwan/application_priority/traffic_policy.py
index d6ba947e..1057742c 100644
--- a/catalystwan/models/configuration/feature_profile/sdwan/application_priority/traffic_policy.py
+++ b/catalystwan/models/configuration/feature_profile/sdwan/application_priority/traffic_policy.py
@@ -31,13 +31,18 @@
     VersionedField,
 )
 from catalystwan.models.configuration.feature_profile.common import RefIdItem
-from catalystwan.models.policy.centralized import TrafficDataDirection
 from catalystwan.models.policy.policy_definition import DNSTypeEntryType, LossProtectionType
 
+TrafficPolicyDirection = Literal[
+    "service",
+    "tunnel",
+    "all",
+]
+
 
 class TrafficPolicyTarget(BaseModel):
     model_config = ConfigDict(populate_by_name=True, extra="forbid")
-    direction: Global[TrafficDataDirection]
+    direction: Global[TrafficPolicyDirection]
     vpn: Global[List[str]]