Update features/insight_and_tasks/data_models/tasks.py

features/insight_and_tasks/data_models/tasks.py

@@ -18,6 +18,10 @@ class TaskType(str, Enum):
     INITIATIVE = "initiative"  # Action-oriented, new projects/changes
     TRACKING = "tracking"      # Measurement-focused, monitoring existing metrics/processes
 
+class KeyResultType(str, Enum):
+    PERFORMANCE = "performance"
+    COMPLETION = "completion"
+
 class DataSubject(str, Enum):
     """Specifies the primary data domain a tracking task relates to."""
     FOLLOWER_STATS = "follower_stats"
@@ -132,6 +136,23 @@ class KeyResult(BaseModel):
         default=None,
         description="The specific target value for the metric (e.g., '5%', '1000 new followers')."
     )
+    key_result_type: KeyResultType = Field(
+        description=(
+            "Indicates the nature of the Key Result. "
+            "PERFORMANCE: Focused on achieving a specific, measurable level for a defined metric. "
+            "Its core metric can typically be extracted and monitored directly via a data source, such as the LinkedIn API "
+            "(e.g., monthly post count, engagement rate, follower gains, impressions, CTR, mention volume). "
+            "The goal is to hit or exceed a target for this metric. "
+            "COMPLETION: Focused on finishing a distinct project, delivering a specific output, or establishing a new process. "
+            "Progress is primarily tracked by the successful completion of the defined scope of work. "
+            "Generally, the primary outcome of a COMPLETION Key Result is not a metric continuously tracked via an automated "
+            "data source like the LinkedIn API, or the 'metric' itself describes the state of completion (e.g., 'report delivered', 'process established')."
+        )
+    )
+    data_subject: Optional[DataSubject] = Field(
+        default=None, 
+        description="For 'performance' key results, specifies the primary data subject (e.g., follower_stats, posts, mentions). Can be 'general' or null for 'completion' tasks."
+    )
 
 class OKR(BaseModel):
     """